From 5b240092718b60bdc304f1d8872d5ffdbb5f6528 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Mon, 13 Apr 2026 09:48:45 -0700
Subject: [PATCH] test: mock model fallback source check

---
 src/agents/model-fallback.probe.test.ts | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/agents/model-fallback.probe.test.ts b/src/agents/model-fallback.probe.test.ts
index 63c1c1fb627..b60cf2e96b0 100644
--- a/src/agents/model-fallback.probe.test.ts
+++ b/src/agents/model-fallback.probe.test.ts
@@ -8,10 +8,13 @@ import { makeModelFallbackCfg } from "./test-helpers/model-fallback-config-fixtu
 // Mock auth-profile submodules — must be before importing model-fallback
 vi.mock("./auth-profiles/store.js", () => ({
   ensureAuthProfileStore: vi.fn(),
-  hasAnyAuthProfileStoreSource: vi.fn(),
   loadAuthProfileStoreForRuntime: vi.fn(),
 }));
 
+vi.mock("./auth-profiles/source-check.js", () => ({
+  hasAnyAuthProfileStoreSource: vi.fn(),
+}));
+
 vi.mock("./auth-profiles/usage.js", () => ({
   getSoonestCooldownExpiry: vi.fn(),
   isProfileInCooldown: vi.fn(),
@@ -27,6 +30,7 @@ vi.mock("./auth-profiles/source-check.js", () => ({
 }));
 
 type AuthProfilesStoreModule = typeof import("./auth-profiles/store.js");
+type AuthProfilesSourceCheckModule = typeof import("./auth-profiles/source-check.js");
 type AuthProfilesUsageModule = typeof import("./auth-profiles/usage.js");
 type AuthProfilesOrderModule = typeof import("./auth-profiles/order.js");
 type ModelFallbackModule = typeof import("./model-fallback.js");
@@ -36,7 +40,7 @@ let mockedEnsureAuthProfileStore: ReturnType<
   typeof vi.mocked<AuthProfilesStoreModule["ensureAuthProfileStore"]>
 >;
 let mockedHasAnyAuthProfileStoreSource: ReturnType<
-  typeof vi.mocked<AuthProfilesStoreModule["hasAnyAuthProfileStoreSource"]>
+  typeof vi.mocked<AuthProfilesSourceCheckModule["hasAnyAuthProfileStoreSource"]>
 >;
 let mockedGetSoonestCooldownExpiry: ReturnType<
   typeof vi.mocked<AuthProfilesUsageModule["getSoonestCooldownExpiry"]>
@@ -61,13 +65,14 @@ let unregisterLogTransport: (() => void) | undefined;
 
 async function loadModelFallbackProbeModules() {
   const authProfilesStoreModule = await import("./auth-profiles/store.js");
+  const authProfilesSourceCheckModule = await import("./auth-profiles/source-check.js");
   const authProfilesUsageModule = await import("./auth-profiles/usage.js");
   const authProfilesOrderModule = await import("./auth-profiles/order.js");
   const loggerModule = await import("../logging/logger.js");
   const modelFallbackModule = await import("./model-fallback.js");
   mockedEnsureAuthProfileStore = vi.mocked(authProfilesStoreModule.ensureAuthProfileStore);
   mockedHasAnyAuthProfileStoreSource = vi.mocked(
-    authProfilesStoreModule.hasAnyAuthProfileStoreSource,
+    authProfilesSourceCheckModule.hasAnyAuthProfileStoreSource,
   );
   mockedGetSoonestCooldownExpiry = vi.mocked(authProfilesUsageModule.getSoonestCooldownExpiry);
   mockedIsProfileInCooldown = vi.mocked(authProfilesUsageModule.isProfileInCooldown);