From 16485afa96de04f1a8e4f7d29b3a13bed8cd6df2 Mon Sep 17 00:00:00 2001
From: Keith Guerin <keithguerin@gmail.com>
Date: Fri, 20 Mar 2026 13:17:18 -0700
Subject: [PATCH] feat(ui): finalize compression UX with threshold hint and
 large prompt override

---
 packages/cli/src/ui/hooks/useGeminiStream.test.tsx | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
index dc0d7f537d..98228b9a2f 100644
--- a/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
+++ b/packages/cli/src/ui/hooks/useGeminiStream.test.tsx
@@ -52,6 +52,7 @@ import {
   MCPDiscoveryState,
   GeminiCliOperation,
   getPlanModeExitMessage,
+  CompressionStatus,
 } from '@google/gemini-cli-core';
 import type { Part, PartListUnion } from '@google/genai';
 import type { UseHistoryManagerReturn } from './useHistoryManager.js';
@@ -2598,7 +2599,7 @@ describe('useGeminiStream', () => {
             value: {
               originalTokenCount: 1000,
               newTokenCount: 500,
-              compressionStatus: 'compressed',
+              compressionStatus: CompressionStatus.COMPRESSED,
             },
           };
           yield {
@@ -2635,7 +2636,7 @@ describe('useGeminiStream', () => {
               isPending: false,
               beforePercentage: 10,
               afterPercentage: 5,
-              compressionStatus: 'compressed',
+              compressionStatus: CompressionStatus.COMPRESSED,
               isManual: false,
               thresholdPercentage: 20,
             },
@@ -2659,7 +2660,7 @@ describe('useGeminiStream', () => {
             value: {
               originalTokenCount: 1000,
               newTokenCount: 500,
-              compressionStatus: 'compressed',
+              compressionStatus: CompressionStatus.COMPRESSED,
             },
           };
           yield {
@@ -2712,7 +2713,7 @@ describe('useGeminiStream', () => {
             value: {
               originalTokenCount: 1000,
               newTokenCount: 500,
-              compressionStatus: 'compressed',
+              compressionStatus: CompressionStatus.COMPRESSED,
               requestTokenCount: 2500, // 25% > 20%
             },
           };
@@ -2749,7 +2750,7 @@ describe('useGeminiStream', () => {
             compression: expect.objectContaining({
               beforePercentage: 10,
               afterPercentage: 5,
-              compressionStatus: 'compressed',
+              compressionStatus: CompressionStatus.COMPRESSED,
               isManual: false,
               thresholdPercentage: 20,
             }),