summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/testing
diff options
context:
space:
mode:
authorFederico Caselli <cfederico87@gmail.com>2022-01-19 23:41:36 +0100
committerMike Bayer <mike_mp@zzzcomputing.com>2022-01-21 17:36:01 -0500
commit9b2cd1ede5951fff7180d64bb39aa3a601ec1900 (patch)
tree5170f93ac7c9ccb350ff9fc8edbaeab400e2803a /lib/sqlalchemy/testing
parent77d8853623037972d851f1ff053f98eb1a236084 (diff)
downloadsqlalchemy-9b2cd1ede5951fff7180d64bb39aa3a601ec1900.tar.gz
Remove dispose warning on async engines when running tests
Co-authored-by: Mike Bayer <mike_mp@zzzcomputing.com> Change-Id: Ia3357959ed286dc7d2ce264b5ddcadf309351ff7
Diffstat (limited to 'lib/sqlalchemy/testing')
-rw-r--r--lib/sqlalchemy/testing/engines.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/lib/sqlalchemy/testing/engines.py b/lib/sqlalchemy/testing/engines.py
index 8ce70d9c2..0310825bc 100644
--- a/lib/sqlalchemy/testing/engines.py
+++ b/lib/sqlalchemy/testing/engines.py
@@ -19,6 +19,7 @@ from .util import decorator
from .util import gc_collect
from .. import event
from .. import pool
+from ..util import await_only
from ..util.typing import Literal
@@ -105,7 +106,10 @@ class ConnectionKiller:
and proxy_ref._pool is rec.pool
):
self._safe(proxy_ref._checkin)
- rec.dispose()
+ if hasattr(rec, "sync_engine"):
+ await_only(rec.dispose())
+ else:
+ rec.dispose()
eng.clear()
def after_test(self):
@@ -332,6 +336,7 @@ def testing_engine(
from sqlalchemy.pool import StaticPool
if config.db is not None and isinstance(config.db.pool, StaticPool):
+ use_reaper = False
engine.pool._transfer_from(config.db.pool)
if scope == "global":