import time

from fixtures.neon_fixtures import NeonEnv


def test_migrations(neon_simple_env: NeonEnv):
    env = neon_simple_env
    env.neon_cli.create_branch("test_migrations", "empty")

    endpoint = env.endpoints.create("test_migrations")
    log_path = endpoint.endpoint_path() / "compute.log"

    endpoint.respec(skip_pg_catalog_updates=False, features=["migrations"])
    endpoint.start()

    time.sleep(1)  # Sleep to let migrations run

    with endpoint.cursor() as cur:
        cur.execute("SELECT id FROM neon_migration.migration_id")
        migration_id = cur.fetchall()
        assert migration_id[0][0] == 2

    with open(log_path, "r") as log_file:
        logs = log_file.read()
        assert "INFO handle_migrations: Ran 2 migrations" in logs

    endpoint.stop()
    endpoint.start()
    time.sleep(1)  # Sleep to let migrations run
    with endpoint.cursor() as cur:
        cur.execute("SELECT id FROM neon_migration.migration_id")
        migration_id = cur.fetchall()
        assert migration_id[0][0] == 2

    with open(log_path, "r") as log_file:
        logs = log_file.read()
        assert "INFO handle_migrations: Ran 0 migrations" in logs