WIP: Firefox support (#810)

e2e/helpers.ts

@@ -11,6 +11,7 @@ import {
   until,
 } from 'selenium-webdriver';
 import chrome from 'selenium-webdriver/chrome';
+import firefox from 'selenium-webdriver/firefox';
 import { expect } from 'vitest';
 import { erc20ABI } from 'wagmi';
 
@@ -22,13 +23,23 @@ const BINARY_PATHS = {
   mac: {
     chrome: '/Applications/Google Chrome.app/Contents/MacOS/Google Chrome',
     brave: '/Applications/Brave Browser.app/Contents/MacOS/Brave Browser',
+    firefox: '/Applications/Firefox.app/Contents/MacOS/Firefox',
   },
   linux: {
     chrome: process.env.CHROMIUM_BIN,
     brave: process.env.BRAVE_BIN,
+    firefox: process.env.FIREFOX_BIN,
   },
 };
 
+export const getRootUrl = () => {
+  const browser = process.env.BROWSER || 'chrome';
+  if (browser === 'firefox') {
+    return 'moz-extension://';
+  }
+  return 'chrome-extension://';
+};
+
 export const byTestId = (id: string) => By.css(`[data-testid="${id}"]`);
 export const byText = (text: string) =>
   By.xpath(`//*[contains(text(),"${text}")]`);
@@ -90,18 +101,34 @@ export async function initDriverWithOptions(opts) {
     '--enable-logging',
   ];
 
-  const options = new chrome.Options()
-    .setChromeBinaryPath(BINARY_PATHS[opts.os][opts.browser])
-    .addArguments(...args);
-  options.setAcceptInsecureCerts(true);
-
-  const service = new chrome.ServiceBuilder().setStdio('inherit');
-
-  return await new Builder()
-    .setChromeService(service)
-    .forBrowser('chrome')
-    .setChromeOptions(options)
-    .build();
+  if (opts.browser === 'firefox') {
+    const options = new firefox.Options()
+      .setBinary(BINARY_PATHS[opts.os][opts.browser])
+      .addArguments(...args.slice(1))
+      .addExtensions('rainbowbx.xpi');
+    options.setAcceptInsecureCerts(true);
+
+    const service = new firefox.ServiceBuilder().setStdio('inherit');
+
+    return await new Builder()
+      .setFirefoxService(service)
+      .forBrowser('firefox')
+      .setFirefoxOptions(options)
+      .build();
+  } else {
+    const options = new chrome.Options()
+      .setChromeBinaryPath(BINARY_PATHS[opts.os][opts.browser])
+      .addArguments(...args);
+    options.setAcceptInsecureCerts(true);
+
+    const service = new chrome.ServiceBuilder().setStdio('inherit');
+
+    return await new Builder()
+      .setChromeService(service)
+      .forBrowser('chrome')
+      .setChromeOptions(options)
+      .build();
+  }
 }
 
 export async function getExtensionIdByName(driver, extensionName) {

e2e/parallel/importWalletFlow.test.ts

@@ -6,12 +6,13 @@ import { afterAll, beforeAll, describe, it } from 'vitest';
 import {
   checkWalletName,
   getExtensionIdByName,
+  getRootUrl,
   importWalletFlow,
   initDriverWithOptions,
 } from '../helpers';
 import { TEST_VARIABLES } from '../walletVariables';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/parallel/importWalletFlowPkey.test.ts

@@ -6,12 +6,13 @@ import { afterAll, beforeAll, describe, it } from 'vitest';
 import {
   checkWalletName,
   getExtensionIdByName,
+  getRootUrl,
   importWalletFlow,
   initDriverWithOptions,
 } from '../helpers';
 import { TEST_VARIABLES } from '../walletVariables';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/parallel/newWalletFlow.test.ts

@@ -9,6 +9,7 @@ import {
   findElementByTestIdAndClick,
   findElementByText,
   getExtensionIdByName,
+  getRootUrl,
   goToPopup,
   goToWelcome,
   initDriverWithOptions,
@@ -18,7 +19,7 @@ import {
   waitAndClick,
 } from '../helpers';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/parallel/settingsFlows.test.ts

@@ -9,6 +9,7 @@ import {
   findElementByTestIdAndClick,
   findElementByTextAndClick,
   getExtensionIdByName,
+  getRootUrl,
   goBackTwice,
   importWalletFlow,
   initDriverWithOptions,
@@ -18,7 +19,7 @@ import {
 } from '../helpers';
 import { TEST_VARIABLES } from '../walletVariables';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/parallel/settingsPrivacyFlows.test.ts

@@ -12,6 +12,7 @@ import {
   findElementByTextAndClick,
   getExtensionIdByName,
   getNumberOfWallets,
+  getRootUrl,
   goToPopup,
   importWalletFlow,
   initDriverWithOptions,
@@ -22,7 +23,7 @@ import {
 } from '../helpers';
 import { TEST_VARIABLES } from '../walletVariables';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/serial/dappInteractions/1_appInteractionsFlow.test.ts

@@ -14,6 +14,7 @@ import {
   findElementByText,
   getAllWindowHandles,
   getExtensionIdByName,
+  getRootUrl,
   getTextFromText,
   getWindowHandle,
   goToPopup,
@@ -60,7 +61,7 @@ const TYPED_MESSAGE = {
 };
 const MESSAGE = 'rainbow rocks 🌈';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/serial/dappInteractions/2_dappInteractionFlow.test.ts

@@ -16,6 +16,7 @@ import {
   getAllWindowHandles,
   getExtensionIdByName,
   getOnchainBalance,
+  getRootUrl,
   getTextFromDappText,
   getWindowHandle,
   goToPopup,
@@ -28,7 +29,7 @@ import {
 } from '../../helpers';
 import { TEST_VARIABLES } from '../../walletVariables';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/serial/send/1_sendFlow.test.ts

@@ -15,6 +15,7 @@ import {
   findElementByTestIdAndClick,
   findElementByText,
   getExtensionIdByName,
+  getRootUrl,
   goToPopup,
   goToWelcome,
   initDriverWithOptions,
@@ -24,7 +25,7 @@ import {
 } from '../../helpers';
 import { TEST_VARIABLES } from '../../walletVariables';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/serial/swap/1_swapFlow1.test.ts

@@ -18,6 +18,7 @@ import {
   findElementByTestIdAndDoubleClick,
   findElementByText,
   getExtensionIdByName,
+  getRootUrl,
   getTextFromText,
   getTextFromTextInput,
   goToPopup,
@@ -30,7 +31,7 @@ import {
 import { convertRawAmountToDecimalFormat, subtract } from '../../numbers';
 import { SWAP_VARIABLES, TEST_VARIABLES } from '../../walletVariables';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/serial/swap/2_swapFlow2.test.ts

@@ -18,6 +18,7 @@ import {
   findElementByTestIdAndClick,
   findElementByText,
   getExtensionIdByName,
+  getRootUrl,
   getTextFromText,
   goToPopup,
   goToWelcome,
@@ -30,7 +31,7 @@ import {
 import { convertRawAmountToDecimalFormat, subtract } from '../../numbers';
 import { SWAP_VARIABLES, TEST_VARIABLES } from '../../walletVariables';
 
-let rootURL = 'chrome-extension://';
+let rootURL = getRootUrl();
 let driver: WebDriver;
 
 const browser = process.env.BROWSER || 'chrome';

e2e/vitest.config.ts

@@ -8,5 +8,6 @@ export default defineConfig({
     watch: false,
     retry: 2,
     bail: 1,
+    hookTimeout: 30_000,
   },
 }) as UserConfig;