pytest-alembic

import pytest from alembic.config import Config from sqlalchemy import create_engine from pytest_alembic.runner import AlembicRunner # conftest.py # This file should be in your tests directory or root project directory @pytest.fixture(scope="session") def alembic_config(): """Override this fixture to provide your alembic configuration.""" # Adjust 'alembic.ini' and 'migrations' to your project's paths config = Config("alembic.ini") config.set_main_option("script_location", "migrations") return config @pytest.fixture(scope="session") def alembic_engine(alembic_config): """Override this fixture to provide your database engine.""" # Use an in-memory SQLite database for quick tests or a separate test database # For real applications, consider a dedicated test database service. return create_engine("sqlite:///:memory/") # test_migrations.py # This file would be in your tests directory def test_alembic_history_consistent(alembic_runner: AlembicRunner): """Verify that the migration history is linear and consistent.""" alembic_runner.check_current_revision_is_linear() def test_all_migrations_can_upgrade_and_downgrade(alembic_runner: AlembicRunner): """Verify that all migrations can be applied and reverted successfully.""" alembic_runner.upgrade_downgrade()