From 4c4ae9edc0a23b82a8a63c59969c609a32dfac16 Mon Sep 17 00:00:00 2001
From: Karl Lum <klum@labkey.com>
Date: Tue, 16 Jul 2024 08:42:03 -0700
Subject: [PATCH] enable the new experimental feature flag for the tests (#185)

---
 .../labkey/test/tests/viscstudies/CAVDStudyTest.java   | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/viscstudies/test/src/org/labkey/test/tests/viscstudies/CAVDStudyTest.java b/viscstudies/test/src/org/labkey/test/tests/viscstudies/CAVDStudyTest.java
index 788d955..7047b25 100644
--- a/viscstudies/test/src/org/labkey/test/tests/viscstudies/CAVDStudyTest.java
+++ b/viscstudies/test/src/org/labkey/test/tests/viscstudies/CAVDStudyTest.java
@@ -22,6 +22,7 @@
 import org.labkey.test.BaseWebDriverTest;
 import org.labkey.test.Locator;
 import org.labkey.test.TestFileUtils;
+import org.labkey.test.TestTimeoutException;
 import org.labkey.test.categories.CustomModules;
 import org.labkey.test.pages.ImportDataPage;
 import org.labkey.test.params.FieldDefinition;
@@ -30,6 +31,7 @@
 import org.labkey.test.tests.StudyBaseTest;
 import org.labkey.test.util.DataRegionExportHelper;
 import org.labkey.test.util.DataRegionTable;
+import org.labkey.test.util.ExperimentalFeaturesHelper;
 import org.labkey.test.util.Ext4Helper;
 import org.labkey.test.util.LogMethod;
 import org.labkey.test.util.LoggedParam;
@@ -80,9 +82,17 @@ public List<String> getAssociatedModules()
         return Arrays.asList("viscstudies");
     }
 
+    @Override
+    protected void doCleanup(boolean afterTest) throws TestTimeoutException
+    {
+        ExperimentalFeaturesHelper.disableExperimentalFeature(createDefaultConnection(), "GWTStudyDesign");
+        super.doCleanup(afterTest);
+    }
+
     @Override @LogMethod
     protected void doCreateSteps()
     {
+        ExperimentalFeaturesHelper.enableExperimentalFeature(createDefaultConnection(), "GWTStudyDesign");
         _containerHelper.createProject(PROJECT_NAME, "None");
         _containerHelper.createSubfolder(PROJECT_NAME, PROJECT_NAME, FOLDER_NAME, "CAVD Study", null);