Reconnect on EOF (#141)

Author: vrslev

packages/stompman/stompman/connection.py

@@ -85,10 +85,8 @@ async def write_frame(self, frame: AnyClientFrame) -> None:
             await self.writer.drain()
 
     async def _read_non_empty_bytes(self, max_chunk_size: int) -> bytes:
-        while (  # noqa: ASYNC110
-            chunk := await self.reader.read(max_chunk_size)
-        ) == b"":  # pragma: no cover (it definitely happens)
-            await asyncio.sleep(0)
+        if (chunk := await self.reader.read(max_chunk_size)) == b"":
+            raise ConnectionLostError
         return chunk
 
     async def read_frames(self) -> AsyncGenerator[AnyServerFrame, None]: