Merge pull request #16 from bimberlabinternal/21.7_fb_merge

Merge discvr-21.3 to discvr-21.7

Author: bbimber

LDK/test/src/org/labkey/test/util/external/labModules/LabModuleHelper.java

@@ -20,18 +20,21 @@
 import org.junit.Assert;
 import org.labkey.test.BaseWebDriverTest;
 import org.labkey.test.Locator;
+import org.labkey.test.WebDriverWrapper;
 import org.labkey.test.util.DataRegionTable;
 import org.labkey.test.util.Ext4Helper;
 import org.labkey.test.util.UIAssayHelper;
 import org.labkey.test.util.ext4cmp.Ext4CmpRef;
 import org.labkey.test.util.ext4cmp.Ext4ComboRef;
 import org.labkey.test.util.ext4cmp.Ext4FieldRef;
 import org.labkey.test.util.ext4cmp.Ext4GridRef;
+import org.openqa.selenium.TimeoutException;
 
 import java.io.File;
 import java.text.SimpleDateFormat;
 import java.util.List;
 import java.util.Random;
+import java.util.Set;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
@@ -249,10 +252,33 @@ public void goToAssayResultImport(String assayName, boolean supportsTemplates)
         _test.waitForText("Data Import");
     }
 
+    public String clickAndGetExampleData()
+    {
+        Locator btn = Locator.linkContainingText("Download Example Data");
+        _test.waitForElement(btn);
+        _test.waitAndClick(btn);
+
+        int iterations = 0;
+        while (_test.getDriver().getWindowHandles().size() < 2)
+        {
+            iterations++;
+            WebDriverWrapper.sleep(100);
+
+            if (iterations > 20)
+            {
+                throw new TimeoutException("Timed out waiting for second browser window");
+            }
+        }
+
+        return getExampleData();
+    }
+
     public String getExampleData()
     {
         String ret = null;
 
+        Set<String> handles = _test.getDriver().getWindowHandles();
+        Assert.assertTrue("Expected more than one open window, was: " + handles.size(), handles.size() > 1);
         String currentWindow = _test.getDriver().getWindowHandle();
         for (String handle : _test.getDriver().getWindowHandles())
         {