From 74801e900413c2ca48a2977680e715cd28aa76d9 Mon Sep 17 00:00:00 2001
From: Brandon Keiji <brandonkeiji@google.com>
Date: Tue, 3 Jun 2025 02:10:54 +0000
Subject: refactor: maintain 1 GeminiChat per GeminiClient (#710)

---
 packages/cli/src/ui/hooks/useGeminiStream.test.tsx | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

(limited to 'packages/cli/src/ui/hooks/useGeminiStream.test.tsx')

diff --git a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
index 3a421ebf..d46fab9e 100644
--- a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
+++ b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
@@ -405,10 +405,9 @@ describe('useGeminiStream', () => {
       } as TrackedCancelledToolCall,
     ];
 
-    let hookResult: any;
-    await act(async () => {
-      hookResult = renderTestHook(simplifiedToolCalls);
-    });
+    const hookResult = await act(async () =>
+      renderTestHook(simplifiedToolCalls),
+    );
 
     const {
       mockMarkToolsAsSubmitted,
@@ -431,9 +430,8 @@ describe('useGeminiStream', () => {
       toolCall2ResponseParts,
     ]);
     expect(localMockSendMessageStream).toHaveBeenCalledWith(
-      expect.anything(),
       expectedMergedResponse,
-      expect.anything(),
+      expect.any(AbortSignal),
     );
   });
 });
-- 
cgit v1.2.3