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 modelTest(final String name) throws Exception { + return modelTest(name, getApiRunner()); } - protected void modelTest(final String name, ApiRunner apiRunner) throws Exception { + @NonNull protected Map modelTest(final String name, ApiRunner apiRunner) throws Exception { logout(); apiRunner.run(include(name)); + return apiRunner.getContext(); } @Override public List getIncludePaths() {