diff options
Diffstat (limited to 'packages/cli/src/ui/App.test.tsx')
| -rw-r--r-- | packages/cli/src/ui/App.test.tsx | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/packages/cli/src/ui/App.test.tsx b/packages/cli/src/ui/App.test.tsx index 4c98827e..ceab46b1 100644 --- a/packages/cli/src/ui/App.test.tsx +++ b/packages/cli/src/ui/App.test.tsx @@ -138,7 +138,9 @@ vi.mock('@google/gemini-cli-core', async (importOriginal) => { getShowMemoryUsage: vi.fn(() => opts.showMemoryUsage ?? false), getAccessibility: vi.fn(() => opts.accessibility ?? {}), getProjectRoot: vi.fn(() => opts.targetDir), - getGeminiClient: vi.fn(() => ({})), + getGeminiClient: vi.fn(() => ({ + getUserTier: vi.fn(), + })), getCheckpointingEnabled: vi.fn(() => opts.checkpointing ?? true), getAllGeminiMdFilenames: vi.fn(() => ['GEMINI.md']), setFlashFallbackHandler: vi.fn(), @@ -639,6 +641,7 @@ describe('App UI', () => { mockConfig.getGeminiClient.mockReturnValue({ isInitialized: vi.fn(() => true), + getUserTier: vi.fn(), } as unknown as GeminiClient); const { unmount, rerender } = render( |