Refactor building of e2e web driver (#7841)

The switch case has been moved to a separate function so that the
initialization steps following the web driver instantiation could more
easily be deduplicated.

test/e2e/webdriver/index.js

@@ -8,28 +8,26 @@ async function buildWebDriver ({ responsive, port } = {}) {
   const browser = process.env.SELENIUM_BROWSER
   const extensionPath = `dist/${browser}`
 
+  const { driver: seleniumDriver, extensionId, extensionUrl } = await buildBrowserWebDriver(browser, { extensionPath, responsive, port })
+  setupFetchMocking(seleniumDriver)
+  await seleniumDriver.get(extensionUrl)
+
+  const driver = new Driver(seleniumDriver, browser)
+
+  return {
+    driver,
+    extensionId,
+    extensionUrl,
+  }
+}
+
+async function buildBrowserWebDriver (browser, webDriverOptions) {
   switch (browser) {
     case Browser.CHROME: {
-      const { driver, extensionId, extensionUrl } = await ChromeDriver.build({ extensionPath, responsive, port })
+      return await ChromeDriver.build(webDriverOptions)
-      setupFetchMocking(driver)
-      await driver.get(extensionUrl)
-
-      return {
-        driver: new Driver(driver, browser),
-        extensionId,
-        extensionUrl,
-      }
     }
     case Browser.FIREFOX: {
-      const { driver, extensionId, extensionUrl } = await FirefoxDriver.build({ extensionPath, responsive, port })
+      return await FirefoxDriver.build(webDriverOptions)
-      setupFetchMocking(driver)
-      await driver.get(extensionUrl)
-
-      return {
-        driver: new Driver(driver, browser),
-        extensionId,
-        extensionUrl,
-      }
     }
     default: {
       throw new Error(`Unrecognized browser: ${browser}`)