diff --git a/tests/utils.py b/tests/utils.py index 47dd213530..909064651d 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -9,13 +9,13 @@ @asynccontextmanager async def run_server(config: Config, sockets=None): server = Server(config=config) - cancel_handle = asyncio.ensure_future(server.serve(sockets=sockets)) + task = asyncio.create_task(server.serve(sockets=sockets)) await asyncio.sleep(0.1) try: yield server finally: await server.shutdown() - cancel_handle.cancel() + task.cancel() @contextmanager