@jsenv/test 3.7.35 → 3.7.37

package/src/runtime_browsers/using_playwright.js

@@ -45,6 +45,7 @@ export const createRuntimeUsingPlaywright = ({
     onConsole,
     onRuntimeStarted,
     onRuntimeStopped,
+    onAllocatedMsRequested = () => {},
     teardownCallbackSet,
     isTestPlan,
 
@@ -166,6 +167,11 @@ ${webServer.rootDirectoryUrl}`);
     }
 
     const page = await browserContext.newPage();
+    // the only thing the page can tell node before it is done executing;
+    // see runtime_browsers/client/request_allocated_ms.js
+    await page.exposeFunction("__jsenv_request_allocated_ms__", (ms) => {
+      onAllocatedMsRequested(ms);
+    });
     if (!isBrowserDedicatedToExecution) {
       page.on("close", () => {
         onRuntimeStopped();

package/src/runtime_node/node_child_process.js

@@ -55,6 +55,7 @@ export const nodeChildProcess = ({
       onConsole,
       onRuntimeStarted,
       onRuntimeStopped,
+      onAllocatedMsRequested = () => {},
 
       measureMemoryUsage,
       onMeasureMemoryAvailable,
@@ -162,6 +163,15 @@ export const nodeChildProcess = ({
           onRuntimeStopped();
         }),
       );
+      cleanupCallbackSet.add(
+        onChildProcessMessage(
+          childProcess,
+          "allocated-ms-request",
+          ({ ms }) => {
+            onAllocatedMsRequested(ms);
+          },
+        ),
+      );
 
       const removeOutputListener = installChildProcessOutputListener(
         childProcess,

package/src/runtime_node/node_worker_thread.js

@@ -57,6 +57,7 @@ export const nodeWorkerThread = ({
       onConsole,
       onRuntimeStarted,
       onRuntimeStopped,
+      onAllocatedMsRequested = () => {},
 
       measureMemoryUsage,
       onMeasureMemoryAvailable,
@@ -143,6 +144,15 @@ export const nodeWorkerThread = ({
           onRuntimeStopped();
         }),
       );
+      cleanupCallbackSet.add(
+        onWorkerThreadMessage(
+          workerThread,
+          "allocated-ms-request",
+          ({ ms }) => {
+            onAllocatedMsRequested(ms);
+          },
+        ),
+      );
 
       const stop = memoize(async () => {
         // read all stdout before terminating

package/src/runtime_node/request_allocated_ms.js

@@ -0,0 +1,36 @@
+/*
+ * Asks the test runner for more time from a test file, when the amount is
+ * computed rather than known in advance (it depends on the platform, on how
+ * many fixtures were found...). A fixed amount belongs in a directive instead,
+ * which the runner reads without executing the file:
+ *
+ *   "jsenv:allocate 90s";
+ *
+ * The request is sent to the process running the test plan, which restarts the
+ * timeout with the requested duration and remembers it: a file asking for more
+ * time than the others is also a file worth starting early when parallelizing.
+ *
+ * Node.js runtimes only; a browser has no channel to reach the test plan while
+ * the file is executing.
+ */
+
+import { parentPort } from "node:worker_threads";
+
+export const requestAllocatedMs = (ms) => {
+  if (typeof ms !== "number") {
+    throw new TypeError(`requestAllocatedMs expects a number, got ${ms}`);
+  }
+  const message = {
+    __jsenv__: "allocated-ms-request",
+    data: JSON.stringify({ ms }),
+  };
+  if (parentPort) {
+    parentPort.postMessage(message);
+    return;
+  }
+  if (process.send && process.connected) {
+    process.send(message);
+    return;
+  }
+  // file executed on its own (node ./file.test.mjs): there is no allocated time
+};