Tweak puppeteer script.

Briskine · Sep 24, 2024 · 58d8506 · 58d8506
1 parent 8f721d1
commit 58d8506
Showing 1 changed file with 17 additions and 15 deletions.
diff --git a/test/test-server.js b/test/test-server.js
@@ -160,22 +160,24 @@ function run () {
 
   puppeteer
     .launch(options)
-    .then(browser => browser.pages()
-    .then(pages => pages.pop())
-    .then(page => {
-      page.on('console', handleConsole)
-      page.on('dialog', dialog => dialog.dismiss())
-      page.on('pageerror', err => console.error(err))
-
-      return page.evaluateOnNewDocument(initMocha)
-        .then(() => page.goto(url))
-        .then(() => page.waitForFunction(() => window.__mochaResult__, { timeout }))
-        .then(() => page.evaluate(() => window.__mochaResult__))
-        .then(obj => {
-          browser.close()
-          return obj
+    .then(browser => {
+      return browser.pages()
+        .then(pages => pages.pop())
+        .then(page => {
+          page.on('console', handleConsole)
+          page.on('dialog', dialog => dialog.dismiss())
+          page.on('pageerror', err => console.error(err))
+
+          return page.evaluateOnNewDocument(initMocha)
+            .then(() => page.goto(url))
+            .then(() => page.waitForFunction(() => window.__mochaResult__, { timeout }))
+            .then(() => page.evaluate(() => window.__mochaResult__))
+            .then(obj => {
+              browser.close()
+              return obj
+            })
         })
-    }))
+    })
 }
 
 run()