From 61f5e77b99b36952d62a03d3340694d7279d9b66 Mon Sep 17 00:00:00 2001
From: Dom Sekotill <dom.sekotill@kodo.org.uk>
Date: Thu, 3 Aug 2023 02:21:39 +0100
Subject: [PATCH] Handle a race condition in mock_stream tests

---
 tests/mock_stream.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/mock_stream.py b/tests/mock_stream.py
index ad66404..c2ccca5 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]:
 		"""
-- 
GitLab