Improve test coverage for cli/src/ui/components (#13598)

packages/cli/src/ui/components/GeminiRespondingSpinner.test.tsx

@@ -0,0 +1,72 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { render } from '../../test-utils/render.js';
+import { GeminiRespondingSpinner } from './GeminiRespondingSpinner.js';
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { useStreamingContext } from '../contexts/StreamingContext.js';
+import { useIsScreenReaderEnabled } from 'ink';
+import { StreamingState } from '../types.js';
+import {
+  SCREEN_READER_LOADING,
+  SCREEN_READER_RESPONDING,
+} from '../textConstants.js';
+
+vi.mock('../contexts/StreamingContext.js');
+vi.mock('ink', async () => {
+  const actual = await vi.importActual('ink');
+  return {
+    ...actual,
+    useIsScreenReaderEnabled: vi.fn(),
+  };
+});
+
+describe('GeminiRespondingSpinner', () => {
+  const mockUseStreamingContext = vi.mocked(useStreamingContext);
+  const mockUseIsScreenReaderEnabled = vi.mocked(useIsScreenReaderEnabled);
+
+  beforeEach(() => {
+    vi.clearAllMocks();
+    mockUseIsScreenReaderEnabled.mockReturnValue(false);
+  });
+
+  it('renders spinner when responding', () => {
+    mockUseStreamingContext.mockReturnValue(StreamingState.Responding);
+    const { lastFrame } = render(<GeminiRespondingSpinner />);
+    // Spinner output varies, but it shouldn't be empty
+    expect(lastFrame()).not.toBe('');
+  });
+
+  it('renders screen reader text when responding and screen reader enabled', () => {
+    mockUseStreamingContext.mockReturnValue(StreamingState.Responding);
+    mockUseIsScreenReaderEnabled.mockReturnValue(true);
+    const { lastFrame } = render(<GeminiRespondingSpinner />);
+    expect(lastFrame()).toContain(SCREEN_READER_RESPONDING);
+  });
+
+  it('renders nothing when not responding and no non-responding display', () => {
+    mockUseStreamingContext.mockReturnValue(StreamingState.Idle);
+    const { lastFrame } = render(<GeminiRespondingSpinner />);
+    expect(lastFrame()).toBe('');
+  });
+
+  it('renders non-responding display when provided', () => {
+    mockUseStreamingContext.mockReturnValue(StreamingState.Idle);
+    const { lastFrame } = render(
+      <GeminiRespondingSpinner nonRespondingDisplay="Waiting..." />,
+    );
+    expect(lastFrame()).toContain('Waiting...');
+  });
+
+  it('renders screen reader loading text when non-responding display provided and screen reader enabled', () => {
+    mockUseStreamingContext.mockReturnValue(StreamingState.Idle);
+    mockUseIsScreenReaderEnabled.mockReturnValue(true);
+    const { lastFrame } = render(
+      <GeminiRespondingSpinner nonRespondingDisplay="Waiting..." />,
+    );
+    expect(lastFrame()).toContain(SCREEN_READER_LOADING);
+  });
+});