Add more tests

Author: thewhaleking

tests/integration_tests/test_disk_cache.py

@@ -1,5 +1,5 @@
 import pytest
-
+import time
 from async_substrate_interface.async_substrate import (
     DiskCachedAsyncSubstrateInterface,
     AsyncSubstrateInterface,
@@ -41,6 +41,7 @@ async def test_disk_cache():
     assert parent_block_hash == parent_block_hash_from_cache
     assert block_runtime_info == block_runtime_info_from_cache
     assert block_runtime_version_for == block_runtime_version_from_cache
+    # Verify data integrity with non-disk cached Async Substrate Interface
     async with AsyncSubstrateInterface(entrypoint) as non_cache_substrate:
         block_hash_non_cache = await non_cache_substrate.get_block_hash(current_block)
         parent_block_hash_non_cache = await non_cache_substrate.get_parent_block_hash(
@@ -58,6 +59,7 @@ async def test_disk_cache():
     assert parent_block_hash == parent_block_hash_non_cache
     assert block_runtime_info == block_runtime_info_non_cache
     assert block_runtime_version_for == block_runtime_version_for_non_cache
+    # Verify data integrity with sync Substrate Interface
     with SubstrateInterface(entrypoint) as sync_substrate:
         block_hash_sync = sync_substrate.get_block_hash(current_block)
         parent_block_hash_sync = sync_substrate.get_parent_block_hash(
@@ -73,4 +75,55 @@ async def test_disk_cache():
     assert parent_block_hash == parent_block_hash_sync
     assert block_runtime_info == block_runtime_info_sync
     assert block_runtime_version_for == block_runtime_version_for_sync
-    print("test_disk_cache succeeded")
+    # Verify data is pulling from disk cache
+    async with DiskCachedAsyncSubstrateInterface(entrypoint) as disk_cached_substrate:
+        start = time.monotonic()
+        new_block_hash = await disk_cached_substrate.get_block_hash(current_block)
+        new_time = time.monotonic()
+        assert new_time - start < 0.001
+
+        start = time.monotonic()
+        new_parent_block_hash = await disk_cached_substrate.get_parent_block_hash(
+            block_hash
+        )
+        new_time = time.monotonic()
+        assert new_time - start < 0.001
+        start = time.monotonic()
+        new_block_runtime_info = await disk_cached_substrate.get_block_runtime_info(
+            block_hash
+        )
+        new_time = time.monotonic()
+        assert new_time - start < 0.001
+        start = time.monotonic()
+        new_block_runtime_version_for = (
+            await disk_cached_substrate.get_block_runtime_version_for(block_hash)
+        )
+        new_time = time.monotonic()
+        assert new_time - start < 0.001
+        start = time.monotonic()
+        new_block_hash_from_cache = await disk_cached_substrate.get_block_hash(
+            current_block
+        )
+        new_time = time.monotonic()
+        assert new_time - start < 0.001
+        start = time.monotonic()
+        new_parent_block_hash_from_cache = (
+            await disk_cached_substrate.get_parent_block_hash(block_hash_from_cache)
+        )
+        new_time = time.monotonic()
+        assert new_time - start < 0.001
+        start = time.monotonic()
+        new_block_runtime_info_from_cache = (
+            await disk_cached_substrate.get_block_runtime_info(block_hash_from_cache)
+        )
+        new_time = time.monotonic()
+        assert new_time - start < 0.001
+        start = time.monotonic()
+        new_block_runtime_version_from_cache = (
+            await disk_cached_substrate.get_block_runtime_version_for(
+                block_hash_from_cache
+            )
+        )
+        new_time = time.monotonic()
+        assert new_time - start < 0.001
+    print("Disk Cache tests passed")