main commit

venv/lib/python3.12/site-packages/sqlalchemy/testing/asyncio.py

@@ -1,5 +1,5 @@
 # testing/asyncio.py
-# Copyright (C) 2005-2025 the SQLAlchemy authors and contributors
+# Copyright (C) 2005-2023 the SQLAlchemy authors and contributors
 # <see AUTHORS file>
 #
 # This module is part of SQLAlchemy and is released under
@@ -24,21 +24,16 @@ from functools import wraps
 import inspect
 
 from . import config
-from ..util.concurrency import _AsyncUtil
+from ..util.concurrency import _util_async_run
+from ..util.concurrency import _util_async_run_coroutine_function
 
 # may be set to False if the
 # --disable-asyncio flag is passed to the test runner.
 ENABLE_ASYNCIO = True
-_async_util = _AsyncUtil()  # it has lazy init so just always create one
-
-
-def _shutdown():
-    """called when the test finishes"""
-    _async_util.close()
 
 
 def _run_coroutine_function(fn, *args, **kwargs):
-    return _async_util.run(fn, *args, **kwargs)
+    return _util_async_run_coroutine_function(fn, *args, **kwargs)
 
 
 def _assume_async(fn, *args, **kwargs):
@@ -55,7 +50,7 @@ def _assume_async(fn, *args, **kwargs):
     if not ENABLE_ASYNCIO:
         return fn(*args, **kwargs)
 
-    return _async_util.run_in_greenlet(fn, *args, **kwargs)
+    return _util_async_run(fn, *args, **kwargs)
 
 
 def _maybe_async_provisioning(fn, *args, **kwargs):
@@ -74,7 +69,7 @@ def _maybe_async_provisioning(fn, *args, **kwargs):
         return fn(*args, **kwargs)
 
     if config.any_async:
-        return _async_util.run_in_greenlet(fn, *args, **kwargs)
+        return _util_async_run(fn, *args, **kwargs)
     else:
         return fn(*args, **kwargs)
 
@@ -94,7 +89,7 @@ def _maybe_async(fn, *args, **kwargs):
     is_async = config._current.is_async
 
     if is_async:
-        return _async_util.run_in_greenlet(fn, *args, **kwargs)
+        return _util_async_run(fn, *args, **kwargs)
     else:
         return fn(*args, **kwargs)