diff --git a/packages/cli/src/ui/components/UserIdentity.test.tsx b/packages/cli/src/ui/components/UserIdentity.test.tsx index 8e63415f5c..5391944d26 100644 --- a/packages/cli/src/ui/components/UserIdentity.test.tsx +++ b/packages/cli/src/ui/components/UserIdentity.test.tsx @@ -47,6 +47,7 @@ describe('', () => { const output = lastFrame(); expect(output).toContain('test@example.com'); expect(output).toContain('/auth'); + expect(output).not.toContain('/upgrade'); unmount(); }); @@ -74,6 +75,7 @@ describe('', () => { const output = lastFrame(); expect(output).toContain('Logged in with Google'); expect(output).toContain('/auth'); + expect(output).not.toContain('/upgrade'); unmount(); }); @@ -130,6 +132,26 @@ describe('', () => { const output = lastFrame(); expect(output).toContain(`Authenticated with ${AuthType.USE_GEMINI}`); expect(output).toContain('/auth'); + expect(output).not.toContain('/upgrade'); + unmount(); + }); + + it('should render specific tier name when provided', async () => { + const mockConfig = makeFakeConfig(); + vi.spyOn(mockConfig, 'getContentGeneratorConfig').mockReturnValue({ + authType: AuthType.LOGIN_WITH_GOOGLE, + model: 'gemini-pro', + } as unknown as ContentGeneratorConfig); + vi.spyOn(mockConfig, 'getUserTierName').mockReturnValue('Enterprise Tier'); + + const { lastFrame, waitUntilReady, unmount } = renderWithProviders( + , + ); + await waitUntilReady(); + + const output = lastFrame(); + expect(output).toContain('Enterprise Tier'); + expect(output).toContain('/upgrade'); unmount(); }); }); diff --git a/packages/cli/src/ui/components/UserIdentity.tsx b/packages/cli/src/ui/components/UserIdentity.tsx index 08c82573d9..98c62ec68f 100644 --- a/packages/cli/src/ui/components/UserIdentity.tsx +++ b/packages/cli/src/ui/components/UserIdentity.tsx @@ -53,12 +53,14 @@ export const UserIdentity: React.FC = ({ config }) => { {/* Tier Name /upgrade */} - - - {tierName ?? 'Gemini Code Assist for individuals'} - - /upgrade - + {tierName && ( + + + {tierName} + + /upgrade + + )} ); };