[WIP] tests

Author: thewhaleking

tests/conftest.py

@@ -0,0 +1,24 @@
+import subprocess
+import time
+
+CONTAINER_NAME_PREFIX = "test_local_chain_"
+LOCALNET_IMAGE_NAME = "ghcr.io/opentensor/subtensor-localnet:devnet-ready"
+
+
+def start_docker_container(exposed_port, name_salt: str):
+    container_name = f"{CONTAINER_NAME_PREFIX}{name_salt}"
+
+    # Command to start container
+    cmds = [
+        "docker",
+        "run",
+        "--rm",
+        "--name",
+        container_name,
+        "-p",
+        f"{exposed_port}:9945",
+        LOCALNET_IMAGE_NAME,
+    ]
+
+    proc = subprocess.Popen(cmds, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    return proc, container_name

tests/test_substrate_addons.py

@@ -0,0 +1,36 @@
+import threading
+import subprocess
+
+import pytest
+import time
+
+from async_substrate_interface.substrate_addons import RetrySyncSubstrate
+from tests.conftest import start_docker_container
+
+
+@pytest.fixture(scope="function")
+def start_containers():
+    # Store our subprocesses globally
+    processes = (start_docker_container(9945, 9945), start_docker_container(9946, 9946))
+    yield processes
+
+    # To stop the instances, you can iterate over the processes and kill them:
+    for process in processes:
+        subprocess.run(["docker", "kill", process[1]])
+        process[0].kill()
+
+
+def test_retry_sync_substrate(start_containers):
+    container1, container2 = start_containers
+    time.sleep(10)
+    with RetrySyncSubstrate(
+        "ws://127.0.0.1:9945", fallback_chains=["ws://127.0.0.1:9946"]
+    ) as substrate:
+        for i in range(10):
+            assert substrate.get_chain_head().startswith("0x")
+            if i == 8:
+                subprocess.run(["docker", "kill", container1[1]])
+                time.sleep(10)
+            if i > 8:
+                assert substrate.chain_endpoint == "ws://127.0.0.1:9946"
+            time.sleep(2)