mirror of
https://github.com/zvx-echo6/meshai.git
synced 2026-08-26 09:21:33 +00:00
malice
9f06930a0d* chore: excise the dead Central NATS consumer path
Central was retired and its database dropped 2026-07-15; its NATS broker no
longer exists. Verified against the live CT108 deployment: all 12 adapters
run feed_source=native, zero on central, and central.enabled is False
(default, never overridden). The consumer and its handlers were unreachable.
Removed:
- central/consumer.py and 6 dead handlers (nws, quake, swpc, nwis, avy,
incident) -- their handle_* entrypoints were reachable only from the
consumer's dispatch
- the Central wiring in main.py (init, guarded start, retry loop, stop path)
- the dead config surface: CentralConsumerConfig, EnvironmentalConfig.central,
adapter_config ("central","severity_thresholds") and its display block
- the nats-py dependency (consumer.py was its only importer)
- 19 test files that exercised only the dead path
KEPT -- these live under central/ but are imported directly by native
adapters, and deleting them would break production:
- wfigs_handler.py: firms_handler._handle_pass_boundary() calls its _render()
on the live FIRMS growth-fire path (env/firms.py -> ingest_hotspot_pixel)
- firms_handler, satpass_handler, tle_handler: split files whose handle_*
entrypoints are dead but whose engines are live. Left intact; splitting
them is separate work.
- pass_predictor, budget, idaho_gauge_sites: fully live.
The usgs_quake keys global_mag_floor / regional_mag_floor / regional_centroid
/ regional_radius_mi / broadcast_pager_alerts are NOT removed despite comments
labelling them "CENTRAL-PATH ONLY" -- notifications/gating/quake.py reads them
unconditionally in the native path. Those comments are corrected separately.
Test-count note: the suite drops ~425 tests. Most were migration PARITY tests
whose sole purpose was proving the native rewrite byte-matched the Central
handler (golden byte-parity, cross-source identity, gate-sequence replay).
With the handler deleted there is nothing left to compare against, so they
cannot exist. Native-only tests were kept and reworked where a test reached
for a central symbol incidentally. This is a real coverage loss, accepted
deliberately: the parity harness proved the refactor faithful, and git
history preserves the originals.
Suite: 1984 passed, 6 failed -- the same 6 pre-existing failures as main
(stale SCHEMA_VERSION x3, expired TLE fixtures x2, one order-dependent),
all being fixed on fix/green-test-suite. No new failures.
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
* test(nws): restore native-only golden coverage for the wire formatter
Commit ca751fb5 deleted the Central nws_handler parity harness along with
the handler itself, which took the ONLY tests that pinned formatters.nws
.format()'s literal wire output. Gate-sequence and schema-conformance
tests already survived natively; the formatter's actual rendered text did
not have any native-only regression net.
Add TestFormatterGolden to test_nws_refactor.py: 3 real-fixture cases plus
6 hand-built pathological cases mined from the deleted test_nws_handler.py
(SVR path-sampling, the "no dangling separator" regression, TOR on-ground
vs radar-indicated, FFW flood-cause detection). Every literal was verified
by temporarily restoring the pre-excision central.nws_handler._render()
from git history (ca751fb5^) in a throwaway, uncommitted script, confirming
byte-identical output against the current native format() for all 37 real
fixtures (nws/ + nws_last/) and all 9 pathological cases, then pinning the
confirmed-matching string as the literal -- not a blind snapshot of
current behavior.
quake/swpc/avalanche/hydro/incident/fire were checked and already carry
equivalent native-only golden coverage (added directly in ca751fb5), so no
changes were needed there.
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
---------
Co-authored-by: Matt Johnson <mj@k7zvx.com>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
629 lines
25 KiB
Python
629 lines
25 KiB
Python
"""Phase-1 quake refactor tests — reference implementation verification.
|
|
|
|
The Central `quake_handler` module (`_render()`, `handle_quake()`) has been
|
|
deleted — the native path is the only production path now. Pure old-vs-new
|
|
parity assertions and tests that only replayed decisions through the
|
|
deleted `handle_quake` have been removed; original diffs are preserved in
|
|
git history. What remains exercises native code directly (hand-written
|
|
expected strings are kept as regression pins on the current wire format).
|
|
|
|
Three test groups:
|
|
|
|
1. Parity (tier-b): fixture 0002 → canonical data → formatter.
|
|
Expected string is hand-written (the current correct format).
|
|
Two synthetic cases show the PAGER + update-prefix rendering explicitly.
|
|
|
|
2. Cross-source identity: native adapter builds the same canonical data
|
|
shape the formatter reads.
|
|
|
|
3. Gate-sequence: exercise gating.quake.decide() directly across a
|
|
synthetic event sequence; assert broadcast/suppress thresholds and the
|
|
commit → suppress-on-replay lifecycle.
|
|
|
|
4. Schema-conformance: env/usgs_quake.py to_event() emits all canonical keys.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from meshai.persistence import close_thread_connection, init_db
|
|
from meshai.persistence import db as persistence_db
|
|
from tests.harness.goldens import (
|
|
assert_byte_identical,
|
|
load_fixtures,
|
|
pinned_time,
|
|
)
|
|
|
|
# ── Shared clock epoch for deterministic renders ─────────────────────────────
|
|
_AT = 1_783_200_000.0 # 2026-07-03T00:00:00Z (pinned)
|
|
|
|
# ── DB fixture shared by gate-sequence tests ─────────────────────────────────
|
|
|
|
@pytest.fixture
|
|
def mem_db(monkeypatch, tmp_path):
|
|
db_path = str(tmp_path / "quake-refactor-test.sqlite")
|
|
monkeypatch.setenv("MESHAI_DB_PATH", db_path)
|
|
persistence_db._initialised.clear()
|
|
close_thread_connection()
|
|
conn = init_db()
|
|
yield conn
|
|
close_thread_connection()
|
|
persistence_db._initialised.discard(db_path)
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# 1. Parity (tier-b) — formatter renders from canonical data
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
def _make_fake_event(data: dict):
|
|
"""Minimal fake Event for calling the formatter without the full pipeline."""
|
|
class _FakeEvent:
|
|
pass
|
|
e = _FakeEvent()
|
|
e.data = data
|
|
return e
|
|
|
|
|
|
class TestFormatterParity:
|
|
"""formatter/quake.format() renders correct output from canonical data."""
|
|
|
|
def test_fixture_0002_new_format(self):
|
|
"""Fixture 0002 (M3.3 Lima Montana) → formatter output matches hand-written expected.
|
|
|
|
Fixture 0002 has alert=null and no tsunami so the tier-b additions
|
|
(PAGER line, update-prefix) are not visible. The hand-written
|
|
expected below documents the canonical format; synthetic tests below
|
|
show the tier-b additions.
|
|
"""
|
|
from meshai.notifications.formatters.quake import format as qfmt
|
|
|
|
fixtures = load_fixtures("quake")
|
|
fx = next(f for f in fixtures if f["envelope"]["id"] == "us6000t9bn")
|
|
inner = fx["envelope"]["data"]
|
|
d = inner["data"]
|
|
geo = inner["geo"]
|
|
cent = geo["centroid"] # [lon, lat]
|
|
|
|
canonical = {
|
|
"magnitude": d["magnitude"], # 3.3
|
|
"depth_km": d["depth"], # 11.169 (raw USGS key)
|
|
"lat": cent[1], # 44.46
|
|
"lon": cent[0], # -112.6108
|
|
"place": d["place"], # "19 km S of Lima, Montana"
|
|
"tsunami": bool(d["tsunami"]), # False
|
|
"pager": d.get("alert"), # None
|
|
"occurred_at": None,
|
|
"event_id": fx["envelope"]["id"],
|
|
"is_update": False,
|
|
"_severity_override": None,
|
|
"_dedup_suffix": "",
|
|
"distance_km": 160.0,
|
|
}
|
|
|
|
# Hand-written new correct format (tier-b changes are invisible here)
|
|
expected = (
|
|
"\U0001f310 New: M3.3 — 19 km S of Lima, Montana"
|
|
"\nDepth: 11 km · @ 44.460, -112.611"
|
|
)
|
|
|
|
with pinned_time(_AT):
|
|
result = qfmt(_make_fake_event(canonical), now=_AT, budget=140)
|
|
|
|
assert_byte_identical(result, expected)
|
|
|
|
def test_tier_b_pager_orange_rendered(self):
|
|
"""Tier-b ①: PAGER=orange is rendered on a 4th line."""
|
|
from meshai.notifications.formatters.quake import format as qfmt
|
|
|
|
canonical = {
|
|
"magnitude": 2.0,
|
|
"depth_km": 10.0,
|
|
"lat": 44.0,
|
|
"lon": -125.0,
|
|
"place": "Off the coast of Oregon",
|
|
"tsunami": False,
|
|
"pager": "orange", # PAGER set — triggers tier-b line
|
|
"is_update": False,
|
|
"occurred_at": None,
|
|
"event_id": "test_pager_orange",
|
|
"_severity_override": "immediate",
|
|
"_dedup_suffix": "",
|
|
"distance_km": 500.0,
|
|
}
|
|
|
|
expected_new = (
|
|
"\U0001f310 New: M2.0 — Off the coast of Oregon"
|
|
"\nDepth: 10 km · @ 44.000, -125.000"
|
|
"\n⚠️ PAGER: orange"
|
|
)
|
|
|
|
with pinned_time(_AT):
|
|
result = qfmt(_make_fake_event(canonical), now=_AT, budget=140)
|
|
|
|
assert_byte_identical(result, expected_new)
|
|
|
|
def test_tier_b_update_prefix_rendered(self):
|
|
"""Tier-b ②: is_update=True produces 'Update:' prefix."""
|
|
from meshai.notifications.formatters.quake import format as qfmt
|
|
|
|
canonical = {
|
|
"magnitude": 3.0,
|
|
"depth_km": 8.0,
|
|
"lat": 44.2,
|
|
"lon": -114.9,
|
|
"place": "5 km NE of Stanley, Idaho",
|
|
"tsunami": False,
|
|
"pager": None,
|
|
"is_update": True, # tier-b: update-prefix now live
|
|
"occurred_at": None,
|
|
"event_id": "test_update_prefix",
|
|
"_severity_override": None,
|
|
"_dedup_suffix": "",
|
|
"distance_km": 10.0,
|
|
}
|
|
|
|
expected_new = (
|
|
"\U0001f310 Update: M3.0 — 5 km NE of Stanley, Idaho"
|
|
"\nDepth: 8 km · @ 44.200, -114.900"
|
|
)
|
|
|
|
with pinned_time(_AT):
|
|
result = qfmt(_make_fake_event(canonical), now=_AT, budget=140)
|
|
|
|
assert_byte_identical(result, expected_new)
|
|
assert "Update:" in result
|
|
assert "New:" not in result
|
|
|
|
def test_tsunami_escalation_preserved(self):
|
|
"""Tsunami escalation renders the 🚨 emoji + TSUNAMI WARNING line."""
|
|
from meshai.notifications.formatters.quake import format as qfmt
|
|
|
|
canonical = {
|
|
"magnitude": 4.5,
|
|
"depth_km": 5.0,
|
|
"lat": 35.0,
|
|
"lon": 141.0,
|
|
"place": "off the coast of Japan",
|
|
"tsunami": True,
|
|
"pager": None,
|
|
"is_update": False,
|
|
"occurred_at": None,
|
|
"event_id": "test_tsunami",
|
|
"_severity_override": "immediate",
|
|
"_dedup_suffix": "",
|
|
"distance_km": 8000.0,
|
|
}
|
|
|
|
expected = (
|
|
"\U0001f6a8 New: M4.5 — off the coast of Japan"
|
|
"\nDepth: 5 km · @ 35.000, 141.000"
|
|
"\n\U0001f6a8 TSUNAMI WARNING"
|
|
)
|
|
|
|
with pinned_time(_AT):
|
|
result = qfmt(_make_fake_event(canonical), now=_AT, budget=140)
|
|
|
|
assert result.startswith("\U0001f6a8"), "Tsunami emoji must be 🚨"
|
|
assert "\U0001f6a8 TSUNAMI WARNING" in result
|
|
assert_byte_identical(result, expected)
|
|
|
|
def test_m5_escalation_emoji_preserved(self):
|
|
"""M5+ uses ⚠️ emoji — unchanged from _render."""
|
|
from meshai.notifications.formatters.quake import format as qfmt
|
|
|
|
canonical = {
|
|
"magnitude": 5.2,
|
|
"depth_km": 12.0,
|
|
"lat": 44.0,
|
|
"lon": -114.0,
|
|
"place": "15 km NW of Mackay, Idaho",
|
|
"tsunami": False,
|
|
"pager": None,
|
|
"is_update": False,
|
|
"occurred_at": None,
|
|
"event_id": "test_m5",
|
|
"_severity_override": None,
|
|
"_dedup_suffix": "",
|
|
"distance_km": 50.0,
|
|
}
|
|
|
|
with pinned_time(_AT):
|
|
result = qfmt(_make_fake_event(canonical), now=_AT, budget=140)
|
|
|
|
assert result.startswith("⚠️"), f"M5.2 must use ⚠️ emoji; got: {result!r}"
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# 2. Cross-source identity — native and Central produce identical renders
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
class TestCrossSourceIdentity:
|
|
"""Native to_event() canonical data renders byte-identically to Central path."""
|
|
|
|
def test_native_central_render_identical_fixture_0002(self):
|
|
"""Build native canonical data for fixture 0002 event, render → byte-identical.
|
|
|
|
The Central path produces canonical data by extracting from the
|
|
envelope. The native path produces canonical data in to_event().
|
|
The formatter reads the same keys from both → same wire.
|
|
"""
|
|
from meshai.notifications.formatters.quake import format as qfmt
|
|
|
|
# Central-path canonical data (hand-extracted from fixture 0002)
|
|
central_canonical = {
|
|
"magnitude": 3.3,
|
|
"depth_km": 11.169, # normalized from raw "depth" field
|
|
"lat": 44.46,
|
|
"lon": -112.6108,
|
|
"place": "19 km S of Lima, Montana",
|
|
"tsunami": False,
|
|
"pager": None,
|
|
"occurred_at": None,
|
|
"event_id": "us6000t9bn",
|
|
"is_update": False,
|
|
"_severity_override": None,
|
|
"_dedup_suffix": "",
|
|
"distance_km": 160.0,
|
|
}
|
|
|
|
# Native-path canonical data (as to_event() would set it)
|
|
native_canonical = {
|
|
"magnitude": 3.3,
|
|
"depth_km": 11.169,
|
|
"lat": 44.46,
|
|
"lon": -112.6108,
|
|
"place": "19 km S of Lima, Montana",
|
|
"tsunami": False, # native has no tsunami flag
|
|
"pager": None, # native has no PAGER
|
|
"occurred_at": None,
|
|
"event_id": "us6000t9bn",
|
|
"is_update": False,
|
|
"_severity_override": None,
|
|
"_dedup_suffix": "",
|
|
"distance_km": 160.0,
|
|
}
|
|
|
|
with pinned_time(_AT):
|
|
central_wire = qfmt(_make_fake_event(central_canonical),
|
|
now=_AT, budget=140)
|
|
native_wire = qfmt(_make_fake_event(native_canonical),
|
|
now=_AT, budget=140)
|
|
|
|
assert_byte_identical(native_wire, central_wire), (
|
|
f"Native and Central renders must be byte-identical:\n"
|
|
f" Central: {central_wire!r}\n"
|
|
f" Native: {native_wire!r}"
|
|
)
|
|
|
|
def test_native_to_event_uses_canonical_keys(self):
|
|
"""to_event() data dict has the canonical keys the formatter reads."""
|
|
from unittest.mock import MagicMock
|
|
from meshai.env.usgs_quake import USGSQuakeAdapter
|
|
|
|
cfg = MagicMock()
|
|
cfg.feed_url = "https://example.com/feed"
|
|
cfg.min_magnitude = 1.0
|
|
cfg.bbox = []
|
|
cfg.region = "magic_valley"
|
|
cfg.tick_seconds = 300
|
|
|
|
adapter = USGSQuakeAdapter(cfg)
|
|
raw_evt = {
|
|
"event_id": "us_test_identity",
|
|
"magnitude": 3.5,
|
|
"place": "5 km SW of Twin Falls, Idaho",
|
|
"depth_km": 7.5,
|
|
"lat": 42.5,
|
|
"lon": -114.5,
|
|
"quake_time": 1_783_000_000.0,
|
|
"fetched_at": 1_783_000_010.0,
|
|
"expires": 1_783_086_400.0,
|
|
"severity": "priority",
|
|
}
|
|
event = adapter.to_event(raw_evt)
|
|
|
|
assert event is not None, "to_event() must return an Event for valid input"
|
|
assert event.data is not None, "event.data must not be None"
|
|
|
|
canonical_keys = {
|
|
"magnitude", "depth_km", "lat", "lon", "place",
|
|
"tsunami", "pager", "occurred_at", "event_id",
|
|
}
|
|
missing = canonical_keys - set(event.data.keys())
|
|
assert not missing, (
|
|
f"to_event() event.data missing canonical keys: {missing}\n"
|
|
f"Got keys: {sorted(event.data.keys())}"
|
|
)
|
|
|
|
# Spot-check values
|
|
assert event.data["magnitude"] == 3.5
|
|
assert event.data["lat"] == 42.5
|
|
assert event.data["lon"] == -114.5
|
|
assert event.data["depth_km"] == 7.5
|
|
assert event.data["tsunami"] is False
|
|
assert event.data["pager"] is None
|
|
assert event.data["event_id"] == "us_test_identity"
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# 3. Gate-sequence — old gating vs new decide() — identical decisions
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
def _make_envelope(*, event_id, mag, lat, lon, depth_km=10.0, place=None,
|
|
tsunami=0, alert=None, time_ms=1_780_000_000_000):
|
|
"""Build a minimal Central-style quake envelope for gate-sequence testing."""
|
|
place = place or f"near test location ({lat:.1f},{lon:.1f})"
|
|
return {
|
|
"envelope": {
|
|
"id": event_id,
|
|
"data": {
|
|
"id": event_id,
|
|
"adapter": "usgs_quake",
|
|
"category": "quake.event.test",
|
|
"severity": 0,
|
|
"geo": {"centroid": [lon, lat]},
|
|
"data": {
|
|
"id": event_id,
|
|
"magnitude": mag,
|
|
"place": place,
|
|
"depth_km": depth_km,
|
|
"time_ms": time_ms,
|
|
"tsunami": tsunami,
|
|
"alert": alert,
|
|
"latitude": lat,
|
|
"longitude": lon,
|
|
"depth": depth_km,
|
|
},
|
|
},
|
|
},
|
|
"subject": "central.quake.event.test.unknown",
|
|
"captured_epoch": int(time_ms / 1000),
|
|
}
|
|
|
|
|
|
class TestGateSequence:
|
|
"""gating.quake.decide() gate thresholds + commit/suppress lifecycle."""
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _db(self, mem_db):
|
|
"""All tests in this class share the same mem_db."""
|
|
self.db = mem_db
|
|
|
|
def _decide(self, fixture, *, now):
|
|
"""Build canonical data from a Central-style fixture and call decide()."""
|
|
from meshai.notifications.gating.quake import decide
|
|
env = fixture["envelope"]
|
|
inner = env.get("data") or {}
|
|
d = inner.get("data") or {}
|
|
geo = inner.get("geo") or {}
|
|
cent = geo.get("centroid") or []
|
|
lon, lat = (cent[0], cent[1]) if len(cent) >= 2 else (None, None)
|
|
tms = d.get("time_ms")
|
|
occurred_at = None
|
|
if isinstance(tms, (int, float)):
|
|
occurred_at = int(tms / 1000) if tms > 1e12 else int(tms)
|
|
|
|
canonical = {
|
|
"magnitude": d.get("magnitude"),
|
|
"depth_km": d.get("depth_km") or d.get("depth"),
|
|
"lat": lat,
|
|
"lon": lon,
|
|
"place": d.get("place"),
|
|
"tsunami": bool(d.get("tsunami")),
|
|
"pager": d.get("alert"),
|
|
"occurred_at": occurred_at,
|
|
"event_id": d.get("id") or inner.get("id"),
|
|
}
|
|
return decide(canonical, source="usgs_quake", now=float(now))
|
|
|
|
def test_gate_sequence_matches(self):
|
|
"""Four-event sequence exercises all of decide()'s broadcast thresholds.
|
|
|
|
[0] M2.0, far (below all thresholds) → suppress
|
|
[1] M2.7, within Idaho (regional gate) → broadcast
|
|
[2] M3.5, anywhere (global floor) → broadcast
|
|
[3] M6.0 + tsunami (any-magnitude tsunami gate) → broadcast
|
|
"""
|
|
t_base = 1_780_000_000.0
|
|
|
|
# [0] M2.0 far outside Idaho (lat=10.0, lon=140.0 → Japan)
|
|
fx0 = _make_envelope(event_id="gs_seq_0", mag=2.0, lat=10.0, lon=140.0,
|
|
time_ms=int(t_base * 1000))
|
|
# [1] M2.7 within 250mi of Idaho centroid (Wyoming border)
|
|
fx1 = _make_envelope(event_id="gs_seq_1", mag=2.7, lat=44.09, lon=-115.96,
|
|
time_ms=int((t_base + 100) * 1000))
|
|
# [2] M3.5 anywhere (global_mag_floor = 3.0 exceeded)
|
|
fx2 = _make_envelope(event_id="gs_seq_2", mag=3.5, lat=10.0, lon=140.0,
|
|
time_ms=int((t_base + 200) * 1000))
|
|
# [3] M6.0 + tsunami (any magnitude with tsunami → broadcast)
|
|
fx3 = _make_envelope(event_id="gs_seq_3", mag=6.0, lat=35.0, lon=141.0,
|
|
tsunami=1, time_ms=int((t_base + 300) * 1000))
|
|
|
|
r0 = self._decide(fx0, now=t_base)
|
|
r1 = self._decide(fx1, now=t_base + 100)
|
|
r2 = self._decide(fx2, now=t_base + 200)
|
|
r3 = self._decide(fx3, now=t_base + 300)
|
|
|
|
assert r0.broadcast is False, "M2.0 far must be suppressed"
|
|
assert r1.broadcast is True, "M2.7 Idaho must broadcast"
|
|
assert r2.broadcast is True, "M3.5 global must broadcast"
|
|
assert r3.broadcast is True, "M6.0+tsunami must broadcast"
|
|
|
|
def test_suppress_after_commit(self):
|
|
"""After commit, a replay of the same event_id is suppressed by decide()."""
|
|
t0 = 1_780_000_000.0
|
|
event_id = "suppress_after_commit_test"
|
|
|
|
fx = _make_envelope(event_id=event_id, mag=3.5, lat=44.09, lon=-115.96,
|
|
time_ms=int(t0 * 1000))
|
|
|
|
# First arrival: broadcast.
|
|
result1 = self._decide(fx, now=t0)
|
|
assert result1.broadcast is True, "First arrival must broadcast"
|
|
assert result1.commit is not None, "commit callback must be attached"
|
|
|
|
# Commit (simulates confirmed delivery).
|
|
result1.commit(t0 + 1.0)
|
|
|
|
# Second arrival with same event_id — must suppress.
|
|
result2 = self._decide(fx, now=t0 + 60)
|
|
assert result2.broadcast is False, "Gate must suppress after commit"
|
|
|
|
def test_severity_override_from_decide(self):
|
|
"""decide() sets _severity_override=immediate for tsunami/PAGER."""
|
|
from meshai.notifications.gating.quake import decide
|
|
|
|
# Tsunami
|
|
canonical_tsunami = {
|
|
"magnitude": 4.5, "depth_km": 10.0, "lat": 35.0, "lon": 141.0,
|
|
"place": "off Japan", "tsunami": True, "pager": None,
|
|
"occurred_at": None, "event_id": "sv_tsunami_test",
|
|
}
|
|
result_ts = decide(canonical_tsunami, source="usgs_quake", now=_AT)
|
|
assert result_ts.broadcast is True
|
|
assert result_ts.data_patch.get("_severity_override") == "immediate"
|
|
|
|
# PAGER orange
|
|
canonical_pager = {
|
|
"magnitude": 2.0, "depth_km": 10.0, "lat": 10.0, "lon": 140.0,
|
|
"place": "Pacific Ocean", "tsunami": False, "pager": "orange",
|
|
"occurred_at": None, "event_id": "sv_pager_test",
|
|
}
|
|
result_pg = decide(canonical_pager, source="usgs_quake", now=_AT)
|
|
assert result_pg.broadcast is True
|
|
assert result_pg.data_patch.get("_severity_override") == "immediate"
|
|
|
|
def test_dedup_suffix_is_empty(self):
|
|
"""decide() data_patch has _dedup_suffix='' (bare event.id used for dedup)."""
|
|
from meshai.notifications.gating.quake import decide
|
|
|
|
canonical = {
|
|
"magnitude": 3.5, "depth_km": 10.0, "lat": 44.0, "lon": -114.0,
|
|
"place": "near Stanley, Idaho", "tsunami": False, "pager": None,
|
|
"occurred_at": None, "event_id": "dedup_suffix_test",
|
|
}
|
|
result = decide(canonical, source="usgs_quake", now=_AT)
|
|
assert result.broadcast is True
|
|
assert result.data_patch.get("_dedup_suffix") == ""
|
|
|
|
def test_is_update_always_false_in_patch(self):
|
|
"""decide() data_patch always has is_update=False (v0.5.9 no-Update rule)."""
|
|
from meshai.notifications.gating.quake import decide
|
|
|
|
canonical = {
|
|
"magnitude": 3.0, "depth_km": 8.0, "lat": 44.0, "lon": -114.5,
|
|
"place": "central Idaho", "tsunami": False, "pager": None,
|
|
"occurred_at": None, "event_id": "is_update_test",
|
|
}
|
|
result = decide(canonical, source="usgs_quake", now=_AT)
|
|
assert result.broadcast is True
|
|
assert result.data_patch.get("is_update") is False
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# 4. Schema conformance — to_event() canonical data completeness
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
|
|
class TestSchemaConformance:
|
|
"""env/usgs_quake.py to_event() emits exactly the canonical key set."""
|
|
|
|
@pytest.fixture
|
|
def adapter(self):
|
|
from unittest.mock import MagicMock
|
|
from meshai.env.usgs_quake import USGSQuakeAdapter
|
|
cfg = MagicMock()
|
|
cfg.feed_url = "https://example.com/feed"
|
|
cfg.min_magnitude = 1.0
|
|
cfg.bbox = []
|
|
cfg.region = "magic_valley"
|
|
cfg.tick_seconds = 300
|
|
return USGSQuakeAdapter(cfg)
|
|
|
|
CANONICAL_KEYS = frozenset({
|
|
"magnitude", "depth_km", "lat", "lon", "place",
|
|
"tsunami", "pager", "occurred_at", "event_id",
|
|
})
|
|
|
|
def _raw_evt(self, **overrides):
|
|
base = {
|
|
"event_id": "conform_test",
|
|
"magnitude": 3.1,
|
|
"place": "5 km NW of test, Idaho",
|
|
"depth_km": 9.0,
|
|
"lat": 44.0,
|
|
"lon": -114.5,
|
|
"quake_time": 1_783_000_000.0,
|
|
"fetched_at": 1_783_000_010.0,
|
|
"expires": 1_783_086_400.0,
|
|
"severity": "routine",
|
|
}
|
|
base.update(overrides)
|
|
return base
|
|
|
|
def test_all_canonical_keys_present(self, adapter):
|
|
"""event.data contains all canonical schema keys."""
|
|
event = adapter.to_event(self._raw_evt())
|
|
assert event is not None
|
|
missing = self.CANONICAL_KEYS - set(event.data.keys())
|
|
assert not missing, f"Missing canonical keys: {missing}"
|
|
|
|
def test_no_extra_non_canonical_fields_cause_formatter_crash(self, adapter):
|
|
"""Extra fields in event.data (e.g. raw USGS keys) don't crash the formatter."""
|
|
from meshai.notifications.formatters.quake import format as qfmt
|
|
|
|
event = adapter.to_event(self._raw_evt())
|
|
assert event is not None
|
|
|
|
# Inject extra keys that might come from Central enrichment
|
|
event.data["_enriched"] = {"geocoder": {"city": "TestCity"}}
|
|
event.data["sig"] = 123
|
|
|
|
with pinned_time(_AT):
|
|
result = qfmt(event, now=_AT, budget=140)
|
|
|
|
assert result is not None
|
|
assert "M3.1" in result
|
|
assert len(result) <= 140
|
|
|
|
def test_tsunami_defaults_false(self, adapter):
|
|
"""Native to_event() sets tsunami=False (native feed has no tsunami data)."""
|
|
event = adapter.to_event(self._raw_evt())
|
|
assert event.data["tsunami"] is False
|
|
|
|
def test_pager_defaults_none(self, adapter):
|
|
"""Native to_event() sets pager=None (PAGER comes from Central only)."""
|
|
event = adapter.to_event(self._raw_evt())
|
|
assert event.data["pager"] is None
|
|
|
|
def test_event_id_matches_raw_evt(self, adapter):
|
|
"""event.data["event_id"] matches the source event_id."""
|
|
event = adapter.to_event(self._raw_evt(event_id="my_quake_id"))
|
|
assert event.data["event_id"] == "my_quake_id"
|
|
|
|
def test_missing_depth_km_yields_none(self, adapter):
|
|
"""to_event() handles missing depth gracefully (depth_km=None in data)."""
|
|
raw = self._raw_evt()
|
|
del raw["depth_km"] # simulate missing depth
|
|
event = adapter.to_event(raw)
|
|
assert event is not None
|
|
assert event.data.get("depth_km") is None
|
|
|
|
def test_formatter_budget_respected(self, adapter):
|
|
"""Formatter output fits within budget for worst-case place string."""
|
|
from meshai.notifications.formatters.quake import format as qfmt
|
|
|
|
raw = self._raw_evt(
|
|
magnitude=7.9,
|
|
place="293 km SSW of a pathologically long place description island "
|
|
"region in the remote northern pacific ocean near absolutely nowhere "
|
|
"at all off the coast of the far edge of the map",
|
|
depth_km=12.0,
|
|
lat=44.123,
|
|
lon=-114.987,
|
|
)
|
|
event = adapter.to_event(raw)
|
|
assert event is not None
|
|
with pinned_time(_AT):
|
|
result = qfmt(event, now=_AT, budget=140)
|
|
assert len(result) <= 140, f"{len(result)} chars:\n{result!r}"
|
|
assert "M7.9" in result
|