@openrewrite/recipes-testing 0.1.0-20260313-092538

package/dist/index.d.ts

@@ -0,0 +1,6 @@
+import { CategoryDescriptor, RecipeMarketplace } from "@openrewrite/rewrite";
+export declare const Testing: CategoryDescriptor[];
+export declare const Migrate: CategoryDescriptor[];
+export declare function activate(marketplace: RecipeMarketplace): Promise<void>;
+export { JestToVitest } from "./jest-to-vitest";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,kBAAkB,EAAE,iBAAiB,EAAC,MAAM,sBAAsB,CAAC;AAG3E,eAAO,MAAM,OAAO,EAAE,kBAAkB,EAA+B,CAAC;AACxE,eAAO,MAAM,OAAO,EAAE,kBAAkB,EAA2C,CAAC;AAEpF,wBAAsB,QAAQ,CAAC,WAAW,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,CAE5E;AAED,OAAO,EAAC,YAAY,EAAC,MAAM,kBAAkB,CAAC"}

package/dist/index.js

@@ -0,0 +1,24 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.JestToVitest = exports.Migrate = exports.Testing = void 0;
+exports.activate = activate;
+const jest_to_vitest_1 = require("./jest-to-vitest");
+exports.Testing = [{ displayName: "Testing" }];
+exports.Migrate = [...exports.Testing, { displayName: "Migrate" }];
+function activate(marketplace) {
+    return __awaiter(this, void 0, void 0, function* () {
+        yield marketplace.install(jest_to_vitest_1.JestToVitest, exports.Migrate);
+    });
+}
+var jest_to_vitest_2 = require("./jest-to-vitest");
+Object.defineProperty(exports, "JestToVitest", { enumerable: true, get: function () { return jest_to_vitest_2.JestToVitest; } });
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;AAMA,4BAEC;AAPD,qDAA8C;AAEjC,QAAA,OAAO,GAAyB,CAAC,EAAC,WAAW,EAAE,SAAS,EAAC,CAAC,CAAC;AAC3D,QAAA,OAAO,GAAyB,CAAC,GAAG,eAAO,EAAE,EAAC,WAAW,EAAE,SAAS,EAAC,CAAC,CAAC;AAEpF,SAAsB,QAAQ,CAAC,WAA8B;;QACzD,MAAM,WAAW,CAAC,OAAO,CAAC,6BAAY,EAAE,eAAO,CAAC,CAAC;IACrD,CAAC;CAAA;AAED,mDAA8C;AAAtC,8GAAA,YAAY,OAAA"}

package/dist/jest-to-vitest.d.ts

@@ -0,0 +1,8 @@
+import { ExecutionContext, Recipe, TreeVisitor } from "@openrewrite/rewrite";
+export declare class JestToVitest extends Recipe {
+    name: string;
+    displayName: string;
+    description: string;
+    editor(): Promise<TreeVisitor<any, ExecutionContext>>;
+}
+//# sourceMappingURL=jest-to-vitest.d.ts.map

package/dist/jest-to-vitest.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"jest-to-vitest.d.ts","sourceRoot":"","sources":["../src/jest-to-vitest.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,gBAAgB,EAAE,MAAM,EAAE,WAAW,EAAC,MAAM,sBAAsB,CAAC;AAyC3E,qBAAa,YAAa,SAAQ,MAAM;IACpC,IAAI,SAAqD;IACzD,WAAW,SAA4B;IACvC,WAAW,SAEiD;IAEtD,MAAM,IAAI,OAAO,CAAC,WAAW,CAAC,GAAG,EAAE,gBAAgB,CAAC,CAAC;CAG9D"}

package/dist/jest-to-vitest.js

@@ -0,0 +1,492 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.JestToVitest = void 0;
+const rewrite_1 = require("@openrewrite/rewrite");
+const javascript_1 = require("@openrewrite/rewrite/javascript");
+const java_1 = require("@openrewrite/rewrite/java");
+const JEST_CONFIG = {
+    dependencies: {
+        '@types/jest': '^29.5.13'
+    },
+};
+const VITEST_CONFIG = {
+    dependencies: {
+        'vitest': '^2.0.0'
+    },
+    context: ['import { vi } from "vitest";',]
+};
+class JestToVitest extends rewrite_1.Recipe {
+    constructor() {
+        super(...arguments);
+        this.name = "org.openrewrite.javascript.testing.JestToVitest";
+        this.displayName = "Migrate Jest to Vitest";
+        this.description = "Migrates Jest test files to use Vitest as the testing framework. " +
+            "This includes updating imports, transforming global functions, and converting " +
+            "Jest-specific mocking patterns to Vitest equivalents.";
+    }
+    editor() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return new JestToVitestVisitor();
+        });
+    }
+}
+exports.JestToVitest = JestToVitest;
+const transformationRules = {
+    fn: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.fn(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.fn(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    spyOn: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.spyOn(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.spyOn(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    mock: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const path = (0, javascript_1.capture)('path');
+            const literalBody = (0, javascript_1.capture)({
+                name: 'literalBody',
+                constraint: (node) => node.kind === java_1.J.Kind.Literal
+            });
+            return {
+                before: (0, javascript_1.pattern) `jest.mock(${path}, () => ${literalBody})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.mock(${path}, () => ({ default: ${literalBody} }))`.configure(VITEST_CONFIG)
+            };
+        }).orElse((0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.mock(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.mock(${args})`.configure(VITEST_CONFIG)
+            };
+        })),
+        imports: ['vi']
+    },
+    unmock: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.unmock(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.unmock(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    clearAllMocks: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.clearAllMocks()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.clearAllMocks()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    resetAllMocks: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.resetAllMocks()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.resetAllMocks()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    restoreAllMocks: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.restoreAllMocks()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.restoreAllMocks()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    useFakeTimers: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.useFakeTimers(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.useFakeTimers(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    useRealTimers: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.useRealTimers()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.useRealTimers()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    runAllTimers: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.runAllTimers()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.runAllTimers()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    runOnlyPendingTimers: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.runOnlyPendingTimers()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.runOnlyPendingTimers()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    advanceTimersByTime: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.advanceTimersByTime(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.advanceTimersByTime(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    advanceTimersToNextTimer: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.advanceTimersToNextTimer(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.advanceTimersToNextTimer(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    clearAllTimers: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.clearAllTimers()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.clearAllTimers()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    getTimerCount: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.getTimerCount()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.getTimerCount()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    setSystemTime: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.setSystemTime(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.setSystemTime(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    getRealSystemTime: {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.getRealSystemTime()`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `vi.getRealSystemTime()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    mocked: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.mocked(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.mocked(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    isMockFunction: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.isMockFunction(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.isMockFunction(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    doMock: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.doMock(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.doMock(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    doUnmock: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.doUnmock(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.doUnmock(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    requireActual: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.requireActual(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `await vi.importActual(${args})`.configure(VITEST_CONFIG),
+                preMatch: (_node, { cursor }) => {
+                    var _a, _b;
+                    const enclosingMethod = cursor.firstEnclosing((n) => n.kind === java_1.J.Kind.MethodDeclaration);
+                    if ((_a = enclosingMethod === null || enclosingMethod === void 0 ? void 0 : enclosingMethod.modifiers) === null || _a === void 0 ? void 0 : _a.some(m => m.keyword === 'async')) {
+                        return true;
+                    }
+                    const enclosingArrow = cursor.firstEnclosing((n) => n.kind === javascript_1.JS.Kind.ArrowFunction);
+                    if ((_b = enclosingArrow === null || enclosingArrow === void 0 ? void 0 : enclosingArrow.modifiers) === null || _b === void 0 ? void 0 : _b.some(m => m.keyword === 'async')) {
+                        return true;
+                    }
+                    return false;
+                }
+            };
+        }).orElse((0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `jest.requireActual(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.importActual(${args})`.configure(VITEST_CONFIG)
+            };
+        })),
+        imports: ['vi']
+    },
+    setTimeout: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const timeout = (0, javascript_1.capture)('timeout');
+            return {
+                before: (0, javascript_1.pattern) `jest.setTimeout(${timeout})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `vi.setConfig({ testTimeout: ${timeout} })`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    describe: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `describe(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `describe(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['describe']
+    },
+    it: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `it(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `it(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['it']
+    },
+    test: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `test(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `test(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['test']
+    },
+    expect: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `expect(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `expect(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['expect']
+    },
+    beforeEach: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `beforeEach(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `beforeEach(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['beforeEach']
+    },
+    afterEach: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `afterEach(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `afterEach(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['afterEach']
+    },
+    beforeAll: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `beforeAll(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `beforeAll(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['beforeAll']
+    },
+    afterAll: {
+        rule: (0, javascript_1.rewrite)(() => {
+            const args = (0, javascript_1.capture)({ variadic: true });
+            return {
+                before: (0, javascript_1.pattern) `afterAll(${args})`.configure(JEST_CONFIG),
+                after: (0, javascript_1.template) `afterAll(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['afterAll']
+    },
+};
+const transformationRuleKeys = new Set(Object.keys(transformationRules));
+function vitestTypeConfig(vitestType) {
+    return {
+        dependencies: { 'vitest': '^2.0.0' },
+        context: [`import { ${vitestType} } from "vitest";`]
+    };
+}
+const typeTransformationRules = [
+    {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.SpyInstance`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `MockInstance`.configure(vitestTypeConfig('MockInstance'))
+        })),
+        vitestImport: 'MockInstance'
+    },
+    {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.Mock`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `Mock`.configure(vitestTypeConfig('Mock'))
+        })),
+        vitestImport: 'Mock'
+    },
+    {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.Mocked`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `Mocked`.configure(vitestTypeConfig('Mocked'))
+        })),
+        vitestImport: 'Mocked'
+    },
+    {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.MockedFunction`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `MockedFunction`.configure(vitestTypeConfig('MockedFunction'))
+        })),
+        vitestImport: 'MockedFunction'
+    },
+    {
+        rule: (0, javascript_1.rewrite)(() => ({
+            before: (0, javascript_1.pattern) `jest.MockedClass`.configure(JEST_CONFIG),
+            after: (0, javascript_1.template) `MockedClass`.configure(vitestTypeConfig('MockedClass'))
+        })),
+        vitestImport: 'MockedClass'
+    },
+];
+class JestToVitestVisitor extends javascript_1.JavaScriptVisitor {
+    constructor() {
+        super(...arguments);
+        this.requiredImports = new Set();
+        this.requiredTypeImports = new Set();
+    }
+    tryTransformJestType(node) {
+        return __awaiter(this, void 0, void 0, function* () {
+            for (const typeRule of typeTransformationRules) {
+                const transformed = yield typeRule.rule.tryOn(this.cursor, node);
+                if (transformed) {
+                    this.requiredTypeImports.add(typeRule.vitestImport);
+                    return transformed;
+                }
+            }
+            return undefined;
+        });
+    }
+    visitFieldAccess(fieldAccess, ctx) {
+        const _super = Object.create(null, {
+            visitFieldAccess: { get: () => super.visitFieldAccess }
+        });
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            fieldAccess = (yield _super.visitFieldAccess.call(this, fieldAccess, ctx));
+            if (fieldAccess.target.kind === java_1.J.Kind.Identifier &&
+                fieldAccess.target.simpleName === 'jest') {
+                return (_a = yield this.tryTransformJestType(fieldAccess)) !== null && _a !== void 0 ? _a : fieldAccess;
+            }
+            return fieldAccess;
+        });
+    }
+    visitParameterizedType(parameterizedType, ctx) {
+        const _super = Object.create(null, {
+            visitParameterizedType: { get: () => super.visitParameterizedType }
+        });
+        return __awaiter(this, void 0, void 0, function* () {
+            parameterizedType = (yield _super.visitParameterizedType.call(this, parameterizedType, ctx));
+            if (parameterizedType.class.kind === java_1.J.Kind.FieldAccess) {
+                const fieldAccess = parameterizedType.class;
+                if (fieldAccess.target.kind === java_1.J.Kind.Identifier &&
+                    fieldAccess.target.simpleName === 'jest') {
+                    const transformed = yield this.tryTransformJestType(fieldAccess);
+                    if (transformed) {
+                        const updated = Object.assign(Object.assign({}, parameterizedType), { class: transformed });
+                        return updated;
+                    }
+                }
+            }
+            return parameterizedType;
+        });
+    }
+    visitMethodInvocation(method, ctx) {
+        const _super = Object.create(null, {
+            visitMethodInvocation: { get: () => super.visitMethodInvocation }
+        });
+        return __awaiter(this, void 0, void 0, function* () {
+            method = (yield _super.visitMethodInvocation.call(this, method, ctx));
+            const simpleName = method.name.simpleName;
+            if (transformationRuleKeys.has(simpleName)) {
+                const transformation = transformationRules[simpleName];
+                const transformed = yield transformation.rule.tryOn(this.cursor, method);
+                if (transformed) {
+                    for (const imp of transformation.imports) {
+                        this.requiredImports.add(imp);
+                    }
+                    return transformed;
+                }
+            }
+            return method;
+        });
+    }
+    visitJsCompilationUnit(cu, ctx) {
+        const _super = Object.create(null, {
+            visitJsCompilationUnit: { get: () => super.visitJsCompilationUnit }
+        });
+        return __awaiter(this, void 0, void 0, function* () {
+            this.requiredImports.clear();
+            this.requiredTypeImports.clear();
+            (0, javascript_1.maybeRemoveImport)(this, 'jest');
+            (0, javascript_1.maybeRemoveImport)(this, '@jest/globals');
+            cu = (yield _super.visitJsCompilationUnit.call(this, cu, ctx));
+            for (const member of this.requiredImports) {
+                (0, javascript_1.maybeAddImport)(this, { module: 'vitest', member });
+            }
+            for (const member of this.requiredTypeImports) {
+                (0, javascript_1.maybeAddImport)(this, { module: 'vitest', member, onlyIfReferenced: false });
+            }
+            return cu;
+        });
+    }
+}
+//# sourceMappingURL=jest-to-vitest.js.map

package/dist/jest-to-vitest.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"jest-to-vitest.js","sourceRoot":"","sources":["../src/jest-to-vitest.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,kDAA2E;AAC3E,gEASyC;AACzC,oDAA4C;AAI5C,MAAM,WAAW,GAAG;IAChB,YAAY,EAAE;QACV,aAAa,EAAE,UAAU;KAC5B;CACJ,CAAC;AAIF,MAAM,aAAa,GAAG;IAClB,YAAY,EAAE;QACV,QAAQ,EAAE,QAAQ;KACrB;IACD,OAAO,EAAE,CAAC,8BAA8B,EAAE;CAC7C,CAAC;AAaF,MAAa,YAAa,SAAQ,gBAAM;IAAxC;;QACI,SAAI,GAAG,iDAAiD,CAAC;QACzD,gBAAW,GAAG,wBAAwB,CAAC;QACvC,gBAAW,GAAG,mEAAmE;YAC7E,gFAAgF;YAChF,uDAAuD,CAAC;IAKhE,CAAC;IAHS,MAAM;;YACR,OAAO,IAAI,mBAAmB,EAAE,CAAC;QACrC,CAAC;KAAA;CACJ;AAVD,oCAUC;AAYD,MAAM,mBAAmB,GAAuC;IAE5D,EAAE,EAAE;QACA,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,WAAW,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACxD,KAAK,EAAE,IAAA,qBAAQ,EAAA,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC3D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,KAAK,EAAE;QACH,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,cAAc,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC3D,KAAK,EAAE,IAAA,qBAAQ,EAAA,YAAY,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC9D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,IAAI,EAAE;QACF,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,MAAM,CAAC,CAAC;YAC7B,MAAM,WAAW,GAAG,IAAA,oBAAO,EAAY;gBACnC,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,QAAC,CAAC,IAAI,CAAC,OAAO;aACrD,CAAC,CAAC;YAGH,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,aAAa,IAAI,WAAW,WAAW,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAChF,KAAK,EAAE,IAAA,qBAAQ,EAAA,WAAW,IAAI,uBAAuB,WAAW,MAAM,CAAC,SAAS,CAAC,aAAa,CAAC;aAClG,CAAC;QACN,CAAC,CAAC,CAAC,MAAM,CAAC,IAAA,oBAAO,EAAC,GAAG,EAAE;YAEnB,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC1D,KAAK,EAAE,IAAA,qBAAQ,EAAA,WAAW,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC7D,CAAC;QACN,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,MAAM,EAAE;QACJ,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,eAAe,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC5D,KAAK,EAAE,IAAA,qBAAQ,EAAA,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC/D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,aAAa,EAAE;QACX,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,sBAAsB,CAAC,SAAS,CAAC,WAAW,CAAC;YAC5D,KAAK,EAAE,IAAA,qBAAQ,EAAA,oBAAoB,CAAC,SAAS,CAAC,aAAa,CAAC;SAC/D,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,aAAa,EAAE;QACX,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,sBAAsB,CAAC,SAAS,CAAC,WAAW,CAAC;YAC5D,KAAK,EAAE,IAAA,qBAAQ,EAAA,oBAAoB,CAAC,SAAS,CAAC,aAAa,CAAC;SAC/D,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,eAAe,EAAE;QACb,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,wBAAwB,CAAC,SAAS,CAAC,WAAW,CAAC;YAC9D,KAAK,EAAE,IAAA,qBAAQ,EAAA,sBAAsB,CAAC,SAAS,CAAC,aAAa,CAAC;SACjE,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,aAAa,EAAE;QACX,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,sBAAsB,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACnE,KAAK,EAAE,IAAA,qBAAQ,EAAA,oBAAoB,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aACtE,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,aAAa,EAAE;QACX,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,sBAAsB,CAAC,SAAS,CAAC,WAAW,CAAC;YAC5D,KAAK,EAAE,IAAA,qBAAQ,EAAA,oBAAoB,CAAC,SAAS,CAAC,aAAa,CAAC;SAC/D,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,YAAY,EAAE;QACV,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,qBAAqB,CAAC,SAAS,CAAC,WAAW,CAAC;YAC3D,KAAK,EAAE,IAAA,qBAAQ,EAAA,mBAAmB,CAAC,SAAS,CAAC,aAAa,CAAC;SAC9D,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,oBAAoB,EAAE;QAClB,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,6BAA6B,CAAC,SAAS,CAAC,WAAW,CAAC;YACnE,KAAK,EAAE,IAAA,qBAAQ,EAAA,2BAA2B,CAAC,SAAS,CAAC,aAAa,CAAC;SACtE,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,mBAAmB,EAAE;QACjB,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,4BAA4B,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACzE,KAAK,EAAE,IAAA,qBAAQ,EAAA,0BAA0B,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC5E,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,wBAAwB,EAAE;QACtB,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,iCAAiC,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC9E,KAAK,EAAE,IAAA,qBAAQ,EAAA,+BAA+B,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aACjF,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,cAAc,EAAE;QACZ,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,uBAAuB,CAAC,SAAS,CAAC,WAAW,CAAC;YAC7D,KAAK,EAAE,IAAA,qBAAQ,EAAA,qBAAqB,CAAC,SAAS,CAAC,aAAa,CAAC;SAChE,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,aAAa,EAAE;QACX,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,sBAAsB,CAAC,SAAS,CAAC,WAAW,CAAC;YAC5D,KAAK,EAAE,IAAA,qBAAQ,EAAA,oBAAoB,CAAC,SAAS,CAAC,aAAa,CAAC;SAC/D,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,aAAa,EAAE;QACX,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,sBAAsB,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACnE,KAAK,EAAE,IAAA,qBAAQ,EAAA,oBAAoB,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aACtE,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,iBAAiB,EAAE;QACf,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,0BAA0B,CAAC,SAAS,CAAC,WAAW,CAAC;YAChE,KAAK,EAAE,IAAA,qBAAQ,EAAA,wBAAwB,CAAC,SAAS,CAAC,aAAa,CAAC;SACnE,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,MAAM,EAAE;QACJ,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,eAAe,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC5D,KAAK,EAAE,IAAA,qBAAQ,EAAA,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC/D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,cAAc,EAAE;QACZ,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,uBAAuB,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACpE,KAAK,EAAE,IAAA,qBAAQ,EAAA,qBAAqB,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aACvE,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,MAAM,EAAE;QACJ,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,eAAe,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC5D,KAAK,EAAE,IAAA,qBAAQ,EAAA,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC/D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,QAAQ,EAAE;QACN,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,iBAAiB,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC9D,KAAK,EAAE,IAAA,qBAAQ,EAAA,eAAe,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aACjE,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IAGD,aAAa,EAAE;QACX,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,sBAAsB,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACnE,KAAK,EAAE,IAAA,qBAAQ,EAAA,yBAAyB,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;gBACxE,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAC,MAAM,EAAC,EAAE,EAAE;;oBAE1B,MAAM,eAAe,GAAG,MAAM,CAAC,cAAc,CACzC,CAAC,CAAC,EAA4B,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAC,CAAC,IAAI,CAAC,iBAAiB,CACvE,CAAC;oBACF,IAAI,MAAA,eAAe,aAAf,eAAe,uBAAf,eAAe,CAAE,SAAS,0CAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,OAAO,CAAC,EAAE,CAAC;wBAC/D,OAAO,IAAI,CAAC;oBAChB,CAAC;oBACD,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,CACxC,CAAC,CAAC,EAAyB,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,eAAE,CAAC,IAAI,CAAC,aAAa,CACjE,CAAC;oBACF,IAAI,MAAA,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,SAAS,0CAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,OAAO,CAAC,EAAE,CAAC;wBAC9D,OAAO,IAAI,CAAC;oBAChB,CAAC;oBACD,OAAO,KAAK,CAAC;gBACjB,CAAC;aACJ,CAAC;QACN,CAAC,CAAC,CAAC,MAAM,CAAC,IAAA,oBAAO,EAAC,GAAG,EAAE;YAEnB,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,sBAAsB,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACnE,KAAK,EAAE,IAAA,qBAAQ,EAAA,mBAAmB,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aACrE,CAAC;QACN,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,UAAU,EAAE;QACR,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,OAAO,GAAG,IAAA,oBAAO,EAAC,SAAS,CAAC,CAAC;YACnC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,mBAAmB,OAAO,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACnE,KAAK,EAAE,IAAA,qBAAQ,EAAA,+BAA+B,OAAO,KAAK,CAAC,SAAS,CAAC,aAAa,CAAC;aACtF,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IAGD,QAAQ,EAAE;QACN,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,YAAY,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACzD,KAAK,EAAE,IAAA,qBAAQ,EAAA,YAAY,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC9D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,UAAU,CAAC;KACxB;IACD,EAAE,EAAE;QACA,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,MAAM,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACnD,KAAK,EAAE,IAAA,qBAAQ,EAAA,MAAM,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aACxD,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,CAAC;KAClB;IACD,IAAI,EAAE;QACF,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,QAAQ,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACrD,KAAK,EAAE,IAAA,qBAAQ,EAAA,QAAQ,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC1D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,MAAM,CAAC;KACpB;IACD,MAAM,EAAE;QACJ,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,UAAU,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACvD,KAAK,EAAE,IAAA,qBAAQ,EAAA,UAAU,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC5D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,QAAQ,CAAC;KACtB;IACD,UAAU,EAAE;QACR,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,cAAc,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC3D,KAAK,EAAE,IAAA,qBAAQ,EAAA,cAAc,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAChE,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,YAAY,CAAC;KAC1B;IACD,SAAS,EAAE;QACP,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC1D,KAAK,EAAE,IAAA,qBAAQ,EAAA,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC/D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,WAAW,CAAC;KACzB;IACD,SAAS,EAAE;QACP,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBAC1D,KAAK,EAAE,IAAA,qBAAQ,EAAA,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC/D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,WAAW,CAAC;KACzB;IACD,QAAQ,EAAE;QACN,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE;YACf,MAAM,IAAI,GAAG,IAAA,oBAAO,EAAC,EAAC,QAAQ,EAAE,IAAI,EAAC,CAAC,CAAC;YACvC,OAAO;gBACH,MAAM,EAAE,IAAA,oBAAO,EAAA,YAAY,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC;gBACzD,KAAK,EAAE,IAAA,qBAAQ,EAAA,YAAY,IAAI,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;aAC9D,CAAC;QACN,CAAC,CAAC;QACF,OAAO,EAAE,CAAC,UAAU,CAAC;KACxB;CACJ,CAAC;AAGF,MAAM,sBAAsB,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC;AAGzE,SAAS,gBAAgB,CAAC,UAAkB;IACxC,OAAO;QACH,YAAY,EAAE,EAAC,QAAQ,EAAE,QAAQ,EAAC;QAClC,OAAO,EAAE,CAAC,YAAY,UAAU,mBAAmB,CAAC;KACvD,CAAC;AACN,CAAC;AASD,MAAM,uBAAuB,GAA6B;IACtD;QACI,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,kBAAkB,CAAC,SAAS,CAAC,WAAW,CAAC;YACxD,KAAK,EAAE,IAAA,qBAAQ,EAAA,cAAc,CAAC,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;SAC5E,CAAC,CAAC;QACH,YAAY,EAAE,cAAc;KAC/B;IACD;QACI,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,WAAW,CAAC,SAAS,CAAC,WAAW,CAAC;YACjD,KAAK,EAAE,IAAA,qBAAQ,EAAA,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;SAC5D,CAAC,CAAC;QACH,YAAY,EAAE,MAAM;KACvB;IACD;QACI,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,aAAa,CAAC,SAAS,CAAC,WAAW,CAAC;YACnD,KAAK,EAAE,IAAA,qBAAQ,EAAA,QAAQ,CAAC,SAAS,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;SAChE,CAAC,CAAC;QACH,YAAY,EAAE,QAAQ;KACzB;IACD;QACI,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,qBAAqB,CAAC,SAAS,CAAC,WAAW,CAAC;YAC3D,KAAK,EAAE,IAAA,qBAAQ,EAAA,gBAAgB,CAAC,SAAS,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;SAChF,CAAC,CAAC;QACH,YAAY,EAAE,gBAAgB;KACjC;IACD;QACI,IAAI,EAAE,IAAA,oBAAO,EAAC,GAAG,EAAE,CAAC,CAAC;YACjB,MAAM,EAAE,IAAA,oBAAO,EAAA,kBAAkB,CAAC,SAAS,CAAC,WAAW,CAAC;YACxD,KAAK,EAAE,IAAA,qBAAQ,EAAA,aAAa,CAAC,SAAS,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;SAC1E,CAAC,CAAC;QACH,YAAY,EAAE,aAAa;KAC9B;CACJ,CAAC;AAEF,MAAM,mBAAoB,SAAQ,8BAAmC;IAArE;;QAEY,oBAAe,GAAG,IAAI,GAAG,EAAU,CAAC;QAEpC,wBAAmB,GAAG,IAAI,GAAG,EAAU,CAAC;IA0GpD,CAAC;IAxGiB,oBAAoB,CAAC,IAAO;;YACtC,KAAK,MAAM,QAAQ,IAAI,uBAAuB,EAAE,CAAC;gBAC7C,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;gBACjE,IAAI,WAAW,EAAE,CAAC;oBACd,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;oBACpD,OAAO,WAAW,CAAC;gBACvB,CAAC;YACL,CAAC;YACD,OAAO,SAAS,CAAC;QACrB,CAAC;KAAA;IAEe,gBAAgB,CAC5B,WAA0B,EAC1B,GAAqB;;;;;;YAErB,WAAW,IAAG,MAAM,OAAM,gBAAgB,YAAC,WAAW,EAAE,GAAG,CAAkB,CAAA,CAAC;YAG9E,IAAI,WAAW,CAAC,MAAM,CAAC,IAAI,KAAK,QAAC,CAAC,IAAI,CAAC,UAAU;gBAC5C,WAAW,CAAC,MAAuB,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;gBAC7D,OAAO,MAAA,MAAM,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC,mCAAI,WAAW,CAAC;YACvE,CAAC;YAED,OAAO,WAAW,CAAC;QACvB,CAAC;KAAA;IAEe,sBAAsB,CAClC,iBAAsC,EACtC,GAAqB;;;;;YAErB,iBAAiB,IAAG,MAAM,OAAM,sBAAsB,YAAC,iBAAiB,EAAE,GAAG,CAAwB,CAAA,CAAC;YAGtG,IAAI,iBAAiB,CAAC,KAAK,CAAC,IAAI,KAAK,QAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACtD,MAAM,WAAW,GAAG,iBAAiB,CAAC,KAAsB,CAAC;gBAC7D,IAAI,WAAW,CAAC,MAAM,CAAC,IAAI,KAAK,QAAC,CAAC,IAAI,CAAC,UAAU;oBAC5C,WAAW,CAAC,MAAuB,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;oBAC7D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;oBACjE,IAAI,WAAW,EAAE,CAAC;wBACd,MAAM,OAAO,mCACN,iBAAwC,KAC3C,KAAK,EAAE,WAA2C,GACrD,CAAC;wBACF,OAAO,OAAO,CAAC;oBACnB,CAAC;gBACL,CAAC;YACL,CAAC;YAED,OAAO,iBAAiB,CAAC;QAC7B,CAAC;KAAA;IAEe,qBAAqB,CACjC,MAA0B,EAC1B,GAAqB;;;;;YAGrB,MAAM,IAAG,MAAM,OAAM,qBAAqB,YAAC,MAAM,EAAE,GAAG,CAAuB,CAAA,CAAC;YAG9E,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC;YAG1C,IAAI,sBAAsB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;gBACzC,MAAM,cAAc,GAAG,mBAAmB,CAAC,UAA8C,CAAC,CAAC;gBAC3F,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACzE,IAAI,WAAW,EAAE,CAAC;oBAEd,KAAK,MAAM,GAAG,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC;wBACvC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oBAClC,CAAC;oBACD,OAAO,WAAW,CAAC;gBACvB,CAAC;YACL,CAAC;YAED,OAAO,MAAM,CAAC;QAClB,CAAC;KAAA;IAEe,sBAAsB,CAClC,EAAsB,EACtB,GAAqB;;;;;YAGrB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC7B,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAGjC,IAAA,8BAAiB,EAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAChC,IAAA,8BAAiB,EAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YAGzC,EAAE,IAAG,MAAM,OAAM,sBAAsB,YAAC,EAAE,EAAE,GAAG,CAAuB,CAAA,CAAC;YAGvE,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBACxC,IAAA,2BAAc,EAAC,IAAI,EAAE,EAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAC,CAAC,CAAC;YACrD,CAAC;YAGD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC5C,IAAA,2BAAc,EAAC,IAAI,EAAE,EAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAC,CAAC,CAAC;YAC9E,CAAC;YAED,OAAO,EAAE,CAAC;QACd,CAAC;KAAA;CACJ"}

package/package.json

@@ -0,0 +1,39 @@
+{
+  "name": "@openrewrite/recipes-testing",
+  "version": "0.1.0-20260313-092538",
+  "license": "Moderne Source Available License",
+  "description": "OpenRewrite recipes for JavaScript/TypeScript testing frameworks.",
+  "homepage": "https://github.com/moderneinc/recipes-javascript",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist/**",
+    "src/**"
+  ],
+  "exports": {
+    ".": "./dist/index.js"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "rm -rf ./dist tsconfig.build.tsbuildinfo && tsc --build tsconfig.build.json",
+    "dev": "tsc --watch -p tsconfig.json",
+    "test": "npm run build && jest",
+    "ci:test": "jest"
+  },
+  "dependencies": {
+    "@openrewrite/rewrite": "next",
+    "immer": "^10.1.1"
+  },
+  "devDependencies": {
+    "@types/jest": "^29.5.13",
+    "@types/node": "^22.5.4",
+    "jest": "^29.7.0",
+    "jest-junit": "^16.0.0",
+    "tmp-promise": "^3.0.3",
+    "ts-jest": "^29.2.5",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.6.2"
+  }
+}

package/src/index.ts

@@ -0,0 +1,11 @@
+import {CategoryDescriptor, RecipeMarketplace} from "@openrewrite/rewrite";
+import {JestToVitest} from "./jest-to-vitest";
+
+export const Testing: CategoryDescriptor[] = [{displayName: "Testing"}];
+export const Migrate: CategoryDescriptor[] = [...Testing, {displayName: "Migrate"}];
+
+export async function activate(marketplace: RecipeMarketplace): Promise<void> {
+    await marketplace.install(JestToVitest, Migrate);
+}
+
+export {JestToVitest} from "./jest-to-vitest";

package/src/jest-to-vitest.ts

@@ -0,0 +1,557 @@
+import {ExecutionContext, Recipe, TreeVisitor} from "@openrewrite/rewrite";
+import {
+    capture,
+    JavaScriptVisitor,
+    JS,
+    maybeAddImport,
+    maybeRemoveImport,
+    pattern,
+    rewrite,
+    template
+} from "@openrewrite/rewrite/javascript";
+import {J} from "@openrewrite/rewrite/java";
+
+// Configuration for Jest patterns - for type attribution
+// Note: Jest globals are available without imports, so we only specify dependencies
+const JEST_CONFIG = {
+    dependencies: {
+        '@types/jest': '^29.5.13'
+    },
+};
+
+// Configuration for Vitest templates - for type attribution
+// Note: Vitest requires explicit imports
+const VITEST_CONFIG = {
+    dependencies: {
+        'vitest': '^2.0.0'
+    },
+    context: ['import { vi } from "vitest";',]
+};
+
+/**
+ * Migrates Jest test files to Vitest.
+ *
+ * This recipe handles:
+ * - Renaming jest imports to vitest
+ * - Transforming Jest global functions (describe, it, test, expect, etc.) to use Vitest
+ * - Converting Jest mock patterns to Vitest equivalents
+ * - Updating jest.fn(), jest.spyOn(), etc. to vi.fn(), vi.spyOn()
+ * - Transforming jest.mock() to vi.mock()
+ * - Migrating Jest type references (jest.SpyInstance, jest.Mock, jest.Mocked, etc.) to Vitest equivalents
+ */
+export class JestToVitest extends Recipe {
+    name = "org.openrewrite.javascript.testing.JestToVitest";
+    displayName = "Migrate Jest to Vitest";
+    description = "Migrates Jest test files to use Vitest as the testing framework. " +
+        "This includes updating imports, transforming global functions, and converting " +
+        "Jest-specific mocking patterns to Vitest equivalents.";
+
+    async editor(): Promise<TreeVisitor<any, ExecutionContext>> {
+        return new JestToVitestVisitor();
+    }
+}
+
+/**
+ * A transformation rule with its associated import requirements
+ */
+interface TransformationRule {
+    rule: ReturnType<typeof rewrite>;
+    imports: string[];  // Vitest members to import when this rule matches
+}
+
+// Transformation rules organized by the method/function name being invoked
+// The key is the simple name of the method (e.g., 'fn' for jest.fn(), 'describe' for describe())
+const transformationRules: Record<string, TransformationRule> = {
+    // Jest object method transformations (jest.X() -> vi.X())
+    fn: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.fn(${args})`.configure(JEST_CONFIG),
+                after: template`vi.fn(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    spyOn: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.spyOn(${args})`.configure(JEST_CONFIG),
+                after: template`vi.spyOn(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    mock: {
+        rule: rewrite(() => {
+            const path = capture('path');
+            const literalBody = capture<J.Literal>({
+                name: 'literalBody',
+                constraint: (node) => node.kind === J.Kind.Literal
+            });
+            // Match mock with arrow factory that returns a literal value
+            // jest.mock('./path', () => 'value') -> vi.mock('./path', () => ({ default: 'value' }))
+            return {
+                before: pattern`jest.mock(${path}, () => ${literalBody})`.configure(JEST_CONFIG),
+                after: template`vi.mock(${path}, () => ({ default: ${literalBody} }))`.configure(VITEST_CONFIG)
+            };
+        }).orElse(rewrite(() => {
+            // Fallback: general mock transformation for other cases (including object literals)
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.mock(${args})`.configure(JEST_CONFIG),
+                after: template`vi.mock(${args})`.configure(VITEST_CONFIG)
+            };
+        })),
+        imports: ['vi']
+    },
+    unmock: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.unmock(${args})`.configure(JEST_CONFIG),
+                after: template`vi.unmock(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    clearAllMocks: {
+        rule: rewrite(() => ({
+            before: pattern`jest.clearAllMocks()`.configure(JEST_CONFIG),
+            after: template`vi.clearAllMocks()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    resetAllMocks: {
+        rule: rewrite(() => ({
+            before: pattern`jest.resetAllMocks()`.configure(JEST_CONFIG),
+            after: template`vi.resetAllMocks()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    restoreAllMocks: {
+        rule: rewrite(() => ({
+            before: pattern`jest.restoreAllMocks()`.configure(JEST_CONFIG),
+            after: template`vi.restoreAllMocks()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    useFakeTimers: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.useFakeTimers(${args})`.configure(JEST_CONFIG),
+                after: template`vi.useFakeTimers(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    useRealTimers: {
+        rule: rewrite(() => ({
+            before: pattern`jest.useRealTimers()`.configure(JEST_CONFIG),
+            after: template`vi.useRealTimers()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    runAllTimers: {
+        rule: rewrite(() => ({
+            before: pattern`jest.runAllTimers()`.configure(JEST_CONFIG),
+            after: template`vi.runAllTimers()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    runOnlyPendingTimers: {
+        rule: rewrite(() => ({
+            before: pattern`jest.runOnlyPendingTimers()`.configure(JEST_CONFIG),
+            after: template`vi.runOnlyPendingTimers()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    advanceTimersByTime: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.advanceTimersByTime(${args})`.configure(JEST_CONFIG),
+                after: template`vi.advanceTimersByTime(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    advanceTimersToNextTimer: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.advanceTimersToNextTimer(${args})`.configure(JEST_CONFIG),
+                after: template`vi.advanceTimersToNextTimer(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    clearAllTimers: {
+        rule: rewrite(() => ({
+            before: pattern`jest.clearAllTimers()`.configure(JEST_CONFIG),
+            after: template`vi.clearAllTimers()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    getTimerCount: {
+        rule: rewrite(() => ({
+            before: pattern`jest.getTimerCount()`.configure(JEST_CONFIG),
+            after: template`vi.getTimerCount()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    setSystemTime: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.setSystemTime(${args})`.configure(JEST_CONFIG),
+                after: template`vi.setSystemTime(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    getRealSystemTime: {
+        rule: rewrite(() => ({
+            before: pattern`jest.getRealSystemTime()`.configure(JEST_CONFIG),
+            after: template`vi.getRealSystemTime()`.configure(VITEST_CONFIG)
+        })),
+        imports: ['vi']
+    },
+    mocked: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.mocked(${args})`.configure(JEST_CONFIG),
+                after: template`vi.mocked(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    isMockFunction: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.isMockFunction(${args})`.configure(JEST_CONFIG),
+                after: template`vi.isMockFunction(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    doMock: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.doMock(${args})`.configure(JEST_CONFIG),
+                after: template`vi.doMock(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    doUnmock: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.doUnmock(${args})`.configure(JEST_CONFIG),
+                after: template`vi.doUnmock(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+    // vi.importActual() is async unlike jest.requireActual()
+    // When in async context, we add await; otherwise just rename and let developer handle it
+    requireActual: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.requireActual(${args})`.configure(JEST_CONFIG),
+                after: template`await vi.importActual(${args})`.configure(VITEST_CONFIG),
+                preMatch: (_node, {cursor}) => {
+                    // Check if we're inside an async function or async arrow function
+                    const enclosingMethod = cursor.firstEnclosing(
+                        (n): n is J.MethodDeclaration => n.kind === J.Kind.MethodDeclaration
+                    );
+                    if (enclosingMethod?.modifiers?.some(m => m.keyword === 'async')) {
+                        return true;
+                    }
+                    const enclosingArrow = cursor.firstEnclosing(
+                        (n): n is JS.ArrowFunction => n.kind === JS.Kind.ArrowFunction
+                    );
+                    if (enclosingArrow?.modifiers?.some(m => m.keyword === 'async')) {
+                        return true;
+                    }
+                    return false;
+                }
+            };
+        }).orElse(rewrite(() => {
+            // Fallback: not in async context, just rename without await
+            const args = capture({variadic: true});
+            return {
+                before: pattern`jest.requireActual(${args})`.configure(JEST_CONFIG),
+                after: template`vi.importActual(${args})`.configure(VITEST_CONFIG)
+            };
+        })),
+        imports: ['vi']
+    },
+    setTimeout: {
+        rule: rewrite(() => {
+            const timeout = capture('timeout');
+            return {
+                before: pattern`jest.setTimeout(${timeout})`.configure(JEST_CONFIG),
+                after: template`vi.setConfig({ testTimeout: ${timeout} })`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['vi']
+    },
+
+    // Jest global function transformations (re-attribute types from Jest to Vitest)
+    describe: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`describe(${args})`.configure(JEST_CONFIG),
+                after: template`describe(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['describe']
+    },
+    it: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`it(${args})`.configure(JEST_CONFIG),
+                after: template`it(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['it']
+    },
+    test: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`test(${args})`.configure(JEST_CONFIG),
+                after: template`test(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['test']
+    },
+    expect: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`expect(${args})`.configure(JEST_CONFIG),
+                after: template`expect(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['expect']
+    },
+    beforeEach: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`beforeEach(${args})`.configure(JEST_CONFIG),
+                after: template`beforeEach(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['beforeEach']
+    },
+    afterEach: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`afterEach(${args})`.configure(JEST_CONFIG),
+                after: template`afterEach(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['afterEach']
+    },
+    beforeAll: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`beforeAll(${args})`.configure(JEST_CONFIG),
+                after: template`beforeAll(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['beforeAll']
+    },
+    afterAll: {
+        rule: rewrite(() => {
+            const args = capture({variadic: true});
+            return {
+                before: pattern`afterAll(${args})`.configure(JEST_CONFIG),
+                after: template`afterAll(${args})`.configure(VITEST_CONFIG)
+            };
+        }),
+        imports: ['afterAll']
+    },
+};
+
+// Pre-compute the set of valid transformation rule keys to avoid prototype chain issues
+const transformationRuleKeys = new Set(Object.keys(transformationRules));
+
+// Helper to create a Vitest config with the right import context for a given type
+function vitestTypeConfig(vitestType: string) {
+    return {
+        dependencies: {'vitest': '^2.0.0'},
+        context: [`import { ${vitestType} } from "vitest";`]
+    };
+}
+
+// Jest type reference transformation rules (jest.X → VitestType)
+// These handle type annotations like `let spy: jest.SpyInstance`
+interface TypeTransformationRule {
+    rule: ReturnType<typeof rewrite>;
+    vitestImport: string;  // The vitest type to import
+}
+
+const typeTransformationRules: TypeTransformationRule[] = [
+    {
+        rule: rewrite(() => ({
+            before: pattern`jest.SpyInstance`.configure(JEST_CONFIG),
+            after: template`MockInstance`.configure(vitestTypeConfig('MockInstance'))
+        })),
+        vitestImport: 'MockInstance'
+    },
+    {
+        rule: rewrite(() => ({
+            before: pattern`jest.Mock`.configure(JEST_CONFIG),
+            after: template`Mock`.configure(vitestTypeConfig('Mock'))
+        })),
+        vitestImport: 'Mock'
+    },
+    {
+        rule: rewrite(() => ({
+            before: pattern`jest.Mocked`.configure(JEST_CONFIG),
+            after: template`Mocked`.configure(vitestTypeConfig('Mocked'))
+        })),
+        vitestImport: 'Mocked'
+    },
+    {
+        rule: rewrite(() => ({
+            before: pattern`jest.MockedFunction`.configure(JEST_CONFIG),
+            after: template`MockedFunction`.configure(vitestTypeConfig('MockedFunction'))
+        })),
+        vitestImport: 'MockedFunction'
+    },
+    {
+        rule: rewrite(() => ({
+            before: pattern`jest.MockedClass`.configure(JEST_CONFIG),
+            after: template`MockedClass`.configure(vitestTypeConfig('MockedClass'))
+        })),
+        vitestImport: 'MockedClass'
+    },
+];
+
+class JestToVitestVisitor extends JavaScriptVisitor<ExecutionContext> {
+    // Track which imports are needed based on transformations made
+    private requiredImports = new Set<string>();
+    // Track type imports separately (need onlyIfReferenced: false since type attribution may differ)
+    private requiredTypeImports = new Set<string>();
+
+    private async tryTransformJestType(node: J): Promise<J | undefined> {
+        for (const typeRule of typeTransformationRules) {
+            const transformed = await typeRule.rule.tryOn(this.cursor, node);
+            if (transformed) {
+                this.requiredTypeImports.add(typeRule.vitestImport);
+                return transformed;
+            }
+        }
+        return undefined;
+    }
+
+    protected async visitFieldAccess(
+        fieldAccess: J.FieldAccess,
+        ctx: ExecutionContext
+    ): Promise<J | undefined> {
+        fieldAccess = await super.visitFieldAccess(fieldAccess, ctx) as J.FieldAccess;
+
+        // Only try type transformations on jest.* field accesses
+        if (fieldAccess.target.kind === J.Kind.Identifier &&
+            (fieldAccess.target as J.Identifier).simpleName === 'jest') {
+            return await this.tryTransformJestType(fieldAccess) ?? fieldAccess;
+        }
+
+        return fieldAccess;
+    }
+
+    protected async visitParameterizedType(
+        parameterizedType: J.ParameterizedType,
+        ctx: ExecutionContext
+    ): Promise<J | undefined> {
+        parameterizedType = await super.visitParameterizedType(parameterizedType, ctx) as J.ParameterizedType;
+
+        // Check if the class name part is a jest type (e.g., jest.Mocked in jest.Mocked<T>)
+        if (parameterizedType.class.kind === J.Kind.FieldAccess) {
+            const fieldAccess = parameterizedType.class as J.FieldAccess;
+            if (fieldAccess.target.kind === J.Kind.Identifier &&
+                (fieldAccess.target as J.Identifier).simpleName === 'jest') {
+                const transformed = await this.tryTransformJestType(fieldAccess);
+                if (transformed) {
+                    const updated: J.ParameterizedType = {
+                        ...parameterizedType as J.ParameterizedType,
+                        class: transformed as J.ParameterizedType['class']
+                    };
+                    return updated;
+                }
+            }
+        }
+
+        return parameterizedType;
+    }
+
+    protected async visitMethodInvocation(
+        method: J.MethodInvocation,
+        ctx: ExecutionContext
+    ): Promise<J | undefined> {
+        // Visit children first
+        method = await super.visitMethodInvocation(method, ctx) as J.MethodInvocation;
+
+        // Get the simple name of the method being invoked
+        const simpleName = method.name.simpleName;
+
+        // Look up the transformation rule by name (using Set to avoid prototype chain issues)
+        if (transformationRuleKeys.has(simpleName)) {
+            const transformation = transformationRules[simpleName as keyof typeof transformationRules];
+            const transformed = await transformation.rule.tryOn(this.cursor, method);
+            if (transformed) {
+                // Track the imports needed for this transformation
+                for (const imp of transformation.imports) {
+                    this.requiredImports.add(imp);
+                }
+                return transformed;
+            }
+        }
+
+        return method;
+    }
+
+    protected async visitJsCompilationUnit(
+        cu: JS.CompilationUnit,
+        ctx: ExecutionContext
+    ): Promise<J | undefined> {
+        // Reset state for this file
+        this.requiredImports.clear();
+        this.requiredTypeImports.clear();
+
+        // Remove Jest imports
+        maybeRemoveImport(this, 'jest');
+        maybeRemoveImport(this, '@jest/globals');
+
+        // Visit the tree to collect required imports
+        cu = await super.visitJsCompilationUnit(cu, ctx) as JS.CompilationUnit;
+
+        // Add the required Vitest imports
+        for (const member of this.requiredImports) {
+            maybeAddImport(this, {module: 'vitest', member});
+        }
+        // Type imports use onlyIfReferenced: false because the type attribution
+        // on the replacement identifier may not match what AddImport expects
+        for (const member of this.requiredTypeImports) {
+            maybeAddImport(this, {module: 'vitest', member, onlyIfReferenced: false});
+        }
+
+        return cu;
+    }
+}