Close unclosed MemoryObjectReceiveStream in TestClient (encode#2693)

nixroxursox · Sep 30, 2024 · d7ff369 · d7ff369
1 parent 8248d07
commit d7ff369
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/starlette/testclient.py b/starlette/testclient.py
@@ -784,7 +784,7 @@ async def receive() -> typing.Any:
                 self.task.result()
             return message
 
-        async with self.stream_send:
+        async with self.stream_send, self.stream_receive:
             await self.stream_receive.send({"type": "lifespan.shutdown"})
             message = await receive()
             assert message["type"] in (