66} from '@metamask/keyring-api' ;
77import { InternalAccount } from '@metamask/keyring-internal-api' ;
88import { SnapId } from '@metamask/snaps-sdk' ;
9+ import { RemoteFeatureFlagControllerState } from '@metamask/remote-feature-flag-controller' ;
910import { getRootMessenger } from '../messenger' ;
1011import { SNAP_MANAGE_ACCOUNTS_CONFIRMATION_TYPES } from '../../../../shared/constants/app' ;
1112import {
@@ -43,6 +44,7 @@ const mockLocale = 'en';
4344const mockPreferencesControllerGetState = jest . fn ( ) ;
4445const mockSnapControllerGet = jest . fn ( ) ;
4546const mockSnapControllerHandleRequest = jest . fn ( ) ;
47+ const mockRemoteFeatureFlagsGetStateRequest = jest . fn ( ) ;
4648
4749const mockFlowId = '123' ;
4850const address = '0x2a4d4b667D5f12C3F9Bf8F14a7B9f8D8d9b8c8fA' ;
@@ -114,6 +116,7 @@ const createControllerMessenger = ({
114116 'PreferencesController:getState' ,
115117 'SnapController:get' ,
116118 'SnapController:handleRequest' ,
119+ 'RemoteFeatureFlagController:getState' ,
117120 ] ,
118121 } ) ;
119122
@@ -171,6 +174,9 @@ const createControllerMessenger = ({
171174 case 'SnapController:handleRequest' :
172175 return mockSnapControllerHandleRequest ( params ) ;
173176
177+ case 'RemoteFeatureFlagController:getState' :
178+ return mockRemoteFeatureFlagsGetStateRequest ( params ) ;
179+
174180 default :
175181 throw new Error (
176182 `MOCK_FAIL - unsupported messenger call: ${ actionType } ` ,
@@ -191,6 +197,11 @@ const createSnapKeyringBuilder = ({
191197 jest . mocked ( isSnapPreinstalled ) . mockReturnValue ( snapPreinstalled ) ;
192198 jest . mocked ( getSnapName ) . mockReturnValue ( snapName ) ;
193199
200+ // Needed now to know if state 2 is enabled or not.
201+ mockRemoteFeatureFlagsGetStateRequest . mockReturnValue ( {
202+ remoteFeatureFlags : { } ,
203+ } as RemoteFeatureFlagControllerState ) ;
204+
194205 return snapKeyringBuilder ( createControllerMessenger ( ) , {
195206 persistKeyringHelper : mockPersistKeyringHelper ,
196207 removeAccountHelper : mockRemoveAccountHelper ,
0 commit comments