From 9eee55ef9c06100a1ba76794be10b95a7f2e9105 Mon Sep 17 00:00:00 2001
From: moataze <motaz_1993@hotmail.com>
Date: Tue, 17 Sep 2024 17:05:09 +0100
Subject: [PATCH] Resolve memory leak in TestClient

---
 starlette/testclient.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/starlette/testclient.py b/starlette/testclient.py
index fcf392e33..1a2d101a0 100644
--- 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 (