From ff50d3303d83fd88f237668372433f8174f68121 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Wed, 18 Feb 2026 13:17:44 +0000
Subject: [PATCH] test(memory): dedupe model-auth mock setup

---
 src/memory/embeddings-voyage.test.ts | 13 ++++---------
 src/memory/embeddings.test.ts        | 13 ++++---------
 src/test-utils/model-auth-mock.ts    | 13 +++++++++++++
 3 files changed, 21 insertions(+), 18 deletions(-)
 create mode 100644 src/test-utils/model-auth-mock.ts

diff --git a/src/memory/embeddings-voyage.test.ts b/src/memory/embeddings-voyage.test.ts
index 7c4b1e643e7..e47710689e6 100644
--- a/src/memory/embeddings-voyage.test.ts
+++ b/src/memory/embeddings-voyage.test.ts
@@ -3,15 +3,10 @@ import * as authModule from "../agents/model-auth.js";
 import { type FetchMock, withFetchPreconnect } from "../test-utils/fetch-mock.js";
 import { createVoyageEmbeddingProvider, normalizeVoyageModel } from "./embeddings-voyage.js";
 
-vi.mock("../agents/model-auth.js", () => ({
-  resolveApiKeyForProvider: vi.fn(),
-  requireApiKey: (auth: { apiKey?: string; mode?: string }, provider: string) => {
-    if (auth?.apiKey) {
-      return auth.apiKey;
-    }
-    throw new Error(`No API key resolved for provider "${provider}" (auth mode: ${auth?.mode}).`);
-  },
-}));
+vi.mock("../agents/model-auth.js", async () => {
+  const { createModelAuthMockModule } = await import("../test-utils/model-auth-mock.js");
+  return createModelAuthMockModule();
+});
 
 const createFetchMock = () => {
   const fetchMock = vi.fn<FetchMock>(
diff --git a/src/memory/embeddings.test.ts b/src/memory/embeddings.test.ts
index 94ae24dcee6..9f1fd146047 100644
--- a/src/memory/embeddings.test.ts
+++ b/src/memory/embeddings.test.ts
@@ -3,15 +3,10 @@ import * as authModule from "../agents/model-auth.js";
 import { DEFAULT_GEMINI_EMBEDDING_MODEL } from "./embeddings-gemini.js";
 import { createEmbeddingProvider, DEFAULT_LOCAL_MODEL } from "./embeddings.js";
 
-vi.mock("../agents/model-auth.js", () => ({
-  resolveApiKeyForProvider: vi.fn(),
-  requireApiKey: (auth: { apiKey?: string; mode?: string }, provider: string) => {
-    if (auth?.apiKey) {
-      return auth.apiKey;
-    }
-    throw new Error(`No API key resolved for provider "${provider}" (auth mode: ${auth?.mode}).`);
-  },
-}));
+vi.mock("../agents/model-auth.js", async () => {
+  const { createModelAuthMockModule } = await import("../test-utils/model-auth-mock.js");
+  return createModelAuthMockModule();
+});
 
 const importNodeLlamaCppMock = vi.fn();
 vi.mock("./node-llama.js", () => ({
diff --git a/src/test-utils/model-auth-mock.ts b/src/test-utils/model-auth-mock.ts
new file mode 100644
index 00000000000..8737fd31e6c
--- /dev/null
+++ b/src/test-utils/model-auth-mock.ts
@@ -0,0 +1,13 @@
+import { vi } from "vitest";
+
+export function createModelAuthMockModule() {
+  return {
+    resolveApiKeyForProvider: vi.fn(),
+    requireApiKey: (auth: { apiKey?: string; mode?: string }, provider: string) => {
+      if (auth?.apiKey) {
+        return auth.apiKey;
+      }
+      throw new Error(`No API key resolved for provider "${provider}" (auth mode: ${auth?.mode}).`);
+    },
+  };
+}