gemini-cli/packages/cli/src/ui/hooks/useShellInactivityStatus.test.ts

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

import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import { act } from 'react';
import { renderHook } from '../../test-utils/render.js';
import { useShellInactivityStatus } from './useShellInactivityStatus.js';
import { useTurnActivityMonitor } from './useTurnActivityMonitor.js';
import { StreamingState } from '../types.js';

vi.mock('./useTurnActivityMonitor.js', () => ({
  useTurnActivityMonitor: vi.fn(),
}));

describe('useShellInactivityStatus', () => {
  beforeEach(() => {
    vi.useFakeTimers();
    vi.mocked(useTurnActivityMonitor).mockReturnValue({
      operationStartTime: 1000,
      isRedirectionActive: false,
    });
  });

  afterEach(() => {
    vi.clearAllMocks();
    vi.useRealTimers();
  });

  const defaultProps = {
    activePtyId: 'pty-1',
    lastOutputTime: 1001,
    streamingState: StreamingState.Responding,
    pendingToolCalls: [],
    embeddedShellFocused: false,
    isInteractiveShellEnabled: true,
  };

  it('should show action_required status after 30s when output has been produced', async () => {
    const { result } = renderHook(() => useShellInactivityStatus(defaultProps));

    expect(result.current.inactivityStatus).toBe('none');

    await act(async () => {
      await vi.advanceTimersByTimeAsync(30000);
    });
    expect(result.current.inactivityStatus).toBe('action_required');
  });

  it('should show silent_working status after 60s when no output has been produced (silent)', async () => {
    const { result } = renderHook(() =>
      useShellInactivityStatus({ ...defaultProps, lastOutputTime: 500 }),
    );

    await act(async () => {
      await vi.advanceTimersByTimeAsync(30000);
    });
    expect(result.current.inactivityStatus).toBe('none');

    await act(async () => {
      await vi.advanceTimersByTimeAsync(30000);
    });
    expect(result.current.inactivityStatus).toBe('silent_working');
  });

  it('should show silent_working status after 2 mins for redirected commands', async () => {
    vi.mocked(useTurnActivityMonitor).mockReturnValue({
      operationStartTime: 1000,
      isRedirectionActive: true,
    });

    const { result } = renderHook(() => useShellInactivityStatus(defaultProps));

    // Should NOT show action_required even after 60s
    await act(async () => {
      await vi.advanceTimersByTimeAsync(60000);
    });
    expect(result.current.inactivityStatus).toBe('none');

    // Should show silent_working after 2 mins (120000ms)
    await act(async () => {
      await vi.advanceTimersByTimeAsync(60000);
    });
    expect(result.current.inactivityStatus).toBe('silent_working');
  });

  it('should suppress focus hint when redirected', async () => {
    vi.mocked(useTurnActivityMonitor).mockReturnValue({
      operationStartTime: 1000,
      isRedirectionActive: true,
    });

    const { result } = renderHook(() => useShellInactivityStatus(defaultProps));

    // Even after delay, focus hint should be suppressed
    await act(async () => {
      await vi.advanceTimersByTimeAsync(20000);
    });
    expect(result.current.shouldShowFocusHint).toBe(false);
  });
});