Fix mock

MetaMask · Nov 4, 2022 · 0fc83df · 0fc83df
1 parent 0f1bead
commit 0fc83df
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/controllers/src/test-utils/controller.ts b/packages/controllers/src/test-utils/controller.ts
@@ -4,6 +4,7 @@ import {
   AllowedActions,
   AllowedEvents,
   CheckSnapBlockListArg,
+  handlerEndowments,
   PersistedSnapControllerState,
   SnapController,
   SnapControllerActions,
@@ -166,7 +167,7 @@ export const getSnapControllerWithEESOptions = (
 
       if (
         method === 'PermissionController:hasPermission' &&
-        args[1] === SnapEndowments.Rpc
+        Object.values(handlerEndowments).includes(args[1])
       ) {
         return true;
       }