diff options
Diffstat (limited to 'packages/cli/src/ui/components')
4 files changed, 20 insertions, 0 deletions
diff --git a/packages/cli/src/ui/components/ModelStatsDisplay.test.tsx b/packages/cli/src/ui/components/ModelStatsDisplay.test.tsx index 6c41b775..57382d91 100644 --- a/packages/cli/src/ui/components/ModelStatsDisplay.test.tsx +++ b/packages/cli/src/ui/components/ModelStatsDisplay.test.tsx @@ -27,7 +27,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => { sessionStartTime: new Date(), metrics, lastPromptTokenCount: 0, + promptCount: 5, }, + + getPromptCount: () => 5, + startNewPrompt: vi.fn(), }); return render(<ModelStatsDisplay />); diff --git a/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx b/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx index f3c0764e..38400caf 100644 --- a/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx +++ b/packages/cli/src/ui/components/SessionSummaryDisplay.test.tsx @@ -26,7 +26,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => { sessionStartTime: new Date(), metrics, lastPromptTokenCount: 0, + promptCount: 5, }, + + getPromptCount: () => 5, + startNewPrompt: vi.fn(), }); return render(<SessionSummaryDisplay duration="1h 23m 45s" />); diff --git a/packages/cli/src/ui/components/StatsDisplay.test.tsx b/packages/cli/src/ui/components/StatsDisplay.test.tsx index a62815d9..a0ed3858 100644 --- a/packages/cli/src/ui/components/StatsDisplay.test.tsx +++ b/packages/cli/src/ui/components/StatsDisplay.test.tsx @@ -27,7 +27,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => { sessionStartTime: new Date(), metrics, lastPromptTokenCount: 0, + promptCount: 5, }, + + getPromptCount: () => 5, + startNewPrompt: vi.fn(), }); return render(<StatsDisplay duration="1s" />); @@ -288,7 +292,11 @@ describe('<StatsDisplay />', () => { sessionStartTime: new Date(), metrics: zeroMetrics, lastPromptTokenCount: 0, + promptCount: 5, }, + + getPromptCount: () => 5, + startNewPrompt: vi.fn(), }); const { lastFrame } = render( diff --git a/packages/cli/src/ui/components/ToolStatsDisplay.test.tsx b/packages/cli/src/ui/components/ToolStatsDisplay.test.tsx index 54902788..e48fcc83 100644 --- a/packages/cli/src/ui/components/ToolStatsDisplay.test.tsx +++ b/packages/cli/src/ui/components/ToolStatsDisplay.test.tsx @@ -27,7 +27,11 @@ const renderWithMockedStats = (metrics: SessionMetrics) => { sessionStartTime: new Date(), metrics, lastPromptTokenCount: 0, + promptCount: 5, }, + + getPromptCount: () => 5, + startNewPrompt: vi.fn(), }); return render(<ToolStatsDisplay />); |