move_detect/tests/test_occupancy.py

from edge.detectors.zone import Zone
from edge.detectors.occupancy import OccupancyWatchdog, OccupancyConfig

_POLY = [(0.0, 0.0), (1.0, 0.0), (1.0, 1.0), (0.0, 1.0)]


def _zone(zone_id="z1", **rules):
    return Zone(zone_id, "restricted", _POLY, rules or None)


def _watch(**cfg):
    return OccupancyWatchdog(OccupancyConfig(**cfg))


def test_no_rules_emits_nothing():
    w = _watch()
    zones = [Zone("z1", "restricted", _POLY)]
    assert w.update(zones, {"z1": {1}}, now=0.0) == []
    assert w.update(zones, {"z1": {1}}, now=999.0) == []


def test_lone_worker_fires_after_sustained_solo():
    w = _watch(lone_confirm_s=30.0)
    zones = [_zone(min_workers=2)]
    assert w.update(zones, {"z1": {7}}, now=0.0) == []       # 방금 혼자가 됨
    assert w.update(zones, {"z1": {7}}, now=29.0) == []      # 아직 유예
    out = w.update(zones, {"z1": {7}}, now=31.0)
    assert out == [{"kind": "lone_worker", "zone_id": "z1", "track_id": 7,
                    "metrics": {"workers": 1, "required": 2, "alone_s": 31.0}}]
    # 조건이 계속돼도 재발보 없음 (해소 후 재충족으로만)
    assert w.update(zones, {"z1": {7}}, now=60.0) == []


def test_colleague_return_resets_lone_timer():
    w = _watch(lone_confirm_s=30.0)
    zones = [_zone(min_workers=2)]
    w.update(zones, {"z1": {7}}, now=0.0)
    w.update(zones, {"z1": {7, 8}}, now=20.0)                # 동료 복귀 → 리셋
    assert w.update(zones, {"z1": {7}}, now=40.0) == []      # 새로 30초 필요
    assert len(w.update(zones, {"z1": {7}}, now=71.0)) == 1


def test_empty_zone_is_not_a_lone_violation():
    w = _watch(lone_confirm_s=0.0)
    zones = [_zone(min_workers=2)]
    assert w.update(zones, {}, now=0.0) == []
    assert w.update(zones, {}, now=100.0) == []


def test_overcapacity_fires_after_sustained_excess():
    w = _watch(over_confirm_s=10.0)
    zones = [_zone(max_workers=2)]
    assert w.update(zones, {"z1": {1, 2, 3}}, now=0.0) == []
    out = w.update(zones, {"z1": {1, 2, 3}}, now=11.0)
    assert out == [{"kind": "overcapacity", "zone_id": "z1", "track_id": None,
                    "metrics": {"workers": 3, "limit": 2}}]
    # 해소 후 재충족이면 다시 발보
    w.update(zones, {"z1": {1, 2}}, now=20.0)
    w.update(zones, {"z1": {1, 2, 3}}, now=30.0)
    assert len(w.update(zones, {"z1": {1, 2, 3}}, now=41.0)) == 1


def test_overstay_fires_per_track():
    w = _watch()
    zones = [_zone(max_dwell_s=60)]
    w.update(zones, {"z1": {1}}, now=0.0)
    w.update(zones, {"z1": {1, 2}}, now=30.0)                # 2번은 30초에 진입
    out = w.update(zones, {"z1": {1, 2}}, now=61.0)          # 1번만 60초 초과
    assert [o["track_id"] for o in out] == [1]
    assert out[0]["kind"] == "overstay"
    out = w.update(zones, {"z1": {1, 2}}, now=91.0)          # 이제 2번도 초과
    assert [o["track_id"] for o in out] == [2]


def test_dwell_survives_short_detection_dropout():
    # run_once의 stale 필터로 1~2틱 미검출이 흔하다. grace 안의 부재는 체류 연속으로
    # 간주해야 10분 규칙이 실사용에서 무력화되지 않는다.
    w = _watch(dwell_absence_grace_s=5.0)
    zones = [_zone(max_dwell_s=60)]
    w.update(zones, {"z1": {1}}, now=0.0)
    w.update(zones, {}, now=30.0)                            # 드롭아웃 (grace보다 김? 아님: 아래 재검출로 판단)
    out = w.update(zones, {"z1": {1}}, now=61.0)
    # 30.0의 부재는 last_seen(0.0)에서 5초 grace를 초과 → 새 체류로 리셋되었다.
    assert out == []
    # 연속 체류가 실제로 60초를 넘기면 발보
    w.update(zones, {"z1": {1}}, now=90.0)
    w.update(zones, {"z1": {1}}, now=120.0)
    assert len(w.update(zones, {"z1": {1}}, now=122.0)) == 1


def test_dwell_short_gap_within_grace_keeps_timer():
    w = _watch(dwell_absence_grace_s=5.0)
    zones = [_zone(max_dwell_s=60)]
    w.update(zones, {"z1": {1}}, now=0.0)
    w.update(zones, {}, now=2.0)                             # 2초 드롭아웃 — grace 이내
    out = w.update(zones, {"z1": {1}}, now=61.0)
    assert len(out) == 1 and out[0]["metrics"]["dwell_s"] == 61.0


def test_forget_clears_dwell_state():
    w = _watch()
    zones = [_zone(max_dwell_s=60)]
    w.update(zones, {"z1": {1}}, now=0.0)
    w.forget(1)
    assert w.update(zones, {"z1": {1}}, now=61.0) == []      # 새 체류로 시작


def test_deleted_zone_state_is_pruned():
    w = _watch(lone_confirm_s=0.0, over_confirm_s=0.0)
    zones = [_zone(min_workers=2, max_workers=1)]
    w.update(zones, {"z1": {1}}, now=0.0)
    w.update([], {}, now=1.0)                                # 존 삭제 (config 동기)
    assert w._lone_since == {} and w._over_since == {}
    assert w._lone_fired == set() and w._over_fired == set()


# --- "아무도 있으면 안 되는 구역" ------------------------------------------------------
#
# 크레인 짐 아래, 굴착기 선회 반경처럼 인원 상한이 0인 구역이 있다. 깔림 사고의 전형이다.
# max_workers=0 을 falsy 로 걸러내면 운영자가 규칙을 저장해 놓고도 아무 일이 일어나지 않는다
# — 화면에는 규칙이 있는데 감지기는 꺼져 있는, 이 프로젝트가 반복해서 잡아온 조용한 실패다.

def test_zero_capacity_zone_fires_on_a_single_person():
    w = _watch(over_confirm_s=10.0)
    zones = [_zone(max_workers=0)]
    assert w.update(zones, {"z1": {1}}, now=0.0) == []        # 유예 안
    out = w.update(zones, {"z1": {1}}, now=11.0)
    assert out == [{"kind": "overcapacity", "zone_id": "z1", "track_id": None,
                    "metrics": {"workers": 1, "limit": 0}}]


def test_zero_capacity_zone_stays_quiet_when_empty():
    w = _watch(over_confirm_s=10.0)
    zones = [_zone(max_workers=0)]
    assert w.update(zones, {"z1": set()}, now=0.0) == []
    assert w.update(zones, {"z1": set()}, now=999.0) == []


def test_absent_max_workers_still_disables_the_rule():
    # 규칙을 안 넣은 것과 0 을 넣은 것은 다르다. 전자는 "제한 없음", 후자는 "아무도 안 됨".
    w = _watch(over_confirm_s=10.0)
    zones = [_zone(min_workers=2)]           # max_workers 키 자체가 없음
    assert w.update(zones, {"z1": {1, 2, 3}}, now=0.0) == []
    assert [o["kind"] for o in w.update(zones, {"z1": {1, 2, 3}}, now=11.0)] == []