MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-03-16 00:51:25 -07:00
Code Issues Packages Projects Releases Wiki Activity

Feature/quota visibility 16795 (#18203)

Browse Source
This commit is contained in:
Spencer
2026-02-09 21:53:10 -05:00
committed by GitHub
parent 0a3ecf3a75
commit 6dae3a5402
43 changed files with 1315 additions and 317 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
packages/cli/src/ui/components/ModelStatsDisplay.test.tsx
View File

@@ -1,6 +1,6 @@
/**
* @license
* Copyright 2025 Google LLC
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
@@ -33,7 +33,11 @@ vi.mock('../contexts/SettingsContext.js', async (importOriginal) => {
const useSessionStatsMock = vi.mocked(SessionContext.useSessionStats);
const useSettingsMock = vi.mocked(SettingsContext.useSettings);
const renderWithMockedStats = (metrics: SessionMetrics, width?: number) => {
const renderWithMockedStats = (
metrics: SessionMetrics,
width?: number,
currentModel: string = 'gemini-2.5-pro',
) => {
useSessionStatsMock.mockReturnValue({
stats: {
sessionId: 'test-session',
@@ -55,7 +59,7 @@ const renderWithMockedStats = (metrics: SessionMetrics, width?: number) => {
},
} as unknown as LoadedSettings);
return render(<ModelStatsDisplay />, width);
return render(<ModelStatsDisplay currentModel={currentModel} />, width);
};
describe('<ModelStatsDisplay />', () => {
@@ -380,6 +384,7 @@ describe('<ModelStatsDisplay />', () => {
},
},
80,
'auto-gemini-3',
);
const output = lastFrame();