chore: use vitest in tinybench-plugin

Author: GuillaumeLagrange

packages/tinybench-plugin/moon.yml

@@ -7,3 +7,10 @@ tasks:
     platform: "system"
     options:
       cache: false
+  test:
+    command: vitest --run
+    inputs:
+      - "./vitest.config.ts"
+
+  test/integ:
+    command: noop

packages/tinybench-plugin/package.json

@@ -8,9 +8,15 @@
     "tinybench",
     "performance"
   ],
-  "main": "dist/index.cjs.js",
+  "main": "dist/index.cjs",
   "module": "dist/index.es5.js",
   "types": "dist/index.d.ts",
+  "type": "module",
+  "exports": {
+    "import": "./dist/index.es5.js",
+    "require": "./dist/index.cjs",
+    "types": "./dist/index.d.ts"
+  },
   "files": [
     "dist"
   ],
@@ -20,8 +26,8 @@
   "license": "Apache-2.0",
   "devDependencies": {
     "@types/stack-trace": "^0.0.30",
-    "jest-mock-extended": "^3.0.4",
-    "tinybench": "^2.5.0"
+    "tinybench": "^2.5.0",
+    "vitest": "^1.2.2"
   },
   "dependencies": {
     "@codspeed/core": "workspace:^4.0.1",

packages/tinybench-plugin/tests/__snapshots__/index.integ.test.ts.snap

@@ -1,6 +1,6 @@
-// Jest Snapshot v1, https://goo.gl/fbAQLP
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
 
-exports[`Benchmark.Suite check console output(instrumented=false)  1`] = `
+exports[`Benchmark.Suite > check console output(instrumented=%p)  false 1`] = `
 {
   "log": [],
   "warn": [
@@ -11,7 +11,7 @@ exports[`Benchmark.Suite check console output(instrumented=false)  1`] = `
 }
 `;
 
-exports[`Benchmark.Suite check console output(instrumented=true)  1`] = `
+exports[`Benchmark.Suite > check console output(instrumented=%p)  true 1`] = `
 {
   "log": [
     [

packages/tinybench-plugin/tests/index.integ.test.ts

@@ -1,27 +1,48 @@
-import { mockDeep, mockReset } from "jest-mock-extended";
-const mockCore = mockDeep<typeof core>();
-
-import * as core from "@codspeed/core";
 import { Bench } from "tinybench";
-import { withCodSpeed } from "..";
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { withCodSpeed } from "../src";
 import { registerBenchmarks } from "./registerBenchmarks";
 import { registerOtherBenchmarks } from "./registerOtherBenchmarks";
 
-jest.mock("@codspeed/core", () => {
-  mockCore.getGitDir = jest.requireActual("@codspeed/core").getGitDir;
-  return mockCore;
+const mockCore = vi.hoisted(() => {
+  return {
+    mongoMeasurement: {
+      start: vi.fn(),
+      stop: vi.fn(),
+      setupInstruments: vi.fn(),
+    },
+    Measurement: {
+      isInstrumented: vi.fn(),
+      startInstrumentation: vi.fn(),
+      stopInstrumentation: vi.fn(),
+    },
+    optimizeFunction: vi
+      .fn()
+      .mockImplementation(async (fn: () => Promise<void>) => {
+        await fn();
+      }),
+    setupCore: vi.fn(),
+    teardownCore: vi.fn(),
+  };
+});
+
+vi.mock("@codspeed/core", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("@codspeed/core")>();
+  return {
+    ...actual,
+    ...mockCore,
+  };
 });
 
 beforeEach(() => {
-  mockReset(mockCore);
-  jest.clearAllMocks();
+  vi.clearAllMocks();
 });
 
 describe("Benchmark.Suite", () => {
   it("simple suite", async () => {
     mockCore.Measurement.isInstrumented.mockReturnValue(false);
     const bench = withCodSpeed(new Bench({ time: 100 }));
-    const onComplete = jest.fn();
+    const onComplete = vi.fn();
     bench.add("RegExp", function () {
       /o/.test("Hello World!");
     });
@@ -87,8 +108,8 @@ describe("Benchmark.Suite", () => {
   it.each([true, false])(
     "check console output(instrumented=%p) ",
     async (instrumented) => {
-      const logSpy = jest.spyOn(console, "log");
-      const warnSpy = jest.spyOn(console, "warn");
+      const logSpy = vi.spyOn(console, "log");
+      const warnSpy = vi.spyOn(console, "warn");
       mockCore.Measurement.isInstrumented.mockReturnValue(instrumented);
       await withCodSpeed(new Bench({ time: 100 }))
         .add("RegExp", function () {
@@ -128,8 +149,8 @@ describe("Benchmark.Suite", () => {
     );
   });
   // TODO: this is not supported at the moment as tinybench does not support tasks with same name
-  // remove `.failing` when tinybench supports it
-  it.failing(
+  // remove `.fails` when tinybench supports it
+  it.fails(
     "check that benchmarks with same name have different URIs when registered in different files",
     async () => {
       mockCore.Measurement.isInstrumented.mockReturnValue(true);
@@ -152,10 +173,10 @@ describe("Benchmark.Suite", () => {
     mockCore.optimizeFunction.mockImplementation(async (fn) => {
       await fn();
     });
-    const beforeAll = jest.fn();
-    const beforeEach = jest.fn();
-    const afterEach = jest.fn();
-    const afterAll = jest.fn();
+    const beforeAll = vi.fn();
+    const beforeEach = vi.fn();
+    const afterEach = vi.fn();
+    const afterAll = vi.fn();
 
     await withCodSpeed(new Bench())
       .add(

packages/tinybench-plugin/tsconfig.test.json

@@ -1,7 +1,7 @@
 {
   "extends": "./tsconfig.json",
   "compilerOptions": {
-    "types": ["jest", "node"]
+    "types": ["vitest/globals", "node"]
   },
   "include": ["tests/**/*.ts", "benches/**/*.ts", "jest.config.ts"]
 }

packages/tinybench-plugin/vitest.config.ts

@@ -0,0 +1,11 @@
+import { defineConfig } from "vitest/config";
+
+export default defineConfig({
+  define: {
+    __VERSION__: JSON.stringify("1.0.0"),
+  },
+  test: {
+    exclude: ["**/node_modules/**", "**/.rollup.cache/**"],
+    mockReset: true,
+  },
+});