diff --git a/starlette/testclient.py b/starlette/testclient.py index a4c7dffc6a..b25b14a0ba 100644 --- 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"}