packages/cli/src/ui/components/messages/SubagentProgressDisplay.test.tsx

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

import { render, cleanup } from '../../../test-utils/render.js';
import { SubagentProgressDisplay } from './SubagentProgressDisplay.js';
import { type SubagentProgress, SubagentState } from '@google/gemini-cli-core';
import { describe, it, expect, vi, afterEach } from 'vitest';

describe('<SubagentProgressDisplay />', () => {
  afterEach(() => {
    vi.restoreAllMocks();
    cleanup();
  });

  it('renders correctly with description in args', async () => {
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [
        {
          id: '1',
          type: 'tool_call',
          content: 'run_shell_command',
          args: '{"command": "echo hello", "description": "Say hello"}',
          status: SubagentState.RUNNING,
        },
      ],
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });

  it('renders correctly with displayName and description from item', async () => {
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [
        {
          id: '1',
          type: 'tool_call',
          content: 'run_shell_command',
          displayName: 'RunShellCommand',
          description: 'Executing echo hello',
          args: '{"command": "echo hello"}',
          status: SubagentState.RUNNING,
        },
      ],
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });

  it('renders correctly with command fallback', async () => {
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [
        {
          id: '2',
          type: 'tool_call',
          content: 'run_shell_command',
          args: '{"command": "echo hello"}',
          status: SubagentState.RUNNING,
        },
      ],
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });

  it('renders correctly with file_path', async () => {
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [
        {
          id: '3',
          type: 'tool_call',
          content: 'write_file',
          args: '{"file_path": "/tmp/test.txt", "content": "foo"}',
          status: SubagentState.COMPLETED,
        },
      ],
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });

  it('truncates long args', async () => {
    const longDesc =
      'This is a very long description that should definitely be truncated because it exceeds the limit of sixty characters.';
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [
        {
          id: '4',
          type: 'tool_call',
          content: 'run_shell_command',
          args: JSON.stringify({ description: longDesc }),
          status: SubagentState.RUNNING,
        },
      ],
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });

  it('renders thought bubbles correctly', async () => {
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [
        {
          id: '5',
          type: 'thought',
          content: 'Thinking about life',
          status: SubagentState.RUNNING,
        },
      ],
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });

  it('renders cancelled state correctly', async () => {
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [],
      state: SubagentState.CANCELLED,
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });

  it('renders "Request cancelled." with the info icon', async () => {
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [
        {
          id: '6',
          type: 'thought',
          content: 'Request cancelled.',
          status: SubagentState.ERROR,
        },
      ],
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });

  it('renders error tool status correctly', async () => {
    const progress: SubagentProgress = {
      isSubagentProgress: true,
      agentName: 'TestAgent',
      recentActivity: [
        {
          id: '7',
          type: 'tool_call',
          content: 'run_shell_command',
          args: '{"command": "echo hello"}',
          status: SubagentState.ERROR,
        },
      ],
    };

    const { lastFrame } = await render(
      <SubagentProgressDisplay progress={progress} terminalWidth={80} />,
    );
    expect(lastFrame()).toMatchSnapshot();
  });
});