|
import { pathToFileURL } from 'node:url'; |
|
import { workerId } from 'tinypool'; |
|
import { ViteNodeRunner, ModuleCacheMap } from 'vite-node/client'; |
|
import { readFileSync } from 'node:fs'; |
|
import { resolve, normalize } from 'pathe'; |
|
import { e as environments } from './vendor/index.GVFv9dZ0.js'; |
|
import { i as isChildProcess, s as setProcessTitle } from './vendor/base.5NT-gWu5.js'; |
|
import { s as setupInspect } from './vendor/inspector.IgLX3ur5.js'; |
|
import { c as createRuntimeRpc, a as rpcDone } from './vendor/rpc.joBhAkyK.js'; |
|
import 'node:console'; |
|
import '@vitest/utils'; |
|
import 'node:module'; |
|
import './vendor/index.8bPxjt7g.js'; |
|
import './vendor/global.CkGT_TMy.js'; |
|
|
|
function isBuiltinEnvironment(env) { |
|
return env in environments; |
|
} |
|
const _loaders = new Map(); |
|
async function createEnvironmentLoader(options) { |
|
if (!_loaders.has(options.root)) { |
|
const loader = new ViteNodeRunner(options); |
|
await loader.executeId("/@vite/env"); |
|
_loaders.set(options.root, loader); |
|
} |
|
return _loaders.get(options.root); |
|
} |
|
async function loadEnvironment(ctx, rpc) { |
|
var _a; |
|
const name = ctx.environment.name; |
|
if (isBuiltinEnvironment(name)) |
|
return environments[name]; |
|
const loader = await createEnvironmentLoader({ |
|
root: ctx.config.root, |
|
fetchModule: async (id) => { |
|
const result = await rpc.fetch(id, "ssr"); |
|
if (result.id) |
|
return { code: readFileSync(result.id, "utf-8") }; |
|
return result; |
|
}, |
|
resolveId: (id, importer) => rpc.resolveId(id, importer, "ssr") |
|
}); |
|
const root = loader.root; |
|
const packageId = name[0] === "." || name[0] === "/" ? resolve(root, name) : ((_a = await rpc.resolveId(`vitest-environment-${name}`, void 0, "ssr")) == null ? void 0 : _a.id) ?? resolve(root, name); |
|
const pkg = await loader.executeId(normalize(packageId)); |
|
if (!pkg || !pkg.default || typeof pkg.default !== "object") { |
|
throw new TypeError( |
|
`Environment "${name}" is not a valid environment. Path "${packageId}" should export default object with a "setup" or/and "setupVM" method.` |
|
); |
|
} |
|
const environment = pkg.default; |
|
if (environment.transformMode !== "web" && environment.transformMode !== "ssr") { |
|
throw new TypeError( |
|
`Environment "${name}" is not a valid environment. Path "${packageId}" should export default object with a "transformMode" method equal to "ssr" or "web".` |
|
); |
|
} |
|
return environment; |
|
} |
|
|
|
if (isChildProcess()) |
|
setProcessTitle(`vitest ${workerId}`); |
|
async function run(ctx) { |
|
const prepareStart = performance.now(); |
|
const inspectorCleanup = setupInspect(ctx); |
|
process.env.VITEST_WORKER_ID = String(ctx.workerId); |
|
process.env.VITEST_POOL_ID = String(workerId); |
|
try { |
|
if (ctx.worker[0] === ".") |
|
throw new Error(`Path to the test runner cannot be relative, received "${ctx.worker}"`); |
|
const file = ctx.worker.startsWith("file:") ? ctx.worker : pathToFileURL(ctx.worker).toString(); |
|
const testRunnerModule = await import(file); |
|
if (!testRunnerModule.default || typeof testRunnerModule.default !== "object") |
|
throw new TypeError(`Test worker object should be exposed as a default export. Received "${typeof testRunnerModule.default}"`); |
|
const worker = testRunnerModule.default; |
|
if (!worker.getRpcOptions || typeof worker.getRpcOptions !== "function") |
|
throw new TypeError(`Test worker should expose "getRpcOptions" method. Received "${typeof worker.getRpcOptions}".`); |
|
const { rpc, onCancel } = createRuntimeRpc(worker.getRpcOptions(ctx)); |
|
const beforeEnvironmentTime = performance.now(); |
|
const environment = await loadEnvironment(ctx, rpc); |
|
if (ctx.environment.transformMode) |
|
environment.transformMode = ctx.environment.transformMode; |
|
const state = { |
|
ctx, |
|
|
|
moduleCache: new ModuleCacheMap(), |
|
mockMap: new Map(), |
|
config: ctx.config, |
|
onCancel, |
|
environment, |
|
durations: { |
|
environment: beforeEnvironmentTime, |
|
prepare: prepareStart |
|
}, |
|
rpc, |
|
providedContext: ctx.providedContext |
|
}; |
|
if (!worker.runTests || typeof worker.runTests !== "function") |
|
throw new TypeError(`Test worker should expose "runTests" method. Received "${typeof worker.runTests}".`); |
|
await worker.runTests(state); |
|
} finally { |
|
await rpcDone().catch(() => { |
|
}); |
|
inspectorCleanup(); |
|
} |
|
} |
|
|
|
export { run }; |
|
|