@@ -21,7 +21,6 @@ import {
2121 PREVIEW_GEMINI_FLASH_MODEL ,
2222 PREVIEW_GEMINI_3_1_FLASH_LITE_MODEL ,
2323 AuthType ,
24- UserTierId ,
2524} from '@google/gemini-cli-core' ;
2625import type { Config , ModelSlashCommandEvent } from '@google/gemini-cli-core' ;
2726
@@ -56,7 +55,6 @@ describe('<ModelDialog />', () => {
5655 const mockGetGemini31FlashLiteLaunchedSync = vi . fn ( ) ;
5756 const mockGetProModelNoAccess = vi . fn ( ) ;
5857 const mockGetProModelNoAccessSync = vi . fn ( ) ;
59- const mockGetUserTier = vi . fn ( ) ;
6058
6159 interface MockConfig extends Partial < Config > {
6260 setModel : ( model : string , isTemporary ?: boolean ) => void ;
@@ -67,7 +65,6 @@ describe('<ModelDialog />', () => {
6765 getGemini31FlashLiteLaunchedSync : ( ) => boolean ;
6866 getProModelNoAccess : ( ) => Promise < boolean > ;
6967 getProModelNoAccessSync : ( ) => boolean ;
70- getUserTier : ( ) => UserTierId | undefined ;
7168 }
7269
7370 const mockConfig : MockConfig = {
@@ -79,7 +76,6 @@ describe('<ModelDialog />', () => {
7976 getGemini31FlashLiteLaunchedSync : mockGetGemini31FlashLiteLaunchedSync ,
8077 getProModelNoAccess : mockGetProModelNoAccess ,
8178 getProModelNoAccessSync : mockGetProModelNoAccessSync ,
82- getUserTier : mockGetUserTier ,
8379 } ;
8480
8581 beforeEach ( ( ) => {
@@ -90,7 +86,6 @@ describe('<ModelDialog />', () => {
9086 mockGetGemini31FlashLiteLaunchedSync . mockReturnValue ( false ) ;
9187 mockGetProModelNoAccess . mockResolvedValue ( false ) ;
9288 mockGetProModelNoAccessSync . mockReturnValue ( false ) ;
93- mockGetUserTier . mockReturnValue ( UserTierId . STANDARD ) ;
9489
9590 // Default implementation for getDisplayString
9691 mockGetDisplayString . mockImplementation ( ( val : string ) => {
@@ -136,7 +131,6 @@ describe('<ModelDialog />', () => {
136131 mockGetProModelNoAccess . mockResolvedValue ( true ) ;
137132 mockGetHasAccessToPreviewModel . mockReturnValue ( true ) ;
138133 mockGetGemini31FlashLiteLaunchedSync . mockReturnValue ( true ) ;
139- mockGetUserTier . mockReturnValue ( UserTierId . FREE ) ;
140134 mockGetDisplayString . mockImplementation ( ( val : string ) => val ) ;
141135
142136 const { lastFrame, unmount } = await renderComponent ( ) ;
@@ -442,34 +436,11 @@ describe('<ModelDialog />', () => {
442436 unmount ( ) ;
443437 } ) ;
444438
445- it ( 'hides Flash Lite Preview model for users with pro access' , async ( ) => {
446- mockGetProModelNoAccessSync . mockReturnValue ( false ) ;
447- mockGetProModelNoAccess . mockResolvedValue ( false ) ;
448- mockGetHasAccessToPreviewModel . mockReturnValue ( true ) ;
449- const { lastFrame, stdin, waitUntilReady, unmount } =
450- await renderComponent ( ) ;
451-
452- // Go to manual view
453- await act ( async ( ) => {
454- stdin . write ( '\u001B[B' ) ; // Manual
455- } ) ;
456- await waitUntilReady ( ) ;
457- await act ( async ( ) => {
458- stdin . write ( '\r' ) ;
459- } ) ;
460- await waitUntilReady ( ) ;
461-
462- const output = lastFrame ( ) ;
463- expect ( output ) . not . toContain ( PREVIEW_GEMINI_3_1_FLASH_LITE_MODEL ) ;
464- unmount ( ) ;
465- } ) ;
466-
467- it ( 'shows Flash Lite Preview model for free tier users' , async ( ) => {
439+ it ( 'shows Flash Lite Preview model regardless of tier when flag is enabled' , async ( ) => {
468440 mockGetProModelNoAccessSync . mockReturnValue ( false ) ;
469441 mockGetProModelNoAccess . mockResolvedValue ( false ) ;
470442 mockGetHasAccessToPreviewModel . mockReturnValue ( true ) ;
471443 mockGetGemini31FlashLiteLaunchedSync . mockReturnValue ( true ) ;
472- mockGetUserTier . mockReturnValue ( UserTierId . FREE ) ;
473444 const { lastFrame, stdin, waitUntilReady, unmount } =
474445 await renderComponent ( ) ;
475446
0 commit comments