"""Unit tests for the WC2 canonical registry (runner/harness/canonical.py).

Pure parts: enrollment (recipe_meta.WARM_CANONICAL), stable domain, registry read/write. The
data-warm lifecycle (deploy/undeploy-keep-volume/seed) is integration, proven live on a real recipe
canonical (W1).
"""

from __future__ import annotations

import json
import os
import sys

sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "runner"))
from harness import canonical, warm  # noqa: E402


def test_canonical_domain():
    assert canonical.canonical_domain("custom-html") == "warm-custom-html.ci.commoninternet.net"
    assert warm.stable_domain("keycloak") == "warm-keycloak.ci.commoninternet.net"
    # stable_domain matches the live-warm keycloak's mapping (no divergence)
    assert warm.stable_domain("keycloak") == warm.WARM_DOMAINS["keycloak"]


def test_is_enrolled_missing_meta_false(tmp_path, monkeypatch):
    # A recipe with no recipe_meta.py is not enrolled.
    assert canonical.is_enrolled("definitely-not-a-recipe-xyz") is False


def test_is_enrolled_reads_flag(tmp_path, monkeypatch):
    # Point the module's tests/<recipe>/ lookup at a temp recipe by monkeypatching __file__ dir.
    recipe = "tmpwarm"
    tests_dir = tmp_path / "tests" / recipe
    tests_dir.mkdir(parents=True)
    (tests_dir / "recipe_meta.py").write_text("WARM_CANONICAL = True\n")
    # canonical.is_enrolled builds the path from canonical.__file__/../../tests/<recipe>; emulate by
    # creating the layout under a fake harness dir and pointing __file__ there.
    fake_harness = tmp_path / "runner" / "harness"
    fake_harness.mkdir(parents=True)
    monkeypatch.setattr(canonical, "__file__", str(fake_harness / "canonical.py"))
    assert canonical.is_enrolled(recipe) is True
    (tests_dir / "recipe_meta.py").write_text("WARM_CANONICAL = False\n")
    assert canonical.is_enrolled(recipe) is False
    (tests_dir / "recipe_meta.py").write_text("DEPS = ['keycloak']\n")  # flag absent
    assert canonical.is_enrolled(recipe) is False


def test_registry_roundtrip(tmp_path, monkeypatch):
    monkeypatch.setenv("CCCI_WARM_ROOT", str(tmp_path))
    assert canonical.read_registry("custom-html") is None
    rec = canonical.write_registry("custom-html", version="1.10.0+x", commit="abc123", status="idle")
    assert rec["domain"] == "warm-custom-html.ci.commoninternet.net"
    assert rec["version"] == "1.10.0+x" and rec["commit"] == "abc123" and rec["status"] == "idle"
    back = canonical.read_registry("custom-html")
    assert back == rec
    # atomic overwrite to a new status
    canonical.write_registry("custom-html", version="1.10.0+x", commit="abc123", status="warm")
    assert canonical.read_registry("custom-html")["status"] == "warm"
    # the file is valid JSON on disk
    with open(canonical.registry_path("custom-html")) as f:
        assert json.load(f)["status"] == "warm"


def test_enrolled_recipes_scans_meta(tmp_path, monkeypatch):
    # enrolled_recipes() lists recipes whose tests/<r>/recipe_meta.py sets WARM_CANONICAL=True.
    fake_harness = tmp_path / "runner" / "harness"
    fake_harness.mkdir(parents=True)
    monkeypatch.setattr(canonical, "__file__", str(fake_harness / "canonical.py"))
    for name, body in (("aaa", "WARM_CANONICAL = True\n"),
                       ("bbb", "DEPS=['x']\n"),
                       ("ccc", "WARM_CANONICAL = True\n")):
        d = tmp_path / "tests" / name
        d.mkdir(parents=True)
        (d / "recipe_meta.py").write_text(body)
    (tmp_path / "tests" / "ddd").mkdir(parents=True)  # no recipe_meta.py at all
    assert canonical.enrolled_recipes() == ["aaa", "ccc"]


def test_prune_stale_drops_deenrolled_only(tmp_path, monkeypatch):
    # prune_stale removes <recipe>/ dirs that have a canonical.json but aren't enrolled; keeps
    # enrolled canonicals, reconciler dirs (no canonical.json), and alerts/.
    monkeypatch.setenv("CCCI_WARM_ROOT", str(tmp_path))
    monkeypatch.setattr(canonical, "enrolled_recipes", lambda: ["keepme"])
    monkeypatch.setattr(canonical.warmsnap, "stack_volumes", lambda d: [])  # no docker in unit
    # enrolled canonical (keep), de-enrolled canonical (prune), reconciler dir (keep), alerts (keep)
    for name in ("keepme", "gone"):
        (tmp_path / name).mkdir()
        (tmp_path / name / "canonical.json").write_text('{"recipe":"%s"}' % name)
    (tmp_path / "keycloak").mkdir(); (tmp_path / "keycloak" / "last_good").write_text("v1")  # reconciler
    (tmp_path / "alerts").mkdir()
    pruned = canonical.prune_stale()
    assert pruned == ["gone"]
    assert not (tmp_path / "gone").exists()
    assert (tmp_path / "keepme").exists()
    assert (tmp_path / "keycloak").exists()   # no canonical.json → not a canonical → kept
    assert (tmp_path / "alerts").exists()