fix: correctly resolve external dependencies loaded by custom environments (#4196)

Author: sheremet-va

packages/vitest/src/runtime/child.ts

@@ -39,9 +39,8 @@ async function init(ctx: ChildContext) {
 
   const environment = await loadEnvironment(ctx.environment.name, {
     root: ctx.config.root,
-    fetchModule(id) {
-      return rpc.fetch(id, 'ssr')
-    },
+    fetchModule: id => rpc.fetch(id, 'ssr'),
+    resolveId: (id, importer) => rpc.resolveId(id, importer, 'ssr'),
   })
   if (ctx.environment.transformMode)
     environment.transformMode = ctx.environment.transformMode

packages/vitest/src/runtime/vm.ts

@@ -39,9 +39,8 @@ export async function run(ctx: WorkerContext) {
 
   const environment = await loadEnvironment(ctx.environment.name, {
     root: ctx.config.root,
-    fetchModule(id) {
-      return rpc.fetch(id, 'ssr')
-    },
+    fetchModule: id => rpc.fetch(id, 'ssr'),
+    resolveId: (id, importer) => rpc.resolveId(id, importer, 'ssr'),
   })
 
   if (!environment.setupVM) {

packages/vitest/src/runtime/worker.ts

@@ -37,9 +37,8 @@ async function init(ctx: WorkerContext) {
 
   const environment = await loadEnvironment(ctx.environment.name, {
     root: ctx.config.root,
-    fetchModule(id) {
-      return rpc.fetch(id, 'ssr')
-    },
+    fetchModule: id => rpc.fetch(id, 'ssr'),
+    resolveId: (id, importer) => rpc.resolveId(id, importer, 'ssr'),
   })
   if (ctx.environment.transformMode)
     environment.transformMode = ctx.environment.transformMode

pnpm-lock.yaml

@@ -6,6 +6,8 @@
     "coverage": "vitest run --coverage"
   },
   "devDependencies": {
+    "@types/debug": "^4.1.8",
+    "debug": "^4.3.4",
     "vitest": "workspace:*",
     "vitest-environment-custom": "file:./vitest-environment-custom"
   }

test/env-custom/package.json

@@ -1,6 +1,10 @@
 import vm from 'node:vm'
+import debug from 'debug'
 import type { Environment } from 'vitest'
 
+// test that external packages (debug) are loaded correctly
+const log = debug('test:env')
+
 export default <Environment>{
   name: 'custom',
   transformMode: 'ssr',
@@ -28,6 +32,9 @@ export default <Environment>{
       teardown() {
         delete global.testEnvironment
         delete global.option
+
+        if (global.__exists)
+          log('should not log')
       },
     }
   },