diff --git a/src/test-utils/model-auth-mock.ts b/src/test-utils/model-auth-mock.ts deleted file mode 100644 index 92553062273..00000000000 --- a/src/test-utils/model-auth-mock.ts +++ /dev/null @@ -1,18 +0,0 @@ -import { vi } from "vitest"; - -type ModelAuthMockModule = { - resolveApiKeyForProvider: (...args: unknown[]) => unknown; - requireApiKey: (auth: { apiKey?: string; mode?: string }, provider: string) => string; -}; - -export function createModelAuthMockModule(): ModelAuthMockModule { - return { - resolveApiKeyForProvider: vi.fn() as (...args: unknown[]) => unknown, - 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}).`); - }, - }; -} diff --git a/src/test-utils/session-write-lock-module-mock.ts b/src/test-utils/session-write-lock-module-mock.ts index a13d8019fbc..0d4013ae84e 100644 --- a/src/test-utils/session-write-lock-module-mock.ts +++ b/src/test-utils/session-write-lock-module-mock.ts @@ -1,4 +1,3 @@ -import { vi } from "vitest"; import type * as SessionWriteLockModule from "../agents/session-write-lock.js"; type SessionWriteLockModuleShape = typeof SessionWriteLockModule; @@ -13,16 +12,3 @@ export async function buildSessionWriteLockModuleMock( acquireSessionWriteLock, }; } - -export function resetModulesWithSessionWriteLockDoMock( - modulePath: string, - acquireSessionWriteLock: SessionWriteLockModuleShape["acquireSessionWriteLock"], -): void { - vi.resetModules(); - vi.doMock(modulePath, () => - buildSessionWriteLockModuleMock( - () => vi.importActual(modulePath), - acquireSessionWriteLock, - ), - ); -}