cleanup injected js

Author: greggman

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();