@google/gemini-cli 0.12.0-nightly.20251027.cb0947c5 → 0.12.0-preview.0

package/dist/src/ui/hooks/useGeminiStream.test.js

@@ -4,12 +4,13 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { renderHook, act, waitFor } from '@testing-library/react';
+import { act } from 'react';
+import { renderHook } from '../../test-utils/render.js';
 import { useGeminiStream } from './useGeminiStream.js';
 import { useKeypress } from './useKeypress.js';
 import * as atCommandProcessor from './atCommandProcessor.js';
 import { useReactToolScheduler } from './useReactToolScheduler.js';
-import { ApprovalMode, AuthType, GeminiEventType as ServerGeminiEventType, ToolErrorType, ToolConfirmationOutcome, tokenLimit, } from '@google/gemini-cli-core';
+import { ApprovalMode, AuthType, GeminiEventType as ServerGeminiEventType, ToolErrorType, ToolConfirmationOutcome, tokenLimit, debugLogger, } from '@google/gemini-cli-core';
 import { MessageType, StreamingState } from '../types.js';
 // --- MOCKS ---
 const mockSendMessageStream = vi
@@ -378,7 +379,7 @@ describe('useGeminiStream', () => {
                 await capturedOnComplete(completedToolCalls);
             }
         });
-        await waitFor(() => {
+        await vi.waitFor(() => {
             expect(mockMarkToolsAsSubmitted).toHaveBeenCalledTimes(1);
             expect(mockSendMessageStream).toHaveBeenCalledTimes(1);
         });
@@ -427,7 +428,7 @@ describe('useGeminiStream', () => {
                 await capturedOnComplete(cancelledToolCalls);
             }
         });
-        await waitFor(() => {
+        await vi.waitFor(() => {
             expect(mockMarkToolsAsSubmitted).toHaveBeenCalledWith(['1']);
             expect(client.addHistory).toHaveBeenCalledWith({
                 role: 'user',
@@ -510,7 +511,7 @@ describe('useGeminiStream', () => {
                 await capturedOnComplete(allCancelledTools);
             }
         });
-        await waitFor(() => {
+        await vi.waitFor(() => {
             // The tools should be marked as submitted locally
             expect(mockMarkToolsAsSubmitted).toHaveBeenCalledWith([
                 'cancel-1',
@@ -610,7 +611,7 @@ describe('useGeminiStream', () => {
             }
         });
         // 5. Wait for submitQuery to be called
-        await waitFor(() => {
+        await vi.waitFor(() => {
             expect(mockSendMessageStream).toHaveBeenCalledWith(toolCallResponseParts, expect.any(AbortSignal), 'prompt-id-4');
         });
         // 6. After submission, the state should remain Responding until the stream completes.
@@ -648,13 +649,13 @@ describe('useGeminiStream', () => {
                 result.current.submitQuery('test query');
             });
             // Wait for the first part of the response
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(result.current.streamingState).toBe(StreamingState.Responding);
             });
             // Simulate escape key press
             simulateEscapeKeyPress();
             // Verify cancellation message is added
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockAddItem).toHaveBeenCalledWith({
                     type: MessageType.INFO,
                     text: 'Request cancelled.',
@@ -720,7 +721,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 result.current.submitQuery('long running query');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(result.current.streamingState).toBe(StreamingState.Responding);
             });
             // Cancel the request
@@ -808,7 +809,7 @@ describe('useGeminiStream', () => {
             // The imperative cancel function should be called on the scheduler
             expect(mockCancelAllToolCalls).toHaveBeenCalled();
             // A cancellation message should be added to history
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
                     text: 'Request cancelled.',
                 }), expect.any(Number));
@@ -829,7 +830,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('/memory add "test fact"');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockScheduleToolCalls).toHaveBeenCalledWith([
                     expect.objectContaining({
                         name: 'save_memory',
@@ -849,7 +850,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('/help');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockHandleSlashCommand).toHaveBeenCalledWith('/help');
                 expect(mockScheduleToolCalls).not.toHaveBeenCalled();
                 expect(mockSendMessageStream).not.toHaveBeenCalled(); // No LLM call made
@@ -865,7 +866,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('/my-custom-command');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockHandleSlashCommand).toHaveBeenCalledWith('/my-custom-command');
                 expect(localMockSendMessageStream).not.toHaveBeenCalledWith('/my-custom-command', expect.anything(), expect.anything());
                 expect(localMockSendMessageStream).toHaveBeenCalledWith('This is the actual prompt from the command file.', expect.any(AbortSignal), expect.any(String));
@@ -882,7 +883,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('/emptycmd');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockHandleSlashCommand).toHaveBeenCalledWith('/emptycmd');
                 expect(localMockSendMessageStream).toHaveBeenCalledWith('', expect.any(AbortSignal), expect.any(String));
             });
@@ -892,7 +893,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('// This is a line comment');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockHandleSlashCommand).not.toHaveBeenCalled();
                 expect(localMockSendMessageStream).toHaveBeenCalledWith('// This is a line comment', expect.any(AbortSignal), expect.any(String));
             });
@@ -902,7 +903,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('/* This is a block comment */');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockHandleSlashCommand).not.toHaveBeenCalled();
                 expect(localMockSendMessageStream).toHaveBeenCalledWith('/* This is a block comment */', expect.any(AbortSignal), expect.any(String));
             });
@@ -912,7 +913,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('/about');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockHandleSlashCommand).not.toHaveBeenCalled();
             });
         });
@@ -960,7 +961,7 @@ describe('useGeminiStream', () => {
                     await capturedOnComplete([completedToolCall]);
                 }
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockPerformMemoryRefresh).toHaveBeenCalledTimes(1);
             });
         });
@@ -988,7 +989,7 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('test query');
             });
             // 3. Assertion
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockParseAndFormatApiError).toHaveBeenCalledWith('Rate limit exceeded', mockAuthType, undefined, 'gemini-2.5-pro', 'gemini-2.5-flash');
             });
         });
@@ -1209,8 +1210,8 @@ describe('useGeminiStream', () => {
             expect(mockOnConfirm).not.toHaveBeenCalled();
         });
         it('should handle errors gracefully when auto-approving tool calls', async () => {
-            const consoleSpy = vi
-                .spyOn(console, 'error')
+            const debuggerSpy = vi
+                .spyOn(debugLogger, 'warn')
                 .mockImplementation(() => { });
             const mockOnConfirmSuccess = vi.fn().mockResolvedValue(undefined);
             const mockOnConfirmError = vi
@@ -1286,8 +1287,8 @@ describe('useGeminiStream', () => {
             expect(mockOnConfirmSuccess).toHaveBeenCalledTimes(1);
             expect(mockOnConfirmError).toHaveBeenCalledTimes(1);
             // Error should be logged
-            expect(consoleSpy).toHaveBeenCalledWith('Failed to auto-approve tool call call2:', expect.any(Error));
-            consoleSpy.mockRestore();
+            expect(debuggerSpy).toHaveBeenCalledWith('Failed to auto-approve tool call call2:', expect.any(Error));
+            debuggerSpy.mockRestore();
         });
         it('should skip tool calls without confirmationDetails', async () => {
             const awaitingApprovalToolCalls = [
@@ -1443,7 +1444,7 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('Generate long text');
             });
             // Check that the info message was added
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockAddItem).toHaveBeenCalledWith({
                     type: 'info',
                     text: '⚠️  Response truncated due to token limits.',
@@ -1472,7 +1473,7 @@ describe('useGeminiStream', () => {
                     await result.current.submitQuery('Test overflow');
                 });
                 // Check that the message was added without suggestion
-                await waitFor(() => {
+                await vi.waitFor(() => {
                     expect(mockAddItem).toHaveBeenCalledWith({
                         type: 'info',
                         text: `Sending this message (20 tokens) might exceed the remaining context window limit (80 tokens).`,
@@ -1497,7 +1498,7 @@ describe('useGeminiStream', () => {
                     await result.current.submitQuery('Test overflow');
                 });
                 // Check that the message was added with suggestion
-                await waitFor(() => {
+                await vi.waitFor(() => {
                     expect(mockAddItem).toHaveBeenCalledWith({
                         type: 'info',
                         text: `Sending this message (30 tokens) might exceed the remaining context window limit (70 tokens). Please try reducing the size of your message or use the \`/compress\` command to compress the chat history.`,
@@ -1523,7 +1524,7 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('Test overflow');
             });
             // Check that onCancelSubmit was called
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(onCancelSubmitSpy).toHaveBeenCalled();
             });
         });
@@ -1633,7 +1634,7 @@ describe('useGeminiStream', () => {
                 await act(async () => {
                     await result.current.submitQuery(`Test ${reason}`);
                 });
-                await waitFor(() => {
+                await vi.waitFor(() => {
                     expect(mockAddItem).toHaveBeenCalledWith({
                         type: 'info',
                         text: message,
@@ -1706,7 +1707,7 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('First query');
             });
             // Wait for the first response to complete
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
                     type: 'gemini',
                     text: 'Some response content',
@@ -1731,7 +1732,7 @@ describe('useGeminiStream', () => {
             // We can verify this by checking that the LoadingIndicator would not show the previous thought
             // The actual thought state is internal to the hook, but we can verify the behavior
             // by ensuring the second response doesn't show the previous thought
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
                     type: 'gemini',
                     text: 'New response content',
@@ -1790,7 +1791,7 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('Test query');
             });
             // Verify cancellation message was added
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
                     type: 'info',
                     text: 'User cancelled the request.',
@@ -1817,7 +1818,7 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('Test query');
             });
             // Verify error message was added
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
                     type: 'error',
                 }), expect.any(Number));
@@ -1851,7 +1852,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('test query');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
                 expect(typeof result.current.loopDetectionConfirmationRequest?.onComplete).toBe('function');
             });
@@ -1887,7 +1888,7 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('test query');
             });
             // Wait for confirmation request to be set
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
             });
             // Simulate user selecting "disable"
@@ -1906,7 +1907,7 @@ describe('useGeminiStream', () => {
                 text: 'Loop detection has been disabled for this session. Retrying request...',
             }, expect.any(Number));
             // Verify that the request was retried
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockSendMessageStream).toHaveBeenCalledTimes(2);
                 expect(mockSendMessageStream).toHaveBeenNthCalledWith(2, 'test query', expect.any(AbortSignal), expect.any(String));
             });
@@ -1930,7 +1931,7 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('test query');
             });
             // Wait for confirmation request to be set
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
             });
             // Simulate user selecting "keep"
@@ -1963,7 +1964,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('first query');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
             });
             // Simulate user selecting "keep" for first request
@@ -1999,7 +2000,7 @@ describe('useGeminiStream', () => {
             await act(async () => {
                 await result.current.submitQuery('second query');
             });
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
             });
             // Simulate user selecting "disable" for second request
@@ -2015,7 +2016,7 @@ describe('useGeminiStream', () => {
                 text: 'Loop detection has been disabled for this session. Retrying request...',
             }, expect.any(Number));
             // Verify that the request was retried
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockSendMessageStream).toHaveBeenCalledTimes(3); // 1st query, 2nd query, retry of 2nd query
                 expect(mockSendMessageStream).toHaveBeenNthCalledWith(3, 'second query', expect.any(AbortSignal), expect.any(String));
             });
@@ -2035,14 +2036,14 @@ describe('useGeminiStream', () => {
                 await result.current.submitQuery('test query');
             });
             // Verify that the content was added to history before the loop detection dialog
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(mockAddItem).toHaveBeenCalledWith(expect.objectContaining({
                     type: 'gemini',
                     text: 'Some response content',
                 }), expect.any(Number));
             });
             // Then verify loop detection confirmation request was set
-            await waitFor(() => {
+            await vi.waitFor(() => {
                 expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
             });
         });