feat(jest-environment-puppeteer): Add ability to connect to an alread…

…y existing instance of Chrome
argos-ci · Aug 19, 2018 · 9de05f0 · 9de05f0
1 parent 89e72b7
commit 9de05f0
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 6 deletions.
diff --git a/packages/jest-environment-puppeteer/src/PuppeteerEnvironment.js b/packages/jest-environment-puppeteer/src/PuppeteerEnvironment.js
@@ -10,6 +10,27 @@ const handleError = error => {
   process.emit('uncaughtException', error)
 }
 
+const getBrowser = async config => {
+  const wsEndpoint = fs.readFileSync(WS_ENDPOINT_PATH, 'utf8')
+  let slowMo
+  let ignoreHTTPSErrors
+
+  if (!wsEndpoint) {
+    throw new Error('wsEndpoint not found')
+  }
+
+  if (config) {
+    slowMo = (config.connect && config.connect.slowMo) || (config.launch && config.launch.slowMo) 
+    ignoreHTTPSErrors = (config.connect && config.connect.ignoreHTTPSErrors) || (config.launch && config.launch.ignoreHTTPSErrors) 
+  }
+
+  return puppeteer.connect({
+    slowMo,
+    ignoreHTTPSErrors,
+    browserWSEndpoint: wsEndpoint,
+  })
+}
+
 const KEYS = {
   CONTROL_C: '\u0003',
   CONTROL_D: '\u0004',
@@ -36,10 +57,7 @@ class PuppeteerEnvironment extends NodeEnvironment {
     if (!wsEndpoint) {
       throw new Error('wsEndpoint not found')
     }
-    this.global.browser = await puppeteer.connect({
-      ...config.launch,
-      browserWSEndpoint: wsEndpoint,
-    })
+    this.global.browser = await getBrowser(config)
     this.global.page = await this.global.browser.newPage()
     if (config && config.exitOnPageError) {
       this.global.page.addListener('pageerror', handleError)

diff --git a/packages/jest-environment-puppeteer/src/global.js b/packages/jest-environment-puppeteer/src/global.js
@@ -17,7 +17,11 @@ let browser
 
 export async function setup() {
   const config = await readConfig()
-  browser = await puppeteer.launch(config.launch)
+  if (config.connect) {
+    browser = await puppeteer.connect(config.connect)
+  } else {
+    browser = await puppeteer.launch(config.launch)
+  }
   mkdirp.sync(DIR)
   fs.writeFileSync(WS_ENDPOINT_PATH, browser.wsEndpoint())
 

diff --git a/packages/jest-environment-puppeteer/src/readConfig.js b/packages/jest-environment-puppeteer/src/readConfig.js
@@ -38,7 +38,13 @@ async function readConfig() {
   }
 
   // eslint-disable-next-line global-require, import/no-dynamic-require
-  const localConfig = require(absConfigPath)
+  const requiredConfig = require(absConfigPath)
+  let localConfig
+  if (typeof requiredConfig === 'function') {
+    localConfig = await requiredConfig()
+  } else {
+    localConfig = requiredConfig
+  }
   return merge({}, defaultConfig, localConfig)
 }