Add test for Snap calling wallet_switchEthereumChain

Author: Mrtenz

app/scripts/lib/rpc-method-middleware/handlers/switch-ethereum-chain.test.js

@@ -175,4 +175,61 @@ describe('switchEthereumChainHandler', () => {
       },
     );
   });
+
+  it('calls `grantPermittedChainsPermissionIncremental` if the origin is a Snap', async () => {
+    const mocks = makeMocks({
+      overrides: {
+        grantPermittedChainsPermissionIncremental: jest.fn(),
+      },
+    });
+
+    const switchEthereumChainHandler = switchEthereumChain.implementation;
+    await switchEthereumChainHandler(
+      {
+        origin: 'npm:foo-snap',
+        params: [{ chainId: CHAIN_IDS.MAINNET }],
+      },
+      {},
+      jest.fn(),
+      jest.fn(),
+      mocks,
+    );
+
+    expect(
+      mocks.grantPermittedChainsPermissionIncremental,
+    ).toHaveBeenCalledTimes(1);
+    expect(mocks.setActiveNetwork).toHaveBeenCalledTimes(1);
+    expect(mocks.setActiveNetwork).toHaveBeenCalledWith(
+      createMockMainnetConfiguration().rpcEndpoints[0].networkClientId,
+    );
+  });
+
+  it('does not call `grantPermittedChainsPermissionIncremental` if the origin is a Snap, but the permission is already set', async () => {
+    const mocks = makeMocks({
+      permissionedChainIds: [CHAIN_IDS.MAINNET],
+      overrides: {
+        grantPermittedChainsPermissionIncremental: jest.fn(),
+      },
+    });
+
+    const switchEthereumChainHandler = switchEthereumChain.implementation;
+    await switchEthereumChainHandler(
+      {
+        origin: 'npm:foo-snap',
+        params: [{ chainId: CHAIN_IDS.MAINNET }],
+      },
+      {},
+      jest.fn(),
+      jest.fn(),
+      mocks,
+    );
+
+    expect(
+      mocks.grantPermittedChainsPermissionIncremental,
+    ).not.toHaveBeenCalled();
+    expect(mocks.setActiveNetwork).toHaveBeenCalledTimes(1);
+    expect(mocks.setActiveNetwork).toHaveBeenCalledWith(
+      createMockMainnetConfiguration().rpcEndpoints[0].networkClientId,
+    );
+  });
 });