test: Fix incorrect mock for KeyringController:getState

MetaMask · Jan 26, 2024 · 41fee27 · 41fee27
1 parent a590dcc
commit 41fee27
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/packages/assets-controllers/src/TokenDetectionController.test.ts b/packages/assets-controllers/src/TokenDetectionController.test.ts
@@ -6,6 +6,7 @@ import {
   convertHexToDecimal,
   BUILT_IN_NETWORKS,
 } from '@metamask/controller-utils';
+import type { KeyringControllerState } from '@metamask/keyring-controller';
 import {
   defaultState as defaultNetworkState,
   type NetworkConfiguration,
@@ -1303,9 +1304,9 @@ async function withController<ReturnValue>(
     });
   controllerMessenger.registerActionHandler(
     'KeyringController:getState',
-    jest.fn().mockReturnValue({
-      isActive: true,
-    }),
+    jest.fn<KeyringControllerState, []>().mockReturnValue({
+      isUnlocked: true,
+    } as unknown as KeyringControllerState),
   );
   controllerMessenger.registerActionHandler(
     'NetworkController:getNetworkConfigurationByNetworkClientId',