central/tests/test_satpass_predict.py

"""Tests for the v0.11.1 satpass_predict adapter.

Deterministic via a fixed ISS TLE + fixed observer + pinned reference time.
The TLE comes from the v0.11.0 stations fixture (epoch 2026-06-08T19:17 UTC);
reference time pinned at 2026-06-09T07:00 UTC; observer is Treasure Valley
(43.6, -116.2, 0m elev). This combination produces a known ISS pass starting
at ~15:36 UTC the same day (verified via the sgp4 sanity script during Phase
A of v0.11.1).
"""

from __future__ import annotations

from datetime import datetime, timezone
from pathlib import Path
from unittest.mock import AsyncMock, MagicMock

import pytest

from central.adapter import SourceAdapter
from central.adapters.satpass_predict import (
    Observer,
    SatpassPredictAdapter,
    SatpassPredictSettings,
    _gmst_rad,
    _next_passes,
    _observer_ecef,
    _severity_from_elev,
    _topocentric_az_el,
)
from central.config_models import AdapterConfig


# Live TLE from the v0.11.0 stations fixture, ISS (NORAD 25544).
_ISS_L1 = "1 25544U 98067A   26159.80410962  .00007129  00000+0  13425-3 0  9999"
_ISS_L2 = "2 25544  51.6336 341.5878 0006923 148.5365 211.6039 15.49672912570453"

# Pinned observer + reference time.
_OBS = Observer(name="Treasure Valley", slug="treasure-valley",
                state="ID", lat=43.6, lon=-116.2, elev_m=0.0)
_REF = datetime(2026, 6, 9, 7, 0, 0, tzinfo=timezone.utc)


@pytest.fixture
def adapter(tmp_path: Path) -> SatpassPredictAdapter:
    cfg = AdapterConfig(
        name="satpass_predict",
        enabled=True,
        cadence_s=3600,
        settings={"observers": [_OBS.model_dump()],
                  "min_elevation_deg": 10.0, "horizon_hours": 24},
        updated_at=datetime.now(timezone.utc),
    )
    return SatpassPredictAdapter(cfg, MagicMock(), tmp_path / "cursors.db")


# --- Pure math helpers ------------------------------------------------------


def test_gmst_rad_returns_radians_in_canonical_range():
    """GMST output must wrap into [0, 2π)."""
    import math as m
    val = _gmst_rad(2460835.0, 0.5)  # arbitrary post-2000 JD
    assert 0.0 <= val < 2.0 * m.pi


def test_observer_ecef_for_north_pole_and_equator():
    """Sanity: north pole sits on z-axis; equator at lon=0 sits on x-axis."""
    pole = _observer_ecef(90.0, 0.0, 0.0)
    assert abs(pole[0]) < 1e-6 and abs(pole[1]) < 1e-6
    assert pole[2] > 6378.0  # ~6378.137 km

    eq_zero = _observer_ecef(0.0, 0.0, 0.0)
    assert eq_zero[0] > 6378.0 and abs(eq_zero[1]) < 1e-6 and abs(eq_zero[2]) < 1e-6


def test_topocentric_zenith_satellite_returns_90_elevation():
    """A satellite directly overhead must read elevation 90°, any azimuth."""
    obs_lat, obs_lon = 43.6, -116.2
    obs = _observer_ecef(obs_lat, obs_lon, 0.0)
    # 400km straight up = scale observer position vector by (R+400)/R
    import math as m
    r_obs = m.sqrt(sum(c * c for c in obs))
    r_sat = r_obs + 400.0
    scale = r_sat / r_obs
    sat_ecef = (obs[0] * scale, obs[1] * scale, obs[2] * scale)
    az, el = _topocentric_az_el(sat_ecef, obs, obs_lat, obs_lon)
    assert abs(el - 90.0) < 0.01, f"expected zenith elevation, got {el}"


def test_topocentric_below_horizon_returns_negative_elevation():
    """Satellite on the opposite side of the earth = below horizon."""
    obs = _observer_ecef(0.0, 0.0, 0.0)  # equator, prime meridian
    antipode = (-obs[0] * 2.0, 0.0, 0.0)  # other side, well below
    _, el = _topocentric_az_el(antipode, obs, 0.0, 0.0)
    assert el < -10.0


# --- Severity bucketing -----------------------------------------------------


@pytest.mark.parametrize("max_elev, expected", [
    (90.0, 4),    # zenith
    (60.0, 4),    # boundary -> 4
    (59.99, 3),
    (30.0, 3),    # boundary -> 3
    (29.99, 2),
    (10.0, 2),    # boundary -> 2 (gate threshold; emit)
    (9.99, 1),    # below gate -> 1 (should never emit in practice)
    (0.0, 1),
])
def test_severity_from_elev_buckets(max_elev, expected):
    assert _severity_from_elev(max_elev) == expected


# --- Pass detection (the load-bearing math test) ---------------------------


def test_iss_next_pass_over_treasure_valley_is_chronologically_sane():
    """Pinned TLE + observer + ref time produces ONE known ISS pass in 24h.
    AOS < peak < LOS, max_elev in (10, 90), positive duration."""
    passes = _next_passes(
        _ISS_L1, _ISS_L2, _OBS,
        ref_time=_REF, horizon_hours=24, min_elevation_deg=10.0,
    )
    assert len(passes) > 0, "expected at least one ISS pass over Boise in next 24h"
    p = passes[0]
    assert p["aos"] < p["peak"] <= p["los"]
    assert 10.0 < p["max_elev_deg"] < 90.0
    assert (p["los"] - p["aos"]).total_seconds() > 0
    # And the pass must lie inside the 24h horizon (ref + 24h = 2026-06-10T07:00 UTC).
    horizon_end = datetime(2026, 6, 10, 7, 0, 0, tzinfo=timezone.utc)
    assert p["aos"] >= _REF
    assert p["los"] <= horizon_end


def test_iss_pass_has_plausible_azimuths():
    """Azimuth at AOS and LOS should be valid 0-360° readings."""
    passes = _next_passes(
        _ISS_L1, _ISS_L2, _OBS,
        ref_time=_REF, horizon_hours=24, min_elevation_deg=10.0,
    )
    p = passes[0]
    assert 0.0 <= p["aos_az"] < 360.0
    # los_az may be None if the pass ran to the horizon edge, but for ISS
    # against the pinned ref it completes within 24h.
    if p["los_az"] is not None:
        assert 0.0 <= p["los_az"] < 360.0


def test_min_elevation_gate_filters_lower_passes():
    """Same TLE, raise the gate to 80° -- now zero passes (ISS at 51.6°
    inclination from latitude 43.6° can't reach 80° often)."""
    passes_low = _next_passes(_ISS_L1, _ISS_L2, _OBS, _REF, 24, 10.0)
    passes_high = _next_passes(_ISS_L1, _ISS_L2, _OBS, _REF, 24, 80.0)
    assert len(passes_low) > 0
    # No 80°+ passes today (would require near-overhead crossing).
    for p in passes_high:
        assert p["max_elev_deg"] >= 80.0


def test_malformed_tle_returns_empty_pass_list():
    """A garbage TLE must not crash; just yield no passes."""
    passes = _next_passes("not a tle", "also not", _OBS, _REF, 24, 10.0)
    assert passes == []


# --- _build_event / _pass_to_event ------------------------------------------


def _row_for_iss():
    return {
        "norad_id": 25544, "satellite_name": "ISS (ZARYA)",
        "tle_line1": _ISS_L1, "tle_line2": _ISS_L2,
        "tle_epoch": "2026-06-08T19:17:55+00:00",
    }


def test_pass_event_shape(adapter):
    passes = _next_passes(_ISS_L1, _ISS_L2, _OBS, _REF, 24, 10.0)
    assert passes
    ev = adapter._pass_to_event(passes[0], _row_for_iss(), _OBS)
    # Identity
    assert ev.adapter == "satpass_predict"
    assert ev.category == "pass.satpass_predict"
    # Dedup id shape: {observer_slug}:{norad_id}:{aos_iso}
    assert ev.id.startswith("treasure-valley:25544:")
    assert ":2026-06-" in ev.id  # AOS within the same UTC day window
    # Severity bucket maps from peak elevation
    assert ev.severity == _severity_from_elev(passes[0]["max_elev_deg"])
    # Geo: centroid at the observer point
    assert ev.geo.centroid == (-116.2, 43.6)
    assert ev.geo.primary_region == "US-ID"
    # data fields per spec
    assert ev.data["observer_name"] == "Treasure Valley"
    assert ev.data["observer_slug"] == "treasure-valley"
    assert ev.data["observer_state"] == "ID"
    assert ev.data["norad_id"] == 25544
    assert ev.data["satellite_name"] == "ISS (ZARYA)"
    assert ev.data["max_elevation_deg"] == round(passes[0]["max_elev_deg"], 2)
    assert ev.data["duration_s"] > 0
    assert ev.data["tle_epoch"] == "2026-06-08T19:17:55+00:00"


def test_subject_for_uses_observer_state_and_slug(adapter):
    passes = _next_passes(_ISS_L1, _ISS_L2, _OBS, _REF, 24, 10.0)
    ev = adapter._pass_to_event(passes[0], _row_for_iss(), _OBS)
    assert adapter.subject_for(ev) == "central.sat.pass.us.id.treasure-valley"


def test_subject_for_falls_back_when_state_or_slug_missing(adapter):
    from central.models import Event, Geo
    ev = Event(
        id="x", adapter="satpass_predict", category="pass.satpass_predict",
        time=datetime.now(timezone.utc), severity=2, geo=Geo(), data={},
    )
    assert adapter.subject_for(ev) == "central.sat.pass.us.unknown.unknown"


# --- poll() integration with mocked pool ------------------------------------


def _mock_pool_returning(rows):
    """Build a MagicMock pool that yields ``rows`` from any SELECT."""
    pool = MagicMock()
    conn = MagicMock()
    conn.fetch = AsyncMock(return_value=rows)
    pool.acquire.return_value.__aenter__ = AsyncMock(return_value=conn)
    pool.acquire.return_value.__aexit__ = AsyncMock(return_value=None)
    return pool


@pytest.mark.asyncio
async def test_poll_empty_tles_table_logs_and_yields_zero(tmp_path):
    """v0.11.1 spec: empty TLE table -> 0 events, INFO log, no exception."""
    cfg = AdapterConfig(
        name="satpass_predict", enabled=True, cadence_s=3600,
        settings={"observers": [_OBS.model_dump()],
                  "min_elevation_deg": 10.0, "horizon_hours": 24},
        updated_at=datetime.now(timezone.utc),
    )
    config_store = MagicMock()
    config_store.get_pool.return_value = _mock_pool_returning([])
    adapter = SatpassPredictAdapter(cfg, config_store, tmp_path / "cursors.db")
    await adapter.startup()
    try:
        events = [e async for e in adapter.poll()]
        assert events == []
    finally:
        await adapter.shutdown()


@pytest.mark.asyncio
async def test_poll_multi_observer_yields_per_observer_pass_list(tmp_path):
    """Two observers in settings → each observer gets its own pass list against
    the same TLE. Boise (43.6, -116.2) and Salt Lake City (40.76, -111.89)
    both see ISS but with slightly different AOS times -> different events."""
    boise = _OBS
    slc = Observer(name="Salt Lake City", slug="slc",
                   state="UT", lat=40.76, lon=-111.89, elev_m=0.0)
    cfg = AdapterConfig(
        name="satpass_predict", enabled=True, cadence_s=3600,
        settings={"observers": [boise.model_dump(), slc.model_dump()],
                  "min_elevation_deg": 10.0, "horizon_hours": 24},
        updated_at=datetime.now(timezone.utc),
    )
    config_store = MagicMock()
    config_store.get_pool.return_value = _mock_pool_returning([_row_for_iss()])
    adapter = SatpassPredictAdapter(cfg, config_store, tmp_path / "cursors.db")
    await adapter.startup()
    try:
        events = [e async for e in adapter.poll()]
        # We don't pin counts (number of passes per 24h varies with the pinned
        # ref time), but each observer must have at least one event distinct
        # from the other.
        boise_evs = [e for e in events if e.data["observer_slug"] == "treasure-valley"]
        slc_evs = [e for e in events if e.data["observer_slug"] == "slc"]
        assert boise_evs, "no Boise passes"
        assert slc_evs, "no Salt Lake City passes"
        # Subject routing differs by state.
        assert adapter.subject_for(boise_evs[0]) == "central.sat.pass.us.id.treasure-valley"
        assert adapter.subject_for(slc_evs[0]) == "central.sat.pass.us.ut.slc"
    finally:
        await adapter.shutdown()


# --- Settings / apply_config / dedup-mixin regression ----------------------


def test_default_settings_match_spec():
    s = SatpassPredictSettings()
    assert s.min_elevation_deg == 10.0
    assert s.horizon_hours == 24
    assert len(s.observers) == 1
    assert s.observers[0].slug == "treasure-valley"


def test_inherits_dedup_mixin_from_source_adapter(tmp_path):
    """v0.9.1 regression guard."""
    assert issubclass(SatpassPredictAdapter, SourceAdapter)
    a = SatpassPredictAdapter(
        AdapterConfig(
            name="satpass_predict", enabled=False, cadence_s=3600,
            settings={}, updated_at=datetime.now(timezone.utc),
        ),
        MagicMock(),
        tmp_path / "cursors.db",
    )
    assert callable(a.is_published)
    assert callable(a.mark_published)
    assert callable(a.sweep_old_ids)


@pytest.mark.asyncio
async def test_apply_config_updates_observers_and_threshold(adapter):
    new_obs = Observer(name="Sandpoint", slug="sandpoint",
                       state="ID", lat=48.27, lon=-116.55, elev_m=600.0)
    new_cfg = AdapterConfig(
        name="satpass_predict", enabled=True, cadence_s=3600,
        settings={"observers": [new_obs.model_dump()],
                  "min_elevation_deg": 25.0, "horizon_hours": 12},
        updated_at=datetime.now(timezone.utc),
    )
    await adapter.apply_config(new_cfg)
    assert len(adapter._observers) == 1
    assert adapter._observers[0].slug == "sandpoint"
    assert adapter._min_elev == 25.0
    assert adapter._horizon_h == 12.0


# --- Stream registry + family map + GUI wiring ----------------------------


def test_central_sat_family_includes_pass_token():
    """v0.11.1: pass.* categories also route to CENTRAL_SAT."""
    from central.supervisor import STREAM_CATEGORY_DOMAINS
    assert STREAM_CATEGORY_DOMAINS["CENTRAL_SAT"] == ("tle", "pass")


def test_satpass_predict_in_space_adapter_group():
    from central.gui.routes import ADAPTER_GROUPS
    assert "satpass_predict" in ADAPTER_GROUPS["Space"]


# --- Partials render cleanly (v0.10.0 pattern) ------------------------------


def test_summary_partial_renders_cleanly_with_real_pass(adapter):
    from jinja2 import Environment, FileSystemLoader
    templates_dir = Path(__file__).parent.parent / "src" / "central" / "gui" / "templates"
    env = Environment(loader=FileSystemLoader(str(templates_dir)), autoescape=True)
    tmpl = env.get_template("_event_summaries/satpass_predict.html")
    passes = _next_passes(_ISS_L1, _ISS_L2, _OBS, _REF, 24, 10.0)
    ev = adapter._pass_to_event(passes[0], _row_for_iss(), _OBS)
    rendered = tmpl.render(event={
        "data": {"data": {"data": ev.model_dump(mode="json")["data"]}}
    }).strip()
    assert "ISS (ZARYA)" in rendered, f"got: {rendered!r}"
    assert "max elevation" in rendered
    assert "UTC" in rendered


def test_row_partial_renders_cleanly(adapter):
    from jinja2 import Environment, FileSystemLoader
    templates_dir = Path(__file__).parent.parent / "src" / "central" / "gui" / "templates"
    env = Environment(loader=FileSystemLoader(str(templates_dir)), autoescape=True)
    tmpl = env.get_template("_event_rows/satpass_predict.html")
    passes = _next_passes(_ISS_L1, _ISS_L2, _OBS, _REF, 24, 10.0)
    ev = adapter._pass_to_event(passes[0], _row_for_iss(), _OBS)
    rendered = tmpl.render(event={
        "data": {"data": {"data": ev.model_dump(mode="json")["data"]}}
    })
    assert "<dt>Satellite</dt>" in rendered and "ISS (ZARYA)" in rendered
    assert "<dt>Observer</dt>" in rendered and "Treasure Valley" in rendered
    assert "<dt>AOS (rise)</dt>" in rendered
    assert "<dt>Peak</dt>" in rendered
    assert "<dt>LOS (set)</dt>" in rendered
    assert "<dt>Duration</dt>" in rendered