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
+
+ )}
);
};