diff --git a/tests/mock_stream.py b/tests/mock_stream.py index ad664040b7c3e832a4a00e69104964dcaa11ffd3..c2ccca59452fa765aa6a540a4549b9782ae13d25 100644 --- a/tests/mock_stream.py +++ b/tests/mock_stream.py @@ -104,10 +104,13 @@ class MockMessageStream: """ if self.closed: return - resp = await self.send_msg(Close()) + self.closed = True + try: + resp = await self.send_msg(Close()) + except anyio.BrokenResourceError: + return assert len(resp) == 0, resp await self._stream.aclose() - self.closed = True async def send_msg(self, msg: Message) -> list[Message]: """