Improve mocks for 'clear network status' test

MetaMask · May 8, 2023 · afb95de · afb95de
1 parent a634e4c
commit afb95de
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/packages/network-controller/tests/NetworkController.test.ts b/packages/network-controller/tests/NetworkController.test.ts
@@ -3529,9 +3529,13 @@ function refreshNetworkTests({
             request: {
               method: 'net_version',
             },
-            response: {
-              result: '1',
+            response: SUCCESSFUL_NET_VERSION_RESPONSE,
+          },
+          {
+            request: {
+              method: 'eth_getBlockByNumber',
             },
+            response: SUCCESSFUL_ETH_GET_BLOCK_BY_NUMBER_RESPONSE,
           },
           // Called during network lookup after resetting connection.
           // Delayed to ensure that we can check the network status
@@ -3541,9 +3545,14 @@ function refreshNetworkTests({
             request: {
               method: 'net_version',
             },
-            response: {
-              result: '1',
+            response: SUCCESSFUL_NET_VERSION_RESPONSE,
+          },
+          {
+            delay: 1,
+            request: {
+              method: 'eth_getBlockByNumber',
             },
+            response: SUCCESSFUL_ETH_GET_BLOCK_BY_NUMBER_RESPONSE,
           },
         ]);
         const fakeNetworkClient = buildFakeClient(fakeProvider);