Detect unreported websocket closures in async client (Fixes miguelgri…

…nberg/python-socketio#401)
miguelgrinberg · Dec 22, 2019 · a67b3d7 · a67b3d7
1 parent 33c7cf1
commit a67b3d7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/engineio/asyncio_client.py b/engineio/asyncio_client.py
@@ -485,6 +485,8 @@ async def _read_loop_websocket(self):
             p = None
             try:
                 p = (await self.ws.receive()).data
+                if p is None:  # pragma: no cover
+                    raise RuntimeError('WebSocket read returned None')
             except aiohttp.client_exceptions.ServerDisconnectedError:
                 self.logger.info(
                     'Read loop: WebSocket connection was closed, aborting')