fix tests

MetaMask · Jan 23, 2023 · 548ad1e · 548ad1e
1 parent 4943056
commit 548ad1e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/shared/constants/permissions.test.js b/shared/constants/permissions.test.js
@@ -11,7 +11,10 @@ describe('EndowmentPermissions', () => {
   it('has the expected permission keys', () => {
     expect(Object.keys(EndowmentPermissions).sort()).toStrictEqual(
       Object.keys(endowmentPermissionBuilders)
-        .filter((targetKey) => !ExcludedSnapEndowments.has(targetKey))
+        .filter(
+          (targetKey) =>
+            !Object.keys(ExcludedSnapEndowments).includes(targetKey),
+        )
         .sort(),
     );
   });
@@ -23,7 +26,8 @@ describe('RestrictedMethods', () => {
       [
         'eth_accounts',
         ...Object.keys(restrictedMethodPermissionBuilders).filter(
-          (targetKey) => !ExcludedSnapPermissions.has(targetKey),
+          (targetKey) =>
+            !Object.keys(ExcludedSnapPermissions).includes(targetKey),
         ),
       ].sort(),
     );

diff --git a/shared/constants/permissions.ts b/shared/constants/permissions.ts
@@ -24,7 +24,15 @@ export const PermissionNamespaces = Object.freeze({
   wallet_snap_: 'wallet_snap_*',
 } as const);
 
-export const EndowmentPermissions = Object.freeze({
+export const StableEndowmentPermissions = Object.freeze({
+  'endowment:network-access': 'endowment:network-access',
+  'endowment:transaction-insight': 'endowment:transaction-insight',
+  'endowment:cronjob': 'endowment:cronjob',
+  'endowment:ethereum-provider': 'endowment:ethereum-provider',
+  'endowment:rpc': 'endowment:rpc',
+} as const);
+
+export const FlaskEndowmentPermissions = Object.freeze({
   'endowment:network-access': 'endowment:network-access',
   'endowment:long-running': 'endowment:long-running',
   'endowment:transaction-insight': 'endowment:transaction-insight',
@@ -52,6 +60,10 @@ export const ExcludedFlaskSnapEndowments = {
   'endowment:keyring': 'This endowment is not available',
 };
 
+export const EndowmentPermissions = isMain
+  ? StableEndowmentPermissions
+  : FlaskEndowmentPermissions;
+
 export const ExcludedSnapPermissions = isMain
   ? ExcludedStableSnapPermissions
   : ExcludedFlaskSnapPermissions;