@vitest/runner 2.0.0-beta.9 → 2.0.1

package/dist/chunk-tasks.js

@@ -13,11 +13,39 @@ function partitionSuiteChildren(suite) {
       tasksGroup = [c];
     }
   }
-  if (tasksGroup.length > 0)
+  if (tasksGroup.length > 0) {
     tasksGroups.push(tasksGroup);
+  }
   return tasksGroups;
 }
 
+function limitConcurrency(concurrency = Infinity) {
+  let count = 0;
+  let head;
+  let tail;
+  const finish = () => {
+    count--;
+    if (head) {
+      head[0]();
+      head = head[1];
+      tail = head && tail;
+    }
+  };
+  return (func, ...args) => {
+    return new Promise((resolve) => {
+      if (count++ < concurrency) {
+        resolve();
+      } else if (tail) {
+        tail = tail[1] = [resolve];
+      } else {
+        head = tail = [resolve];
+      }
+    }).then(() => {
+      return func(...args);
+    }).finally(finish);
+  };
+}
+
 function interpretTaskModes(suite, namePattern, onlyMode, parentIsOnly, allowOnly) {
   const suiteIsOnly = parentIsOnly || suite.mode === "only";
   suite.tasks.forEach((t) => {
@@ -36,39 +64,50 @@ function interpretTaskModes(suite, namePattern, onlyMode, parentIsOnly, allowOnl
       }
     }
     if (t.type === "test") {
-      if (namePattern && !getTaskFullName(t).match(namePattern))
+      if (namePattern && !getTaskFullName(t).match(namePattern)) {
         t.mode = "skip";
+      }
     } else if (t.type === "suite") {
-      if (t.mode === "skip")
+      if (t.mode === "skip") {
         skipAllTasks(t);
-      else
+      } else {
         interpretTaskModes(t, namePattern, onlyMode, includeTask, allowOnly);
+      }
     }
   });
   if (suite.mode === "run") {
-    if (suite.tasks.length && suite.tasks.every((i) => i.mode !== "run"))
+    if (suite.tasks.length && suite.tasks.every((i) => i.mode !== "run")) {
       suite.mode = "skip";
+    }
   }
 }
 function getTaskFullName(task) {
   return `${task.suite ? `${getTaskFullName(task.suite)} ` : ""}${task.name}`;
 }
 function someTasksAreOnly(suite) {
-  return suite.tasks.some((t) => t.mode === "only" || t.type === "suite" && someTasksAreOnly(t));
+  return suite.tasks.some(
+    (t) => t.mode === "only" || t.type === "suite" && someTasksAreOnly(t)
+  );
 }
 function skipAllTasks(suite) {
   suite.tasks.forEach((t) => {
     if (t.mode === "run") {
       t.mode = "skip";
-      if (t.type === "suite")
+      if (t.type === "suite") {
         skipAllTasks(t);
+      }
     }
   });
 }
 function checkAllowOnly(task, allowOnly) {
-  if (allowOnly)
+  if (allowOnly) {
     return;
-  const error = processError(new Error("[Vitest] Unexpected .only modifier. Remove it or pass --allowOnly argument to bypass this error"));
+  }
+  const error = processError(
+    new Error(
+      "[Vitest] Unexpected .only modifier. Remove it or pass --allowOnly argument to bypass this error"
+    )
+  );
   task.result = {
     state: "fail",
     errors: [error]
@@ -76,8 +115,9 @@ function checkAllowOnly(task, allowOnly) {
 }
 function generateHash(str) {
   let hash = 0;
-  if (str.length === 0)
+  if (str.length === 0) {
     return `${hash}`;
+  }
   for (let i = 0; i < str.length; i++) {
     const char = str.charCodeAt(i);
     hash = (hash << 5) - hash + char;
@@ -88,11 +128,12 @@ function generateHash(str) {
 function calculateSuiteHash(parent) {
   parent.tasks.forEach((t, idx) => {
     t.id = `${parent.id}_${idx}`;
-    if (t.type === "suite")
+    if (t.type === "suite") {
       calculateSuiteHash(t);
+    }
   });
 }
-function createFileTask(filepath, root, projectName) {
+function createFileTask(filepath, root, projectName, pool) {
   const path = relative(root, filepath);
   const file = {
     id: generateHash(`${path}${projectName || ""}`),
@@ -103,7 +144,8 @@ function createFileTask(filepath, root, projectName) {
     tasks: [],
     meta: /* @__PURE__ */ Object.create(null),
     projectName,
-    file: void 0
+    file: void 0,
+    pool
   };
   file.file = file;
   return file;
@@ -151,8 +193,9 @@ function getTests(suite) {
           tests.push(task);
         } else {
           const taskTests = getTests(task);
-          for (const test of taskTests)
+          for (const test of taskTests) {
             tests.push(test);
+          }
         }
       }
     }
@@ -160,31 +203,41 @@ function getTests(suite) {
   return tests;
 }
 function getTasks(tasks = []) {
-  return toArray(tasks).flatMap((s) => isAtomTest(s) ? [s] : [s, ...getTasks(s.tasks)]);
+  return toArray(tasks).flatMap(
+    (s) => isAtomTest(s) ? [s] : [s, ...getTasks(s.tasks)]
+  );
 }
 function getSuites(suite) {
-  return toArray(suite).flatMap((s) => s.type === "suite" ? [s, ...getSuites(s.tasks)] : []);
+  return toArray(suite).flatMap(
+    (s) => s.type === "suite" ? [s, ...getSuites(s.tasks)] : []
+  );
 }
 function hasTests(suite) {
-  return toArray(suite).some((s) => s.tasks.some((c) => isAtomTest(c) || hasTests(c)));
+  return toArray(suite).some(
+    (s) => s.tasks.some((c) => isAtomTest(c) || hasTests(c))
+  );
 }
 function hasFailed(suite) {
-  return toArray(suite).some((s) => {
-    var _a;
-    return ((_a = s.result) == null ? void 0 : _a.state) === "fail" || s.type === "suite" && hasFailed(s.tasks);
-  });
+  return toArray(suite).some(
+    (s) => {
+      var _a;
+      return ((_a = s.result) == null ? void 0 : _a.state) === "fail" || s.type === "suite" && hasFailed(s.tasks);
+    }
+  );
 }
 function getNames(task) {
   const names = [task.name];
   let current = task;
   while (current == null ? void 0 : current.suite) {
     current = current.suite;
-    if (current == null ? void 0 : current.name)
+    if (current == null ? void 0 : current.name) {
       names.unshift(current.name);
+    }
   }
-  if (current !== task.file)
+  if (current !== task.file) {
     names.unshift(task.file.name);
+  }
   return names;
 }
 
-export { createFileTask as a, getTests as b, calculateSuiteHash as c, getTasks as d, getSuites as e, hasFailed as f, generateHash as g, hasTests as h, interpretTaskModes as i, getNames as j, createChainable as k, partitionSuiteChildren as p, someTasksAreOnly as s };
+export { createFileTask as a, isAtomTest as b, calculateSuiteHash as c, getTests as d, getTasks as e, getSuites as f, generateHash as g, hasTests as h, interpretTaskModes as i, hasFailed as j, getNames as k, createChainable as l, limitConcurrency as m, partitionSuiteChildren as p, someTasksAreOnly as s };

package/dist/index.d.ts

@@ -1,13 +1,14 @@
 import { VitestRunner } from './types.js';
 export { CancelReason, VitestRunnerConfig, VitestRunnerConstructor, VitestRunnerImportSource } from './types.js';
-import { T as Task, F as File, d as SuiteAPI, e as TestAPI, f as SuiteCollector, g as CustomAPI, h as SuiteHooks, O as OnTestFailedHandler, i as OnTestFinishedHandler, a as Test, C as Custom, S as Suite } from './tasks-Cjrz1dUg.js';
-export { D as DoneCallback, E as ExtendedContext, t as Fixture, s as FixtureFn, r as FixtureOptions, u as Fixtures, v as HookCleanupCallback, H as HookListener, I as InferFixturesTypes, R as RunMode, y as RuntimeContext, B as SequenceHooks, G as SequenceSetupFiles, x as SuiteFactory, k as TaskBase, A as TaskContext, w as TaskCustomOptions, m as TaskMeta, l as TaskPopulated, n as TaskResult, o as TaskResultPack, j as TaskState, z as TestContext, p as TestFunction, q as TestOptions, U as Use } from './tasks-Cjrz1dUg.js';
+import { T as Task, F as File, d as SuiteAPI, e as TestAPI, f as SuiteCollector, g as CustomAPI, h as SuiteHooks, O as OnTestFailedHandler, i as OnTestFinishedHandler, a as Test, C as Custom, S as Suite } from './tasks-WAKtRuk9.js';
+export { D as DoneCallback, E as ExtendedContext, t as Fixture, s as FixtureFn, r as FixtureOptions, u as Fixtures, v as HookCleanupCallback, H as HookListener, I as InferFixturesTypes, R as RunMode, y as RuntimeContext, B as SequenceHooks, G as SequenceSetupFiles, x as SuiteFactory, k as TaskBase, A as TaskContext, w as TaskCustomOptions, m as TaskMeta, l as TaskPopulated, n as TaskResult, o as TaskResultPack, j as TaskState, z as TestContext, p as TestFunction, q as TestOptions, U as Use } from './tasks-WAKtRuk9.js';
 import { Awaitable } from '@vitest/utils';
 export { processError } from '@vitest/utils/error';
 import '@vitest/utils/diff';
 
 declare function updateTask(task: Task, runner: VitestRunner): void;
 declare function startTests(paths: string[], runner: VitestRunner): Promise<File[]>;
+declare function publicCollect(paths: string[], runner: VitestRunner): Promise<File[]>;
 
 declare const suite: SuiteAPI;
 declare const test: TestAPI;
@@ -23,11 +24,11 @@ declare function afterEach<ExtraContext = {}>(fn: SuiteHooks<ExtraContext>['afte
 declare const onTestFailed: (fn: OnTestFailedHandler) => void;
 declare const onTestFinished: (fn: OnTestFinishedHandler) => void;
 
-declare function setFn(key: Test | Custom, fn: (() => Awaitable<void>)): void;
-declare function getFn<Task = Test | Custom>(key: Task): (() => Awaitable<void>);
+declare function setFn(key: Test | Custom, fn: () => Awaitable<void>): void;
+declare function getFn<Task = Test | Custom>(key: Task): () => Awaitable<void>;
 declare function setHooks(key: Suite, hooks: SuiteHooks): void;
 declare function getHooks(key: Suite): SuiteHooks;
 
 declare function getCurrentTest<T extends Test | Custom | undefined>(): T;
 
-export { Custom, CustomAPI, File, OnTestFailedHandler, OnTestFinishedHandler, Suite, SuiteAPI, SuiteCollector, SuiteHooks, Task, Test, TestAPI, VitestRunner, afterAll, afterEach, beforeAll, beforeEach, createTaskCollector, describe, getCurrentSuite, getCurrentTest, getFn, getHooks, it, onTestFailed, onTestFinished, setFn, setHooks, startTests, suite, test, updateTask };
+export { Custom, CustomAPI, File, OnTestFailedHandler, OnTestFinishedHandler, Suite, SuiteAPI, SuiteCollector, SuiteHooks, Task, Test, TestAPI, VitestRunner, afterAll, afterEach, beforeAll, beforeEach, publicCollect as collectTests, createTaskCollector, describe, getCurrentSuite, getCurrentTest, getFn, getHooks, it, onTestFailed, onTestFinished, setFn, setHooks, startTests, suite, test, updateTask };