Initial public release

tests/test_queueing.py

@@ -0,0 +1,89 @@
+from __future__ import annotations
+
+from ha_deconz_bridge.events import DeconzEvent, DeconzSensorEvent, HaCommandEvent
+from ha_deconz_bridge.queueing import EventBroker
+
+
+def test_event_broker_coalesces_latest_ha_command() -> None:
+    broker = EventBroker()
+    broker.put_ha(HaCommandEvent("light", on=True, brightness=10))
+    broker.put_ha(HaCommandEvent("light", on=True, brightness=200))
+    broker.put_deconz(DeconzEvent("1", {"on": True}))
+
+    first = broker.get()
+    second = broker.get()
+
+    assert isinstance(first, HaCommandEvent)
+    assert isinstance(second, DeconzEvent)
+    assert first.brightness == 200
+
+
+def test_event_broker_merges_partial_ha_commands() -> None:
+    broker = EventBroker()
+    broker.put_ha(HaCommandEvent("light", on=True, brightness=120, color_temp=400))
+    broker.put_ha(HaCommandEvent("light", rgb=(10, 20, 30)))
+
+    event = broker.get()
+
+    assert isinstance(event, HaCommandEvent)
+    assert event.on is True
+    assert event.brightness == 120
+    assert event.rgb == (10, 20, 30)
+    assert event.color_temp is None
+
+
+def test_event_broker_moves_merged_ha_command_to_back() -> None:
+    broker = EventBroker()
+    broker.put_ha(HaCommandEvent("first", brightness=1))
+    broker.put_ha(HaCommandEvent("second", brightness=2))
+    broker.put_ha(HaCommandEvent("first", rgb=(1, 2, 3)))
+
+    first = broker.get()
+    second = broker.get()
+
+    assert isinstance(first, HaCommandEvent)
+    assert isinstance(second, HaCommandEvent)
+    assert first.light_name == "second"
+    assert second.light_name == "first"
+    assert second.brightness == 1
+    assert second.rgb == (1, 2, 3)
+
+
+def test_event_broker_deduplicates_pending_deconz_events_by_controller() -> None:
+    broker = EventBroker()
+    broker.put_deconz(DeconzEvent("1", {"bri": 10}))
+    broker.put_deconz(DeconzEvent("2", {"bri": 20}))
+    broker.put_deconz(DeconzEvent("1", {"bri": 30}))
+
+    first = broker.get()
+    second = broker.get()
+
+    assert isinstance(first, DeconzEvent)
+    assert isinstance(second, DeconzEvent)
+    assert first.controller_id == "2"
+    assert first.raw_state == {"bri": 20}
+    assert second.controller_id == "1"
+    assert second.raw_state == {"bri": 30}
+
+
+def test_event_broker_prioritizes_sensor_then_ha_then_light_events() -> None:
+    broker = EventBroker()
+    broker.put_ha(HaCommandEvent("light", on=True, brightness=10))
+    broker.put_deconz(DeconzEvent("1", {"on": True}))
+    broker.put_sensor(DeconzSensorEvent("4", {"buttonevent": 1002}))
+
+    assert isinstance(broker.get(), DeconzSensorEvent)
+    assert isinstance(broker.get(), HaCommandEvent)
+    assert isinstance(broker.get(), DeconzEvent)
+
+
+def test_event_broker_does_not_let_light_readbacks_starve_ha_commands() -> None:
+    broker = EventBroker()
+    broker.put_deconz(DeconzEvent("1", {"bri": 10}))
+    broker.put_deconz(DeconzEvent("2", {"bri": 20}))
+    broker.put_ha(HaCommandEvent("light", on=True, brightness=255))
+
+    event = broker.get()
+
+    assert isinstance(event, HaCommandEvent)
+    assert event.brightness == 255