diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2021-11-04 23:49:15 -0400 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2021-11-04 23:50:11 -0400 |
| commit | 427fcde8c9b4eb62fa2861093de90c8df2e3f305 (patch) | |
| tree | a718220228494c6718530343b3df5bb7c8e25428 /alembic/util | |
| parent | 23f05cd8391bd3600578f14db7b670841f527290 (diff) | |
| download | alembic-427fcde8c9b4eb62fa2861093de90c8df2e3f305.tar.gz | |
rollback connection before teardown
this test needs to clear out the transaction
if running on SQLAlchemy 2.0.
Change-Id: Ic3f76433ba9b9ea5485a81caae745d3c7f2a4ee8
Diffstat (limited to 'alembic/util')
| -rw-r--r-- | alembic/util/sqla_compat.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/alembic/util/sqla_compat.py b/alembic/util/sqla_compat.py index 21dee89..6f98af4 100644 --- a/alembic/util/sqla_compat.py +++ b/alembic/util/sqla_compat.py @@ -122,6 +122,14 @@ def _safe_begin_connection_transaction( return connection.begin() +def _safe_rollback_connection_transaction( + connection: "Connection", +) -> "Transaction": + transaction = _get_connection_transaction(connection) + if transaction: + transaction.rollback() + + def _get_connection_in_transaction(connection: Optional["Connection"]) -> bool: try: in_transaction = connection.in_transaction # type: ignore |