Close unclosed MemoryObjectReceiveStream in TestClient (#2693)

encode · Sep 22, 2024 · 6d70ad3 · 6d70ad3
1 parent 2d0dde8
commit 6d70ad3
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 (