chore: add tests for bridge selectors

MetaMask · Oct 2, 2024 · f4e0114 · f4e0114
1 parent ea07b02
commit f4e0114
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 0 deletions.
diff --git a/test/jest/mock-store.js b/test/jest/mock-store.js
@@ -681,6 +681,8 @@ export const createBridgeMockStore = (
           extensionSupport: false,
           srcNetworkAllowlist: [],
           destNetworkAllowlist: [],
+          approvalGasMultiplier: {},
+          bridgeGasMultiplier: {},
           ...featureFlagOverrides,
         },
         ...bridgeStateOverrides,

diff --git a/ui/ducks/bridge/selectors.test.ts b/ui/ducks/bridge/selectors.test.ts
@@ -8,6 +8,8 @@ import { ALLOWED_BRIDGE_CHAIN_IDS } from '../../../shared/constants/bridge';
 import { mockNetworkState } from '../../../test/stub/networks';
 import {
   getAllBridgeableNetworks,
+  getApprovalGasMultipliers,
+  getBridgeGasMultipliers,
   getFromAmount,
   getFromChain,
   getFromChains,
@@ -491,4 +493,59 @@ describe('Bridge selectors', () => {
       expect(result).toStrictEqual([{ address: '0x00', symbol: 'TEST' }]);
     });
   });
+
+  describe('getApprovalGasMultipliers', () => {
+    it('returns approval gas multipliers when present', () => {
+      const state = createBridgeMockStore(
+        {
+          approvalGasMultiplier: {
+            [CHAIN_IDS.MAINNET]: 1.1,
+            [CHAIN_IDS.LINEA_MAINNET]: 1.2,
+          },
+        },
+        {},
+        {},
+        {},
+      );
+      const result = getApprovalGasMultipliers(state as never);
+      expect(result).toEqual({
+        [CHAIN_IDS.MAINNET]: 1.1,
+        [CHAIN_IDS.LINEA_MAINNET]: 1.2,
+      });
+    });
+
+    it('returns an empty object when approval gas multipliers are not present', () => {
+      const state = createBridgeMockStore();
+      const result = getApprovalGasMultipliers(state as never);
+      expect(result).toEqual({});
+    });
+  });
+
+  describe('getBridgeGasMultipliers', () => {
+    it('should return bridge gas multipliers when present', () => {
+      const state = createBridgeMockStore(
+        {
+          bridgeGasMultiplier: {
+            [CHAIN_IDS.MAINNET]: 1.1,
+            [CHAIN_IDS.LINEA_MAINNET]: 1.2,
+          },
+        },
+        {},
+        {},
+        {},
+      );
+
+      const result = getBridgeGasMultipliers(state as never);
+      expect(result).toEqual({
+        [CHAIN_IDS.MAINNET]: 1.1,
+        [CHAIN_IDS.LINEA_MAINNET]: 1.2,
+      });
+    });
+
+    it('should return an empty object when bridge gas multipliers are not present', () => {
+      const state = createBridgeMockStore();
+      const result = getBridgeGasMultipliers(state as never);
+      expect(result).toEqual({});
+    });
+  });
 });