Fix/selenium (#461)

WebKit · Dec 9, 2024 · e2e3c49 · e2e3c49
1 parent 67bc21f
commit e2e3c49
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/tests/index.html b/tests/index.html
@@ -64,11 +64,13 @@
 
                 runner.on("end", function () {
                     window.suite = createReport(runner.suite);
-                    const event = new Event("test-complete");
-                    window.dispatchEvent(event);
+                    window.dispatchEvent(new Event("test-complete"));
                 });
             }
-            window.addEventListener("start-test", startTest);
+
+            window.addEventListener("start-test", () => startTest(), { once: true });
+            window.benchmarkReady = true;
+            window.dispatchEvent(new Event("benchmark-ready"));
         </script>
     </body>
 </html>
diff --git a/tests/run.mjs b/tests/run.mjs
@@ -95,9 +95,16 @@ function printTree(node) {
 
 async function test() {
     driver = await new Builder().withCapabilities(capabilities).build();
-
     try {
         await driver.get(`http://localhost:${PORT}/tests/index.html`);
+
+        await driver.executeAsyncScript((callback) => {
+            if (window.benchmarkReady)
+                callback();
+
+            window.addEventListener("benchmark-ready", () => callback(), { once: true });
+        });
+
         const result = await driver.executeAsyncScript(function (callback) {
             window.addEventListener(
                 "test-complete",
@@ -108,8 +115,7 @@ async function test() {
                     }),
                 { once: true }
             );
-            const event = new Event("start-test");
-            window.dispatchEvent(event);
+            window.dispatchEvent(new Event("start-test"));
         });
 
         printTree(result.suite);