Added test

Author: thewhaleking

async_substrate_interface/async_substrate.py

@@ -618,7 +618,9 @@ async def _handler(self, ws: ClientConnection) -> None:
         for task in pending:
             task.cancel()
         for task in done:
-            if isinstance(task.result(), (asyncio.TimeoutError, ConnectionClosed, TimeoutError)):
+            if isinstance(
+                task.result(), (asyncio.TimeoutError, ConnectionClosed, TimeoutError)
+            ):
                 should_reconnect = True
         if should_reconnect is True:
             for original_id, payload in list(self._inflight.items()):
@@ -692,7 +694,9 @@ async def _start_receiving(self, ws: ClientConnection) -> Exception:
         except Exception as e:
             if isinstance(e, ssl.SSLError):
                 e = ConnectionClosed
-            if not isinstance(e, (asyncio.TimeoutError, TimeoutError, ConnectionClosed)):
+            if not isinstance(
+                e, (asyncio.TimeoutError, TimeoutError, ConnectionClosed)
+            ):
                 logger.exception("Websocket receiving exception", exc_info=e)
                 for fut in self._received.values():
                     if not fut.done():
@@ -717,7 +721,9 @@ async def _start_sending(self, ws) -> Exception:
         except Exception as e:
             if isinstance(e, ssl.SSLError):
                 e = ConnectionClosed
-            if not isinstance(e, (asyncio.TimeoutError, TimeoutError, ConnectionClosed)):
+            if not isinstance(
+                e, (asyncio.TimeoutError, TimeoutError, ConnectionClosed)
+            ):
                 logger.exception("Websocket sending exception", exc_info=e)
                 if to_send is not None:
                     self._received[to_send["id"]].set_exception(e)

tests/integration_tests/test_async_substrate_interface.py

@@ -1,3 +1,4 @@
+import asyncio
 import time
 
 import pytest
@@ -149,3 +150,14 @@ async def test_query_multiple():
             storage_function="OwnedHotkeys",
             block_hash=block_hash,
         )
+
+
+@pytest.mark.asyncio
+async def test_reconnection():
+    async with AsyncSubstrateInterface(
+        ARCHIVE_ENTRYPOINT, ss58_format=42, retry_timeout=8.0
+    ) as substrate:
+        await asyncio.sleep(9)  # sleep for longer than the retry timeout
+        bh = await substrate.get_chain_finalised_head()
+        assert isinstance(bh, str)
+        assert isinstance(await substrate.get_block_number(bh), int)