Improve mocks for 'clear network status' test

MetaMask · May 8, 2023 · 821d791 · 821d791
1 parent 9d04302
commit 821d791
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
@@ -3531,9 +3531,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
@@ -3543,9 +3547,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);