722 lines
26 KiB
JavaScript
722 lines
26 KiB
JavaScript
import fs from 'node:fs';
|
|
import { isBareImport } from '@vitest/utils/helpers';
|
|
import { i as isBuiltin, a as isBrowserExternal, t as toBuiltin } from './modules.BJuCwlRJ.js';
|
|
import { E as EnvironmentTeardownError, a as getSafeWorkerState } from './utils.BX5Fg8C4.js';
|
|
import { pathToFileURL } from 'node:url';
|
|
import { normalize, join } from 'pathe';
|
|
import { distDir } from '../path.js';
|
|
import { VitestModuleEvaluator, unwrapId } from '../module-evaluator.js';
|
|
import { isAbsolute, resolve } from 'node:path';
|
|
import vm from 'node:vm';
|
|
import { MockerRegistry, mockObject, RedirectedModule, AutomockedModule } from '@vitest/mocker';
|
|
import { findMockRedirect } from '@vitest/mocker/redirect';
|
|
import * as viteModuleRunner from 'vite/module-runner';
|
|
import { T as Traces } from './traces.CCmnQaNT.js';
|
|
|
|
class BareModuleMocker {
|
|
static pendingIds = [];
|
|
spyModule;
|
|
primitives;
|
|
registries = /* @__PURE__ */ new Map();
|
|
mockContext = { callstack: null };
|
|
_otel;
|
|
constructor(options) {
|
|
this.options = options;
|
|
this._otel = options.traces;
|
|
this.primitives = {
|
|
Object,
|
|
Error,
|
|
Function,
|
|
RegExp,
|
|
Symbol: globalThis.Symbol,
|
|
Array,
|
|
Map
|
|
};
|
|
if (options.spyModule) this.spyModule = options.spyModule;
|
|
}
|
|
get root() {
|
|
return this.options.root;
|
|
}
|
|
get moduleDirectories() {
|
|
return this.options.moduleDirectories || [];
|
|
}
|
|
getMockerRegistry() {
|
|
const suite = this.getSuiteFilepath();
|
|
if (!this.registries.has(suite)) this.registries.set(suite, new MockerRegistry());
|
|
return this.registries.get(suite);
|
|
}
|
|
reset() {
|
|
this.registries.clear();
|
|
}
|
|
invalidateModuleById(_id) {
|
|
// implemented by mockers that control the module runner
|
|
}
|
|
isModuleDirectory(path) {
|
|
return this.moduleDirectories.some((dir) => path.includes(dir));
|
|
}
|
|
getSuiteFilepath() {
|
|
return this.options.getCurrentTestFilepath() || "global";
|
|
}
|
|
createError(message, codeFrame) {
|
|
const Error = this.primitives.Error;
|
|
const error = new Error(message);
|
|
Object.assign(error, { codeFrame });
|
|
return error;
|
|
}
|
|
async resolveId(rawId, importer) {
|
|
return this._otel.$("vitest.mocker.resolve_id", { attributes: {
|
|
"vitest.module.raw_id": rawId,
|
|
"vitest.module.importer": rawId
|
|
} }, async (span) => {
|
|
const result = await this.options.resolveId(rawId, importer);
|
|
if (!result) {
|
|
span.addEvent("could not resolve id, fallback to unresolved values");
|
|
const id = normalizeModuleId(rawId);
|
|
span.setAttributes({
|
|
"vitest.module.id": id,
|
|
"vitest.module.url": rawId,
|
|
"vitest.module.external": id,
|
|
"vitest.module.fallback": true
|
|
});
|
|
return {
|
|
id,
|
|
url: rawId,
|
|
external: id
|
|
};
|
|
}
|
|
// external is node_module or unresolved module
|
|
// for example, some people mock "vscode" and don't have it installed
|
|
const external = !isAbsolute(result.file) || this.isModuleDirectory(result.file) ? normalizeModuleId(rawId) : null;
|
|
const id = normalizeModuleId(result.id);
|
|
span.setAttributes({
|
|
"vitest.module.id": id,
|
|
"vitest.module.url": result.url,
|
|
"vitest.module.external": external ?? false
|
|
});
|
|
return {
|
|
...result,
|
|
id,
|
|
external
|
|
};
|
|
});
|
|
}
|
|
async resolveMocks() {
|
|
if (!BareModuleMocker.pendingIds.length) return;
|
|
await Promise.all(BareModuleMocker.pendingIds.map(async (mock) => {
|
|
const { id, url, external } = await this.resolveId(mock.id, mock.importer);
|
|
if (mock.action === "unmock") this.unmockPath(id);
|
|
if (mock.action === "mock") this.mockPath(mock.id, id, url, external, mock.type, mock.factory);
|
|
}));
|
|
BareModuleMocker.pendingIds = [];
|
|
}
|
|
// public method to avoid circular dependency
|
|
getMockContext() {
|
|
return this.mockContext;
|
|
}
|
|
// path used to store mocked dependencies
|
|
getMockPath(dep) {
|
|
return `mock:${dep}`;
|
|
}
|
|
getDependencyMock(id) {
|
|
return this.getMockerRegistry().getById(fixLeadingSlashes(id));
|
|
}
|
|
getDependencyMockByUrl(url) {
|
|
return this.getMockerRegistry().get(url);
|
|
}
|
|
findMockRedirect(mockPath, external) {
|
|
return findMockRedirect(this.root, mockPath, external);
|
|
}
|
|
mockObject(object, mockExportsOrModuleType, moduleType) {
|
|
let mockExports;
|
|
if (mockExportsOrModuleType === "automock" || mockExportsOrModuleType === "autospy") {
|
|
moduleType = mockExportsOrModuleType;
|
|
mockExports = void 0;
|
|
} else mockExports = mockExportsOrModuleType;
|
|
moduleType ??= "automock";
|
|
const createMockInstance = this.spyModule?.createMockInstance;
|
|
if (!createMockInstance) throw this.createError("[vitest] `spyModule` is not defined. This is a Vitest error. Please open a new issue with reproduction.");
|
|
return mockObject({
|
|
globalConstructors: this.primitives,
|
|
createMockInstance,
|
|
type: moduleType
|
|
}, object, mockExports);
|
|
}
|
|
unmockPath(id) {
|
|
this.getMockerRegistry().deleteById(id);
|
|
this.invalidateModuleById(id);
|
|
}
|
|
mockPath(originalId, id, url, external, mockType, factory) {
|
|
const registry = this.getMockerRegistry();
|
|
if (mockType === "manual") registry.register("manual", originalId, id, url, factory);
|
|
else if (mockType === "autospy") registry.register("autospy", originalId, id, url);
|
|
else {
|
|
const redirect = this.findMockRedirect(id, external);
|
|
if (redirect) registry.register("redirect", originalId, id, url, redirect);
|
|
else registry.register("automock", originalId, id, url);
|
|
}
|
|
// every time the mock is registered, we remove the previous one from the cache
|
|
this.invalidateModuleById(id);
|
|
}
|
|
async importActual(_rawId, _importer, _callstack) {
|
|
throw new Error(`importActual is not implemented`);
|
|
}
|
|
async importMock(_rawId, _importer, _callstack) {
|
|
throw new Error(`importMock is not implemented`);
|
|
}
|
|
queueMock(id, importer, factoryOrOptions) {
|
|
const mockType = getMockType(factoryOrOptions);
|
|
BareModuleMocker.pendingIds.push({
|
|
action: "mock",
|
|
id,
|
|
importer,
|
|
factory: typeof factoryOrOptions === "function" ? factoryOrOptions : void 0,
|
|
type: mockType
|
|
});
|
|
}
|
|
queueUnmock(id, importer) {
|
|
BareModuleMocker.pendingIds.push({
|
|
action: "unmock",
|
|
id,
|
|
importer
|
|
});
|
|
}
|
|
}
|
|
function getMockType(factoryOrOptions) {
|
|
if (!factoryOrOptions) return "automock";
|
|
if (typeof factoryOrOptions === "function") return "manual";
|
|
return factoryOrOptions.spy ? "autospy" : "automock";
|
|
}
|
|
// unique id that is not available as "$bare_import" like "test"
|
|
// https://nodejs.org/api/modules.html#built-in-modules-with-mandatory-node-prefix
|
|
const prefixedBuiltins = new Set([
|
|
"node:sea",
|
|
"node:sqlite",
|
|
"node:test",
|
|
"node:test/reporters"
|
|
]);
|
|
const isWindows$1 = process.platform === "win32";
|
|
// transform file url to id
|
|
// virtual:custom -> virtual:custom
|
|
// \0custom -> \0custom
|
|
// /root/id -> /id
|
|
// /root/id.js -> /id.js
|
|
// C:/root/id.js -> /id.js
|
|
// C:\root\id.js -> /id.js
|
|
// TODO: expose this in vite/module-runner
|
|
function normalizeModuleId(file) {
|
|
if (prefixedBuiltins.has(file)) return file;
|
|
// if it's not in the root, keep it as a path, not a URL
|
|
return slash(file).replace(/^\/@fs\//, isWindows$1 ? "" : "/").replace(/^node:/, "").replace(/^\/+/, "/").replace(/^file:\//, "/");
|
|
}
|
|
const windowsSlashRE = /\\/g;
|
|
function slash(p) {
|
|
return p.replace(windowsSlashRE, "/");
|
|
}
|
|
const multipleSlashRe = /^\/+/;
|
|
// module-runner incorrectly replaces file:///path with `///path`
|
|
function fixLeadingSlashes(id) {
|
|
if (id.startsWith("//")) return id.replace(multipleSlashRe, "/");
|
|
return id;
|
|
}
|
|
|
|
// copied from vite/src/shared/utils.ts
|
|
const postfixRE = /[?#].*$/;
|
|
function cleanUrl(url) {
|
|
return url.replace(postfixRE, "");
|
|
}
|
|
function splitFileAndPostfix(path) {
|
|
const file = cleanUrl(path);
|
|
return {
|
|
file,
|
|
postfix: path.slice(file.length)
|
|
};
|
|
}
|
|
function injectQuery(url, queryToInject) {
|
|
const { file, postfix } = splitFileAndPostfix(url);
|
|
return `${file}?${queryToInject}${postfix[0] === "?" ? `&${postfix.slice(1)}` : postfix}`;
|
|
}
|
|
function removeQuery(url, queryToRemove) {
|
|
return url.replace(new RegExp(`[?&]${queryToRemove}(?=[&#]|$)`), "").replace(/\?$/, "");
|
|
}
|
|
|
|
const spyModulePath = resolve(distDir, "spy.js");
|
|
class VitestMocker extends BareModuleMocker {
|
|
filterPublicKeys;
|
|
constructor(moduleRunner, options) {
|
|
super(options);
|
|
this.moduleRunner = moduleRunner;
|
|
this.options = options;
|
|
const context = this.options.context;
|
|
if (context) this.primitives = vm.runInContext("({ Object, Error, Function, RegExp, Symbol, Array, Map })", context);
|
|
const Symbol = this.primitives.Symbol;
|
|
this.filterPublicKeys = [
|
|
"__esModule",
|
|
Symbol.asyncIterator,
|
|
Symbol.hasInstance,
|
|
Symbol.isConcatSpreadable,
|
|
Symbol.iterator,
|
|
Symbol.match,
|
|
Symbol.matchAll,
|
|
Symbol.replace,
|
|
Symbol.search,
|
|
Symbol.split,
|
|
Symbol.species,
|
|
Symbol.toPrimitive,
|
|
Symbol.toStringTag,
|
|
Symbol.unscopables
|
|
];
|
|
}
|
|
get evaluatedModules() {
|
|
return this.moduleRunner.evaluatedModules;
|
|
}
|
|
async initializeSpyModule() {
|
|
if (this.spyModule) return;
|
|
this.spyModule = await this.moduleRunner.import(spyModulePath);
|
|
}
|
|
reset() {
|
|
this.registries.clear();
|
|
}
|
|
invalidateModuleById(id) {
|
|
const mockId = this.getMockPath(id);
|
|
const node = this.evaluatedModules.getModuleById(mockId);
|
|
if (node) {
|
|
this.evaluatedModules.invalidateModule(node);
|
|
node.mockedExports = void 0;
|
|
}
|
|
}
|
|
ensureModule(id, url) {
|
|
const node = this.evaluatedModules.ensureModule(id, url);
|
|
// TODO
|
|
node.meta = {
|
|
id,
|
|
url,
|
|
code: "",
|
|
file: null,
|
|
invalidate: false
|
|
};
|
|
return node;
|
|
}
|
|
async callFunctionMock(id, url, mock) {
|
|
const node = this.ensureModule(id, url);
|
|
if (node.exports) return node.exports;
|
|
const exports$1 = await mock.resolve();
|
|
const moduleExports = new Proxy(exports$1, { get: (target, prop) => {
|
|
const val = target[prop];
|
|
// 'then' can exist on non-Promise objects, need nested instanceof check for logic to work
|
|
if (prop === "then") {
|
|
if (target instanceof Promise) return target.then.bind(target);
|
|
} else if (!(prop in target)) {
|
|
if (this.filterPublicKeys.includes(prop)) return;
|
|
throw this.createError(`[vitest] No "${String(prop)}" export is defined on the "${mock.raw}" mock. Did you forget to return it from "vi.mock"?
|
|
If you need to partially mock a module, you can use "importOriginal" helper inside:
|
|
`, `vi.mock(import("${mock.raw}"), async (importOriginal) => {
|
|
const actual = await importOriginal()
|
|
return {
|
|
...actual,
|
|
// your mocked methods
|
|
}
|
|
})`);
|
|
}
|
|
return val;
|
|
} });
|
|
node.exports = moduleExports;
|
|
return moduleExports;
|
|
}
|
|
async importActual(rawId, importer, callstack) {
|
|
const { url } = await this.resolveId(rawId, importer);
|
|
const actualUrl = injectQuery(url, "_vitest_original");
|
|
const node = await this.moduleRunner.fetchModule(actualUrl, importer);
|
|
return await this.moduleRunner.cachedRequest(node.url, node, callstack || [importer], void 0, true);
|
|
}
|
|
async importMock(rawId, importer) {
|
|
const { id, url, external } = await this.resolveId(rawId, importer);
|
|
let mock = this.getDependencyMock(id);
|
|
if (!mock) {
|
|
const redirect = this.findMockRedirect(id, external);
|
|
if (redirect) mock = new RedirectedModule(rawId, id, rawId, redirect);
|
|
else mock = new AutomockedModule(rawId, id, rawId);
|
|
}
|
|
if (mock.type === "automock" || mock.type === "autospy") {
|
|
const node = await this.moduleRunner.fetchModule(url, importer);
|
|
const mod = await this.moduleRunner.cachedRequest(url, node, [importer], void 0, true);
|
|
const Object = this.primitives.Object;
|
|
return this.mockObject(mod, Object.create(Object.prototype), mock.type);
|
|
}
|
|
if (mock.type === "manual") return this.callFunctionMock(id, url, mock);
|
|
const node = await this.moduleRunner.fetchModule(mock.redirect);
|
|
return this.moduleRunner.cachedRequest(mock.redirect, node, [importer], void 0, true);
|
|
}
|
|
async requestWithMockedModule(url, evaluatedNode, callstack, mock) {
|
|
return this._otel.$("vitest.mocker.evaluate", async (span) => {
|
|
const mockId = this.getMockPath(evaluatedNode.id);
|
|
span.setAttributes({
|
|
"vitest.module.id": mockId,
|
|
"vitest.mock.type": mock.type,
|
|
"vitest.mock.id": mock.id,
|
|
"vitest.mock.url": mock.url,
|
|
"vitest.mock.raw": mock.raw
|
|
});
|
|
if (mock.type === "automock" || mock.type === "autospy") {
|
|
const cache = this.evaluatedModules.getModuleById(mockId);
|
|
if (cache && cache.mockedExports) return cache.mockedExports;
|
|
const Object = this.primitives.Object;
|
|
// we have to define a separate object that will copy all properties into itself
|
|
// and can't just use the same `exports` define automatically by Vite before the evaluator
|
|
const exports$1 = Object.create(null);
|
|
Object.defineProperty(exports$1, Symbol.toStringTag, {
|
|
value: "Module",
|
|
configurable: true,
|
|
writable: true
|
|
});
|
|
const node = this.ensureModule(mockId, this.getMockPath(evaluatedNode.url));
|
|
node.meta = evaluatedNode.meta;
|
|
node.file = evaluatedNode.file;
|
|
node.mockedExports = exports$1;
|
|
const mod = await this.moduleRunner.cachedRequest(url, node, callstack, void 0, true);
|
|
this.mockObject(mod, exports$1, mock.type);
|
|
return exports$1;
|
|
}
|
|
if (mock.type === "manual" && !callstack.includes(mockId) && !callstack.includes(url)) try {
|
|
callstack.push(mockId);
|
|
// this will not work if user does Promise.all(import(), import())
|
|
// we can also use AsyncLocalStorage to store callstack, but this won't work in the browser
|
|
// maybe we should improve mock API in the future?
|
|
this.mockContext.callstack = callstack;
|
|
return await this.callFunctionMock(mockId, this.getMockPath(url), mock);
|
|
} finally {
|
|
this.mockContext.callstack = null;
|
|
const indexMock = callstack.indexOf(mockId);
|
|
callstack.splice(indexMock, 1);
|
|
}
|
|
else if (mock.type === "redirect" && !callstack.includes(mock.redirect)) {
|
|
span.setAttribute("vitest.mock.redirect", mock.redirect);
|
|
return mock.redirect;
|
|
}
|
|
});
|
|
}
|
|
async mockedRequest(url, evaluatedNode, callstack) {
|
|
const mock = this.getDependencyMock(evaluatedNode.id);
|
|
if (!mock) return;
|
|
return this.requestWithMockedModule(url, evaluatedNode, callstack, mock);
|
|
}
|
|
}
|
|
|
|
class VitestTransport {
|
|
constructor(options, evaluatedModules, callstacks) {
|
|
this.options = options;
|
|
this.evaluatedModules = evaluatedModules;
|
|
this.callstacks = callstacks;
|
|
}
|
|
async invoke(event) {
|
|
if (event.type !== "custom") return { error: /* @__PURE__ */ new Error(`Vitest Module Runner doesn't support Vite HMR events.`) };
|
|
if (event.event !== "vite:invoke") return { error: /* @__PURE__ */ new Error(`Vitest Module Runner doesn't support ${event.event} event.`) };
|
|
const { name, data } = event.data;
|
|
if (name === "getBuiltins")
|
|
// we return an empty array here to avoid client-side builtin check,
|
|
// as we need builtins to go through `fetchModule`
|
|
return { result: [] };
|
|
if (name !== "fetchModule") return { error: /* @__PURE__ */ new Error(`Unknown method: ${name}. Expected "fetchModule".`) };
|
|
try {
|
|
return { result: await this.options.fetchModule(...data) };
|
|
} catch (cause) {
|
|
if (cause instanceof EnvironmentTeardownError) {
|
|
const [id, importer] = data;
|
|
let message = `Cannot load '${id}'${importer ? ` imported from ${importer}` : ""} after the environment was torn down. This is not a bug in Vitest.`;
|
|
const moduleNode = importer ? this.evaluatedModules.getModuleById(importer) : void 0;
|
|
const callstack = moduleNode ? this.callstacks.get(moduleNode) : void 0;
|
|
if (callstack) message += ` The last recorded callstack:\n- ${[
|
|
...callstack,
|
|
importer,
|
|
id
|
|
].reverse().join("\n- ")}`;
|
|
const error = new EnvironmentTeardownError(message);
|
|
if (cause.stack) error.stack = cause.stack.replace(cause.message, error.message);
|
|
return { error };
|
|
}
|
|
return { error: cause };
|
|
}
|
|
}
|
|
}
|
|
|
|
const createNodeImportMeta = (modulePath) => {
|
|
if (!viteModuleRunner.createDefaultImportMeta) throw new Error(`createNodeImportMeta is not supported in this version of Vite.`);
|
|
const defaultMeta = viteModuleRunner.createDefaultImportMeta(modulePath);
|
|
const href = defaultMeta.url;
|
|
const importMetaResolver = createImportMetaResolver();
|
|
return {
|
|
...defaultMeta,
|
|
main: false,
|
|
resolve(id, parent) {
|
|
return (importMetaResolver ?? defaultMeta.resolve)(id, parent ?? href);
|
|
}
|
|
};
|
|
};
|
|
function createImportMetaResolver() {
|
|
if (!import.meta.resolve) return;
|
|
return (specifier, importer) => import.meta.resolve(specifier, importer);
|
|
}
|
|
// @ts-expect-error overriding private method
|
|
class VitestModuleRunner extends viteModuleRunner.ModuleRunner {
|
|
mocker;
|
|
moduleExecutionInfo;
|
|
_otel;
|
|
_callstacks;
|
|
constructor(vitestOptions) {
|
|
const options = vitestOptions;
|
|
const evaluatedModules = options.evaluatedModules;
|
|
const callstacks = /* @__PURE__ */ new WeakMap();
|
|
const transport = new VitestTransport(options.transport, evaluatedModules, callstacks);
|
|
super({
|
|
transport,
|
|
hmr: false,
|
|
evaluatedModules,
|
|
sourcemapInterceptor: "prepareStackTrace",
|
|
createImportMeta: vitestOptions.createImportMeta
|
|
}, options.evaluator);
|
|
this.vitestOptions = vitestOptions;
|
|
this._callstacks = callstacks;
|
|
this._otel = vitestOptions.traces || new Traces({ enabled: false });
|
|
this.moduleExecutionInfo = options.getWorkerState().moduleExecutionInfo;
|
|
this.mocker = options.mocker || new VitestMocker(this, {
|
|
spyModule: options.spyModule,
|
|
context: options.vm?.context,
|
|
traces: this._otel,
|
|
resolveId: options.transport.resolveId,
|
|
get root() {
|
|
return options.getWorkerState().config.root;
|
|
},
|
|
get moduleDirectories() {
|
|
return options.getWorkerState().config.deps.moduleDirectories || [];
|
|
},
|
|
getCurrentTestFilepath() {
|
|
return options.getWorkerState().filepath;
|
|
}
|
|
});
|
|
if (options.vm) options.vm.context.__vitest_mocker__ = this.mocker;
|
|
else Object.defineProperty(globalThis, "__vitest_mocker__", {
|
|
configurable: true,
|
|
writable: true,
|
|
value: this.mocker
|
|
});
|
|
}
|
|
/**
|
|
* Vite checks that the module has exports emulating the Node.js behaviour,
|
|
* but Vitest is more relaxed.
|
|
*
|
|
* We should keep the Vite behavour when there is a `strict` flag.
|
|
* @internal
|
|
*/
|
|
processImport(exports$1) {
|
|
return exports$1;
|
|
}
|
|
async import(rawId) {
|
|
const resolved = await this._otel.$("vitest.module.resolve_id", { attributes: { "vitest.module.raw_id": rawId } }, async (span) => {
|
|
const result = await this.vitestOptions.transport.resolveId(rawId);
|
|
if (result) span.setAttributes({
|
|
"vitest.module.url": result.url,
|
|
"vitest.module.file": result.file,
|
|
"vitest.module.id": result.id
|
|
});
|
|
return result;
|
|
});
|
|
return super.import(resolved ? resolved.url : rawId);
|
|
}
|
|
async fetchModule(url, importer) {
|
|
return await this.cachedModule(url, importer);
|
|
}
|
|
_cachedRequest(url, module, callstack = [], metadata) {
|
|
// @ts-expect-error "cachedRequest" is private
|
|
return super.cachedRequest(url, module, callstack, metadata);
|
|
}
|
|
/**
|
|
* @internal
|
|
*/
|
|
async cachedRequest(url, mod, callstack = [], metadata, ignoreMock = false) {
|
|
// Track for a better error message if dynamic import is not resolved properly
|
|
this._callstacks.set(mod, callstack);
|
|
if (ignoreMock) return this._cachedRequest(url, mod, callstack, metadata);
|
|
let mocked;
|
|
if (mod.meta && "mockedModule" in mod.meta) {
|
|
const mockedModule = mod.meta.mockedModule;
|
|
const mockId = this.mocker.getMockPath(mod.id);
|
|
// bypass mock and force "importActual" behavior when:
|
|
// - mock was removed by doUnmock (stale mockedModule in meta)
|
|
// - self-import: mock factory/file is importing the module it's mocking
|
|
const isStale = !this.mocker.getDependencyMock(mod.id);
|
|
const isSelfImport = callstack.includes(mockId) || callstack.includes(url) || "redirect" in mockedModule && callstack.includes(mockedModule.redirect);
|
|
if (isStale || isSelfImport) {
|
|
const node = await this.fetchModule(injectQuery(url, "_vitest_original"));
|
|
return this._cachedRequest(node.url, node, callstack, metadata);
|
|
}
|
|
mocked = await this.mocker.requestWithMockedModule(url, mod, callstack, mockedModule);
|
|
} else mocked = await this.mocker.mockedRequest(url, mod, callstack);
|
|
if (typeof mocked === "string") {
|
|
const node = await this.fetchModule(mocked);
|
|
return this._cachedRequest(mocked, node, callstack, metadata);
|
|
}
|
|
if (mocked != null && typeof mocked === "object") return mocked;
|
|
return this._cachedRequest(url, mod, callstack, metadata);
|
|
}
|
|
/** @internal */
|
|
_invalidateSubTreeById(ids, invalidated = /* @__PURE__ */ new Set()) {
|
|
for (const id of ids) {
|
|
if (invalidated.has(id)) continue;
|
|
const node = this.evaluatedModules.getModuleById(id);
|
|
if (!node) continue;
|
|
invalidated.add(id);
|
|
const subIds = Array.from(this.evaluatedModules.idToModuleMap).filter(([, mod]) => mod.importers.has(id)).map(([key]) => key);
|
|
if (subIds.length) this._invalidateSubTreeById(subIds, invalidated);
|
|
this.evaluatedModules.invalidateModule(node);
|
|
}
|
|
}
|
|
}
|
|
|
|
const bareVitestRegexp = /^@?vitest(?:\/|$)/;
|
|
const normalizedDistDir = normalize(distDir);
|
|
const relativeIds = {};
|
|
const externalizeMap = /* @__PURE__ */ new Map();
|
|
// all Vitest imports always need to be externalized
|
|
function getCachedVitestImport(id, state) {
|
|
if (id.startsWith("/@fs/") || id.startsWith("\\@fs\\")) id = id.slice(process.platform === "win32" ? 5 : 4);
|
|
if (externalizeMap.has(id)) return {
|
|
externalize: externalizeMap.get(id),
|
|
type: "module"
|
|
};
|
|
// always externalize Vitest because we import from there before running tests
|
|
// so we already have it cached by Node.js
|
|
const root = state().config.root;
|
|
const relativeRoot = relativeIds[root] ?? (relativeIds[root] = normalizedDistDir.slice(root.length));
|
|
if (id.includes(distDir) || id.includes(normalizedDistDir)) {
|
|
const externalize = id.startsWith("file://") ? id : pathToFileURL(id).toString();
|
|
externalizeMap.set(id, externalize);
|
|
return {
|
|
externalize,
|
|
type: "module"
|
|
};
|
|
}
|
|
if (relativeRoot && relativeRoot !== "/" && id.startsWith(relativeRoot)) {
|
|
const externalize = pathToFileURL(join(root, id)).toString();
|
|
externalizeMap.set(id, externalize);
|
|
return {
|
|
externalize,
|
|
type: "module"
|
|
};
|
|
}
|
|
if (bareVitestRegexp.test(id)) {
|
|
externalizeMap.set(id, id);
|
|
return {
|
|
externalize: id,
|
|
type: "module"
|
|
};
|
|
}
|
|
return null;
|
|
}
|
|
|
|
const { readFileSync } = fs;
|
|
const VITEST_VM_CONTEXT_SYMBOL = "__vitest_vm_context__";
|
|
const cwd = process.cwd();
|
|
const isWindows = process.platform === "win32";
|
|
function startVitestModuleRunner(options) {
|
|
const traces = options.traces;
|
|
const state = () => getSafeWorkerState() || options.state;
|
|
const rpc = () => state().rpc;
|
|
const environment = () => {
|
|
const environment = state().environment;
|
|
return environment.viteEnvironment || environment.name;
|
|
};
|
|
const vm = options.context && options.externalModulesExecutor ? {
|
|
context: options.context,
|
|
externalModulesExecutor: options.externalModulesExecutor
|
|
} : void 0;
|
|
const evaluator = options.evaluator || new VitestModuleEvaluator(vm, {
|
|
traces,
|
|
evaluatedModules: options.evaluatedModules,
|
|
get moduleExecutionInfo() {
|
|
return state().moduleExecutionInfo;
|
|
},
|
|
get interopDefault() {
|
|
return state().config.deps.interopDefault;
|
|
},
|
|
getCurrentTestFilepath: () => state().filepath
|
|
});
|
|
const moduleRunner = new VitestModuleRunner({
|
|
spyModule: options.spyModule,
|
|
evaluatedModules: options.evaluatedModules,
|
|
evaluator,
|
|
traces,
|
|
mocker: options.mocker,
|
|
transport: {
|
|
async fetchModule(id, importer, options) {
|
|
const resolvingModules = state().resolvingModules;
|
|
if (isWindows) {
|
|
if (id[1] === ":") {
|
|
// The drive letter is different for whatever reason, we need to normalize it to CWD
|
|
if (id[0] !== cwd[0] && id[0].toUpperCase() === cwd[0].toUpperCase()) id = (cwd[0].toUpperCase() === cwd[0] ? id[0].toUpperCase() : id[0].toLowerCase()) + id.slice(1);
|
|
// always mark absolute windows paths, otherwise Vite will externalize it
|
|
id = `/@id/${id}`;
|
|
}
|
|
}
|
|
const vitest = getCachedVitestImport(id, state);
|
|
if (vitest) return vitest;
|
|
// strip _vitest_original query added by importActual so that
|
|
// the plugin pipeline sees the original import id (e.g. virtual modules's load hook)
|
|
const isImportActual = id.includes("_vitest_original");
|
|
if (isImportActual) id = removeQuery(id, "_vitest_original");
|
|
const rawId = unwrapId(id);
|
|
resolvingModules.add(rawId);
|
|
try {
|
|
if (VitestMocker.pendingIds.length) await moduleRunner.mocker.resolveMocks();
|
|
if (!isImportActual) {
|
|
const resolvedMock = moduleRunner.mocker.getDependencyMockByUrl(id);
|
|
if (resolvedMock?.type === "manual" || resolvedMock?.type === "redirect") return {
|
|
code: "",
|
|
file: null,
|
|
id: resolvedMock.id,
|
|
url: resolvedMock.url,
|
|
invalidate: false,
|
|
mockedModule: resolvedMock
|
|
};
|
|
}
|
|
if (isBuiltin(rawId)) return {
|
|
externalize: rawId,
|
|
type: "builtin"
|
|
};
|
|
if (isBrowserExternal(rawId)) return {
|
|
externalize: toBuiltin(rawId),
|
|
type: "builtin"
|
|
};
|
|
// if module is invalidated, the worker will be recreated,
|
|
// so cached is always true in a single worker
|
|
if (options?.cached) return { cache: true };
|
|
const otelCarrier = traces?.getContextCarrier();
|
|
const result = await rpc().fetch(id, importer, environment(), options, otelCarrier);
|
|
if ("cached" in result) return {
|
|
code: readFileSync(result.tmp, "utf-8"),
|
|
...result
|
|
};
|
|
return result;
|
|
} catch (cause) {
|
|
// rethrow vite error if it cannot load the module because it's not resolved
|
|
if (typeof cause === "object" && cause != null && cause.code === "ERR_LOAD_URL" || typeof cause?.message === "string" && cause.message.includes("Failed to load url") || typeof cause?.message === "string" && cause.message.startsWith("Cannot find module '")) {
|
|
const error = new Error(`Cannot find ${isBareImport(id) ? "package" : "module"} '${id}'${importer ? ` imported from ${importer}` : ""}`, { cause });
|
|
error.code = "ERR_MODULE_NOT_FOUND";
|
|
throw error;
|
|
}
|
|
throw cause;
|
|
} finally {
|
|
resolvingModules.delete(rawId);
|
|
}
|
|
},
|
|
resolveId(id, importer) {
|
|
return rpc().resolve(id, importer, environment());
|
|
}
|
|
},
|
|
getWorkerState: state,
|
|
vm,
|
|
createImportMeta: options.createImportMeta
|
|
});
|
|
return moduleRunner;
|
|
}
|
|
|
|
export { BareModuleMocker as B, VITEST_VM_CONTEXT_SYMBOL as V, VitestModuleRunner as a, VitestTransport as b, createNodeImportMeta as c, normalizeModuleId as n, startVitestModuleRunner as s };
|