from __future__ import annotations

import types

import ariadne.migrate as migrate_module


class DummyDatabase:
    def __init__(self, name: str):
        self.name = name
        self.calls: list[tuple[int, bool, bool]] = []
        self.closed = False

    def migrate(self, lock_id: int, *, include_ariadne_tables: bool, include_access_requests: bool) -> None:
        self.calls.append((lock_id, include_ariadne_tables, include_access_requests))

    def close(self) -> None:
        self.closed = True


def test_build_db_uses_settings(monkeypatch) -> None:
    captured = {}
    monkeypatch.setattr(
        migrate_module,
        "settings",
        types.SimpleNamespace(
            ariadne_db_pool_min=1,
            ariadne_db_pool_max=3,
            ariadne_db_connect_timeout_sec=5,
            ariadne_db_lock_timeout_sec=7,
            ariadne_db_statement_timeout_sec=11,
            ariadne_db_idle_in_tx_timeout_sec=13,
        ),
    )

    def fake_database(dsn, config):
        captured["dsn"] = dsn
        captured["config"] = config
        return DummyDatabase("ariadne")

    monkeypatch.setattr(migrate_module, "Database", fake_database)
    migrate_module._build_db("postgresql://db", "ariadne_migrate")

    assert captured["dsn"] == "postgresql://db"
    assert captured["config"].application_name == "ariadne_migrate"
    assert captured["config"].lock_timeout_sec == 7


def test_main_returns_when_migrations_disabled(monkeypatch) -> None:
    monkeypatch.setattr(
        migrate_module,
        "settings",
        types.SimpleNamespace(ariadne_run_migrations=False),
    )
    monkeypatch.setattr(migrate_module, "_build_db", lambda *_args, **_kwargs: (_ for _ in ()).throw(RuntimeError("should not build")))
    migrate_module.main()


def test_main_runs_ariadne_and_portal_migrations(monkeypatch) -> None:
    ariadne_db = DummyDatabase("ariadne")
    portal_db = DummyDatabase("portal")
    databases = [ariadne_db, portal_db]
    monkeypatch.setattr(
        migrate_module,
        "settings",
        types.SimpleNamespace(
            ariadne_run_migrations=True,
            ariadne_database_url="postgresql://ariadne",
            portal_database_url="postgresql://portal",
        ),
    )
    monkeypatch.setattr(migrate_module, "_build_db", lambda *_args, **_kwargs: databases.pop(0))

    migrate_module.main()

    assert ariadne_db.calls == [(migrate_module.ARIADNE_MIGRATION_LOCK_ID, True, False)]
    assert portal_db.calls == [(migrate_module.PORTAL_MIGRATION_LOCK_ID, False, True)]
    assert ariadne_db.closed is True
    assert portal_db.closed is True