Improve code coverage for cli package (#13724)

packages/cli/src/ui/auth/AuthInProgress.test.tsx

@@ -0,0 +1,90 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { render } from 'ink-testing-library';
+import { act } from 'react';
+import { AuthInProgress } from './AuthInProgress.js';
+import { useKeypress, type Key } from '../hooks/useKeypress.js';
+
+// Mock dependencies
+vi.mock('../hooks/useKeypress.js', () => ({
+  useKeypress: vi.fn(),
+}));
+
+vi.mock('../components/CliSpinner.js', () => ({
+  CliSpinner: () => '[Spinner]',
+}));
+
+describe('AuthInProgress', () => {
+  const onTimeout = vi.fn();
+
+  const originalError = console.error;
+
+  beforeEach(() => {
+    vi.clearAllMocks();
+    vi.useFakeTimers();
+    console.error = (...args) => {
+      if (
+        typeof args[0] === 'string' &&
+        args[0].includes('was not wrapped in act')
+      ) {
+        return;
+      }
+      originalError.call(console, ...args);
+    };
+  });
+
+  afterEach(() => {
+    console.error = originalError;
+    vi.useRealTimers();
+  });
+
+  it('renders initial state with spinner', () => {
+    const { lastFrame } = render(<AuthInProgress onTimeout={onTimeout} />);
+    expect(lastFrame()).toContain('[Spinner] Waiting for auth...');
+    expect(lastFrame()).toContain('Press ESC or CTRL+C to cancel');
+  });
+
+  it('calls onTimeout when ESC is pressed', () => {
+    render(<AuthInProgress onTimeout={onTimeout} />);
+    const keypressHandler = vi.mocked(useKeypress).mock.calls[0][0];
+
+    keypressHandler({ name: 'escape' } as unknown as Key);
+    expect(onTimeout).toHaveBeenCalled();
+  });
+
+  it('calls onTimeout when Ctrl+C is pressed', () => {
+    render(<AuthInProgress onTimeout={onTimeout} />);
+    const keypressHandler = vi.mocked(useKeypress).mock.calls[0][0];
+
+    keypressHandler({ name: 'c', ctrl: true } as unknown as Key);
+    expect(onTimeout).toHaveBeenCalled();
+  });
+
+  it('calls onTimeout and shows timeout message after 3 minutes', async () => {
+    const { lastFrame } = render(<AuthInProgress onTimeout={onTimeout} />);
+
+    await act(async () => {
+      vi.advanceTimersByTime(180000);
+    });
+
+    expect(onTimeout).toHaveBeenCalled();
+    await vi.waitUntil(
+      () => lastFrame()?.includes('Authentication timed out'),
+      { timeout: 1000 },
+    );
+  });
+
+  it('clears timer on unmount', () => {
+    const { unmount } = render(<AuthInProgress onTimeout={onTimeout} />);
+    act(() => {
+      unmount();
+    });
+    vi.advanceTimersByTime(180000);
+    expect(onTimeout).not.toHaveBeenCalled();
+  });
+});