From 059c8aa61e7e9ca935a657aa3f9cf17e3d87a2e8 Mon Sep 17 00:00:00 2001
From: "yanlong.wang" <yanlong.wang@naiver.org>
Date: Mon, 29 Apr 2024 15:51:23 +0800
Subject: [PATCH] fix: remove exposed function before cleanup

---
 backend/functions/src/services/puppeteer.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/backend/functions/src/services/puppeteer.ts b/backend/functions/src/services/puppeteer.ts
index 7a4c1bf..45031bc 100644
--- a/backend/functions/src/services/puppeteer.ts
+++ b/backend/functions/src/services/puppeteer.ts
@@ -82,6 +82,7 @@ export class PuppeteerControl extends AsyncService {
             return page;
         },
         destroy: async (page) => {
+            await page.removeExposedFunction('reportSnapshot');
             await page.browserContext().close();
         },
         validate: async (page) => {
@@ -198,6 +199,9 @@ function giveSnapshot() {
         await page.evaluateOnNewDocument(`
 let aftershot = undefined;
 const handlePageLoad = () => {
+    if (window.haltSnapshot) {
+        return;
+    }
     if (document.readyState !== 'complete') {
         return;
     }