@eggjs/tegg-vitest 0.0.0 → 3.73.0-beta.0

package/README.md

@@ -0,0 +1,64 @@
+# @eggjs/tegg-vitest
+
+Vitest adapter that provides tegg context injection and lifecycle handling via a custom Vitest runner.
+
+## Install
+
+This package lives in the tegg monorepo workspace.
+
+## Usage
+
+1. Create a Vitest setup file that calls `configureTeggRunner`:
+
+```ts
+// vitest.setup.ts
+import path from 'path';
+import mm from 'egg-mock';
+import { configureTeggRunner } from '@eggjs/tegg-vitest';
+
+const app = mm.app({
+  baseDir: path.join(__dirname, 'fixtures/apps/my-app'),
+  framework: require.resolve('egg'),
+});
+
+configureTeggRunner({
+  getApp: () => app,
+  restoreMocks: true,
+  parallel: process.env.VITEST_WORKER_ID != null,
+});
+```
+
+2. Wire it in `vitest.config.ts` with the custom runner:
+
+```ts
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    environment: 'node',
+    setupFiles: ['./vitest.setup.ts'],
+    runner: '@eggjs/tegg-vitest/runner',
+  },
+});
+```
+
+## Options
+
+- `getApp`: Provide a custom app getter. Default: `require('egg-mock/bootstrap').app`.
+- `parallel`: Skip auto `app.close()` when running in parallel mode. Default: auto-detected from `VITEST_WORKER_ID`.
+- `restoreMocks`: Restore mocks after each test (defaults to true).
+
+## Lifecycle & Context Injection
+
+The custom runner extends Vitest's `VitestTestRunner` and manages tegg context at the runner level:
+
+- **`importFile` (collection phase)**: Captures per-file config from `configureTeggRunner()` and calls `app.ready()`.
+- **`onBeforeRunSuite` (file suite)**: Creates a suite-scoped `ctx` via `app.mockContext()`, overrides `ctxStorage.getStore()`, and opens a held `beginModuleScope` that stays alive for the entire file.
+- **`onBeforeRunTask` (per test)**: Creates a per-test `ctx` and opens a held `beginModuleScope` for the test.
+- **`onAfterRunTask`**: Releases the test scope, restores mocks, and restores `ctxStorage.getStore()` back to the suite `ctx`.
+- **`onAfterRunSuite`**: Releases the suite scope, restores original `getStore()`, and calls `app.close()` unless `parallel` is true.
+
+## Limitations
+
+- Context is managed at the **file suite** level. `egg-mock`'s Mocha runner patch can switch context at the **`describe` suite** level. If your tests rely on describe-scoped suite context, you must manage that manually.
+- If `getApp` throws or returns `undefined`, the adapter will run tests without context injection.

package/dist/index.d.ts

@@ -0,0 +1 @@
+export * from './src';

package/dist/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./src"), exports);
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsd0NBQXNCIn0=

package/dist/runner.d.ts

@@ -0,0 +1 @@
+export { default } from './src/runner';

package/dist/runner.js

@@ -0,0 +1,9 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.default = void 0;
+var runner_1 = require("./src/runner");
+Object.defineProperty(exports, "default", { enumerable: true, get: function () { return __importDefault(runner_1).default; } });
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicnVubmVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vcnVubmVyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7OztBQUFBLHVDQUF1QztBQUE5QixrSEFBQSxPQUFPLE9BQUEifQ==

package/dist/src/index.d.ts

@@ -0,0 +1,7 @@
+import type { TeggVitestAdapterOptions } from './shared';
+export type { EggMockApp, TeggVitestAdapterOptions } from './shared';
+/**
+ * Configure the custom Vitest runner (used via globalThis.__teggVitestConfig).
+ * Call this in a setupFile and set `runner` in vitest.config.ts to use the runner approach.
+ */
+export declare function configureTeggRunner(options?: TeggVitestAdapterOptions): void;

package/dist/src/index.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.configureTeggRunner = configureTeggRunner;
+const shared_1 = require("./shared");
+/**
+ * Configure the custom Vitest runner (used via globalThis.__teggVitestConfig).
+ * Call this in a setupFile and set `runner` in vitest.config.ts to use the runner approach.
+ */
+function configureTeggRunner(options = {}) {
+    var _a, _b;
+    globalThis.__teggVitestConfig = {
+        restoreMocks: (_a = options.restoreMocks) !== null && _a !== void 0 ? _a : true,
+        getApp: (_b = options.getApp) !== null && _b !== void 0 ? _b : shared_1.defaultGetApp,
+    };
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvaW5kZXgudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7QUFXQSxrREFLQztBQWhCRCxxQ0FFa0I7QUFLbEI7OztHQUdHO0FBQ0gsU0FBZ0IsbUJBQW1CLENBQUMsVUFBb0MsRUFBRTs7SUFDdkUsVUFBa0IsQ0FBQyxrQkFBa0IsR0FBRztRQUN2QyxZQUFZLEVBQUUsTUFBQSxPQUFPLENBQUMsWUFBWSxtQ0FBSSxJQUFJO1FBQzFDLE1BQU0sRUFBRSxNQUFBLE9BQU8sQ0FBQyxNQUFNLG1DQUFJLHNCQUFhO0tBQ3hDLENBQUM7QUFDSixDQUFDIn0=

package/dist/src/runner.d.ts

@@ -0,0 +1,20 @@
+import { VitestTestRunner } from 'vitest/runners';
+import type { Suite, Task } from 'vitest';
+export default class TeggVitestRunner extends VitestTestRunner {
+    private fileScopeMap;
+    private taskScopeMap;
+    private fileAppMap;
+    private warned;
+    /**
+     * Override importFile to capture per-file config set by configureTeggRunner()
+     * and await app.ready() during collection phase.
+     */
+    importFile(filepath: string, source: Parameters<VitestTestRunner['importFile']>[1]): Promise<unknown>;
+    onBeforeRunSuite(suite: Suite): Promise<void>;
+    onAfterRunSuite(suite: Suite): Promise<void>;
+    onBeforeTryTask(test: Task, options?: {
+        retry: number;
+        repeats: number;
+    }): Promise<void>;
+    onAfterRunTask(test: Task): Promise<void>;
+}

package/dist/src/runner.js

@@ -0,0 +1,179 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const runners_1 = require("vitest/runners");
+const shared_1 = require("./shared");
+/**
+ * Create a held beginModuleScope: starts the scope and waits until init() is
+ * complete (the inner fn starts executing), then returns the held scope.
+ * The scope stays alive until endScope() is called.
+ */
+async function createHeldScope(ctx) {
+    let endScope;
+    const gate = new Promise(resolve => {
+        endScope = resolve;
+    });
+    let scopeReady;
+    const readyPromise = new Promise(resolve => {
+        scopeReady = resolve;
+    });
+    const scopePromise = ctx.beginModuleScope(async () => {
+        // init() has completed at this point, signal readiness
+        scopeReady();
+        await gate;
+    });
+    // Race readyPromise against scopePromise: if beginModuleScope rejects
+    // before invoking the callback (so scopeReady is never called), the error
+    // propagates immediately instead of hanging forever on readyPromise.
+    // Promise.race attaches a rejection handler to scopePromise, so there are
+    // no unhandled rejections. scopePromise itself is preserved as-is for
+    // gate/endScope behavior in releaseHeldScope.
+    await Promise.race([readyPromise, scopePromise]);
+    return { scopePromise, endScope };
+}
+async function releaseHeldScope(scope) {
+    if (!scope)
+        return;
+    scope.endScope();
+    await scope.scopePromise;
+}
+function isFileSuite(suite) {
+    return !suite.suite && !!suite.filepath;
+}
+function getTaskFilepath(task) {
+    var _a;
+    return (_a = task.file) === null || _a === void 0 ? void 0 : _a.filepath;
+}
+class TeggVitestRunner extends runners_1.VitestTestRunner {
+    constructor() {
+        super(...arguments);
+        this.fileScopeMap = new Map();
+        this.taskScopeMap = new Map();
+        this.fileAppMap = new Map();
+        this.warned = false;
+    }
+    /**
+     * Override importFile to capture per-file config set by configureTeggRunner()
+     * and await app.ready() during collection phase.
+     */
+    async importFile(filepath, source) {
+        var _a, _b;
+        // Clear stale state for this file before re-collection in watch mode
+        if (source === 'collect') {
+            this.fileAppMap.delete(filepath);
+            this.warned = false;
+        }
+        // Clear any stale config before importing
+        delete globalThis.__teggVitestConfig;
+        const result = await super.importFile(filepath, source);
+        if (source === 'collect') {
+            const rawConfig = globalThis.__teggVitestConfig;
+            if (rawConfig) {
+                delete globalThis.__teggVitestConfig;
+                const config = {
+                    restoreMocks: (_a = rawConfig.restoreMocks) !== null && _a !== void 0 ? _a : true,
+                    getApp: (_b = rawConfig.getApp) !== null && _b !== void 0 ? _b : shared_1.defaultGetApp,
+                };
+                (0, shared_1.debugLog)(`captured config for ${filepath}`);
+                // Resolve app and await ready during collection
+                try {
+                    const app = await config.getApp();
+                    if (app) {
+                        await app.ready();
+                        this.fileAppMap.set(filepath, { app, config });
+                        (0, shared_1.debugLog)(`app ready for ${filepath}`);
+                    }
+                }
+                catch (err) {
+                    if (!this.warned) {
+                        this.warned = true;
+                        // eslint-disable-next-line no-console
+                        console.warn('[tegg-vitest] getApp failed, skip context injection.', err);
+                    }
+                }
+            }
+        }
+        return result;
+    }
+    async onBeforeRunSuite(suite) {
+        if (isFileSuite(suite)) {
+            const filepath = suite.filepath;
+            (0, shared_1.debugLog)(`onBeforeRunSuite (file): ${filepath}`);
+            const fileApp = this.fileAppMap.get(filepath);
+            if (fileApp) {
+                const { app, config } = fileApp;
+                if (typeof app.mockContext === 'function' && app.ctxStorage) {
+                    const suiteCtx = app.mockContext(undefined, {
+                        mockCtxStorage: false,
+                        reuseCtxStorage: false,
+                    });
+                    app.ctxStorage.enterWith(suiteCtx);
+                    let suiteScope = null;
+                    if (typeof suiteCtx.beginModuleScope === 'function') {
+                        suiteScope = await createHeldScope(suiteCtx);
+                        (0, shared_1.debugLog)('suite held scope created');
+                    }
+                    this.fileScopeMap.set(filepath, { app, config, suiteCtx, suiteScope });
+                    (0, shared_1.debugLog)('file suite scope created');
+                }
+            }
+        }
+        await super.onBeforeRunSuite(suite);
+    }
+    async onAfterRunSuite(suite) {
+        if (isFileSuite(suite)) {
+            const filepath = suite.filepath;
+            (0, shared_1.debugLog)(`onAfterRunSuite (file): ${filepath}`);
+            const fileState = this.fileScopeMap.get(filepath);
+            if (fileState) {
+                await releaseHeldScope(fileState.suiteScope);
+                this.fileScopeMap.delete(filepath);
+            }
+            this.fileAppMap.delete(filepath);
+        }
+        await super.onAfterRunSuite(suite);
+    }
+    async onBeforeTryTask(test, options) {
+        const filepath = getTaskFilepath(test);
+        if (filepath) {
+            const fileState = this.fileScopeMap.get(filepath);
+            if (fileState) {
+                // Release previous scope on retry to avoid leaks
+                const existing = this.taskScopeMap.get(test.id);
+                if (existing) {
+                    await releaseHeldScope(existing.testScope);
+                }
+                (0, shared_1.debugLog)(`onBeforeTryTask: ${test.name} (retry=${options === null || options === void 0 ? void 0 : options.retry})`);
+                const testCtx = fileState.app.mockContext(undefined, {
+                    mockCtxStorage: false,
+                    reuseCtxStorage: false,
+                });
+                fileState.app.ctxStorage.enterWith(testCtx);
+                let testScope = null;
+                if (typeof testCtx.beginModuleScope === 'function') {
+                    testScope = await createHeldScope(testCtx);
+                    (0, shared_1.debugLog)('test held scope created');
+                }
+                this.taskScopeMap.set(test.id, { testScope, filepath });
+            }
+        }
+        await super.onBeforeTryTask(test);
+    }
+    async onAfterRunTask(test) {
+        const taskState = this.taskScopeMap.get(test.id);
+        if (taskState) {
+            (0, shared_1.debugLog)(`onAfterRunTask: ${test.name}`);
+            await releaseHeldScope(taskState.testScope);
+            this.taskScopeMap.delete(test.id);
+            const fileState = this.fileScopeMap.get(taskState.filepath);
+            if (fileState) {
+                await (0, shared_1.restoreEggMocksIfNeeded)(fileState.config.restoreMocks);
+                // Restore suite context
+                fileState.app.ctxStorage.enterWith(fileState.suiteCtx);
+                (0, shared_1.debugLog)('restored suite context');
+            }
+        }
+        await super.onAfterRunTask(test);
+    }
+}
+exports.default = TeggVitestRunner;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicnVubmVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL3J1bm5lci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOztBQUFBLDRDQUFrRDtBQUVsRCxxQ0FJa0I7QUE4QmxCOzs7O0dBSUc7QUFDSCxLQUFLLFVBQVUsZUFBZSxDQUFDLEdBQVE7SUFDckMsSUFBSSxRQUFxQixDQUFDO0lBQzFCLE1BQU0sSUFBSSxHQUFHLElBQUksT0FBTyxDQUFPLE9BQU8sQ0FBQyxFQUFFO1FBQ3ZDLFFBQVEsR0FBRyxPQUFPLENBQUM7SUFDckIsQ0FBQyxDQUFDLENBQUM7SUFFSCxJQUFJLFVBQXVCLENBQUM7SUFDNUIsTUFBTSxZQUFZLEdBQUcsSUFBSSxPQUFPLENBQU8sT0FBTyxDQUFDLEVBQUU7UUFDL0MsVUFBVSxHQUFHLE9BQU8sQ0FBQztJQUN2QixDQUFDLENBQUMsQ0FBQztJQUVILE1BQU0sWUFBWSxHQUFHLEdBQUcsQ0FBQyxnQkFBZ0IsQ0FBQyxLQUFLLElBQUksRUFBRTtRQUNuRCx1REFBdUQ7UUFDdkQsVUFBVSxFQUFFLENBQUM7UUFDYixNQUFNLElBQUksQ0FBQztJQUNiLENBQUMsQ0FBQyxDQUFDO0lBRUgsc0VBQXNFO0lBQ3RFLDBFQUEwRTtJQUMxRSxxRUFBcUU7SUFDckUsMEVBQTBFO0lBQzFFLHNFQUFzRTtJQUN0RSw4Q0FBOEM7SUFDOUMsTUFBTSxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUUsWUFBWSxFQUFFLFlBQVksQ0FBRSxDQUFDLENBQUM7SUFFbkQsT0FBTyxFQUFFLFlBQVksRUFBRSxRQUFRLEVBQUUsQ0FBQztBQUNwQyxDQUFDO0FBRUQsS0FBSyxVQUFVLGdCQUFnQixDQUFDLEtBQXVCO0lBQ3JELElBQUksQ0FBQyxLQUFLO1FBQUUsT0FBTztJQUNuQixLQUFLLENBQUMsUUFBUSxFQUFFLENBQUM7SUFDakIsTUFBTSxLQUFLLENBQUMsWUFBWSxDQUFDO0FBQzNCLENBQUM7QUFFRCxTQUFTLFdBQVcsQ0FBQyxLQUFZO0lBQy9CLE9BQU8sQ0FBQyxLQUFLLENBQUMsS0FBSyxJQUFJLENBQUMsQ0FBQyxLQUFLLENBQUMsUUFBUSxDQUFDO0FBQzFDLENBQUM7QUFFRCxTQUFTLGVBQWUsQ0FBQyxJQUFVOztJQUNqQyxPQUFPLE1BQUMsSUFBWSxDQUFDLElBQUksMENBQUUsUUFBUSxDQUFDO0FBQ3RDLENBQUM7QUFFRCxNQUFxQixnQkFBaUIsU0FBUSwwQkFBZ0I7SUFBOUQ7O1FBQ1UsaUJBQVksR0FBRyxJQUFJLEdBQUcsRUFBMEIsQ0FBQztRQUNqRCxpQkFBWSxHQUFHLElBQUksR0FBRyxFQUEwQixDQUFDO1FBQ2pELGVBQVUsR0FBRyxJQUFJLEdBQUcsRUFBd0IsQ0FBQztRQUM3QyxXQUFNLEdBQUcsS0FBSyxDQUFDO0lBb0p6QixDQUFDO0lBbEpDOzs7T0FHRztJQUNILEtBQUssQ0FBQyxVQUFVLENBQUMsUUFBZ0IsRUFBRSxNQUFxRDs7UUFDdEYscUVBQXFFO1FBQ3JFLElBQUksTUFBTSxLQUFLLFNBQVMsRUFBRSxDQUFDO1lBQ3pCLElBQUksQ0FBQyxVQUFVLENBQUMsTUFBTSxDQUFDLFFBQVEsQ0FBQyxDQUFDO1lBQ2pDLElBQUksQ0FBQyxNQUFNLEdBQUcsS0FBSyxDQUFDO1FBQ3RCLENBQUM7UUFDRCwwQ0FBMEM7UUFDMUMsT0FBUSxVQUFrQixDQUFDLGtCQUFrQixDQUFDO1FBRTlDLE1BQU0sTUFBTSxHQUFHLE1BQU0sS0FBSyxDQUFDLFVBQVUsQ0FBQyxRQUFRLEVBQUUsTUFBTSxDQUFDLENBQUM7UUFFeEQsSUFBSSxNQUFNLEtBQUssU0FBUyxFQUFFLENBQUM7WUFDekIsTUFBTSxTQUFTLEdBQUksVUFBa0IsQ0FBQyxrQkFBa0IsQ0FBQztZQUN6RCxJQUFJLFNBQVMsRUFBRSxDQUFDO2dCQUNkLE9BQVEsVUFBa0IsQ0FBQyxrQkFBa0IsQ0FBQztnQkFFOUMsTUFBTSxNQUFNLEdBQXFCO29CQUMvQixZQUFZLEVBQUUsTUFBQSxTQUFTLENBQUMsWUFBWSxtQ0FBSSxJQUFJO29CQUM1QyxNQUFNLEVBQUUsTUFBQSxTQUFTLENBQUMsTUFBTSxtQ0FBSSxzQkFBYTtpQkFDMUMsQ0FBQztnQkFFRixJQUFBLGlCQUFRLEVBQUMsdUJBQXVCLFFBQVEsRUFBRSxDQUFDLENBQUM7Z0JBRTVDLGdEQUFnRDtnQkFDaEQsSUFBSSxDQUFDO29CQUNILE1BQU0sR0FBRyxHQUFHLE1BQU0sTUFBTSxDQUFDLE1BQU0sRUFBRSxDQUFDO29CQUNsQyxJQUFJLEdBQUcsRUFBRSxDQUFDO3dCQUNSLE1BQU0sR0FBRyxDQUFDLEtBQUssRUFBRSxDQUFDO3dCQUNsQixJQUFJLENBQUMsVUFBVSxDQUFDLEdBQUcsQ0FBQyxRQUFRLEVBQUUsRUFBRSxHQUFHLEVBQUUsTUFBTSxFQUFFLENBQUMsQ0FBQzt3QkFDL0MsSUFBQSxpQkFBUSxFQUFDLGlCQUFpQixRQUFRLEVBQUUsQ0FBQyxDQUFDO29CQUN4QyxDQUFDO2dCQUNILENBQUM7Z0JBQUMsT0FBTyxHQUFHLEVBQUUsQ0FBQztvQkFDYixJQUFJLENBQUMsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDO3dCQUNqQixJQUFJLENBQUMsTUFBTSxHQUFHLElBQUksQ0FBQzt3QkFDbkIsc0NBQXNDO3dCQUN0QyxPQUFPLENBQUMsSUFBSSxDQUFDLHNEQUFzRCxFQUFFLEdBQUcsQ0FBQyxDQUFDO29CQUM1RSxDQUFDO2dCQUNILENBQUM7WUFDSCxDQUFDO1FBQ0gsQ0FBQztRQUVELE9BQU8sTUFBTSxDQUFDO0lBQ2hCLENBQUM7SUFFRCxLQUFLLENBQUMsZ0JBQWdCLENBQUMsS0FBWTtRQUNqQyxJQUFJLFdBQVcsQ0FBQyxLQUFLLENBQUMsRUFBRSxDQUFDO1lBQ3ZCLE1BQU0sUUFBUSxHQUFHLEtBQUssQ0FBQyxRQUFTLENBQUM7WUFDakMsSUFBQSxpQkFBUSxFQUFDLDRCQUE0QixRQUFRLEVBQUUsQ0FBQyxDQUFDO1lBRWpELE1BQU0sT0FBTyxHQUFHLElBQUksQ0FBQyxVQUFVLENBQUMsR0FBRyxDQUFDLFFBQVEsQ0FBQyxDQUFDO1lBQzlDLElBQUksT0FBTyxFQUFFLENBQUM7Z0JBQ1osTUFBTSxFQUFFLEdBQUcsRUFBRSxNQUFNLEVBQUUsR0FBRyxPQUFPLENBQUM7Z0JBRWhDLElBQUksT0FBTyxHQUFHLENBQUMsV0FBVyxLQUFLLFVBQVUsSUFBSSxHQUFHLENBQUMsVUFBVSxFQUFFLENBQUM7b0JBQzVELE1BQU0sUUFBUSxHQUFHLEdBQUcsQ0FBQyxXQUFXLENBQUMsU0FBUyxFQUFFO3dCQUMxQyxjQUFjLEVBQUUsS0FBSzt3QkFDckIsZUFBZSxFQUFFLEtBQUs7cUJBQ3ZCLENBQUMsQ0FBQztvQkFDSCxHQUFHLENBQUMsVUFBVSxDQUFDLFNBQVMsQ0FBQyxRQUFRLENBQUMsQ0FBQztvQkFFbkMsSUFBSSxVQUFVLEdBQXFCLElBQUksQ0FBQztvQkFDeEMsSUFBSSxPQUFPLFFBQVEsQ0FBQyxnQkFBZ0IsS0FBSyxVQUFVLEVBQUUsQ0FBQzt3QkFDcEQsVUFBVSxHQUFHLE1BQU0sZUFBZSxDQUFDLFFBQVEsQ0FBQyxDQUFDO3dCQUM3QyxJQUFBLGlCQUFRLEVBQUMsMEJBQTBCLENBQUMsQ0FBQztvQkFDdkMsQ0FBQztvQkFFRCxJQUFJLENBQUMsWUFBWSxDQUFDLEdBQUcsQ0FBQyxRQUFRLEVBQUUsRUFBRSxHQUFHLEVBQUUsTUFBTSxFQUFFLFFBQVEsRUFBRSxVQUFVLEVBQUUsQ0FBQyxDQUFDO29CQUN2RSxJQUFBLGlCQUFRLEVBQUMsMEJBQTBCLENBQUMsQ0FBQztnQkFDdkMsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO1FBRUQsTUFBTSxLQUFLLENBQUMsZ0JBQWdCLENBQUMsS0FBSyxDQUFDLENBQUM7SUFDdEMsQ0FBQztJQUVELEtBQUssQ0FBQyxlQUFlLENBQUMsS0FBWTtRQUNoQyxJQUFJLFdBQVcsQ0FBQyxLQUFLLENBQUMsRUFBRSxDQUFDO1lBQ3ZCLE1BQU0sUUFBUSxHQUFHLEtBQUssQ0FBQyxRQUFTLENBQUM7WUFDakMsSUFBQSxpQkFBUSxFQUFDLDJCQUEyQixRQUFRLEVBQUUsQ0FBQyxDQUFDO1lBRWhELE1BQU0sU0FBUyxHQUFHLElBQUksQ0FBQyxZQUFZLENBQUMsR0FBRyxDQUFDLFFBQVEsQ0FBQyxDQUFDO1lBQ2xELElBQUksU0FBUyxFQUFFLENBQUM7Z0JBQ2QsTUFBTSxnQkFBZ0IsQ0FBQyxTQUFTLENBQUMsVUFBVSxDQUFDLENBQUM7Z0JBQzdDLElBQUksQ0FBQyxZQUFZLENBQUMsTUFBTSxDQUFDLFFBQVEsQ0FBQyxDQUFDO1lBQ3JDLENBQUM7WUFDRCxJQUFJLENBQUMsVUFBVSxDQUFDLE1BQU0sQ0FBQyxRQUFRLENBQUMsQ0FBQztRQUNuQyxDQUFDO1FBRUQsTUFBTSxLQUFLLENBQUMsZUFBZSxDQUFDLEtBQUssQ0FBQyxDQUFDO0lBQ3JDLENBQUM7SUFFRCxLQUFLLENBQUMsZUFBZSxDQUFDLElBQVUsRUFBRSxPQUE0QztRQUM1RSxNQUFNLFFBQVEsR0FBRyxlQUFlLENBQUMsSUFBSSxDQUFDLENBQUM7UUFDdkMsSUFBSSxRQUFRLEVBQUUsQ0FBQztZQUNiLE1BQU0sU0FBUyxHQUFHLElBQUksQ0FBQyxZQUFZLENBQUMsR0FBRyxDQUFDLFFBQVEsQ0FBQyxDQUFDO1lBQ2xELElBQUksU0FBUyxFQUFFLENBQUM7Z0JBQ2QsaURBQWlEO2dCQUNqRCxNQUFNLFFBQVEsR0FBRyxJQUFJLENBQUMsWUFBWSxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsRUFBRSxDQUFDLENBQUM7Z0JBQ2hELElBQUksUUFBUSxFQUFFLENBQUM7b0JBQ2IsTUFBTSxnQkFBZ0IsQ0FBQyxRQUFRLENBQUMsU0FBUyxDQUFDLENBQUM7Z0JBQzdDLENBQUM7Z0JBRUQsSUFBQSxpQkFBUSxFQUFDLG9CQUFvQixJQUFJLENBQUMsSUFBSSxXQUFXLE9BQU8sYUFBUCxPQUFPLHVCQUFQLE9BQU8sQ0FBRSxLQUFLLEdBQUcsQ0FBQyxDQUFDO2dCQUVwRSxNQUFNLE9BQU8sR0FBRyxTQUFTLENBQUMsR0FBRyxDQUFDLFdBQVksQ0FBQyxTQUFTLEVBQUU7b0JBQ3BELGNBQWMsRUFBRSxLQUFLO29CQUNyQixlQUFlLEVBQUUsS0FBSztpQkFDdkIsQ0FBQyxDQUFDO2dCQUNILFNBQVMsQ0FBQyxHQUFHLENBQUMsVUFBVyxDQUFDLFNBQVMsQ0FBQyxPQUFPLENBQUMsQ0FBQztnQkFFN0MsSUFBSSxTQUFTLEdBQXFCLElBQUksQ0FBQztnQkFDdkMsSUFBSSxPQUFPLE9BQU8sQ0FBQyxnQkFBZ0IsS0FBSyxVQUFVLEVBQUUsQ0FBQztvQkFDbkQsU0FBUyxHQUFHLE1BQU0sZUFBZSxDQUFDLE9BQU8sQ0FBQyxDQUFDO29CQUMzQyxJQUFBLGlCQUFRLEVBQUMseUJBQXlCLENBQUMsQ0FBQztnQkFDdEMsQ0FBQztnQkFFRCxJQUFJLENBQUMsWUFBWSxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsRUFBRSxFQUFFLEVBQUUsU0FBUyxFQUFFLFFBQVEsRUFBRSxDQUFDLENBQUM7WUFDMUQsQ0FBQztRQUNILENBQUM7UUFFRCxNQUFNLEtBQUssQ0FBQyxlQUFlLENBQUMsSUFBSSxDQUFDLENBQUM7SUFDcEMsQ0FBQztJQUVELEtBQUssQ0FBQyxjQUFjLENBQUMsSUFBVTtRQUM3QixNQUFNLFNBQVMsR0FBRyxJQUFJLENBQUMsWUFBWSxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsRUFBRSxDQUFDLENBQUM7UUFDakQsSUFBSSxTQUFTLEVBQUUsQ0FBQztZQUNkLElBQUEsaUJBQVEsRUFBQyxtQkFBbUIsSUFBSSxDQUFDLElBQUksRUFBRSxDQUFDLENBQUM7WUFFekMsTUFBTSxnQkFBZ0IsQ0FBQyxTQUFTLENBQUMsU0FBUyxDQUFDLENBQUM7WUFDNUMsSUFBSSxDQUFDLFlBQVksQ0FBQyxNQUFNLENBQUMsSUFBSSxDQUFDLEVBQUUsQ0FBQyxDQUFDO1lBRWxDLE1BQU0sU0FBUyxHQUFHLElBQUksQ0FBQyxZQUFZLENBQUMsR0FBRyxDQUFDLFNBQVMsQ0FBQyxRQUFRLENBQUMsQ0FBQztZQUM1RCxJQUFJLFNBQVMsRUFBRSxDQUFDO2dCQUNkLE1BQU0sSUFBQSxnQ0FBdUIsRUFBQyxTQUFTLENBQUMsTUFBTSxDQUFDLFlBQVksQ0FBQyxDQUFDO2dCQUM3RCx3QkFBd0I7Z0JBQ3hCLFNBQVMsQ0FBQyxHQUFHLENBQUMsVUFBVyxDQUFDLFNBQVMsQ0FBQyxTQUFTLENBQUMsUUFBUSxDQUFDLENBQUM7Z0JBQ3hELElBQUEsaUJBQVEsRUFBQyx3QkFBd0IsQ0FBQyxDQUFDO1lBQ3JDLENBQUM7UUFDSCxDQUFDO1FBRUQsTUFBTSxLQUFLLENBQUMsY0FBYyxDQUFDLElBQUksQ0FBQyxDQUFDO0lBQ25DLENBQUM7Q0FDRjtBQXhKRCxtQ0F3SkMifQ==

package/dist/src/shared.d.ts

@@ -0,0 +1,25 @@
+import type { AsyncLocalStorage } from 'node:async_hooks';
+import type { Application } from 'egg';
+export type EggMockApp = Application & {
+    ctxStorage?: {
+        getStore?: () => any;
+        enterWith?: (store: any) => void;
+    } & AsyncLocalStorage<any>;
+    mockContext?: (data?: any, options?: any) => any;
+};
+export interface TeggVitestAdapterOptions {
+    /**
+     * Resolve app instance.
+     * Default: require('egg-mock/bootstrap').app
+     */
+    getApp?: () => Promise<EggMockApp | undefined> | EggMockApp | undefined;
+    /**
+     * Restore mocks after each test.
+     * Default: true (calls egg-mock.restore())
+     */
+    restoreMocks?: boolean;
+}
+export declare const DEBUG_ENABLED: boolean;
+export declare function debugLog(message: string, extra?: unknown): void;
+export declare function defaultGetApp(): Promise<EggMockApp | undefined>;
+export declare function restoreEggMocksIfNeeded(restoreMocks: boolean): Promise<void>;

package/dist/src/shared.js

@@ -0,0 +1,34 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEBUG_ENABLED = void 0;
+exports.debugLog = debugLog;
+exports.defaultGetApp = defaultGetApp;
+exports.restoreEggMocksIfNeeded = restoreEggMocksIfNeeded;
+exports.DEBUG_ENABLED = process.env.DEBUG_TEGG_VITEST === '1';
+function debugLog(message, extra) {
+    if (!exports.DEBUG_ENABLED)
+        return;
+    if (extra === undefined) {
+        // eslint-disable-next-line no-console
+        console.log(`[tegg-vitest] ${message}`);
+        return;
+    }
+    // eslint-disable-next-line no-console
+    console.log(`[tegg-vitest] ${message}`, extra);
+}
+async function defaultGetApp() {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const bootstrap = require('egg-mock/bootstrap');
+    return bootstrap === null || bootstrap === void 0 ? void 0 : bootstrap.app;
+}
+async function restoreEggMocksIfNeeded(restoreMocks) {
+    if (!restoreMocks)
+        return;
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const eggMock = require('egg-mock');
+    const mm = (eggMock === null || eggMock === void 0 ? void 0 : eggMock.default) || eggMock;
+    if (mm === null || mm === void 0 ? void 0 : mm.restore) {
+        await mm.restore();
+    }
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic2hhcmVkLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL3NoYXJlZC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7QUE2QkEsNEJBU0M7QUFFRCxzQ0FJQztBQUVELDBEQVFDO0FBM0JZLFFBQUEsYUFBYSxHQUFHLE9BQU8sQ0FBQyxHQUFHLENBQUMsaUJBQWlCLEtBQUssR0FBRyxDQUFDO0FBRW5FLFNBQWdCLFFBQVEsQ0FBQyxPQUFlLEVBQUUsS0FBZTtJQUN2RCxJQUFJLENBQUMscUJBQWE7UUFBRSxPQUFPO0lBQzNCLElBQUksS0FBSyxLQUFLLFNBQVMsRUFBRSxDQUFDO1FBQ3hCLHNDQUFzQztRQUN0QyxPQUFPLENBQUMsR0FBRyxDQUFDLGlCQUFpQixPQUFPLEVBQUUsQ0FBQyxDQUFDO1FBQ3hDLE9BQU87SUFDVCxDQUFDO0lBQ0Qsc0NBQXNDO0lBQ3RDLE9BQU8sQ0FBQyxHQUFHLENBQUMsaUJBQWlCLE9BQU8sRUFBRSxFQUFFLEtBQUssQ0FBQyxDQUFDO0FBQ2pELENBQUM7QUFFTSxLQUFLLFVBQVUsYUFBYTtJQUNqQyw4REFBOEQ7SUFDOUQsTUFBTSxTQUFTLEdBQUcsT0FBTyxDQUFDLG9CQUFvQixDQUFDLENBQUM7SUFDaEQsT0FBTyxTQUFTLGFBQVQsU0FBUyx1QkFBVCxTQUFTLENBQUUsR0FBRyxDQUFDO0FBQ3hCLENBQUM7QUFFTSxLQUFLLFVBQVUsdUJBQXVCLENBQUMsWUFBcUI7SUFDakUsSUFBSSxDQUFDLFlBQVk7UUFBRSxPQUFPO0lBQzFCLDhEQUE4RDtJQUM5RCxNQUFNLE9BQU8sR0FBRyxPQUFPLENBQUMsVUFBVSxDQUFDLENBQUM7SUFDcEMsTUFBTSxFQUFFLEdBQUcsQ0FBQSxPQUFPLGFBQVAsT0FBTyx1QkFBUCxPQUFPLENBQUUsT0FBTyxLQUFJLE9BQU8sQ0FBQztJQUN2QyxJQUFJLEVBQUUsYUFBRixFQUFFLHVCQUFGLEVBQUUsQ0FBRSxPQUFPLEVBQUUsQ0FBQztRQUNoQixNQUFNLEVBQUUsQ0FBQyxPQUFPLEVBQUUsQ0FBQztJQUNyQixDQUFDO0FBQ0gsQ0FBQyJ9

package/package.json

@@ -1,31 +1,61 @@
 {
   "name": "@eggjs/tegg-vitest",
-  "version": "0.0.0",
-  "description": "egg logger",
+  "version": "3.73.0-beta.0",
+  "description": "Vitest adapter for tegg context injection",
   "keywords": [
     "egg",
-    "logger"
+    "tegg",
+    "vitest",
+    "test",
+    "adapter"
   ],
-  "homepage": "https://github.com/eggjs/egg/tree/next/packages/logger",
-  "bugs": {
-    "url": "https://github.com/eggjs/egg/issues"
+  "main": "dist/index.js",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    },
+    "./runner": {
+      "types": "./dist/runner.d.ts",
+      "default": "./dist/runner.js"
+    }
+  },
+  "files": [
+    "dist/**/*.js",
+    "dist/**/*.d.ts"
+  ],
+  "typings": "dist/index.d.ts",
+  "scripts": {
+    "clean": "tsc -b --clean",
+    "tsc:pub": "ut run clean && tsc -p ./tsconfig.pub.json",
+    "test": "node --eval \"process.exit(parseInt(process.versions.node) < 18 ? 0 : 1)\" || vitest run"
   },
+  "author": "killagu <killa123@126.com>",
   "license": "MIT",
-  "author": "fengmk2 <fengmk2@gmail.com> (https://github.com/fengmk2)",
+  "homepage": "https://github.com/eggjs/tegg",
+  "bugs": {
+    "url": "https://github.com/eggjs/tegg/issues"
+  },
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/eggjs/egg.git",
-    "directory": "packages/logger"
+    "url": "https://github.com/eggjs/tegg.git",
+    "directory": "core/vitest"
   },
-  "publishConfig": {
-    "access": "public"
+  "engines": {
+    "node": ">=18.0.0"
   },
   "dependencies": {
+    "egg-mock": "^5.5.0"
+  },
+  "peerDependencies": {
+    "vitest": "^1.6.0 || ^2.0.0 || ^3.0.0"
   },
   "devDependencies": {
+    "@types/node": "^20.2.4",
+    "egg": "^3.9.1",
+    "egg-tracer": "^2.0.0",
+    "typescript": "^5.0.4",
+    "vitest": "^1.6.0"
   },
-  "engines": {
-    "node": ">= 22.18.0"
-  }
+  "gitHead": "419d966f3efc649e533f165ba5d15994d7076722"
 }
-