Add Base.close()

battis · Dec 29, 2024 · e66ac2e · e66ac2e
1 parent 6e7e9f8
commit e66ac2e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/datadirect-puppeteer/src/PuppeteerSession/Base.ts b/packages/datadirect-puppeteer/src/PuppeteerSession/Base.ts
@@ -120,4 +120,12 @@ export class Base extends EventEmitter {
       init
     );
   }
+
+  public async close() {
+    if (this.page.browser().pages.length === 1) {
+      await this.page.browser().close();
+    } else {
+      await this.page.close();
+    }
+  }
 }