Fixed mocked data in unit tests

polkascan · Nov 13, 2020 · b24673b · b24673b
1 parent a850826
commit b24673b
Show file tree

Hide file tree

Showing 2 changed files with 62 additions and 1 deletion.
diff --git a/test/test_helper_functions.py b/test/test_helper_functions.py
@@ -41,6 +41,21 @@ def mocked_request(method, params):
                     "result": {"specVersion": 2023},
                     "id": 1
                 }
+            elif method == 'chain_getHeader':
+                return {
+                    "jsonrpc": "2.0",
+                    "result": {
+                        "digest": {
+                            "logs": [
+                            ]
+                        },
+                        "extrinsicsRoot": "0xa94148d938c7b7976abf4272dca95724d7a74da2f3649ec0bd53dc3daaedda44",
+                        "number": "0x4abaaa",
+                        "parentHash": "0xe1781813275653a970b4260298b3858b36d38e072256dad674f7c786a0cae236",
+                        "stateRoot": "0xb6aa468385c82d15b343a676b3488d9f141ac100fc548bb8a546f27a7241c44a"
+                    },
+                    "id": 1
+                }
 
         cls.substrate.rpc_request = MagicMock(side_effect=mocked_request)
 

diff --git a/test/test_runtime_state.py b/test/test_runtime_state.py
@@ -34,18 +34,34 @@ def setUpClass(cls):
     def test_plaintype_call(self):
 
         def mocked_request(method, params):
+
             if method == 'chain_getRuntimeVersion':
                 return {
                     "jsonrpc": "2.0",
                     "result": {"specVersion": 2023},
                     "id": 1
                 }
-            if method == 'state_getStorageAt':
+            elif method == 'state_getStorageAt':
                 return {
                     "jsonrpc": "2.0",
                     "result": '0x0800000000000000482d7c0900000000020000000100000000000000000000000000020000',
                     "id": 1
                 }
+            elif method == 'chain_getHeader':
+                return {
+                    "jsonrpc": "2.0",
+                    "result": {
+                        "digest": {
+                            "logs": [
+                            ]
+                        },
+                        "extrinsicsRoot": "0xa94148d938c7b7976abf4272dca95724d7a74da2f3649ec0bd53dc3daaedda44",
+                        "number": "0x4abaaa",
+                        "parentHash": "0xe1781813275653a970b4260298b3858b36d38e072256dad674f7c786a0cae236",
+                        "stateRoot": "0xb6aa468385c82d15b343a676b3488d9f141ac100fc548bb8a546f27a7241c44a"
+                    },
+                    "id": 1
+                }
 
         metadata_decoder = MetadataDecoder(ScaleBytes(metadata_v12_hex))
         metadata_decoder.decode()
@@ -79,6 +95,21 @@ def mocked_request(method, params):
                     'result': '0x00000000030000c16ff28623000000000000000000000000000000000000000000000000000000c16ff286230000000000000000000000c16ff28623000000000000000000',
                     'id': 1
                 }
+            elif method == 'chain_getHeader':
+                return {
+                    "jsonrpc": "2.0",
+                    "result": {
+                        "digest": {
+                            "logs": [
+                            ]
+                        },
+                        "extrinsicsRoot": "0xa94148d938c7b7976abf4272dca95724d7a74da2f3649ec0bd53dc3daaedda44",
+                        "number": "0x4abaaa",
+                        "parentHash": "0xe1781813275653a970b4260298b3858b36d38e072256dad674f7c786a0cae236",
+                        "stateRoot": "0xb6aa468385c82d15b343a676b3488d9f141ac100fc548bb8a546f27a7241c44a"
+                    },
+                    "id": 1
+                }
 
         self.substrate.rpc_request = MagicMock(side_effect=mocked_request)
         metadata_decoder = MetadataDecoder(ScaleBytes(metadata_v12_hex))
@@ -121,6 +152,21 @@ def mocked_request(method, params):
                     ],
                     "id": 1
                 }
+            elif method == 'chain_getHeader':
+                return {
+                    "jsonrpc": "2.0",
+                    "result": {
+                        "digest": {
+                            "logs": [
+                            ]
+                        },
+                        "extrinsicsRoot": "0xa94148d938c7b7976abf4272dca95724d7a74da2f3649ec0bd53dc3daaedda44",
+                        "number": "0x4abaaa",
+                        "parentHash": "0xe1781813275653a970b4260298b3858b36d38e072256dad674f7c786a0cae236",
+                        "stateRoot": "0xb6aa468385c82d15b343a676b3488d9f141ac100fc548bb8a546f27a7241c44a"
+                    },
+                    "id": 1
+                }
 
         self.substrate.rpc_request = MagicMock(side_effect=mocked_request)
         metadata_decoder = MetadataDecoder(ScaleBytes(metadata_v12_hex))