Connection: Handle websocket exceptions

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "xapi-python"
-version = "0.1.3"
+version = "0.1.4"
 authors = [
   { name="Paweł Knioła", email="[email protected]" },
 ]

setup.py

@@ -20,6 +20,6 @@
         "Programming Language :: Python :: 3.12"
     ],
     python_requires='>=3.7',
-    version="0.1.3",
+    version="0.1.4",
     packages=['xapi'],
 )

tests/test_connect.py

@@ -11,6 +11,14 @@
 
 class TestConnection(unittest.IsolatedAsyncioTestCase):
 
+    async def test_connect_websocket_exception(self):
+        c = Connection()
+        with patch("websockets.client.connect", new_callable=AsyncMock) as mocked_connect:
+            mocked_connect.side_effect = websockets.exceptions.WebSocketException()
+            with self.assertRaises(ConnectionClosed) as cm:
+                await c.connect("ws://127.0.0.1:9000")
+            self.assertEqual(str(cm.exception), "WebSocket exception: ")
+
     async def test_connect_socket_gaierror(self):
         c = Connection()
         with patch("websockets.client.connect", new_callable=AsyncMock) as mocked_connect:

xapi/__init__.py

@@ -3,7 +3,7 @@
 """
 
 name = "xapi"
-__version__ = "0.1.3"
+__version__ = "0.1.4"
 
 from .xapi import XAPI, connect
 from .enums import TradeCmd, TradeType, TradeStatus, PeriodCode

xapi/connection.py

@@ -16,6 +16,9 @@ async def connect(self, url):
         try:
             self._conn = await websockets.client.connect(url, close_timeout=0, max_size=None)
 
+        except websockets.exceptions.WebSocketException as e:
+            raise ConnectionClosed(f"WebSocket exception: {e}")
+
         except socket.gaierror:
             raise ConnectionClosed("Hostname cannot be resolved")