Improve code coverage for cli package (#13724)

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

@@ -97,27 +97,95 @@ const mockConfig = {
 } as unknown as Config;
 
 describe('AlternateBufferQuittingDisplay', () => {
+  const baseUIState = {
+    terminalWidth: 80,
+    mainAreaWidth: 80,
+    slashCommands: [],
+    activePtyId: undefined,
+    embeddedShellFocused: false,
+    renderMarkdown: false,
+    bannerData: {
+      defaultText: '',
+      warningText: '',
+    },
+  };
+
   it('renders with active and pending tool messages', () => {
     const { lastFrame } = renderWithProviders(
       <AlternateBufferQuittingDisplay />,
       {
         uiState: {
+          ...baseUIState,
           history: mockHistory,
           pendingHistoryItems: mockPendingHistoryItems,
-          terminalWidth: 80,
-          mainAreaWidth: 80,
-          slashCommands: [],
-          activePtyId: undefined,
-          embeddedShellFocused: false,
-          renderMarkdown: false,
-          bannerData: {
-            defaultText: '',
-            warningText: '',
-          },
         },
         config: mockConfig,
       },
     );
-    expect(lastFrame()).toMatchSnapshot();
+    expect(lastFrame()).toMatchSnapshot('with_history_and_pending');
+  });
+
+  it('renders with empty history and no pending items', () => {
+    const { lastFrame } = renderWithProviders(
+      <AlternateBufferQuittingDisplay />,
+      {
+        uiState: {
+          ...baseUIState,
+          history: [],
+          pendingHistoryItems: [],
+        },
+        config: mockConfig,
+      },
+    );
+    expect(lastFrame()).toMatchSnapshot('empty');
+  });
+
+  it('renders with history but no pending items', () => {
+    const { lastFrame } = renderWithProviders(
+      <AlternateBufferQuittingDisplay />,
+      {
+        uiState: {
+          ...baseUIState,
+          history: mockHistory,
+          pendingHistoryItems: [],
+        },
+        config: mockConfig,
+      },
+    );
+    expect(lastFrame()).toMatchSnapshot('with_history_no_pending');
+  });
+
+  it('renders with pending items but no history', () => {
+    const { lastFrame } = renderWithProviders(
+      <AlternateBufferQuittingDisplay />,
+      {
+        uiState: {
+          ...baseUIState,
+          history: [],
+          pendingHistoryItems: mockPendingHistoryItems,
+        },
+        config: mockConfig,
+      },
+    );
+    expect(lastFrame()).toMatchSnapshot('with_pending_no_history');
+  });
+
+  it('renders with user and gemini messages', () => {
+    const history: HistoryItem[] = [
+      { id: 1, type: 'user', text: 'Hello Gemini' },
+      { id: 2, type: 'gemini', text: 'Hello User!' },
+    ];
+    const { lastFrame } = renderWithProviders(
+      <AlternateBufferQuittingDisplay />,
+      {
+        uiState: {
+          ...baseUIState,
+          history,
+          pendingHistoryItems: [],
+        },
+        config: mockConfig,
+      },
+    );
+    expect(lastFrame()).toMatchSnapshot('with_user_gemini_messages');
   });
 });