test: Fix incorrect mock for KeyringController:getState

MetaMask · Feb 6, 2024 · 3fd2138 · 3fd2138
1 parent 5df6b81
commit 3fd2138
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',