mirror of
https://github.com/feder-cr/invisible_playwright.git
synced 2026-06-19 09:08:06 +02:00
8f4b20a19d
- BINARY_VERSION -> firefox-12 (self-calibrating font widths, per-family canvas distinctness, render-noise that preserves solid reference renders). - font_pool: the standard Windows fonts (Calibri, Franklin Gothic, Gadugi, Javanese Text, Myanmar Text) move from the per-profile optional set to core, so they are always present and the detected font set matches a real Windows install on every host. Defensive dedup in derive_font_prefs. - GPU persona applied on every platform (Linux/macOS present a coherent Windows GPU + WebGL params); pool re-rooted on a real-device GPU mix; render seeds recalibrated. - prefs: emit absolute per-family font widths that the binary self-calibrates. - geoip: always pull the latest mmdb via the releases/latest permalink, checked each launch, offline-safe (no pinned tag that can 404). - tests: per-font canvas distinctness, solid-readback purity under render-noise, always-present standard-font invariant, no duplicate families.
142 lines
6.5 KiB
Python
142 lines
6.5 KiB
Python
"""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
|