test - e2e - remove usage of extensionId from tests

1 files changed, 6 insertions, 7 deletions

diff --git a/test/e2e/beta/from-import-beta-ui.spec.js b/test/e2e/beta/from-import-beta-ui.spec.js
index d6e976dea..cd7d90e95 100644
--- a/test/e2e/beta/from-import-beta-ui.spec.js
+++ b/test/e2e/beta/from-import-beta-ui.spec.js
@@ -9,17 +9,16 @@ const {
   verboseReportOnFailure,
 } = require('../func')
 const {
-  checkBrowserForConsoleErrors,
-  loadExtension,
   findElement,
   findElements,
+  checkBrowserForConsoleErrors,
 } = require('./helpers')
 
 
 describe('Using MetaMask with an existing account', function () {
   const browser = process.env.SELENIUM_BROWSER
-  let extensionId
   let driver
+  let extensionUri
   let tokenAddress
 
   const testSeedPhrase = 'phrase upgrade clock rough situate wedding elder clever doctor stamp excess tent'
@@ -273,7 +272,7 @@ describe('Using MetaMask with an existing account', function () {
       await delay(regularDelayMs)
 
       await driver.switchTo().window(extension)
-      await loadExtension(driver, extensionId)
+      await driver.get(extensionUri)
       await delay(regularDelayMs)
 
       const confirmButton = await findElement(driver, By.xpath(`//button[contains(text(), 'Confirm')]`), 14000)
@@ -286,7 +285,7 @@ describe('Using MetaMask with an existing account', function () {
       await delay(regularDelayMs)
       await driver.switchTo().window(extension)
       await delay(regularDelayMs)
-      await loadExtension(driver, extensionId)
+      await driver.get(extensionUri)
       await delay(regularDelayMs)
     })
   })
@@ -348,7 +347,7 @@ describe('Using MetaMask with an existing account', function () {
       await delay(regularDelayMs)
 
       await driver.switchTo().window(extension)
-      await loadExtension(driver, extensionId)
+      await driver.get(extensionUri)
       await delay(regularDelayMs)
 
       const confirmButton = await findElement(driver, By.xpath(`//button[contains(text(), 'Confirm')]`))
@@ -361,7 +360,7 @@ describe('Using MetaMask with an existing account', function () {
       tokenAddress = await tokenContactAddress.getText()
       await driver.close()
       await driver.switchTo().window(extension)
-      await loadExtension(driver, extensionId)
+      await driver.get(extensionUri)
       await delay(regularDelayMs)
     })