cleanup injected js

greggman · Jul 2, 2024 · 6dac51e · 6dac51e
1 parent 91d4ea7
commit 6dac51e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/test/puppeteer.js b/test/puppeteer.js
@@ -73,9 +73,9 @@ async function test(port) {
   for (const {url, js, screenshot} of testPages) {
     waitingPromiseInfo = makePromiseInfo();
     console.log(`===== [ ${url} ] =====`);
-    if (js) {
-      await page.evaluateOnNewDocument(js);
-    }
+    const id = js
+      ? await page.evaluateOnNewDocument(js)
+      : undefined;
     await page.goto(url);
     await page.waitForNetworkIdle();
     if (js) {
@@ -93,6 +93,9 @@ async function test(port) {
       const path = `${dir}/${name}.png`;
       await page.screenshot({path});
     }
+    if (id) {
+      await page.removeScriptToEvaluateOnNewDocument(id.identifier);
+    }
   }
 
   await browser.close();