use asyncio.run to run coroutines

kevin1024 · Dec 15, 2023 · 895ae20 · 895ae20
1 parent f075c8b
commit 895ae20
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/integration/test_aiohttp.py b/tests/integration/test_aiohttp.py
@@ -14,10 +14,10 @@
 
 
 def run_in_loop(fn):
- with contextlib.closing(asyncio.new_event_loop()) as loop:
- asyncio.set_event_loop(loop)
- task = loop.create_task(fn(loop))
-  return loop.run_until_complete(task)
+ async def wrapper():
+ return await fn(asyncio.get_running_loop())
+
+ return asyncio.run(wrapper())
 
 
 def request(method, url, output="text", **kwargs):