Verified Commit 61f5e77b authored by Dom Sekotill's avatar Dom Sekotill
Browse files

Handle a race condition in mock_stream tests

parent 7bb78150

tests/mock_stream.py

+5 −2
Original line number Diff line number Diff line
@@ -104,10 +104,13 @@ class MockMessageStream: 
		"""

		if self.closed:

			return

		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]:

		"""