remove hardcoded tiername when missing tier (#21022)

packages/cli/src/ui/components/UserIdentity.test.tsx

@@ -47,6 +47,7 @@ describe('<UserIdentity />', () => {
     const output = lastFrame();
     expect(output).toContain('test@example.com');
     expect(output).toContain('/auth');
+    expect(output).not.toContain('/upgrade');
     unmount();
   });
 
@@ -74,6 +75,7 @@ describe('<UserIdentity />', () => {
     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('<UserIdentity />', () => {
     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(
+      <UserIdentity config={mockConfig} />,
+    );
+    await waitUntilReady();
+
+    const output = lastFrame();
+    expect(output).toContain('Enterprise Tier');
+    expect(output).toContain('/upgrade');
     unmount();
   });
 });

packages/cli/src/ui/components/UserIdentity.tsx

@@ -53,12 +53,14 @@ export const UserIdentity: React.FC<UserIdentityProps> = ({ config }) => {
       </Box>
 
       {/* Tier Name /upgrade */}
+      {tierName && (
         <Box>
           <Text color={theme.text.primary} wrap="truncate-end">
-          {tierName ?? 'Gemini Code Assist for individuals'}
+            {tierName}
           </Text>
           <Text color={theme.text.secondary}> /upgrade</Text>
         </Box>
+      )}
     </Box>
   );
 };