fix: tests transport

signalrcore/transport/websockets/websocket_transport.py

@@ -391,10 +391,12 @@ def on_message(self, app, raw_message):
 
     def on_enter(self) -> None:
         super(WsDisconnectedState, self).on_enter()
-
-        if self.context._on_close is not None and callable(self.context._on_close):
-            self.context._on_close()
-
+        try:
+            if self.context._on_close is not None and callable(self.context._on_close):
+                self.context._on_close()
+        except Exception as ex:
+            self.logger.error(ex)
+
         if self.context.reconnection_handler is not None:
             self.handle_reconnect()
 

test/reconnection_test.py

@@ -42,6 +42,10 @@ def test_reconnect_interval_config(self):
 
         connection.transport._ws.close(status=1000, reason="Closed".encode("utf-8"))
 
+        self.assertRaises(
+            HubConnectionError,
+            lambda: connection.send("DisconnectMe", []))
+
         self.assertTrue(self._locks[identifier].acquire(timeout=30))
 
         connection.stop()
@@ -87,6 +91,10 @@ def test_no_reconnect(self):
 
         connection.transport._ws.close(status=1000, reason="Closed".encode("utf-8"))
 
+        self.assertRaises(
+            HubConnectionError,
+            lambda: connection.send("DisconnectMe", []))
+
         self.assertTrue(self._locks[identifier].acquire(timeout=40)) # Released on close
 
         # reconnecting not configured must not enqueue message