From 7b79579d20bd8995c6d32f69ce7100b5d0fe6687 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Tue, 7 Apr 2026 14:42:21 +0800
Subject: [PATCH] Tests: fix agent runtime drift

---
 src/agents/model-selection.test.ts  | 2 +-
 src/agents/tools/image-tool.test.ts | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/agents/model-selection.test.ts b/src/agents/model-selection.test.ts
index f060c203636..fc2ccefdd2a 100644
--- a/src/agents/model-selection.test.ts
+++ b/src/agents/model-selection.test.ts
@@ -954,7 +954,7 @@ describe("model-selection", () => {
 
       expect(result).toEqual({
         provider: "google-vertex",
-        model: "gemini-3.1-flash-lite",
+        model: "gemini-3.1-flash-lite-preview",
       });
     });
 
diff --git a/src/agents/tools/image-tool.test.ts b/src/agents/tools/image-tool.test.ts
index fc54635d0ee..651799a6410 100644
--- a/src/agents/tools/image-tool.test.ts
+++ b/src/agents/tools/image-tool.test.ts
@@ -107,6 +107,12 @@ vi.mock("../auth-profiles.js", () => ({
       return { version: 1, profiles: {} };
     }
   },
+  hasAnyAuthProfileStoreSource: (agentDir?: string) => {
+    if (!agentDir) {
+      return false;
+    }
+    return fsSync.existsSync(path.join(agentDir, "auth-profiles.json"));
+  },
   listProfilesForProvider: (
     store: { profiles?: Record<string, { provider?: string }> },
     provider: string,