From 5641346aba9abb7d53c09fb1a5cd9016ed3cdf84 Mon Sep 17 00:00:00 2001 From: Ian Thomas Date: Mon, 1 Jul 2024 09:14:09 +0100 Subject: [PATCH] Explicitly close memory object streams (#1253) --- tests/conftest.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/conftest.py b/tests/conftest.py index 61a1ea47..2c266555 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -63,6 +63,12 @@ def __init__(self, sockets, *args, **kwargs): send_stream, receive_stream = create_memory_object_stream(max_buffer_size=inf) self._streams[socket] = {"send": send_stream, "receive": receive_stream} + def close(self): + for streams in self._streams.values(): + for stream in streams.values(): + stream.close() + self._streams.clear() + def send(self, socket, *args, **kwargs): msg = super().send(socket, *args, **kwargs) send_stream: MemoryObjectSendStream[Any] = self._streams[socket]["send"] @@ -102,6 +108,7 @@ async def do_debug_request(self, msg): def destroy(self): self.stop() + self.session.close() for socket in self.test_sockets: socket.close() self.context.destroy()