From 781b1c1e095bc35e57eb665db9d9bf4890142735 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Thu, 19 Feb 2026 07:36:55 +0000 Subject: [PATCH] test(memory): dedupe voyage embedding provider test setup --- src/memory/embeddings-voyage.test.ts | 52 +++++++++++++--------------- 1 file changed, 24 insertions(+), 28 deletions(-) diff --git a/src/memory/embeddings-voyage.test.ts b/src/memory/embeddings-voyage.test.ts index e47710689e..08e59474ae 100644 --- a/src/memory/embeddings-voyage.test.ts +++ b/src/memory/embeddings-voyage.test.ts @@ -19,6 +19,28 @@ const createFetchMock = () => { return withFetchPreconnect(fetchMock); }; +function mockVoyageApiKey() { + vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({ + apiKey: "voyage-key-123", + mode: "api-key", + source: "test", + }); +} + +async function createDefaultVoyageProvider( + model: string, + fetchMock: ReturnType, +) { + vi.stubGlobal("fetch", fetchMock); + mockVoyageApiKey(); + return createVoyageEmbeddingProvider({ + config: {} as never, + provider: "voyage", + model, + fallback: "none", + }); +} + describe("voyage embedding provider", () => { afterEach(() => { vi.resetAllMocks(); @@ -27,20 +49,7 @@ describe("voyage embedding provider", () => { it("configures client with correct defaults and headers", async () => { const fetchMock = createFetchMock(); - vi.stubGlobal("fetch", fetchMock); - - vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({ - apiKey: "voyage-key-123", - mode: "api-key", - source: "test", - }); - - const result = await createVoyageEmbeddingProvider({ - config: {} as never, - provider: "voyage", - model: "voyage-4-large", - fallback: "none", - }); + const result = await createDefaultVoyageProvider("voyage-4-large", fetchMock); await result.provider.embedQuery("test query"); @@ -105,20 +114,7 @@ describe("voyage embedding provider", () => { ), ), ); - vi.stubGlobal("fetch", fetchMock); - - vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({ - apiKey: "voyage-key-123", - mode: "api-key", - source: "test", - }); - - const result = await createVoyageEmbeddingProvider({ - config: {} as never, - provider: "voyage", - model: "voyage-4-large", - fallback: "none", - }); + const result = await createDefaultVoyageProvider("voyage-4-large", fetchMock); await result.provider.embedBatch(["doc1", "doc2"]);