summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/testing/engines.py
diff options
context:
space:
mode:
authorFederico Caselli <cfederico87@gmail.com>2021-02-06 15:17:20 +0100
committerMike Bayer <mike_mp@zzzcomputing.com>2021-03-24 11:45:39 -0400
commit502be87a0b5c7bfa28db62b4af867457cd29a5fa (patch)
treecd54dda88546e34cca1a019fd40e89d91ee703ca /lib/sqlalchemy/testing/engines.py
parent021f2ab7f5e03be3b6a5ad7f2c0e22bd93b83b18 (diff)
downloadsqlalchemy-502be87a0b5c7bfa28db62b4af867457cd29a5fa.tar.gz
Add support for aiosqlite
Added support for the aiosqlite database driver for use with the SQLAlchemy asyncio extension. Fixes: #5920 Change-Id: Id11a320516a44e886a6f518d2866a0f992413e55
Diffstat (limited to 'lib/sqlalchemy/testing/engines.py')
-rw-r--r--lib/sqlalchemy/testing/engines.py14
1 files changed, 13 insertions, 1 deletions
diff --git a/lib/sqlalchemy/testing/engines.py b/lib/sqlalchemy/testing/engines.py
index a313c298a..3faf96857 100644
--- a/lib/sqlalchemy/testing/engines.py
+++ b/lib/sqlalchemy/testing/engines.py
@@ -266,7 +266,13 @@ def reconnecting_engine(url=None, options=None):
return engine
-def testing_engine(url=None, options=None, future=None, asyncio=False):
+def testing_engine(
+ url=None,
+ options=None,
+ future=None,
+ asyncio=False,
+ transfer_staticpool=False,
+):
"""Produce an engine configured by --options with optional overrides."""
if asyncio:
@@ -300,6 +306,12 @@ def testing_engine(url=None, options=None, future=None, asyncio=False):
engine = create_engine(url, **options)
+ if transfer_staticpool:
+ from sqlalchemy.pool import StaticPool
+
+ if config.db is not None and isinstance(config.db.pool, StaticPool):
+ engine.pool._transfer_from(config.db.pool)
+
if scope == "global":
if asyncio:
engine.sync_engine._has_events = True