Skip to content

Commit bfb5fde

Browse files
Merge pull request #8 from Wizard1209/fix/send-no-timeout
2 parents 591c4a7 + 9a2aa0c commit bfb5fde

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/pysignalr/transport/websocket.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,10 @@ async def _set_state(self, state: ConnectionState) -> None:
148148
self._state = state
149149

150150
async def _get_connection(self) -> WebSocketClientProtocol:
151-
await self._connected.wait()
151+
try:
152+
await asyncio.wait_for(self._connected.wait(), self._connection_timeout)
153+
except asyncio.TimeoutError as e:
154+
raise RuntimeError('The socket was never run') from e
152155
if not self._ws or self._ws.state != State.OPEN:
153156
raise RuntimeError('Connection is closed')
154157
return self._ws

0 commit comments

Comments
 (0)