mirror of
https://github.com/zvx-echo6/meshai.git
synced 2026-06-11 01:14:45 +02:00
78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
|
"""v0.5.1: sub-adapter (owned-sources) routing for shared Central subjects."""
|
||
|
|
|
||
|
|
import json
|
||
|
|
|
||
|
|
from meshai.config import EnvironmentalConfig
|
||
|
|
from meshai.central.consumer import CentralConsumer
|
||
|
|
from meshai.notifications.pipeline.bus import EventBus
|
||
|
|
|
||
|
|
|
||
|
|
def _envelope(adapter, category="x.y", eid="e1"):
|
||
|
|
return {"id": eid, "data": {
|
||
|
|
"id": eid, "adapter": adapter, "category": category,
|
||
|
|
"time": "2026-05-28T00:00:00Z", "severity": 1,
|
||
|
|
"geo": {"centroid": [-114.0, 42.0], "primary_region": "US-ID", "regions": ["US-ID"]},
|
||
|
|
"data": {}}}
|
||
|
|
|
||
|
|
|
||
|
|
def _route(central, adapter, subject, category="x.y"):
|
||
|
|
"""Simulate a message arriving on the subscription that matches `subject`,
|
||
|
|
with that subscription's owned-sources, and return the emitted Event (or None)."""
|
||
|
|
env = EnvironmentalConfig()
|
||
|
|
for a in central:
|
||
|
|
getattr(env, a).feed_source = "central"
|
||
|
|
rec = []
|
||
|
|
bus = EventBus(); bus.subscribe(rec.append)
|
||
|
|
c = CentralConsumer(env, bus)
|
||
|
|
so = c._subject_owned()
|
||
|
|
owned = None
|
||
|
|
for filt, o in so.items():
|
||
|
|
prefix = filt[:-1] if filt.endswith(">") else filt
|
||
|
|
if subject == filt or subject.startswith(prefix):
|
||
|
|
owned = o
|
||
|
|
break
|
||
|
|
ev = c._handle(subject, json.dumps(_envelope(adapter, category)).encode(), owned)
|
||
|
|
return ev, rec
|
||
|
|
|
||
|
|
|
||
|
|
def test_roads511_only_drops_wzdx():
|
||
|
|
ev, rec = _route(["roads511"], "wzdx", "central.traffic.work_zone.ok")
|
||
|
|
assert ev is None and rec == []
|
||
|
|
|
||
|
|
|
||
|
|
def test_roads511_only_emits_state_511_atis():
|
||
|
|
ev, rec = _route(["roads511"], "state_511_atis", "central.traffic.event.id.1")
|
||
|
|
assert ev is not None and ev.source == "roads511" and len(rec) == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_both_central_wzdx_routes_to_traffic():
|
||
|
|
ev, rec = _route(["traffic", "roads511"], "wzdx", "central.traffic.work_zone.ok")
|
||
|
|
assert ev is not None and ev.source == "traffic"
|
||
|
|
|
||
|
|
|
||
|
|
def test_both_central_state511_routes_to_roads511():
|
||
|
|
ev, rec = _route(["traffic", "roads511"], "state_511_atis", "central.traffic.event.id.1")
|
||
|
|
assert ev is not None and ev.source == "roads511"
|
||
|
|
|
||
|
|
|
||
|
|
def test_firms_only_drops_wfigs():
|
||
|
|
ev, rec = _route(["firms"], "wfigs_incidents", "central.fire.incident.mt.x")
|
||
|
|
assert ev is None and rec == []
|
||
|
|
|
||
|
|
|
||
|
|
def test_firms_only_emits_firms():
|
||
|
|
ev, rec = _route(["firms"], "firms", "central.fire.hotspot.viirs_noaa20.high")
|
||
|
|
assert ev is not None and ev.source == "firms" and len(rec) == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_tomtom_incidents_remaps_to_traffic():
|
||
|
|
ev, rec = _route(["traffic"], "tomtom_incidents", "central.traffic.incident.x")
|
||
|
|
assert ev is not None and ev.source == "traffic"
|
||
|
|
|
||
|
|
|
||
|
|
def test_subject_owned_shares_traffic_subject():
|
||
|
|
env = EnvironmentalConfig()
|
||
|
|
env.traffic.feed_source = "central"
|
||
|
|
env.roads511.feed_source = "central"
|
||
|
|
so = CentralConsumer(env, None)._subject_owned()
|
||
|
|
assert so.get("central.traffic.>") == {"traffic", "roads511"}
|