test: add actions, events, callbacks to test messenger

MetaMask · Jan 26, 2024 · a590dcc · a590dcc
1 parent 6ba98de
commit a590dcc
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/packages/assets-controllers/src/TokenDetectionController.test.ts b/packages/assets-controllers/src/TokenDetectionController.test.ts
@@ -133,10 +133,14 @@ function buildTokenDetectionControllerMessenger(
   return controllerMessenger.getRestricted({
     name: controllerName,
     allowedActions: [
+      'KeyringController:getState',
       'NetworkController:getNetworkConfigurationByNetworkClientId',
       'TokenListController:getState',
     ],
     allowedEvents: [
+      'AccountsController:selectedAccountChange',
+      'KeyringController:lock',
+      'KeyringController:unlock',
       'NetworkController:stateChange',
       'NetworkController:networkDidChange',
       'TokenListController:stateChange',
@@ -1297,6 +1301,12 @@ async function withController<ReturnValue>(
     .mockImplementation((networkClientId: string) => {
       return mockNetworkConfigurations[networkClientId];
     });
+  controllerMessenger.registerActionHandler(
+    'KeyringController:getState',
+    jest.fn().mockReturnValue({
+      isActive: true,
+    }),
+  );
   controllerMessenger.registerActionHandler(
     'NetworkController:getNetworkConfigurationByNetworkClientId',
     mockGetNetworkConfigurationByNetworkClientId,
@@ -1323,6 +1333,7 @@ async function withController<ReturnValue>(
       ...getDefaultPreferencesState(),
       useTokenDetection: true,
     }),
+    trackMetaMetricsEvent: jest.fn(),
     messenger: buildTokenDetectionControllerMessenger(controllerMessenger),
     ...options,
   });