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

test(cli): fix flaky LoginWithGoogleRestartDialog assertions and prune dead types

Browse Source
This commit is contained in:
mkorwel
2026-04-16 00:24:45 +00:00
parent f31d959095
commit cb66f06883
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx
+7 -7
View File
@@ -8,9 +8,7 @@ import { render } from '../../test-utils/render.js';
import { describe, it, expect, vi, beforeEach, type Mock } from 'vitest';
import { LoginWithGoogleRestartDialog } from './LoginWithGoogleRestartDialog.js';
import { useKeypress } from '../hooks/useKeypress.js';
import { runExitCleanup } from '../../utils/cleanup.js';
import {
RELAUNCH_EXIT_CODE,
_resetRelaunchStateForTesting,
} from '../../utils/processUtils.js';
import { type Config } from '@google/gemini-cli-core';
@@ -20,12 +18,15 @@ vi.mock('../hooks/useKeypress.js', () => ({
useKeypress: vi.fn(),
}));
vi.mock('../../utils/cleanup.js', () => ({
runExitCleanup: vi.fn(),
vi.mock('../../utils/processUtils.js', () => ({
relaunchApp: vi.fn().mockResolvedValue(undefined),
RELAUNCH_EXIT_CODE: 199,
_resetRelaunchStateForTesting: vi.fn(),
}));
const mockedUseKeypress = useKeypress as Mock;
const mockedRunExitCleanup = runExitCleanup as Mock;
import { relaunchApp } from '../../utils/processUtils.js';
const mockedRelaunchApp = relaunchApp as Mock;
describe('LoginWithGoogleRestartDialog', () => {
const onDismiss = vi.fn();
@@ -100,8 +101,7 @@ describe('LoginWithGoogleRestartDialog', () => {
// Advance timers to trigger the setTimeout callback
await vi.runAllTimersAsync();
expect(mockedRunExitCleanup).toHaveBeenCalledTimes(1);
expect(exitSpy).toHaveBeenCalledWith(RELAUNCH_EXIT_CODE);
expect(mockedRelaunchApp).toHaveBeenCalledTimes(1);
vi.useRealTimers();
unmount();