File size: 2,314 Bytes
9332ccf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 | /**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { renderWithProviders } from '../../test-utils/render.js';
import { ContextUsageDisplay } from './ContextUsageDisplay.js';
import { describe, it, expect, vi } from 'vitest';
vi.mock('@google/gemini-cli-core', async (importOriginal) => {
const actual =
await importOriginal<typeof import('@google/gemini-cli-core')>();
return {
...actual,
tokenLimit: () => 10000,
};
});
describe('ContextUsageDisplay', () => {
it('renders correct percentage used', async () => {
const { lastFrame, unmount } = await renderWithProviders(
<ContextUsageDisplay
promptTokenCount={5000}
model="gemini-pro"
terminalWidth={120}
/>,
);
const output = lastFrame();
expect(output).toContain('50% used');
unmount();
});
it('renders correctly when usage is 0%', async () => {
const { lastFrame, unmount } = await renderWithProviders(
<ContextUsageDisplay
promptTokenCount={0}
model="gemini-pro"
terminalWidth={120}
/>,
);
const output = lastFrame();
expect(output).toContain('0% used');
unmount();
});
it('renders abbreviated label when terminal width is small', async () => {
const { lastFrame, unmount } = await renderWithProviders(
<ContextUsageDisplay
promptTokenCount={2000}
model="gemini-pro"
terminalWidth={80}
/>,
{ width: 80 },
);
const output = lastFrame();
expect(output).toContain('20%');
expect(output).not.toContain('context used');
unmount();
});
it('renders 80% correctly', async () => {
const { lastFrame, unmount } = await renderWithProviders(
<ContextUsageDisplay
promptTokenCount={8000}
model="gemini-pro"
terminalWidth={120}
/>,
);
const output = lastFrame();
expect(output).toContain('80% used');
unmount();
});
it('renders 100% when full', async () => {
const { lastFrame, unmount } = await renderWithProviders(
<ContextUsageDisplay
promptTokenCount={10000}
model="gemini-pro"
terminalWidth={120}
/>,
);
const output = lastFrame();
expect(output).toContain('100% used');
unmount();
});
});
|