Add executionContext weakmap bench attempt

bench.js

@@ -3,8 +3,9 @@
 const path = require("path");
 const Benchmark = require("benchmark");
 const puppeteer = require("puppeteer");
-const { queries: currentQueries } = require("./current");
-const { queries: newQueries, getDocument } = require("./");
+const { queries: baseQueries, getDocument } = require("./versions/base");
+const { queries: windowQueries } = require("./versions/window");
+const { queries: execQueries } = require("./versions/executionContext");
 
 var suite = new Benchmark.Suite("pptr-testing-library");
 
@@ -17,19 +18,28 @@ async function main() {
   // add tests
   suite
     .add({
-      name: "current findByRole",
+      name: "base",
       defer: true,
       fn(deferred) {
-        currentQueries
+        baseQueries
           .getByText(document, "Hello h1")
           .then(() => deferred.resolve());
       },
     })
     .add({
-      name: "new findByRole",
+      name: "window check",
       defer: true,
       fn(deferred) {
-        newQueries
+        windowQueries
+          .getByText(document, "Hello h1")
+          .then(() => deferred.resolve());
+      },
+    })
+    .add({
+      name: "execution context",
+      defer: true,
+      fn(deferred) {
+        execQueries
           .getByText(document, "Hello h1")
           .then(() => deferred.resolve());
       },

lib/index.ts

@@ -1,6 +1,6 @@
 import {readFileSync} from 'fs'
 import * as path from 'path'
-import {ElementHandle, EvaluateFn, JSHandle, Page} from 'puppeteer'
+import {ElementHandle, EvaluateFn, ExecutionContext, JSHandle, Page} from 'puppeteer'
 import waitForExpect from 'wait-for-expect'
 
 import {IConfigureOptions, IQueryUtils, IScopedQueryUtils} from './typedefs'
@@ -117,6 +117,8 @@ interface IHandleSet {
   argsToForward: any[]
 }
 
+const executionContextInit = new WeakMap<ExecutionContext, boolean>();
+
 function createDelegateFor<T = DOMReturnType>(
   fnName: keyof IQueryUtils,
   contextFn?: ContextFn,
@@ -129,9 +131,14 @@ function createDelegateFor<T = DOMReturnType>(
     // @ts-ignore
     const containerHandle: ElementHandle = contextFn ? contextFn.apply(this, args) : this
 
-    const shouldIncludeTestingLib = await containerHandle.evaluate(
-      () => (window as any).__dom_testing_library__ == null
-    );
+    // const shouldIncludeTestingLib = await containerHandle.evaluate(
+    //   () => (window as any).__dom_testing_library__ == null
+    // );
+    const executionContext = containerHandle.executionContext();
+    const shouldIncludeTestingLib = !executionContextInit.has(executionContext);
+    if (shouldIncludeTestingLib) {
+      executionContextInit.set(executionContext, true);
+    }
 
     // @ts-ignore
     const evaluateFn: EvaluateFn = new Function(