Use Python 3.12-specific way of fetching all tasks of all loops

Signed-off-by: Sergey Vasilyev <[email protected]>
nolar · Oct 9, 2023 · b626a75 · b626a75
1 parent 03b4885
commit b626a75
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -738,7 +738,10 @@ def _get_all_tasks() -> Set[asyncio.Task]:
     i = 0
     while True:
         try:
-            tasks = list(asyncio.tasks._all_tasks)
+            if sys.version_info >= (3, 12):
+                tasks = asyncio.tasks._eager_tasks | set(asyncio.tasks._scheduled_tasks)
+            else:
+                tasks = list(asyncio.tasks._all_tasks)
         except RuntimeError:
             i += 1
             if i >= 1000: