test_runner: support module detection in module mocks

PR-URL: #53642 Reviewed-By: Colin Ihrig <[email protected]> Reviewed-By: Chemi Atlow <[email protected]> Reviewed-By: Benjamin Gruenbaum <[email protected]> Reviewed-By: James M Snell <[email protected]>
nodejs · Jul 16, 2024 · e6a1eeb · e6a1eeb
1 parent ac9c2e6
commit e6a1eeb
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/lib/internal/test_runner/mock/mock.js b/lib/internal/test_runner/mock/mock.js
@@ -513,7 +513,9 @@ class MockTracker {
       mockSpecifier, caller, null,
     );
     debug('module mock, url = "%s", format = "%s", caller = "%s"', url, format, caller);
-    validateOneOf(format, 'format', kSupportedFormats);
+    if (format) { // Format is not yet known for ambiguous files when detection is enabled.
+      validateOneOf(format, 'format', kSupportedFormats);
+    }
     const baseURL = URL.parse(url);
 
     if (!baseURL) {

diff --git a/lib/test/mock_loader.js b/lib/test/mock_loader.js
@@ -143,26 +143,34 @@ async function load(url, context, nextLoad) {
   const baseURL = parsedURL ? parsedURL.href : url;
   const mock = mocks.get(baseURL);
 
+  const original = await nextLoad(url, context);
   debug('load hook, mock = %o', mock);
   if (mock?.active !== true) {
-    return nextLoad(url);
+    return original;
   }
 
   // Treat builtins as commonjs because customization hooks do not allow a
   // core module to be replaced.
-  const format = mock.format === 'builtin' ? 'commonjs' : mock.format;
+  // Also collapse 'commonjs-sync' and 'require-commonjs' to 'commonjs'.
+  const format = (
+    original.format === 'builtin' ||
+    original.format === 'commonjs-sync' ||
+    original.format === 'require-commonjs') ? 'commonjs' : original.format;
 
-  return {
+  const result = {
     __proto__: null,
     format,
     shortCircuit: true,
-    source: await createSourceFromMock(mock),
+    source: await createSourceFromMock(mock, format),
   };
+
+  debug('load hook finished, result = %o', result);
+  return result;
 }
 
-async function createSourceFromMock(mock) {
+async function createSourceFromMock(mock, format) {
   // Create mock implementation from provided exports.
-  const { exportNames, format, hasDefaultExport, url } = mock;
+  const { exportNames, hasDefaultExport, url } = mock;
   const useESM = format === 'module';
   const source = `${testImportSource(useESM)}
 if (!$__test.mock._mockExports.has('${url}')) {