Add ON_CONN_FAIL signal

Author: IURII Shikanov

tests/test_signal.py

@@ -1,6 +1,6 @@
 import pytest
 
-from aiohttp import WSServerHandshakeError
+from aiohttp import ClientConnectionError, WSServerHandshakeError
 
 from wsrpc_aiohttp import ClientException, WebSocketAsync
 from wsrpc_aiohttp.signal import Signal
@@ -165,3 +165,30 @@ async def on_call_fail(**kwargs):
         assert call_fail
         assert call_fail_args["method"] == "proc_fail"
         assert isinstance(call_fail_args["err"], RuntimeError)
+
+
+async def test_on_conn_fail_signal(client, handler, monkeypatch):
+    conn_fail = False
+    conn_err = None
+
+    async def on_conn_fail(socket, request, err):
+        nonlocal conn_fail
+        nonlocal conn_err
+        conn_fail = True
+        conn_err = err
+
+    handler.ON_CONN_FAIL.connect(on_conn_fail)
+
+    async def prepare_mock(*args, **kwargs):
+        raise RuntimeError("Prepare failed")
+
+    monkeypatch.setattr(
+        "wsrpc_aiohttp.websocket.handler.web.WebSocketResponse.prepare",
+        prepare_mock,
+    )
+    with pytest.raises(Exception):
+        async with client:
+            pass
+
+    assert conn_fail
+    assert isinstance(conn_err, RuntimeError)

wsrpc_aiohttp/websocket/handler.py

@@ -42,6 +42,7 @@ class WebSocketBase(WSRPCBase, AbstractView):
     ON_AUTH_FAIL = Signal()
     ON_CONN_OPEN = Signal()
     ON_CONN_CLOSE = Signal()
+    ON_CONN_FAIL = Signal()
 
     def __init__(self, request):
         AbstractView.__init__(self, request)
@@ -80,6 +81,7 @@ def freeze(cls):
             cls.ON_AUTH_FAIL,
             cls.ON_CONN_OPEN,
             cls.ON_CONN_CLOSE,
+            cls.ON_CONN_FAIL,
             cls.ON_CALL_START,
             cls.ON_CALL_SUCCESS,
             cls.ON_CALL_FAIL,
@@ -123,7 +125,15 @@ async def __handle_request(self):
             request=self.request,
         )
 
-        await self.socket.prepare(self.request)
+        try:
+            await self.socket.prepare(self.request)
+        except Exception as err:
+            await self.ON_CONN_FAIL.call(
+                socket=self.socket,
+                request=self.request,
+                err=err,
+            )
+            raise
 
         try:
             self.clients[self.id] = self