from edge.events import EventType, Severity, Event, severity_for


def test_severity_mapping():
    assert severity_for(EventType.NO_MOTION) == Severity.CRITICAL
    assert severity_for(EventType.ZONE_RESTRICTED) == Severity.HIGH
    assert severity_for(EventType.FALL) == Severity.HIGH
    assert severity_for(EventType.IMMOBILITY) == Severity.WARNING
    assert severity_for(EventType.ZONE_WARNING) == Severity.WARNING


def test_event_to_payload_shape():
    e = Event(EventType.FALL, Severity.HIGH, camera_id="cam-01", track_id=7,
              zone_id=None, occurred_at=1000.0, duration_s=3.2,
              metrics={"torso_angle": 78.0}, spatial=None)
    p = e.to_payload(device_id="pi-01")
    assert p["device_id"] == "pi-01"
    assert p["event_type"] == "fall"
    assert p["severity"] == "high"
    assert p["track_id"] == 7
    assert p["metrics"]["torso_angle"] == 78.0
    assert "snapshots" not in p  # snapshots는 Alerter가 붙임
    assert "spatial" not in p    # None이면 키 생략 (구서버 하위호환)


def test_event_to_payload_includes_spatial_when_present():
    e = Event(EventType.FALL, Severity.HIGH, camera_id="cam-01", track_id=7,
              zone_id=None, occurred_at=1000.0, duration_s=3.2,
              metrics={}, spatial=(0.51234, 0.87))
    p = e.to_payload(device_id="pi-01")
    assert p["spatial"] == [0.5123, 0.87]   # 0..1 정규화 좌표, 4자리 반올림