From de7f146994d9f82cf47a3365c4d10cd56d8460d0 Mon Sep 17 00:00:00 2001
From: Kristijan Mitrovic <kmitrovic@itekako.com>
Date: Tue, 21 Apr 2020 11:57:24 +0200
Subject: [PATCH] Return context from ran script

---
 .../cobbzilla/wizardtest/resources/ApiModelTestBase.java  | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/wizard-server-test/src/main/java/org/cobbzilla/wizardtest/resources/ApiModelTestBase.java b/wizard-server-test/src/main/java/org/cobbzilla/wizardtest/resources/ApiModelTestBase.java
index 1dbb16d..eae9e8c 100644
--- a/wizard-server-test/src/main/java/org/cobbzilla/wizardtest/resources/ApiModelTestBase.java
+++ b/wizard-server-test/src/main/java/org/cobbzilla/wizardtest/resources/ApiModelTestBase.java
@@ -1,6 +1,7 @@
 package org.cobbzilla.wizardtest.resources;
 
 import lombok.Getter;
+import lombok.NonNull;
 import lombok.extern.slf4j.Slf4j;
 import org.cobbzilla.util.collection.SingletonList;
 import org.cobbzilla.util.io.FileUtil;
@@ -271,13 +272,14 @@ public abstract class ApiModelTestBase<C extends PgRestServerConfiguration, S ex
         getConfiguration().pgDump(file);
     }
 
-    protected void modelTest(final String name) throws Exception {
-        modelTest(name, getApiRunner());
+    @NonNull protected Map<String, Object> modelTest(final String name) throws Exception {
+        return modelTest(name, getApiRunner());
     }
 
-    protected void modelTest(final String name, ApiRunner apiRunner) throws Exception {
+    @NonNull protected Map<String, Object> modelTest(final String name, ApiRunner apiRunner) throws Exception {
         logout();
         apiRunner.run(include(name));
+        return apiRunner.getContext();
     }
 
     @Override public List<String> getIncludePaths() {