Skip to content
5 changes: 5 additions & 0 deletions src/org/labkey/test/tests/AdvancedImportOptionsTest.java
Original file line number Diff line number Diff line change
Expand Up @@ -72,6 +72,7 @@ public class AdvancedImportOptionsTest extends BaseWebDriverTest implements Post
private static final int EXPECTED_COMPLETED_IMPORT_JOBS = 1;
private static final int EXPECTED_COMPLETED_MULTI_FOLDER_JOBS = 2;
private Boolean _studyDesignPreviouslyEnabled;
private Boolean _advancedImportOptionsEnabled;

@Override
public List<String> getAssociatedModules()
Expand All @@ -96,6 +97,7 @@ public static void doSetup()
{
AdvancedImportOptionsTest test = getCurrentTest();
test._studyDesignPreviouslyEnabled = OptionalFeatureHelper.enableOptionalFeature(test.createDefaultConnection(), "studyDesignFlag");
test._advancedImportOptionsEnabled = OptionalFeatureHelper.enableOptionalFeature(test.createDefaultConnection(), "advancedImportFlag");
}

@Override
Expand All @@ -109,6 +111,9 @@ public void doCleanup(boolean afterTest) throws TestTimeoutException
if (_studyDesignPreviouslyEnabled != null)
OptionalFeatureHelper.setOptionalFeature(createDefaultConnection(), "studyDesignFlag", _studyDesignPreviouslyEnabled);

if (_advancedImportOptionsEnabled != null)
OptionalFeatureHelper.setOptionalFeature(createDefaultConnection(), "advancedImportFlag", _advancedImportOptionsEnabled);

_userHelper.deleteUser(LIMITED_USER);
}

Expand Down