"""Unit tests for the geoip mmdb auto-update in `download.py`.

daijro/geoip-all-in-one rebuilds weekly and keeps only the latest ~2 releases,
so `ensure_geoip_mmdb` never pins a tag: on every call it resolves the CURRENT
latest tag (from the `releases/latest/download` permalink, no GitHub API) and
downloads it only when it differs from the cache. These tests mock the cache
root, the tag resolver, and the per-tag download so nothing touches the network.
"""
import pytest

import invisible_playwright.download as dl


@pytest.fixture
def cache(tmp_path, monkeypatch):
    """Point the cache at tmp_path and clear the env override."""
    monkeypatch.setattr(dl, "cache_root", lambda: tmp_path)
    monkeypatch.delenv("STEALTHFOX_GEOIP_MMDB", raising=False)
    return tmp_path


def _make_cached(root, tag, name=dl.GEOIP_MMDB_NAME):
    d = root / "geoip" / tag
    d.mkdir(parents=True, exist_ok=True)
    f = d / name
    f.write_bytes(b"FAKE-MMDB")
    return f


# ──────────────────────────────────────────────────────────────────────
#  env override
# ──────────────────────────────────────────────────────────────────────
@pytest.mark.unit
def test_env_override_returns_file(tmp_path, monkeypatch):
    f = tmp_path / "mine.mmdb"
    f.write_bytes(b"X")
    monkeypatch.setenv("STEALTHFOX_GEOIP_MMDB", str(f))
    assert dl.ensure_geoip_mmdb() == f


@pytest.mark.unit
def test_env_override_missing_raises(tmp_path, monkeypatch):
    monkeypatch.setenv("STEALTHFOX_GEOIP_MMDB", str(tmp_path / "nope.mmdb"))
    with pytest.raises(RuntimeError):
        dl.ensure_geoip_mmdb()


# ──────────────────────────────────────────────────────────────────────
#  every-launch latest check
# ──────────────────────────────────────────────────────────────────────
@pytest.mark.unit
def test_cache_is_latest_no_download(cache, monkeypatch):
    f = _make_cached(cache, "2026.06.17")
    monkeypatch.setattr(dl, "_resolve_latest_geoip_tag", lambda: "2026.06.17")
    monkeypatch.setattr(dl, "_download_file", lambda *a, **k: (_ for _ in ()).throw(
        AssertionError("must not download when cache already on the latest tag")))
    assert dl.ensure_geoip_mmdb() == f


@pytest.mark.unit
def test_new_tag_downloads_and_prunes(cache, monkeypatch):
    old = _make_cached(cache, "2026.06.10")
    monkeypatch.setattr(dl, "_resolve_latest_geoip_tag", lambda: "2026.06.17")
    monkeypatch.setattr(dl, "_download_geoip_tag", lambda tag: _make_cached(cache, tag))
    got = dl.ensure_geoip_mmdb()
    assert got.parent.name == "2026.06.17"
    assert not old.parent.exists()  # old tag pruned
    assert got.exists()


@pytest.mark.unit
def test_cold_cache_downloads_latest(cache, monkeypatch):
    monkeypatch.setattr(dl, "_resolve_latest_geoip_tag", lambda: "2026.06.17")
    monkeypatch.setattr(dl, "_download_geoip_tag", lambda tag: _make_cached(cache, tag))
    got = dl.ensure_geoip_mmdb()
    assert got.parent.name == "2026.06.17"
    assert got.exists()


# ──────────────────────────────────────────────────────────────────────
#  offline resilience (no pinned-tag fallback — the pin rots and 404s)
# ──────────────────────────────────────────────────────────────────────
@pytest.mark.unit
def test_offline_with_cache_uses_cache(cache, monkeypatch):
    f = _make_cached(cache, "2026.06.10")
    monkeypatch.setattr(dl, "_resolve_latest_geoip_tag", lambda: None)  # offline
    monkeypatch.setattr(dl, "_download_file", lambda *a, **k: (_ for _ in ()).throw(
        AssertionError("offline → must not attempt a download")))
    assert dl.ensure_geoip_mmdb() == f  # cache reused, no raise


@pytest.mark.unit
def test_cold_cache_offline_raises(cache, monkeypatch):
    monkeypatch.setattr(dl, "_resolve_latest_geoip_tag", lambda: None)  # offline
    with pytest.raises(RuntimeError):
        dl.ensure_geoip_mmdb()


@pytest.mark.unit
def test_download_failure_with_cache_falls_back(cache, monkeypatch):
    f = _make_cached(cache, "2026.06.10")
    monkeypatch.setattr(dl, "_resolve_latest_geoip_tag", lambda: "2026.06.17")

    def boom(tag):
        raise OSError("transient download failure")

    monkeypatch.setattr(dl, "_download_geoip_tag", boom)
    assert dl.ensure_geoip_mmdb() == f  # keeps the old cache rather than failing


# ──────────────────────────────────────────────────────────────────────
#  latest-tag resolution via the permalink 302 (no GitHub API)
# ──────────────────────────────────────────────────────────────────────
@pytest.mark.unit
def test_resolve_tag_from_permalink_redirect(monkeypatch):
    class _Resp:
        headers = {"Location":
                   "https://github.com/daijro/geoip-all-in-one/releases/download/"
                   "2026.06.17/geoip-aio-all.mmdb.zip"}

    monkeypatch.setattr(dl.requests, "head", lambda *a, **k: _Resp())
    assert dl._resolve_latest_geoip_tag() == "2026.06.17"


@pytest.mark.unit
def test_resolve_tag_permalink_fails_falls_back_to_api(monkeypatch):
    def head_boom(*a, **k):
        raise OSError("no network for HEAD")

    monkeypatch.setattr(dl.requests, "head", head_boom)
    monkeypatch.setattr(dl, "_latest_geoip_tag_api", lambda: "2026.06.17")
    assert dl._resolve_latest_geoip_tag() == "2026.06.17"


@pytest.mark.unit
def test_resolve_tag_all_fail_returns_none(monkeypatch):
    def boom(*a, **k):
        raise OSError("offline")

    monkeypatch.setattr(dl.requests, "head", boom)
    monkeypatch.setattr(dl, "_latest_geoip_tag_api", boom)
    assert dl._resolve_latest_geoip_tag() is None