fix: ensure unsubscribe only attempts on open WebSocket connections (#51)

Author: smypmsa

common/metric_types.py

@@ -91,13 +91,16 @@ async def _collect_ws_data():
 
         finally:
             if websocket:
-                try:
-                    await asyncio.shield(self.unsubscribe(websocket))
-                except asyncio.CancelledError:
-                    logging.info("Unsubscribe completed despite cancellation")
-                except Exception as e:
-                    logging.warning(f"Error during unsubscribe (non-critical): {e!s}")
-
+                # Only try to unsubscribe if connection is still open
+                if websocket.open:
+                    try:
+                        await asyncio.shield(self.unsubscribe(websocket))
+                    except asyncio.CancelledError:
+                        logging.info("Unsubscribe completed despite cancellation")
+                    except Exception as e:
+                        logging.warning(f"Error during unsubscribe (non-critical): {e!s}")
+
+                # Close can be called even on closed connections (it's idempotent)
                 try:
                     await asyncio.shield(websocket.close())
                 except asyncio.CancelledError:

tests/test_api_read.py

@@ -29,7 +29,7 @@ def main() -> None:
     setup_environment()
 
     # Import handler after environment setup
-    from api.read.hyperliquid import handler
+    from api.read.arbitrum import handler
 
     server = HTTPServer(("localhost", 8000), handler)
     print("Server started at http://localhost:8000")