def test_get_pending_migrations() -> None:
runner = Runner()
total_migrations = get_total_migration_count()
assert len(runner._get_pending_migrations()) == total_migrations
runner.run_migration(MigrationKey(MigrationGroup.SYSTEM,
"0001_migrations"))
assert len(runner._get_pending_migrations()) == total_migrations - 1
def test_run_all() -> None:
runner = Runner()
assert len(runner._get_pending_migrations()) == get_total_migration_count()
with pytest.raises(MigrationError):
runner.run_all(force=False)
runner.run_all(force=True)
assert runner._get_pending_migrations() == []
def test_reverse_all() -> None:
runner = Runner()
all_migrations = runner._get_pending_migrations()
runner.run_all(force=True)
for migration in reversed(all_migrations):
runner.reverse_migration(migration, force=True)
connection = get_cluster(StorageSetKey.MIGRATIONS).get_query_connection(
ClickhouseClientSettings.MIGRATE)
assert connection.execute(
"SHOW TABLES") == [], "All tables should be deleted"