perf(test): reduce module reload churn in unit suites

This commit is contained in:
Peter Steinberger
2026-02-13 15:19:09 +00:00
parent 6c4c535813
commit faec6ccb1d
4 changed files with 18 additions and 64 deletions

View File

@@ -1,4 +1,6 @@
import { afterEach, describe, expect, it, vi } from "vitest";
import * as authModule from "../agents/model-auth.js";
import { createVoyageEmbeddingProvider, normalizeVoyageModel } from "./embeddings-voyage.js";
vi.mock("../agents/model-auth.js", () => ({
resolveApiKeyForProvider: vi.fn(),
@@ -20,7 +22,6 @@ const createFetchMock = () =>
describe("voyage embedding provider", () => {
afterEach(() => {
vi.resetAllMocks();
vi.resetModules();
vi.unstubAllGlobals();
});
@@ -28,9 +29,6 @@ describe("voyage embedding provider", () => {
const fetchMock = createFetchMock();
vi.stubGlobal("fetch", fetchMock);
const { createVoyageEmbeddingProvider } = await import("./embeddings-voyage.js");
const authModule = await import("../agents/model-auth.js");
vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({
apiKey: "voyage-key-123",
mode: "api-key",
@@ -69,8 +67,6 @@ describe("voyage embedding provider", () => {
const fetchMock = createFetchMock();
vi.stubGlobal("fetch", fetchMock);
const { createVoyageEmbeddingProvider } = await import("./embeddings-voyage.js");
const result = await createVoyageEmbeddingProvider({
config: {} as never,
provider: "voyage",
@@ -103,9 +99,6 @@ describe("voyage embedding provider", () => {
})) as unknown as typeof fetch;
vi.stubGlobal("fetch", fetchMock);
const { createVoyageEmbeddingProvider } = await import("./embeddings-voyage.js");
const authModule = await import("../agents/model-auth.js");
vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({
apiKey: "voyage-key-123",
mode: "api-key",
@@ -131,7 +124,6 @@ describe("voyage embedding provider", () => {
});
it("normalizes model names", async () => {
const { normalizeVoyageModel } = await import("./embeddings-voyage.js");
expect(normalizeVoyageModel("voyage/voyage-large-2")).toBe("voyage-large-2");
expect(normalizeVoyageModel("voyage-4-large")).toBe("voyage-4-large");
expect(normalizeVoyageModel(" voyage-lite ")).toBe("voyage-lite");

View File

@@ -1,4 +1,4 @@
import { afterEach, describe, expect, it, vi } from "vitest";
import { afterEach, beforeAll, describe, expect, it, vi } from "vitest";
import { DEFAULT_GEMINI_EMBEDDING_MODEL } from "./embeddings-gemini.js";
vi.mock("../agents/model-auth.js", () => ({
@@ -21,7 +21,6 @@ const createFetchMock = () =>
describe("embedding provider remote overrides", () => {
afterEach(() => {
vi.resetAllMocks();
vi.resetModules();
vi.unstubAllGlobals();
});
@@ -170,7 +169,6 @@ describe("embedding provider remote overrides", () => {
describe("embedding provider auto selection", () => {
afterEach(() => {
vi.resetAllMocks();
vi.resetModules();
vi.unstubAllGlobals();
});
@@ -266,6 +264,10 @@ describe("embedding provider auto selection", () => {
});
describe("embedding provider local fallback", () => {
beforeAll(() => {
vi.resetModules();
});
afterEach(() => {
vi.resetAllMocks();
vi.resetModules();