From 8f217ff01ef293952da77e81b5814897b05bd725 Mon Sep 17 00:00:00 2001
From: Shreya Keshive <shreyakeshive@google.com>
Date: Tue, 3 Mar 2026 16:15:09 -0500
Subject: [PATCH] fix tests

---
 packages/cli/src/gemini.test.tsx         | 2 ++
 packages/cli/src/gemini_cleanup.test.tsx | 1 +
 2 files changed, 3 insertions(+)

diff --git a/packages/cli/src/gemini.test.tsx b/packages/cli/src/gemini.test.tsx
index c6625cadd3..ca25d09f68 100644
--- a/packages/cli/src/gemini.test.tsx
+++ b/packages/cli/src/gemini.test.tsx
@@ -545,6 +545,7 @@ describe('gemini.tsx main function kitty protocol', () => {
       getQuestion: () => '',
       getSandbox: () => undefined,
       getListExtensions: () => flag === 'listExtensions',
+      getAcpMode: () => false,
       getListSessions: () => flag === 'listSessions',
       getDeleteSession: () => (flag === 'deleteSession' ? value : undefined),
       getExtensions: () => [
@@ -619,6 +620,7 @@ describe('gemini.tsx main function kitty protocol', () => {
     const mockConfig = createMockConfig({
       isInteractive: () => false,
       getQuestion: () => '',
+      getAcpMode: () => false,
       getSandbox: () => ({ command: 'docker', image: 'test-image' }),
     });
 
diff --git a/packages/cli/src/gemini_cleanup.test.tsx b/packages/cli/src/gemini_cleanup.test.tsx
index 5c229d1782..458a350132 100644
--- a/packages/cli/src/gemini_cleanup.test.tsx
+++ b/packages/cli/src/gemini_cleanup.test.tsx
@@ -69,6 +69,7 @@ vi.mock('./config/config.js', () => ({
     getSandbox: vi.fn(() => false),
     getQuestion: vi.fn(() => ''),
     isInteractive: () => false,
+    getAcpMode: vi.fn(() => false),
     storage: { initialize: vi.fn().mockResolvedValue(undefined) },
   } as unknown as Config),
   parseArguments: vi.fn().mockResolvedValue({}),