Add retry to browser setup

compulim · Feb 2, 2019 · e6d1415 · e6d1415
1 parent e98c2eb
commit e6d1415
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 12 deletions.
diff --git a/__tests__/setup/retry.js b/__tests__/setup/retry.js
@@ -0,0 +1,13 @@
+export default async function (fn, retries) {
+  let lastErr;
+
+  for (; retries > 0; retries--) {
+    try {
+      return await fn();
+    } catch (err) {
+      lastErr = err;
+    }
+  }
+
+  throw lastErr;
+}
diff --git a/__tests__/setup/setupTestFramework.js b/__tests__/setup/setupTestFramework.js
@@ -9,6 +9,7 @@ import handler from 'serve-handler';
 import { timeouts } from '../constants.json';
 
 import createPageObjects from './pageObjects/index';
+import retry from './retry';
 import setupTestEnvironment from './setupTestEnvironment';
 import webChatLoaded from './conditions/webChatLoaded';
 
@@ -31,21 +32,23 @@ global.setupWebDriver = async (options = {}) => {
       let { baseURL, builder } = await setupTestEnvironment(BROWSER_NAME, new Builder(), options);
       const driver = builder.build();
 
-      // If the baseURL contains $PORT, it means it requires us to fill-in
-      if (/\$PORT/i.test(baseURL)) {
-        const { port } = await global.setupWebServer();
+      await retry(async () => {
+        // If the baseURL contains $PORT, it means it requires us to fill-in
+        if (/\$PORT/i.test(baseURL)) {
+          const { port } = await global.setupWebServer();
 
-        await driver.get(baseURL.replace(/\$PORT/ig, port));
-      } else {
-        await driver.get(baseURL);
-      }
+          await driver.get(baseURL.replace(/\$PORT/ig, port));
+        } else {
+          await driver.get(baseURL);
+        }
 
-      await driver.executeScript((coverage, props) => {
-        window.__coverage__ = coverage;
-        main({ props });
-      }, global.__coverage__, options.props);
+        await driver.executeScript((coverage, props) => {
+          window.__coverage__ = coverage;
+          main({ props });
+        }, global.__coverage__, options.props);
 
-      await driver.wait(webChatLoaded(), timeouts.navigation);
+        await driver.wait(webChatLoaded(), timeouts.navigation);
+      }, 3);
 
       return { driver, pageObjects: createPageObjects(driver) };
     })();