detect vitest

Author: upsuper

packages/vite/src/node/plugins/wasm.ts

@@ -67,9 +67,17 @@ const instantiateFromFile = async (
 const instantiateFromFileCode = instantiateFromFile.toString()
 
 export const wasmHelperPlugin = (): Plugin => {
+  // Vitest may use the client environment to run tests in Node.js.
+  // Try to detect it explicitly.
+  let isVitest = false
+
   return {
     name: 'vite:wasm-helper',
 
+    configResolved(config) {
+      isVitest = config.plugins.some((plugin) => plugin.name === 'vitest')
+    },
+
     resolveId: {
       filter: { id: exactRegex(wasmHelperId) },
       handler(id) {
@@ -80,10 +88,10 @@ export const wasmHelperPlugin = (): Plugin => {
     load: {
       filter: { id: [exactRegex(wasmHelperId), wasmInitRE] },
       async handler(id) {
-        const isServer = this.environment.config.consumer === 'server'
+        const isNode = this.environment.config.consumer === 'server' || isVitest
 
         if (id === wasmHelperId) {
-          const instantiateFromUrl = isServer
+          const instantiateFromUrl = isNode
             ? instantiateFromFileCode
             : instantiateFromUrlCode
           return `
@@ -93,7 +101,7 @@ export default ${wasmHelperCode}
         }
 
         id = id.split('?')[0]
-        const url = isServer ? fsPathFromId(id) : await fileToUrl(this, id)
+        const url = isNode ? fsPathFromId(id) : await fileToUrl(this, id)
 
         return `
   import initWasm from "${wasmHelperId}"