"""
Tests for Scribe migration 006.

Runs against `core_api_test` only (see conftest.py — skip when
TEST_DATABASE_URL is unset, hard failure if it resolves to `core_api`).
Self-contained: starts and ends at head regardless of what ran before it.
"""
import pytest
from sqlalchemy import text
from sqlalchemy.ext.asyncio import create_async_engine
from sqlalchemy.pool import NullPool

from tests.conftest import run_alembic


async def _table_exists(url: str, table: str) -> bool:
    asyncpg_url = url.replace("postgresql://", "postgresql+asyncpg://", 1)
    engine = create_async_engine(asyncpg_url, poolclass=NullPool)
    try:
        async with engine.connect() as conn:
            result = await conn.execute(
                text("SELECT to_regclass(:name) IS NOT NULL AS present"),
                {"name": table},
            )
            return bool(result.scalar())
    finally:
        await engine.dispose()


class TestMigration006RoundTrip:
    """Migration 006 must upgrade and downgrade cleanly (dossier verification #1)."""

    @pytest.mark.asyncio
    async def test_upgrade_creates_scribe_tables(self, test_database_url):
        result = run_alembic(test_database_url, "upgrade", "head")
        assert result.returncode == 0, result.stderr

        for table in ("scribe_addresses", "scribe_messages", "scribe_mailbox_entries"):
            assert await _table_exists(test_database_url, table), f"{table} missing after upgrade"

    @pytest.mark.asyncio
    async def test_downgrade_then_upgrade_round_trips_cleanly(self, test_database_url):
        baseline = run_alembic(test_database_url, "upgrade", "head")
        assert baseline.returncode == 0, baseline.stderr

        down = run_alembic(test_database_url, "downgrade", "005")
        assert down.returncode == 0, down.stderr
        assert not await _table_exists(test_database_url, "scribe_addresses")
        assert not await _table_exists(test_database_url, "scribe_messages")
        assert not await _table_exists(test_database_url, "scribe_mailbox_entries")
        # Book tables, created by the migration below this one, must survive
        # a downgrade of 006 — proves 006 only touches its own tables.
        assert await _table_exists(test_database_url, "book_records")

        # Leave the schema at head for the rest of the suite.
        up = run_alembic(test_database_url, "upgrade", "006")
        assert up.returncode == 0, up.stderr
        assert await _table_exists(test_database_url, "scribe_addresses")
        assert await _table_exists(test_database_url, "scribe_messages")
        assert await _table_exists(test_database_url, "scribe_mailbox_entries")