development - screens - use localhost for network

2 files changed, 20 insertions, 6 deletions

diff --git a/test/screens/new.spec.js b/test/screens/new.spec.js
index 14daac1c1..2af891060 100644
--- a/test/screens/new.spec.js
+++ b/test/screens/new.spec.js
@@ -35,6 +35,7 @@ async function captureAllScreens() {
   tabs = await driver.getAllWindowHandles()
   await driver.switchTo().window(tabs[0])
   await delay(300)
+  await setProviderType('localhost')
 
   // click try new ui
   await driver.findElement(By.css('#app-content > div > div.app-primary.from-right > div > div.flex-row.flex-center.flex-grow > p')).click()
@@ -42,8 +43,12 @@ async function captureAllScreens() {
 
   // close metamask homepage and extra home.html
   tabs = await driver.getAllWindowHandles()
-  await driver.switchTo().window(tabs[2])
-  driver.close()
+  console.log(tabs)
+  // metamask homepage is opened on prod, not dev
+  if (tabs.length > 2) {
+    await driver.switchTo().window(tabs[2])
+    driver.close()
+  }
   await driver.switchTo().window(tabs[1])
   driver.close()
   await driver.switchTo().window(tabs[0])
@@ -159,7 +164,11 @@ async function captureAllScreens() {
   }
 
   async function setLocale(code) {
-    await driver.executeScript('setLocale(arguments[0])', code)
+    await driver.executeScript('window.metamask.updateCurrentLocale(arguments[0])', code)
+  }
+
+  async function setProviderType(type) {
+    await driver.executeScript('window.metamask.setProviderType(arguments[0])', type)
   }
 
   // cleanup
diff --git a/ui/index.js b/ui/index.js
index 8fb000d85..746e28eab 100644
--- a/ui/index.js
+++ b/ui/index.js
@@ -69,9 +69,14 @@ async function startApp (metamaskState, accountManager, opts) {
     store.dispatch(actions.updateMetamaskState(metamaskState))
   })
 
-  // used by screenshotter tooling
-  global.setLocale = (key) => {
-    store.dispatch(actions.updateCurrentLocale(key))
+  // global metamask api - used by tooling
+  global.metamask = {
+    updateCurrentLocale: (code) => {
+      store.dispatch(actions.updateCurrentLocale(code))
+    },
+    setProviderType: (type) => {
+      store.dispatch(actions.setProviderType(type))
+    },
   }
 
   // start app