feat: Add UI for /stats slash command (#883)

1 files changed, 71 insertions, 0 deletions

diff --git a/packages/cli/src/ui/components/StatsDisplay.test.tsx b/packages/cli/src/ui/components/StatsDisplay.test.tsx
new file mode 100644
index 00000000..c7b574a5
--- /dev/null
+++ b/packages/cli/src/ui/components/StatsDisplay.test.tsx
@@ -0,0 +1,71 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { render } from 'ink-testing-library';
+import { describe, it, expect } from 'vitest';
+import { StatsDisplay } from './StatsDisplay.js';
+import { type CumulativeStats } from '../contexts/SessionContext.js';
+
+describe('<StatsDisplay />', () => {
+  const mockStats: CumulativeStats = {
+    turnCount: 10,
+    promptTokenCount: 1000,
+    candidatesTokenCount: 2000,
+    totalTokenCount: 3500,
+    cachedContentTokenCount: 500,
+    toolUsePromptTokenCount: 200,
+    thoughtsTokenCount: 300,
+    apiTimeMs: 50234,
+  };
+
+  const mockLastTurnStats: CumulativeStats = {
+    turnCount: 1,
+    promptTokenCount: 100,
+    candidatesTokenCount: 200,
+    totalTokenCount: 350,
+    cachedContentTokenCount: 50,
+    toolUsePromptTokenCount: 20,
+    thoughtsTokenCount: 30,
+    apiTimeMs: 1234,
+  };
+
+  const mockDuration = '1h 23m 45s';
+
+  it('renders correctly with given stats and duration', () => {
+    const { lastFrame } = render(
+      <StatsDisplay
+        stats={mockStats}
+        lastTurnStats={mockLastTurnStats}
+        duration={mockDuration}
+      />,
+    );
+
+    expect(lastFrame()).toMatchSnapshot();
+  });
+
+  it('renders zero state correctly', () => {
+    const zeroStats: CumulativeStats = {
+      turnCount: 0,
+      promptTokenCount: 0,
+      candidatesTokenCount: 0,
+      totalTokenCount: 0,
+      cachedContentTokenCount: 0,
+      toolUsePromptTokenCount: 0,
+      thoughtsTokenCount: 0,
+      apiTimeMs: 0,
+    };
+
+    const { lastFrame } = render(
+      <StatsDisplay
+        stats={zeroStats}
+        lastTurnStats={zeroStats}
+        duration="0s"
+      />,
+    );
+
+    expect(lastFrame()).toMatchSnapshot();
+  });
+});