vitest 0.0.109 → 0.0.113

package/dist/worker.js

@@ -1,18 +1,19 @@
-import { s as slash, h as resolve, d as dirname$2, m as mergeSlashes, b as basename$2 } from './utils-d97bd6d9.js';
-import { a as spyOn, s as spies, n as nanoid } from './jest-mock-8498c46d.js';
-import { c as distDir } from './constants-e762cbc5.js';
+import { a as resolve, d as dirname$2, b as basename$2 } from './index-1488b423.js';
+import { a as spyOn, s as spies, n as nanoid } from './jest-mock-a57b745c.js';
+import { c as distDir } from './constants-900abe4a.js';
 import { builtinModules, createRequire } from 'module';
 import { pathToFileURL, fileURLToPath as fileURLToPath$2, URL as URL$1 } from 'url';
-import fs, { promises, realpathSync, statSync, Stats, readdirSync, existsSync } from 'fs';
 import vm from 'vm';
 import path from 'path';
+import fs, { promises, realpathSync, statSync, Stats, readdirSync, existsSync } from 'fs';
 import assert from 'assert';
 import { format as format$2, inspect } from 'util';
+import { m as mergeSlashes, s as slash } from './index-041e627e.js';
 import { s as send } from './rpc-7de86f29.js';
-import 'tty';
-import 'local-pkg';
 import 'chai';
 import 'tinyspy';
+import 'tty';
+import 'local-pkg';
 
 const BUILTIN_MODULES$1 = new Set(builtinModules);
 function normalizeSlash$1(str) {
@@ -9215,6 +9216,105 @@ var __spreadValues = (a, b) => {
     }
   return a;
 };
+function resolveMockPath(mockPath, root, nmName) {
+  if (nmName) {
+    const mockFolder = resolve(root, "__mocks__");
+    const files = readdirSync(mockFolder);
+    for (const file of files) {
+      const [basename2] = file.split(".");
+      if (basename2 === nmName)
+        return resolve(mockFolder, file).replace(root, "");
+    }
+    return null;
+  }
+  const dir = dirname$2(mockPath);
+  const baseId = basename$2(mockPath);
+  const fullPath = resolve(dir, "__mocks__", baseId);
+  return existsSync(fullPath) ? fullPath.replace(root, "") : null;
+}
+function getObjectType(value) {
+  return Object.prototype.toString.apply(value).slice(8, -1);
+}
+function mockPrototype(proto) {
+  if (!proto)
+    return null;
+  const newProto = {};
+  const protoDescr = Object.getOwnPropertyDescriptors(proto);
+  for (const d in protoDescr) {
+    Object.defineProperty(newProto, d, protoDescr[d]);
+    if (typeof protoDescr[d].value === "function")
+      spyOn(newProto, d).mockImplementation(() => {
+      });
+  }
+  return newProto;
+}
+function mockObject(obj) {
+  const type = getObjectType(obj);
+  if (Array.isArray(obj))
+    return [];
+  else if (type !== "Object" && type !== "Module")
+    return obj;
+  const newObj = __spreadValues({}, obj);
+  const proto = mockPrototype(Object.getPrototypeOf(obj));
+  Object.setPrototypeOf(newObj, proto);
+  for (const k in obj) {
+    newObj[k] = mockObject(obj[k]);
+    const type2 = getObjectType(obj[k]);
+    if (type2.includes("Function") && !obj[k].__isSpy) {
+      spyOn(newObj, k).mockImplementation(() => {
+      });
+      Object.defineProperty(newObj[k], "length", { value: 0 });
+    }
+  }
+  return newObj;
+}
+function createMocker(root, mockMap) {
+  function getSuiteFilepath() {
+    var _a;
+    return (_a = process.__vitest_worker__) == null ? void 0 : _a.filepath;
+  }
+  function getActualPath(path, nmName) {
+    return nmName ? mergeSlashes(`/@fs/${path}`) : path.replace(root, "");
+  }
+  function unmockPath(path, nmName) {
+    const suitefile = getSuiteFilepath();
+    if (suitefile) {
+      const fsPath = getActualPath(path, nmName);
+      mockMap[suitefile] ?? (mockMap[suitefile] = {});
+      delete mockMap[suitefile][fsPath];
+    }
+  }
+  function mockPath(path, nmName, factory) {
+    const suitefile = getSuiteFilepath();
+    if (suitefile) {
+      const fsPath = getActualPath(path, nmName);
+      mockMap[suitefile] ?? (mockMap[suitefile] = {});
+      mockMap[suitefile][fsPath] = factory || resolveMockPath(path, root, nmName);
+    }
+  }
+  function clearMocks({ clearMocks: clearMocks2, mockReset, restoreMocks }) {
+    if (!clearMocks2 && !mockReset && !restoreMocks)
+      return;
+    spies.forEach((s) => {
+      if (restoreMocks)
+        s.mockRestore();
+      else if (mockReset)
+        s.mockReset();
+      else if (clearMocks2)
+        s.mockClear();
+    });
+  }
+  return {
+    mockPath,
+    unmockPath,
+    clearMocks,
+    getActualPath,
+    mockObject,
+    getSuiteFilepath,
+    resolveMockPath
+  };
+}
+
 const defaultInline = [
   "vitest/dist",
   /virtual:/,
@@ -9262,84 +9362,46 @@ async function interpretedImport(path, interpretDefault) {
   }
   return mod;
 }
-function resolveMockPath(mockPath, root, nmName) {
-  if (nmName) {
-    const mockFolder = resolve(root, "__mocks__");
-    const files = readdirSync(mockFolder);
-    for (const file of files) {
-      const [basename2] = file.split(".");
-      if (basename2 === nmName)
-        return resolve(mockFolder, file).replace(root, "");
-    }
-    return null;
-  }
-  const dir = dirname$2(mockPath);
-  const baseId = basename$2(mockPath);
-  const fullPath = resolve(dir, "__mocks__", baseId);
-  return existsSync(fullPath) ? fullPath.replace(root, "") : null;
-}
-function mockObject(obj) {
-  const newObj = __spreadValues({}, obj);
-  for (const k in obj) {
-    newObj[k] = obj[k];
-    if (typeof obj[k] === "function" && !obj[k].__isSpy)
-      spyOn(newObj, k);
-  }
-  return newObj;
-}
 async function executeInViteNode(options) {
   const { moduleCache, root, files, fetch, mockMap } = options;
   const externalCache = /* @__PURE__ */ new Map();
   builtinModules.forEach((m) => externalCache.set(m, true));
+  const {
+    getActualPath,
+    getSuiteFilepath,
+    mockObject,
+    mockPath,
+    clearMocks,
+    unmockPath,
+    resolveMockPath
+  } = createMocker(root, mockMap);
   const result = [];
   for (const file of files)
     result.push(await cachedRequest(`/@fs/${slash(resolve(file))}`, []));
   return result;
-  function getSuiteFilepath() {
+  async function callFunctionMock(dep, mock) {
     var _a;
-    return (_a = process.__vitest_worker__) == null ? void 0 : _a.filepath;
-  }
-  function getActualPath(path, nmName) {
-    return nmName ? mergeSlashes(`/@fs/${path}`) : path.replace(root, "");
-  }
-  function unmockPath(path, nmName) {
-    const suitefile = getSuiteFilepath();
-    if (suitefile) {
-      const fsPath = getActualPath(path, nmName);
-      mockMap[suitefile] ?? (mockMap[suitefile] = {});
-      delete mockMap[suitefile][fsPath];
-    }
-  }
-  function mockPath(path, nmName) {
-    const suitefile = getSuiteFilepath();
-    if (suitefile) {
-      const mockPath2 = resolveMockPath(path, root, nmName);
-      const fsPath = getActualPath(path, nmName);
-      mockMap[suitefile] ?? (mockMap[suitefile] = {});
-      mockMap[suitefile][fsPath] = mockPath2;
-    }
-  }
-  function clearMocks({ clearMocks: clearMocks2, mockReset, restoreMocks }) {
-    if (!clearMocks2 && !mockReset && !restoreMocks)
-      return;
-    spies.forEach((s) => {
-      if (restoreMocks)
-        s.mockRestore();
-      else if (mockReset)
-        s.mockReset();
-      else if (clearMocks2)
-        s.mockClear();
-    });
+    const name = `${dep}__mock`;
+    const cached = (_a = moduleCache.get(name)) == null ? void 0 : _a.exports;
+    if (cached)
+      return cached;
+    const exports = await mock();
+    setCache(name, { exports });
+    return exports;
   }
   async function directRequest(id, fsPath, callstack) {
     callstack = [...callstack, id];
     const suite = getSuiteFilepath();
     const request = async (dep, canMock = true) => {
       var _a;
-      const mocks2 = mockMap[suite || ""] || {};
-      const mock = mocks2[dep];
-      if (mock && canMock)
-        dep = mock;
+      if (canMock) {
+        const mocks2 = mockMap[suite || ""] || {};
+        const mock = mocks2[dep];
+        if (typeof mock === "function")
+          return callFunctionMock(dep, mock);
+        if (typeof mock === "string")
+          dep = mock;
+      }
       if (callstack.includes(dep)) {
         const cacheKey = toFilePath(dep, root);
         if (!((_a = moduleCache.get(cacheKey)) == null ? void 0 : _a.exports))
@@ -9372,12 +9434,18 @@ ${[...callstack, dep].reverse().map((p) => `- ${p}`).join("\n")}`);
       return request(getActualPath(path, nmName), false);
     };
     const importMock = async (path, nmName) => {
-      const mockPath2 = resolveMockPath(path, root, nmName);
-      if (mockPath2 === null) {
-        const exports2 = await request(getActualPath(path, nmName), false);
-        return mockObject(exports2);
+      if (!suite)
+        throw new Error("You can import mock only inside of a running test");
+      const mock = (mockMap[suite] || {})[path] || resolveMockPath(path, root, nmName);
+      if (mock === null) {
+        const fsPath2 = getActualPath(path, nmName);
+        const exports2 = mockObject(await request(fsPath2, false));
+        setCache(fsPath2, { exports: exports2 });
+        return exports2;
       }
-      return request(mockPath2, true);
+      if (typeof mock === "function")
+        return callFunctionMock(path, mock);
+      return request(mock, true);
     };
     const context = {
       __vite_ssr_import__: request,
@@ -9523,10 +9591,13 @@ async function startViteNode(ctx) {
   return _viteNode;
 }
 function init(ctx) {
+  if (process.__vitest_worker__ && ctx.config.threads)
+    throw new Error(`worker for ${ctx.files.join(",")} already initialized by ${process.__vitest_worker__.ctx.files.join(",")}. This is probably an internal bug of Vitest.`);
   process.stdout.write("\0");
   const { config, port } = ctx;
   const rpcPromiseMap = /* @__PURE__ */ new Map();
   process.__vitest_worker__ = {
+    ctx,
     moduleCache,
     config,
     rpc: (method, ...args) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vitest",
-  "version": "0.0.109",
+  "version": "0.0.113",
   "description": "A blazing fast unit test framework powered by Vite",
   "keywords": [
     "vite",
@@ -51,16 +51,16 @@
     "@types/chai-subset": "^1.3.3",
     "chai": "^4.3.4",
     "local-pkg": "^0.4.0",
-    "tinypool": "^0.0.5",
-    "tinyspy": "^0.2.5"
+    "tinypool": "^0.0.6",
+    "tinyspy": "^0.2.6"
   },
   "devDependencies": {
     "@antfu/install-pkg": "^0.1.0",
-    "@types/diff": "^5.0.1",
+    "@types/diff": "^5.0.2",
     "@types/jsdom": "^16.2.14",
     "@types/micromatch": "^4.0.2",
     "@types/natural-compare": "^1.4.1",
-    "@types/node": "^17.0.3",
+    "@types/node": "^17.0.4",
     "@types/prompts": "^2.4.0",
     "c8": "^7.10.0",
     "cac": "^6.7.12",
@@ -84,7 +84,7 @@
     "pkg-types": "^0.3.2",
     "pretty-format": "^27.4.2",
     "prompts": "^2.4.2",
-    "rollup": "^2.61.1",
+    "rollup": "^2.62.0",
     "source-map-js": "^1.0.1",
     "strip-ansi": "^7.0.1",
     "typescript": "^4.5.4"