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

test: fix ContextUsageDisplay tests to correctly await waitUntilReady

Browse Source
This commit is contained in:
Keith Guerin
2026-03-01 00:31:02 -08:00
parent dc2cc57ba8
commit 542e77ab58
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
packages/cli/src/ui/components/ContextUsageDisplay.test.tsx
View File

@@ -27,39 +27,42 @@ vi.mock('../../config/settings.js', () => ({
}));
describe('ContextUsageDisplay', () => {
it('renders correct percentage left', () => {
const { lastFrame } = render(
it('renders correct percentage left', async () => {
const { lastFrame, waitUntilReady } = render(
<ContextUsageDisplay
promptTokenCount={5000}
model="gemini-pro"
terminalWidth={120}
/>,
);
await waitUntilReady();
const output = lastFrame();
expect(output).toContain('50% left');
});
it('renders short label when terminal width is small', () => {
const { lastFrame } = render(
it('renders short label when terminal width is small', async () => {
const { lastFrame, waitUntilReady } = render(
<ContextUsageDisplay
promptTokenCount={2000}
model="gemini-pro"
terminalWidth={80}
/>,
);
await waitUntilReady();
const output = lastFrame();
expect(output).toContain('80%');
expect(output).not.toContain('context left');
});
it('renders 0% when full', () => {
const { lastFrame } = render(
it('renders 0% when full', async () => {
const { lastFrame, waitUntilReady } = render(
<ContextUsageDisplay
promptTokenCount={10000}
model="gemini-pro"
terminalWidth={120}
/>,
);
await waitUntilReady();
const output = lastFrame();
expect(output).toContain('0% left');
});