From cc61d0a7f9e65ea65f266718c2af6f2329e1d18b Mon Sep 17 00:00:00 2001 From: mkorwel Date: Mon, 20 Apr 2026 22:13:32 +0000 Subject: [PATCH] test: optimize DenseToolMessage tests with fake timers --- .../components/messages/DenseToolMessage.test.tsx | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/packages/cli/src/ui/components/messages/DenseToolMessage.test.tsx b/packages/cli/src/ui/components/messages/DenseToolMessage.test.tsx index 586ce89ab2..6637e96f99 100644 --- a/packages/cli/src/ui/components/messages/DenseToolMessage.test.tsx +++ b/packages/cli/src/ui/components/messages/DenseToolMessage.test.tsx @@ -5,7 +5,7 @@ */ import { describe, it, expect } from 'vitest'; -import { renderWithProviders } from '../../../test-utils/render.js'; +import { renderWithProviders as originalRenderWithProviders } from '../../../test-utils/render.js'; import { createMockSettings } from '../../../test-utils/settings.js'; import { waitFor } from '../../../test-utils/async.js'; import { DenseToolMessage } from './DenseToolMessage.js'; @@ -23,6 +23,11 @@ import type { ToolResultDisplay, } from '../../types.js'; +const renderWithProviders = async ( + component: React.ReactElement, + options?: Parameters[1], +) => originalRenderWithProviders(component, { height: 40, ...options }); + describe('DenseToolMessage', () => { const defaultProps = { callId: 'call-1', @@ -34,6 +39,14 @@ describe('DenseToolMessage', () => { terminalWidth: 80, }; + beforeEach(() => { + vi.useFakeTimers(); + }); + + afterEach(() => { + vi.useRealTimers(); + }); + it('explicitly renders the filename in the header for FileDiff results', async () => { const fileDiff: FileDiff = { fileName: 'test-file.ts',