@@ -53,6 +53,7 @@ describe('<ModelDialog />', () => {
5353 const mockOnClose = vi . fn ( ) ;
5454 const mockGetHasAccessToPreviewModel = vi . fn ( ) ;
5555 const mockGetGemini31LaunchedSync = vi . fn ( ) ;
56+ const mockGetGemini31FlashLiteLaunchedSync = vi . fn ( ) ;
5657 const mockGetProModelNoAccess = vi . fn ( ) ;
5758 const mockGetProModelNoAccessSync = vi . fn ( ) ;
5859 const mockGetUserTier = vi . fn ( ) ;
@@ -63,6 +64,7 @@ describe('<ModelDialog />', () => {
6364 getHasAccessToPreviewModel : ( ) => boolean ;
6465 getIdeMode : ( ) => boolean ;
6566 getGemini31LaunchedSync : ( ) => boolean ;
67+ getGemini31FlashLiteLaunchedSync : ( ) => boolean ;
6668 getProModelNoAccess : ( ) => Promise < boolean > ;
6769 getProModelNoAccessSync : ( ) => boolean ;
6870 getUserTier : ( ) => UserTierId | undefined ;
@@ -74,6 +76,7 @@ describe('<ModelDialog />', () => {
7476 getHasAccessToPreviewModel : mockGetHasAccessToPreviewModel ,
7577 getIdeMode : ( ) => false ,
7678 getGemini31LaunchedSync : mockGetGemini31LaunchedSync ,
79+ getGemini31FlashLiteLaunchedSync : mockGetGemini31FlashLiteLaunchedSync ,
7780 getProModelNoAccess : mockGetProModelNoAccess ,
7881 getProModelNoAccessSync : mockGetProModelNoAccessSync ,
7982 getUserTier : mockGetUserTier ,
@@ -84,6 +87,7 @@ describe('<ModelDialog />', () => {
8487 mockGetModel . mockReturnValue ( DEFAULT_GEMINI_MODEL_AUTO ) ;
8588 mockGetHasAccessToPreviewModel . mockReturnValue ( false ) ;
8689 mockGetGemini31LaunchedSync . mockReturnValue ( false ) ;
90+ mockGetGemini31FlashLiteLaunchedSync . mockReturnValue ( false ) ;
8791 mockGetProModelNoAccess . mockResolvedValue ( false ) ;
8892 mockGetProModelNoAccessSync . mockReturnValue ( false ) ;
8993 mockGetUserTier . mockReturnValue ( UserTierId . STANDARD ) ;
@@ -131,6 +135,7 @@ describe('<ModelDialog />', () => {
131135 mockGetProModelNoAccessSync . mockReturnValue ( true ) ;
132136 mockGetProModelNoAccess . mockResolvedValue ( true ) ;
133137 mockGetHasAccessToPreviewModel . mockReturnValue ( true ) ;
138+ mockGetGemini31FlashLiteLaunchedSync . mockReturnValue ( true ) ;
134139 mockGetUserTier . mockReturnValue ( UserTierId . FREE ) ;
135140 mockGetDisplayString . mockImplementation ( ( val : string ) => val ) ;
136141
@@ -463,6 +468,7 @@ describe('<ModelDialog />', () => {
463468 mockGetProModelNoAccessSync . mockReturnValue ( false ) ;
464469 mockGetProModelNoAccess . mockResolvedValue ( false ) ;
465470 mockGetHasAccessToPreviewModel . mockReturnValue ( true ) ;
471+ mockGetGemini31FlashLiteLaunchedSync . mockReturnValue ( true ) ;
466472 mockGetUserTier . mockReturnValue ( UserTierId . FREE ) ;
467473 const { lastFrame, stdin, waitUntilReady, unmount } =
468474 await renderComponent ( ) ;
0 commit comments