1a40649d2f
run_loop now accepts a single runtime or a list; it steps every camera each tick (one shared clock read), isolates a per-camera step exception, runs the periodic sync/heartbeat/flush once, and routes zones per camera via a zones_for(cfg, rt) callback. parse_cameras() reads CAMERAS='backend:device:camera_id,...' (single-camera fallback when unset) and refuses to yield zero cameras. zones_from_config() gains a camera_id filter; PipelineRuntime exposes camera_id. Single-camera behavior unchanged.
82 lines
3.3 KiB
Python
82 lines
3.3 KiB
Python
"""TDD for multi-camera support: per-camera zone routing, a run_loop that steps
|
|
N cameras per tick, and CAMERAS env parsing. All with fakes (no hardware)."""
|
|
from edge.zones import zones_from_config
|
|
from edge.daemon import run_loop, parse_cameras
|
|
|
|
|
|
def _cfg_two_cams():
|
|
return {"zones": [
|
|
{"zone_id": "z1", "camera_id": "cam-01", "level": "restricted",
|
|
"polygon_json": "[[0,0],[9,0],[9,9]]"},
|
|
{"zone_id": "z2", "camera_id": "cam-02", "level": "warning",
|
|
"polygon_json": "[[0,0],[5,0],[5,5]]"},
|
|
], "thresholds": {}}
|
|
|
|
|
|
def test_zones_filtered_by_camera():
|
|
cfg = _cfg_two_cams()
|
|
assert [z.zone_id for z in zones_from_config(cfg, "cam-01")] == ["z1"]
|
|
assert [z.zone_id for z in zones_from_config(cfg, "cam-02")] == ["z2"]
|
|
assert len(zones_from_config(cfg)) == 2 # no filter -> all (backward compat)
|
|
assert zones_from_config(cfg, "cam-99") == [] # unknown camera -> none
|
|
|
|
|
|
class _RT:
|
|
def __init__(self, cam): self.camera_id = cam; self.steps = 0; self.applied = None
|
|
def step(self, now): self.steps += 1; return []
|
|
def apply_zones(self, z): self.applied = z
|
|
|
|
|
|
class _Client:
|
|
def __init__(self, cfg): self._cfg = cfg; self.configs = 0; self.beats = 0
|
|
def fetch_config(self): self.configs += 1; return self._cfg
|
|
def send_heartbeat(self, m): self.beats += 1; return True
|
|
|
|
|
|
class _Q:
|
|
def __init__(self): self.flushes = 0
|
|
def flush(self): self.flushes += 1; return 0
|
|
|
|
|
|
def test_run_loop_steps_all_cameras_and_routes_zones_per_camera():
|
|
cfg = _cfg_two_cams()
|
|
r1, r2 = _RT("cam-01"), _RT("cam-02")
|
|
run_loop([r1, r2], _Client(cfg), _Q(), clock=lambda: 0.0, sleep=lambda s: None,
|
|
iterations=5, sync_every=5, heartbeat_every=99, flush_every=99,
|
|
zones_for=lambda c, rt: zones_from_config(c, rt.camera_id))
|
|
assert r1.steps == 5 and r2.steps == 5
|
|
assert [z.zone_id for z in r1.applied] == ["z1"]
|
|
assert [z.zone_id for z in r2.applied] == ["z2"]
|
|
|
|
|
|
def test_run_loop_one_camera_glitch_does_not_stop_the_other():
|
|
class Boom(_RT):
|
|
def step(self, now):
|
|
self.steps += 1
|
|
if self.steps == 2:
|
|
raise RuntimeError("cam glitch")
|
|
return []
|
|
r1, r2 = Boom("cam-01"), _RT("cam-02")
|
|
counters = run_loop([r1, r2], _Client(_cfg_two_cams()), _Q(),
|
|
clock=lambda: 0.0, sleep=lambda s: None, iterations=3,
|
|
sync_every=99, heartbeat_every=99, flush_every=99)
|
|
assert r1.steps == 3 and r2.steps == 3 # both kept going
|
|
assert counters["errors"] == 1
|
|
|
|
|
|
def test_parse_cameras_multi_and_single():
|
|
assert parse_cameras({"CAMERAS": "csi:0:cam-01, uvc:1:cam-02"}, "def") == \
|
|
[("csi", "0", "cam-01"), ("uvc", "1", "cam-02")]
|
|
assert parse_cameras({"CAMERA_BACKEND": "csi", "CSI_DEVICE": "2"}, "cam-09") == \
|
|
[("csi", "2", "cam-09")]
|
|
assert parse_cameras({}, "cam-01") == [("uvc", "0", "cam-01")]
|
|
|
|
|
|
def test_parse_cameras_rejects_blank_entries_never_zero_cameras():
|
|
import pytest
|
|
# A non-empty-but-all-blank CAMERAS must fail loudly, never yield zero cameras
|
|
# (a safety daemon monitoring nothing while looking healthy).
|
|
with pytest.raises(ValueError):
|
|
parse_cameras({"CAMERAS": " , "}, "cam-01")
|
|
with pytest.raises(ValueError):
|
|
parse_cameras({"CAMERAS": "csi:0"}, "cam-01") # wrong arity
|