"""Tests for the migration runner."""

from migrations.migrate import get_applied, get_connection, migrate_down, migrate_up, verify


def test_migrate_up_and_status(tmp_path):
    """Should apply all migrations and track them."""
    db_path = str(tmp_path / "test.db")
    count = migrate_up(db_path)
    assert count >= 2  # At least the 2 existing migrations

    conn = get_connection(db_path)
    applied = get_applied(conn)
    assert "001_initial_schema" in applied
    assert "002_add_sessions_and_audit" in applied


def test_migrate_down(tmp_path):
    """Should rollback the last migration."""
    db_path = str(tmp_path / "test.db")
    migrate_up(db_path)
    result = migrate_down(db_path)
    assert result is True

    conn = get_connection(db_path)
    applied = get_applied(conn)
    assert "002_add_sessions_and_audit" not in applied
    assert "001_initial_schema" in applied


def test_verify():
    """Verify should apply migrations to a temp DB cleanly."""
    result = verify()
    assert result is True