Tests

Author: thewhaleking

async_substrate_interface/types.py

@@ -137,6 +137,7 @@ def __init__(
         self.metadata_v15 = metadata_v15
         self.runtime_info = runtime_info
         self.registry = registry
+        runtime_info = runtime_info or {}
         self.runtime_version = runtime_info.get("specVersion")
         self.transaction_version = runtime_info.get("transactionVersion")
         self.runtime_config = runtime_config or RuntimeConfigurationObject(

tests/unit_tests/asyncio_/test_substrate_interface.py

@@ -6,6 +6,7 @@
 
 from async_substrate_interface.async_substrate import AsyncSubstrateInterface
 from async_substrate_interface.types import ScaleObj
+from tests.helpers.settings import ARCHIVE_ENTRYPOINT
 
 
 @pytest.mark.asyncio
@@ -113,3 +114,21 @@ async def test_websocket_shutdown_timer():
         await substrate.get_chain_head()
         await asyncio.sleep(6)  # same sleep time as before
         assert substrate.ws._initialized is True  # connection should still be open
+
+
+@pytest.mark.asyncio
+async def test_runtime_switching():
+    block = 6067945  # block where a runtime switch happens
+    async with AsyncSubstrateInterface(
+        ARCHIVE_ENTRYPOINT, ss58_format=42, chain_name="Bittensor"
+    ) as substrate:
+        # assures we switch between the runtimes without error
+        assert await substrate.get_extrinsics(block_number=block - 20) is not None
+        assert await substrate.get_extrinsics(block_number=block) is not None
+        assert await substrate.get_extrinsics(block_number=block - 21) is not None
+        one, two = await asyncio.gather(
+            substrate.get_extrinsics(block_number=block - 22),
+            substrate.get_extrinsics(block_number=block + 1),
+        )
+        assert one is not None
+        assert two is not None

tests/unit_tests/sync/test_substrate_interface.py

@@ -3,6 +3,8 @@
 from async_substrate_interface.sync_substrate import SubstrateInterface
 from async_substrate_interface.types import ScaleObj
 
+from tests.helpers.settings import ARCHIVE_ENTRYPOINT
+
 
 def test_runtime_call(monkeypatch):
     substrate = SubstrateInterface("ws://localhost", _mock=True)
@@ -73,3 +75,14 @@ def test_runtime_call(monkeypatch):
         "state_call", ["SubstrateApi_SubstrateMethod", "", None]
     )
     substrate.close()
+
+
+def test_runtime_switching():
+    block = 6067945  # block where a runtime switch happens
+    with SubstrateInterface(
+        ARCHIVE_ENTRYPOINT, ss58_format=42, chain_name="Bittensor"
+    ) as substrate:
+        # assures we switch between the runtimes without error
+        assert substrate.get_extrinsics(block_number=block - 20) is not None
+        assert substrate.get_extrinsics(block_number=block) is not None
+        assert substrate.get_extrinsics(block_number=block - 21) is not None

tests/unit_tests/test_types.py

@@ -1,4 +1,4 @@
-from async_substrate_interface.types import ScaleObj
+from async_substrate_interface.types import ScaleObj, Runtime, RuntimeCache
 
 
 def test_scale_object():
@@ -51,3 +51,34 @@ def test_scale_object():
     assert inst_dict["a"] == 1
     assert inst_dict["b"] == 2
     assert [i for i in inst_dict] == ["a", "b"]
+
+
+def test_runtime_cache():
+    fake_block = 2
+    fake_hash = "0xignore"
+    fake_version = 271
+
+    new_fake_block = 3
+    newer_fake_block = 4
+
+    new_fake_hash = "0xnewfakehash"
+
+    runtime = Runtime("", None, None)
+    runtime_cache = RuntimeCache()
+    # insert our Runtime object into the cache with a set block, hash, and version
+    runtime_cache.add_item(runtime, fake_block, fake_hash, fake_version)
+
+    assert runtime_cache.retrieve(fake_block) is not None
+    # cache does not yet know that new_fake_block has the same runtime
+    assert runtime_cache.retrieve(new_fake_block) is None
+    assert (
+        runtime_cache.retrieve(new_fake_block, runtime_version=fake_version) is not None
+    )
+    # after checking the runtime with the new block, it now knows this runtime should also map to this block
+    assert runtime_cache.retrieve(new_fake_block) is not None
+    assert runtime_cache.retrieve(newer_fake_block) is None
+    assert runtime_cache.retrieve(newer_fake_block, fake_hash) is not None
+    assert runtime_cache.retrieve(newer_fake_block) is not None
+    assert runtime_cache.retrieve(block_hash=new_fake_hash) is None
+    assert runtime_cache.retrieve(fake_block, block_hash=new_fake_hash) is not None
+    assert runtime_cache.retrieve(block_hash=new_fake_hash) is not None