test_migrations.py 625 B

123456789101112131415161718
  1. from pathlib import Path
  2. from alembic.config import Config
  3. from alembic.script import ScriptDirectory
  4. def test_each_database_has_the_expected_independent_migration_head() -> None:
  5. backend_dir = Path(__file__).parents[1]
  6. expected_heads = {
  7. "alembic_core": "core_0011",
  8. "alembic_agent": "agent_0002",
  9. "alembic_analytics": "analytics_0001",
  10. }
  11. for section, expected_head in expected_heads.items():
  12. config = Config(backend_dir / "alembic.ini", ini_section=section)
  13. scripts = ScriptDirectory.from_config(config)
  14. assert scripts.get_current_head() == expected_head