gemini-cli/packages/cli/src/ui/components/NewAgentsNotification.test.tsx

/**
 * @license
 * Copyright 2026 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { describe, it, expect, vi } from 'vitest';
import { renderWithProviders as render } from '../../test-utils/render.js';
import { NewAgentsNotification } from './NewAgentsNotification.js';
import { waitFor } from '../../test-utils/async.js';
import { act } from 'react';

describe('NewAgentsNotification', () => {
  const mockAgents = [
    {
      name: 'Agent A',
      description: 'Description A',
      kind: 'remote' as const,
      agentCardUrl: '',
      inputConfig: { inputSchema: {} },
    },
    {
      name: 'Agent B',
      description: 'Description B',
      kind: 'remote' as const,
      agentCardUrl: '',
      inputConfig: { inputSchema: {} },
    },
  ];
  const onSelect = vi.fn();

  it('renders agent list', async () => {
    const { lastFrame, waitUntilReady, unmount } = render(
      <NewAgentsNotification agents={mockAgents} onSelect={onSelect} />,
    );
    await waitUntilReady();

    const frame = lastFrame();
    expect(frame).toMatchSnapshot();
    unmount();
  });

  it('truncates list if more than 5 agents', async () => {
    const manyAgents = Array.from({ length: 7 }, (_, i) => ({
      name: `Agent ${i}`,
      description: `Description ${i}`,
      kind: 'remote' as const,
      agentCardUrl: '',
      inputConfig: { inputSchema: {} },
    }));

    const { lastFrame, waitUntilReady, unmount } = render(
      <NewAgentsNotification agents={manyAgents} onSelect={onSelect} />,
    );
    await waitUntilReady();

    const frame = lastFrame();
    expect(frame).toMatchSnapshot();
    unmount();
  });

  it('shows processing state when an option is selected', async () => {
    const asyncOnSelect = vi.fn(
      () =>
        new Promise<void>(() => {
          // Never resolve
        }),
    );

    const { lastFrame, stdin, unmount } = render(
      <NewAgentsNotification agents={mockAgents} onSelect={asyncOnSelect} />,
    );

    // Press Enter to select the first option
    await act(async () => {
      stdin.write('\r');
    });

    await waitFor(() => {
      expect(lastFrame()).toContain('Processing...');
    });

    unmount();
  });
});