From a3056f4125d3607f7d6592ddc9934ecc78c1adf6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C3=ABl=20Zasso?= <targos@protonmail.com>
Date: Thu, 19 Jan 2023 01:14:07 +0100
Subject: [PATCH] test: refactor to avoid mutation of global by a loader

This makes the test compatible with off-thread loaders.

Co-Authored-By: Geoffrey Booth <webadmin@geoffreybooth.com>
PR-URL: https://github.com/nodejs/node/pull/46220
Reviewed-By: Antoine du Hamel <duhamelantoine1995@gmail.com>
Reviewed-By: Geoffrey Booth <webadmin@geoffreybooth.com>
---
 .../test-esm-loader-resolve-type.mjs          | 32 ++++++++++---------
 .../es-module-loaders/hook-resolve-type.mjs   | 25 ++++++++++++++-
 2 files changed, 41 insertions(+), 16 deletions(-)

diff --git a/test/es-module/test-esm-loader-resolve-type.mjs b/test/es-module/test-esm-loader-resolve-type.mjs
index 722cf5404d25de..482320c664c5d8 100644
--- a/test/es-module/test-esm-loader-resolve-type.mjs
+++ b/test/es-module/test-esm-loader-resolve-type.mjs
@@ -6,6 +6,9 @@ import * as fs from 'fs';
 
 allowGlobals(global.getModuleTypeStats);
 
+const { importedESM: importedESMBefore,
+        importedCJS: importedCJSBefore } = await global.getModuleTypeStats();
+
 const basePath =
   new URL('./node_modules/', import.meta.url);
 
@@ -17,25 +20,24 @@ const createDir = (path) => {
 };
 
 const moduleName = 'module-counter-by-type';
-
 const moduleDir = rel(`${moduleName}`);
-createDir(basePath);
-createDir(moduleDir);
-fs.cpSync(
-  fixtures.path('es-modules', moduleName),
-  moduleDir,
-  { recursive: true }
-);
-
-const { importedESM: importedESMBefore,
-        importedCJS: importedCJSBefore } = global.getModuleTypeStats();
-
-await import(`${moduleName}`).finally(() => {
+try {
+  createDir(basePath);
+  createDir(moduleDir);
+  fs.cpSync(
+    fixtures.path('es-modules', moduleName),
+    moduleDir,
+    { recursive: true }
+  );
+
+
+  await import(`${moduleName}`);
+} finally {
   fs.rmSync(basePath, { recursive: true, force: true });
-});
+}
 
 const { importedESM: importedESMAfter,
-        importedCJS: importedCJSAfter } = global.getModuleTypeStats();
+        importedCJS: importedCJSAfter } = await global.getModuleTypeStats();
 
 // Dynamic import above should increment ESM counter but not CJS counter
 assert.strictEqual(importedESMBefore + 1, importedESMAfter);
diff --git a/test/fixtures/es-module-loaders/hook-resolve-type.mjs b/test/fixtures/es-module-loaders/hook-resolve-type.mjs
index b1f5606c0e4a0a..a4d87938ad843f 100644
--- a/test/fixtures/es-module-loaders/hook-resolve-type.mjs
+++ b/test/fixtures/es-module-loaders/hook-resolve-type.mjs
@@ -1,6 +1,29 @@
 let importedESM = 0;
 let importedCJS = 0;
-global.getModuleTypeStats = () => { return {importedESM, importedCJS} };
+
+export function globalPreload({ port }) {
+  port.on('message', (int32) => {
+    port.postMessage({ importedESM, importedCJS });
+    Atomics.store(int32, 0, 1);
+    Atomics.notify(int32, 0);
+  });
+  port.unref();
+  return `
+    const { receiveMessageOnPort } = getBuiltin('worker_threads');
+    global.getModuleTypeStats = async function getModuleTypeStats() {
+      const sab = new SharedArrayBuffer(4);
+      const int32 = new Int32Array(sab);
+      port.postMessage(int32);
+      // Artificial timeout to keep the event loop alive.
+      // https://bugs.chromium.org/p/v8/issues/detail?id=13238
+      // TODO(targos) Remove when V8 issue is resolved.
+      const timeout = setTimeout(() => { throw new Error('timeout'); }, 1_000);
+      await Atomics.waitAsync(int32, 0, 0).value;
+      clearTimeout(timeout);
+      return receiveMessageOnPort(port).message;
+    };
+  `;
+}
 
 export async function load(url, context, next) {
   return next(url);