remember to open and close the requests session

encode · Sep 6, 2020 · e6fa6e6 · e6fa6e6
1 parent c923a96
commit e6fa6e6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/starlette/testclient.py b/starlette/testclient.py
@@ -455,11 +455,12 @@ def __enter__(self) -> requests.Session:
         self.receive_queue = asyncio.Queue()  # type: asyncio.Queue
         self.task = loop.create_task(self.lifespan())
         loop.run_until_complete(self.wait_startup())
-        return self
+        return super().__enter__()
 
     def __exit__(self, *args: typing.Any) -> None:
         loop = asyncio.get_event_loop()
         loop.run_until_complete(self.wait_shutdown())
+        return super().__exit__(*args)
 
     async def lifespan(self) -> None:
         scope = {"type": "lifespan"}