MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-07-14 12:00:18 -07:00
Code Issues Packages Projects Releases Wiki Activity

Migrate tests to use avoid jsdom (#12118)

Browse Source
This commit is contained in:
Jacob Richman
2025-10-28 10:32:15 -07:00
committed by GitHub
parent 16a005304e
commit 8edec7612e
31 changed files with 765 additions and 579 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/cli/src/ui/hooks/useGeminiStream.test.tsx
+37 -38
View File
@@ -4,12 +4,11 @@
* SPDX-License-Identifier: Apache-2.0
*/
/** @vitest-environment jsdom */
/* eslint-disable @typescript-eslint/no-explicit-any */
import type { Mock, MockInstance } from 'vitest';
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';
@@ -507,7 +506,7 @@ describe('useGeminiStream', () => {
}
});
await waitFor(() => {
await vi.waitFor(() => {
expect(mockMarkToolsAsSubmitted).toHaveBeenCalledTimes(1);
expect(mockSendMessageStream).toHaveBeenCalledTimes(1);
});
@@ -590,7 +589,7 @@ describe('useGeminiStream', () => {
}
});
await waitFor(() => {
await vi.waitFor(() => {
expect(mockMarkToolsAsSubmitted).toHaveBeenCalledWith(['1']);
expect(client.addHistory).toHaveBeenCalledWith({
role: 'user',
@@ -702,7 +701,7 @@ describe('useGeminiStream', () => {
}
});
await waitFor(() => {
await vi.waitFor(() => {
// The tools should be marked as submitted locally
expect(mockMarkToolsAsSubmitted).toHaveBeenCalledWith([
'cancel-1',
@@ -840,7 +839,7 @@ describe('useGeminiStream', () => {
});
// 5. Wait for submitQuery to be called
await waitFor(() => {
await vi.waitFor(() => {
expect(mockSendMessageStream).toHaveBeenCalledWith(
toolCallResponseParts,
expect.any(AbortSignal),
@@ -889,7 +888,7 @@ describe('useGeminiStream', () => {
});
// Wait for the first part of the response
await waitFor(() => {
await vi.waitFor(() => {
expect(result.current.streamingState).toBe(StreamingState.Responding);
});
@@ -897,7 +896,7 @@ describe('useGeminiStream', () => {
simulateEscapeKeyPress();
// Verify cancellation message is added
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
{
type: MessageType.INFO,
@@ -1030,7 +1029,7 @@ describe('useGeminiStream', () => {
result.current.submitQuery('long running query');
});
await waitFor(() => {
await vi.waitFor(() => {
expect(result.current.streamingState).toBe(StreamingState.Responding);
});
@@ -1138,7 +1137,7 @@ describe('useGeminiStream', () => {
expect(mockCancelAllToolCalls).toHaveBeenCalled();
// A cancellation message should be added to history
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
expect.objectContaining({
text: 'Request cancelled.',
@@ -1167,7 +1166,7 @@ describe('useGeminiStream', () => {
await result.current.submitQuery('/memory add "test fact"');
});
await waitFor(() => {
await vi.waitFor(() => {
expect(mockScheduleToolCalls).toHaveBeenCalledWith(
[
expect.objectContaining({
@@ -1194,7 +1193,7 @@ describe('useGeminiStream', () => {
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
@@ -1215,7 +1214,7 @@ describe('useGeminiStream', () => {
await result.current.submitQuery('/my-custom-command');
});
await waitFor(() => {
await vi.waitFor(() => {
expect(mockHandleSlashCommand).toHaveBeenCalledWith(
'/my-custom-command',
);
@@ -1250,7 +1249,7 @@ describe('useGeminiStream', () => {
await result.current.submitQuery('/emptycmd');
});
await waitFor(() => {
await vi.waitFor(() => {
expect(mockHandleSlashCommand).toHaveBeenCalledWith('/emptycmd');
expect(localMockSendMessageStream).toHaveBeenCalledWith(
'',
@@ -1268,7 +1267,7 @@ describe('useGeminiStream', () => {
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',
@@ -1286,7 +1285,7 @@ describe('useGeminiStream', () => {
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 */',
@@ -1324,7 +1323,7 @@ describe('useGeminiStream', () => {
await result.current.submitQuery('/about');
});
await waitFor(() => {
await vi.waitFor(() => {
expect(mockHandleSlashCommand).not.toHaveBeenCalled();
});
});
@@ -1401,7 +1400,7 @@ describe('useGeminiStream', () => {
}
});
await waitFor(() => {
await vi.waitFor(() => {
expect(mockPerformMemoryRefresh).toHaveBeenCalledTimes(1);
});
});
@@ -1457,7 +1456,7 @@ describe('useGeminiStream', () => {
});
// 3. Assertion
await waitFor(() => {
await vi.waitFor(() => {
expect(mockParseAndFormatApiError).toHaveBeenCalledWith(
'Rate limit exceeded',
mockAuthType,
@@ -1990,7 +1989,7 @@ describe('useGeminiStream', () => {
});
// Check that the info message was added
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
{
type: 'info',
@@ -2050,7 +2049,7 @@ describe('useGeminiStream', () => {
});
// Check that the message was added without suggestion
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
{
type: 'info',
@@ -2105,7 +2104,7 @@ describe('useGeminiStream', () => {
});
// Check that the message was added with suggestion
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
{
type: 'info',
@@ -2161,7 +2160,7 @@ describe('useGeminiStream', () => {
});
// Check that onCancelSubmit was called
await waitFor(() => {
await vi.waitFor(() => {
expect(onCancelSubmitSpy).toHaveBeenCalled();
});
});
@@ -2360,7 +2359,7 @@ describe('useGeminiStream', () => {
await result.current.submitQuery(`Test ${reason}`);
});
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
{
type: 'info',
@@ -2487,7 +2486,7 @@ describe('useGeminiStream', () => {
});
// Wait for the first response to complete
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
expect.objectContaining({
type: 'gemini',
@@ -2520,7 +2519,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',
@@ -2638,7 +2637,7 @@ describe('useGeminiStream', () => {
});
// Verify cancellation message was added
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
expect.objectContaining({
type: 'info',
@@ -2696,7 +2695,7 @@ describe('useGeminiStream', () => {
});
// Verify error message was added
await waitFor(() => {
await vi.waitFor(() => {
expect(mockAddItem).toHaveBeenCalledWith(
expect.objectContaining({
type: 'error',
@@ -2747,7 +2746,7 @@ describe('useGeminiStream', () => {
await result.current.submitQuery('test query');
});
await waitFor(() => {
await vi.waitFor(() => {
expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
expect(
typeof result.current.loopDetectionConfirmationRequest?.onComplete,
@@ -2795,7 +2794,7 @@ describe('useGeminiStream', () => {
});
// Wait for confirmation request to be set
await waitFor(() => {
await vi.waitFor(() => {
expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
});
@@ -2824,7 +2823,7 @@ describe('useGeminiStream', () => {
);
// Verify that the request was retried
await waitFor(() => {
await vi.waitFor(() => {
expect(mockSendMessageStream).toHaveBeenCalledTimes(2);
expect(mockSendMessageStream).toHaveBeenNthCalledWith(
2,
@@ -2860,7 +2859,7 @@ describe('useGeminiStream', () => {
});
// Wait for confirmation request to be set
await waitFor(() => {
await vi.waitFor(() => {
expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
});
@@ -2907,7 +2906,7 @@ describe('useGeminiStream', () => {
await result.current.submitQuery('first query');
});
await waitFor(() => {
await vi.waitFor(() => {
expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
});
@@ -2957,7 +2956,7 @@ describe('useGeminiStream', () => {
await result.current.submitQuery('second query');
});
await waitFor(() => {
await vi.waitFor(() => {
expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
});
@@ -2980,7 +2979,7 @@ describe('useGeminiStream', () => {
);
// 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,
@@ -3011,7 +3010,7 @@ describe('useGeminiStream', () => {
});
// 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',
@@ -3022,7 +3021,7 @@ describe('useGeminiStream', () => {
});
// Then verify loop detection confirmation request was set
await waitFor(() => {
await vi.waitFor(() => {
expect(result.current.loopDetectionConfirmationRequest).not.toBeNull();
});
});