/** * @license * Copyright 2025 Google LLC * SPDX-License-Identifier: Apache-2.0 */ import { renderWithProviders } from '../../test-utils/render.js'; import { createMockSettings } from '../../test-utils/settings.js'; import { makeFakeConfig, ExitCodes } from '@google/gemini-cli-core'; import { waitFor } from '../../test-utils/async.js'; import { act } from 'react'; import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest'; import { FolderTrustDialog } from './FolderTrustDialog.js'; import * as processUtils from '../../utils/processUtils.js'; vi.mock('../../utils/processUtils.js', () => ({ relaunchApp: vi.fn(), })); const mockedExit = vi.hoisted(() => vi.fn()); const mockedCwd = vi.hoisted(() => vi.fn().mockReturnValue('/mock/cwd')); const mockedRows = vi.hoisted(() => ({ current: 24 })); vi.mock('node:process', async () => { const actual = await vi.importActual('node:process'); return { ...actual, exit: mockedExit, cwd: mockedCwd, }; }); vi.mock('../hooks/useTerminalSize.js', () => ({ useTerminalSize: () => ({ columns: 80, terminalHeight: mockedRows.current }), })); describe('FolderTrustDialog', () => { beforeEach(() => { vi.clearAllMocks(); vi.useRealTimers(); mockedCwd.mockReturnValue('/home/user/project'); mockedRows.current = 24; }); afterEach(() => { vi.restoreAllMocks(); }); it('should render the dialog with title and description', async () => { const { lastFrame, unmount } = await renderWithProviders( , ); expect(lastFrame()).toContain('Do you trust the files in this folder?'); expect(lastFrame()).toContain( 'Trusting a folder allows Gemini CLI to load its local configurations', ); unmount(); }); it('should truncate discovery results when they exceed maxDiscoveryHeight', async () => { // maxDiscoveryHeight = 24 - 15 = 9. const discoveryResults = { commands: Array.from({ length: 10 }, (_, i) => `cmd${i}`), mcps: Array.from({ length: 10 }, (_, i) => `mcp${i}`), hooks: Array.from({ length: 10 }, (_, i) => `hook${i}`), skills: Array.from({ length: 10 }, (_, i) => `skill${i}`), agents: [], settings: Array.from({ length: 10 }, (_, i) => `setting${i}`), discoveryErrors: [], securityWarnings: [], }; const { lastFrame, unmount } = await renderWithProviders( , { width: 80, config: makeFakeConfig({ useAlternateBuffer: false }), settings: createMockSettings({ ui: { useAlternateBuffer: false } }), uiState: { constrainHeight: true, terminalHeight: 24 }, }, ); expect(lastFrame()).toContain('This folder contains:'); expect(lastFrame()).not.toContain('cmd9'); unmount(); }); it('should adjust maxHeight based on terminal rows', async () => { mockedRows.current = 14; // maxHeight = 14 - 10 = 4 const discoveryResults = { commands: ['cmd1', 'cmd2', 'cmd3', 'cmd4', 'cmd5'], mcps: [], hooks: [], skills: [], agents: [], settings: [], discoveryErrors: [], securityWarnings: [], }; const { lastFrame, unmount } = await renderWithProviders( , { width: 80, config: makeFakeConfig({ useAlternateBuffer: false }), settings: createMockSettings({ ui: { useAlternateBuffer: false } }), uiState: { constrainHeight: true, terminalHeight: 14 }, }, ); // With maxHeight=4, the intro text (4 lines) will take most of the space. // The discovery results will likely be hidden. expect(lastFrame()).not.toContain('cmd1'); unmount(); }); it('should use minimum maxHeight of 4', async () => { mockedRows.current = 8; // 8 - 10 = -2, should use 4 const discoveryResults = { commands: ['cmd1', 'cmd2', 'cmd3', 'cmd4', 'cmd5'], mcps: [], hooks: [], skills: [], agents: [], settings: [], discoveryErrors: [], securityWarnings: [], }; const { lastFrame, unmount } = await renderWithProviders( , { width: 80, config: makeFakeConfig({ useAlternateBuffer: false }), settings: createMockSettings({ ui: { useAlternateBuffer: false } }), uiState: { constrainHeight: true, terminalHeight: 10 }, }, ); expect(lastFrame()).not.toContain('cmd1'); unmount(); }); it('should toggle expansion when global Ctrl+O is handled', async () => { const discoveryResults = { commands: Array.from({ length: 10 }, (_, i) => `cmd${i}`), mcps: [], hooks: [], skills: [], agents: [], settings: [], discoveryErrors: [], securityWarnings: [], }; const { lastFrame, unmount } = await renderWithProviders( , { width: 80, config: makeFakeConfig({ useAlternateBuffer: false }), settings: createMockSettings({ ui: { useAlternateBuffer: false } }), // Initially constrained uiState: { constrainHeight: true, terminalHeight: 24 }, }, ); // Initial state: truncated await waitFor(() => { expect(lastFrame()).toContain('Do you trust the files in this folder?'); expect(lastFrame()).not.toContain('cmd9'); }); unmount(); // We can't easily simulate global Ctrl+O toggle in this unit test // because it's handled in AppContainer. // But we can re-render with constrainHeight: false. const { lastFrame: lastFrameExpanded, unmount: unmountExpanded } = await renderWithProviders( , { width: 80, config: makeFakeConfig({ useAlternateBuffer: false }), settings: createMockSettings({ ui: { useAlternateBuffer: false } }), uiState: { constrainHeight: false, terminalHeight: 50 }, }, ); await waitFor(() => { expect(lastFrameExpanded()).not.toContain('hidden'); expect(lastFrameExpanded()).toContain('- cmd9'); expect(lastFrameExpanded()).toContain('- cmd4'); }); unmountExpanded(); }); it('should display exit message and call process.exit and not call onSelect when escape is pressed', async () => { const onSelect = vi.fn(); const { lastFrame, stdin, waitUntilReady, unmount } = await renderWithProviders( , ); await act(async () => { stdin.write('\u001b[27u'); // Press kitty escape key }); // Escape key has a 50ms timeout in KeypressContext, so we need to wrap waitUntilReady in act await act(async () => { await waitUntilReady(); }); await waitFor(() => { expect(lastFrame()).toContain( 'A folder trust level must be selected to continue. Exiting since escape was pressed.', ); }); await waitFor(() => { expect(mockedExit).toHaveBeenCalledWith( ExitCodes.FATAL_CANCELLATION_ERROR, ); }); expect(onSelect).not.toHaveBeenCalled(); unmount(); }); it('should display restart message when isRestarting is true', async () => { const { lastFrame, unmount } = await renderWithProviders( , ); expect(lastFrame()).toContain('Gemini CLI is restarting'); unmount(); }); it('should call relaunchApp when isRestarting is true', async () => { vi.useFakeTimers(); const relaunchApp = vi .spyOn(processUtils, 'relaunchApp') .mockResolvedValue(undefined); const { unmount } = await renderWithProviders( , ); await vi.advanceTimersByTimeAsync(250); expect(relaunchApp).toHaveBeenCalled(); unmount(); vi.useRealTimers(); }); it('should not call relaunchApp if unmounted before timeout', async () => { vi.useFakeTimers(); const relaunchApp = vi .spyOn(processUtils, 'relaunchApp') .mockResolvedValue(undefined); const { unmount } = await renderWithProviders( , ); // Unmount immediately (before 250ms) unmount(); await vi.advanceTimersByTimeAsync(250); expect(relaunchApp).not.toHaveBeenCalled(); vi.useRealTimers(); }); it('should not call process.exit when "r" is pressed and isRestarting is false', async () => { const { stdin, waitUntilReady, unmount } = await renderWithProviders( , ); await act(async () => { stdin.write('r'); }); await waitUntilReady(); await waitFor(() => { expect(mockedExit).not.toHaveBeenCalled(); }); unmount(); }); describe('directory display', () => { it('should correctly display the folder name for a nested directory', async () => { mockedCwd.mockReturnValue('/home/user/project'); const { lastFrame, unmount } = await renderWithProviders( , ); expect(lastFrame()).toContain('Trust folder (project)'); unmount(); }); it('should correctly display the parent folder name for a nested directory', async () => { mockedCwd.mockReturnValue('/home/user/project'); const { lastFrame, unmount } = await renderWithProviders( , ); expect(lastFrame()).toContain('Trust parent folder (user)'); unmount(); }); it('should correctly display an empty parent folder name for a directory directly under root', async () => { mockedCwd.mockReturnValue('/project'); const { lastFrame, unmount } = await renderWithProviders( , ); expect(lastFrame()).toContain('Trust parent folder ()'); unmount(); }); it('should display discovery results when provided', async () => { mockedRows.current = 40; // Increase height to show all results const discoveryResults = { commands: ['cmd1', 'cmd2'], mcps: ['mcp1'], hooks: ['hook1'], skills: ['skill1'], agents: ['agent1'], settings: ['general', 'ui'], discoveryErrors: [], securityWarnings: [], }; const { lastFrame, unmount } = await renderWithProviders( , { width: 80 }, ); expect(lastFrame()).toContain('This folder contains:'); expect(lastFrame()).toContain('• Commands (2):'); expect(lastFrame()).toContain('- cmd1'); expect(lastFrame()).toContain('- cmd2'); expect(lastFrame()).toContain('• MCP Servers (1):'); expect(lastFrame()).toContain('- mcp1'); expect(lastFrame()).toContain('• Hooks (1):'); expect(lastFrame()).toContain('- hook1'); expect(lastFrame()).toContain('• Skills (1):'); expect(lastFrame()).toContain('- skill1'); expect(lastFrame()).toContain('• Agents (1):'); expect(lastFrame()).toContain('- agent1'); expect(lastFrame()).toContain('• Setting overrides (2):'); expect(lastFrame()).toContain('- general'); expect(lastFrame()).toContain('- ui'); unmount(); }); it('should display security warnings when provided', async () => { const discoveryResults = { commands: [], mcps: [], hooks: [], skills: [], agents: [], settings: [], discoveryErrors: [], securityWarnings: ['Dangerous setting detected!'], }; const { lastFrame, unmount } = await renderWithProviders( , ); expect(lastFrame()).toContain('Security Warnings:'); expect(lastFrame()).toContain('Dangerous setting detected!'); unmount(); }); it('should display discovery errors when provided', async () => { const discoveryResults = { commands: [], mcps: [], hooks: [], skills: [], agents: [], settings: [], discoveryErrors: ['Failed to load custom commands'], securityWarnings: [], }; const { lastFrame, unmount } = await renderWithProviders( , ); expect(lastFrame()).toContain('Discovery Errors:'); expect(lastFrame()).toContain('Failed to load custom commands'); unmount(); }); it('should use scrolling instead of truncation when alternate buffer is enabled and expanded', async () => { const discoveryResults = { commands: Array.from({ length: 20 }, (_, i) => `cmd${i}`), mcps: [], hooks: [], skills: [], agents: [], settings: [], discoveryErrors: [], securityWarnings: [], }; const { lastFrame, unmount } = await renderWithProviders( , { width: 80, config: makeFakeConfig({ useAlternateBuffer: true }), settings: createMockSettings({ ui: { useAlternateBuffer: true } }), uiState: { constrainHeight: false, terminalHeight: 15 }, }, ); // In alternate buffer + expanded, the title should be visible (StickyHeader) expect(lastFrame()).toContain('Do you trust the files in this folder?'); // And it should NOT use MaxSizedBox truncation expect(lastFrame()).not.toContain('hidden'); unmount(); }); it('should strip ANSI codes from discovery results', async () => { const ansiRed = '\u001b[31m'; const ansiReset = '\u001b[39m'; const discoveryResults = { commands: [`${ansiRed}cmd-with-ansi${ansiReset}`], mcps: [`${ansiRed}mcp-with-ansi${ansiReset}`], hooks: [`${ansiRed}hook-with-ansi${ansiReset}`], skills: [`${ansiRed}skill-with-ansi${ansiReset}`], agents: [], settings: [`${ansiRed}setting-with-ansi${ansiReset}`], discoveryErrors: [`${ansiRed}error-with-ansi${ansiReset}`], securityWarnings: [`${ansiRed}warning-with-ansi${ansiReset}`], }; const { lastFrame, unmount } = await renderWithProviders( , { width: 100, uiState: { terminalHeight: 40 } }, ); const output = lastFrame(); expect(output).toContain('cmd-with-ansi'); expect(output).toContain('mcp-with-ansi'); expect(output).toContain('hook-with-ansi'); expect(output).toContain('skill-with-ansi'); expect(output).toContain('setting-with-ansi'); expect(output).toContain('error-with-ansi'); expect(output).toContain('warning-with-ansi'); unmount(); }); }); });