fix(backend): repair 0006 migration tests for alembic 1.18.x proxy api

Replace deprecated alembic.op._proxy assignment with
ops._install_proxy() / ops._remove_proxy() pattern required
by Alembic >= 1.13. Consistent with test_migration_0007_c2.py.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Knacky
2026-06-10 20:02:46 +02:00
parent 873e52a2a1
commit b83316f715

View File

@@ -92,27 +92,22 @@ def _run_upgrade(engine, migration_mod):
with engine.begin() as conn: with engine.begin() as conn:
ctx = MigrationContext.configure(conn) ctx = MigrationContext.configure(conn)
ops = Operations(ctx) ops = Operations(ctx)
# Patch op module for the migration ops._install_proxy() # type: ignore[attr-defined]
import alembic.op as alembic_op
original_proxy = alembic_op._proxy # type: ignore[attr-defined]
alembic_op._proxy = ops # type: ignore[attr-defined]
try: try:
migration_mod.upgrade() migration_mod.upgrade()
finally: finally:
alembic_op._proxy = original_proxy # type: ignore[attr-defined] ops._remove_proxy() # type: ignore[attr-defined]
def _run_downgrade(engine, migration_mod): def _run_downgrade(engine, migration_mod):
with engine.begin() as conn: with engine.begin() as conn:
ctx = MigrationContext.configure(conn) ctx = MigrationContext.configure(conn)
ops = Operations(ctx) ops = Operations(ctx)
import alembic.op as alembic_op ops._install_proxy() # type: ignore[attr-defined]
original_proxy = alembic_op._proxy # type: ignore[attr-defined]
alembic_op._proxy = ops # type: ignore[attr-defined]
try: try:
migration_mod.downgrade() migration_mod.downgrade()
finally: finally:
alembic_op._proxy = original_proxy # type: ignore[attr-defined] ops._remove_proxy() # type: ignore[attr-defined]
class TestMigration0006Upgrade: class TestMigration0006Upgrade: