from pathlib import Path

from unlimited_ocr_rdna4.constants import MODEL_REVISION
from unlimited_ocr_rdna4.paths import default_cache_dir, default_model_dir


def test_xdg_defaults(monkeypatch) -> None:
    monkeypatch.delenv("UNLIMITED_OCR_MODEL_DIR", raising=False)
    monkeypatch.setenv("XDG_DATA_HOME", "/data-home")
    monkeypatch.setenv("XDG_CACHE_HOME", "/cache-home")
    assert default_model_dir() == Path("/data-home/unlimited-ocr-rdna4/models") / MODEL_REVISION
    assert default_cache_dir() == Path("/cache-home/unlimited-ocr-rdna4")


def test_model_dir_override(monkeypatch) -> None:
    monkeypatch.setenv("UNLIMITED_OCR_MODEL_DIR", "/models/pinned")
    assert default_model_dir() == Path("/models/pinned")