@@ -1363,43 +1363,19 @@ async function withController<ReturnValue>(
13631363 return await fn ( {
13641364 controller,
13651365 mockKeyringGetState : ( state : KeyringControllerState ) => {
1366- controllerMessenger . unregisterActionHandler (
1367- 'KeyringController:getState' ,
1368- ) ;
1369- controllerMessenger . registerActionHandler (
1370- 'KeyringController:getState' ,
1371- mockKeyringState . mockReturnValue ( state ) ,
1372- ) ;
1366+ mockKeyringState . mockReturnValue ( state ) ;
13731367 } ,
13741368 mockPreferencesGetState : ( state : PreferencesState ) => {
1375- controllerMessenger . unregisterActionHandler (
1376- 'PreferencesController:getState' ,
1377- ) ;
1378- controllerMessenger . registerActionHandler (
1379- 'PreferencesController:getState' ,
1380- mockPreferencesState . mockReturnValue ( state ) ,
1381- ) ;
1369+ mockPreferencesState . mockReturnValue ( state ) ;
13821370 } ,
13831371 mockTokenListGetState : ( state : TokenListState ) => {
1384- controllerMessenger . unregisterActionHandler (
1385- 'TokenListController:getState' ,
1386- ) ;
1387- controllerMessenger . registerActionHandler (
1388- 'TokenListController:getState' ,
1389- mockTokenListState . mockReturnValue ( state ) ,
1390- ) ;
1372+ mockTokenListState . mockReturnValue ( state ) ;
13911373 } ,
13921374 mockGetNetworkConfigurationByNetworkClientId : (
13931375 handler : ( networkClientId : string ) => NetworkConfiguration ,
13941376 ) => {
1395- controllerMessenger . unregisterActionHandler (
1396- 'NetworkController:getNetworkConfigurationByNetworkClientId' ,
1397- ) ;
1398- controllerMessenger . registerActionHandler (
1399- 'NetworkController:getNetworkConfigurationByNetworkClientId' ,
1400- mockGetNetworkConfigurationByNetworkClientId . mockImplementation (
1401- handler ,
1402- ) ,
1377+ mockGetNetworkConfigurationByNetworkClientId . mockImplementation (
1378+ handler ,
14031379 ) ;
14041380 } ,
14051381 triggerKeyringUnlock : ( ) => {
0 commit comments