Remove timeout breadcrumb from ConnectionLostError (#20)

Author: vrslev

stompman/connection.py

@@ -83,6 +83,14 @@ async def read_frame_of_type(self, type_: type[FrameT]) -> FrameT:
                     return frame
 
 
+@contextmanager
+def _reraise_connection_lost(*causes: type[Exception]) -> Generator[None, None, None]:
+    try:
+        yield
+    except causes as exception:
+        raise ConnectionLostError from exception
+
+
 @dataclass
 class Connection(AbstractConnection):
     connection_parameters: ConnectionParameters
@@ -102,24 +110,17 @@ async def connect(self) -> bool:
             return False
         return True
 
-    @contextmanager
-    def _reraise_connection_lost(self, *causes: type[Exception]) -> Generator[None, None, None]:
-        try:
-            yield
-        except causes as exception:
-            raise ConnectionLostError(self.read_timeout) from exception
-
     async def close(self) -> None:
         self.writer.close()
-        with self._reraise_connection_lost(ConnectionError):
+        with _reraise_connection_lost(ConnectionError):
             await self.writer.wait_closed()
 
     def write_heartbeat(self) -> None:
         return self.writer.write(NEWLINE)
 
     async def write_frame(self, frame: AnyClientFrame) -> None:
         self.writer.write(dump_frame(frame))
-        with self._reraise_connection_lost(ConnectionError):
+        with _reraise_connection_lost(ConnectionError):
             await self.writer.drain()
 
     async def _read_non_empty_bytes(self) -> bytes:
@@ -133,7 +134,7 @@ async def read_frames(self) -> AsyncGenerator[AnyServerFrame, None]:
         parser = Parser()
 
         while True:
-            with self._reraise_connection_lost(ConnectionError, TimeoutError):
+            with _reraise_connection_lost(ConnectionError, TimeoutError):
                 raw_frames = await asyncio.wait_for(self._read_non_empty_bytes(), timeout=self.read_timeout)
 
             for frame in cast(Iterator[AnyServerFrame], parser.load_frames(raw_frames)):

stompman/errors.py

@@ -31,5 +31,4 @@ class FailedAllConnectAttemptsError(Error):
 
 
 @dataclass
-class ConnectionLostError(Error):
-    timeout: int
+class ConnectionLostError(Error): ...