Skip to content

Commit 5641346

Browse files
authored
Explicitly close memory object streams (#1253)
1 parent 3116467 commit 5641346

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/conftest.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,12 @@ def __init__(self, sockets, *args, **kwargs):
6363
send_stream, receive_stream = create_memory_object_stream(max_buffer_size=inf)
6464
self._streams[socket] = {"send": send_stream, "receive": receive_stream}
6565

66+
def close(self):
67+
for streams in self._streams.values():
68+
for stream in streams.values():
69+
stream.close()
70+
self._streams.clear()
71+
6672
def send(self, socket, *args, **kwargs):
6773
msg = super().send(socket, *args, **kwargs)
6874
send_stream: MemoryObjectSendStream[Any] = self._streams[socket]["send"]
@@ -102,6 +108,7 @@ async def do_debug_request(self, msg):
102108

103109
def destroy(self):
104110
self.stop()
111+
self.session.close()
105112
for socket in self.test_sockets:
106113
socket.close()
107114
self.context.destroy()

0 commit comments

Comments
 (0)