You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
3.1 KiB
92 lines
3.1 KiB
"""Unit tests for bounded report queue and queued sender."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from adn_server.application.report.queue import BoundedReportQueue, QueuedReportSender
|
|
|
|
|
|
class RecordingSender:
|
|
def __init__(self) -> None:
|
|
self.systems: dict[str, Any] | None = None
|
|
self.bridges: dict[str, Any] | None = None
|
|
self.config_calls: list[tuple[dict, bool]] = []
|
|
self.bridge_calls: list[tuple[dict, bool]] = []
|
|
self.events: list[str] = []
|
|
|
|
def set_systems(self, systems: dict[str, Any]) -> None:
|
|
self.systems = systems
|
|
|
|
def set_bridges(self, bridges: dict[str, Any]) -> None:
|
|
self.bridges = bridges
|
|
|
|
def send_config(self, systems: dict[str, Any], *, incremental: bool = False) -> None:
|
|
self.config_calls.append((systems, incremental))
|
|
|
|
def send_bridge(self, bridges: dict[str, Any], *, incremental: bool = False) -> None:
|
|
self.bridge_calls.append((bridges, incremental))
|
|
|
|
def send_bridge_event(self, event: str) -> None:
|
|
self.events.append(event)
|
|
|
|
|
|
def test_enqueue_event_is_non_blocking():
|
|
inner = RecordingSender()
|
|
queue = BoundedReportQueue(max_events=4, max_drain_per_tick=10)
|
|
sender = QueuedReportSender(queue, inner)
|
|
sender.send_bridge_event("GROUP VOICE,START,RX,SYS,1,2,3,1,4")
|
|
assert inner.events == []
|
|
assert queue.pending_count() == 1
|
|
|
|
|
|
def test_drain_delivers_events_before_snapshots():
|
|
inner = RecordingSender()
|
|
queue = BoundedReportQueue(max_drain_per_tick=10)
|
|
sender = QueuedReportSender(queue, inner)
|
|
sender.send_bridge_event("e1")
|
|
sender.send_config({"SYS": {}}, incremental=True)
|
|
sender.send_bridge({"99": []}, incremental=False)
|
|
queue.drain(inner)
|
|
assert inner.events == ["e1"]
|
|
assert len(inner.config_calls) == 1
|
|
assert inner.config_calls[0][1] is True
|
|
assert len(inner.bridge_calls) == 1
|
|
|
|
|
|
def test_config_and_bridge_coalesce_to_latest():
|
|
inner = RecordingSender()
|
|
queue = BoundedReportQueue()
|
|
sender = QueuedReportSender(queue, inner)
|
|
sender.send_config({"A": 1})
|
|
sender.send_config({"B": 2}, incremental=True)
|
|
sender.send_bridge({"old": []})
|
|
sender.send_bridge({"new": []}, incremental=True)
|
|
queue.drain(inner)
|
|
assert inner.config_calls == [({"B": 2}, True)]
|
|
assert inner.bridge_calls == [({"new": []}, True)]
|
|
|
|
|
|
def test_bounded_queue_drops_oldest_events():
|
|
inner = RecordingSender()
|
|
queue = BoundedReportQueue(max_events=2, max_drain_per_tick=10)
|
|
sender = QueuedReportSender(queue, inner)
|
|
sender.send_bridge_event("e1")
|
|
sender.send_bridge_event("e2")
|
|
sender.send_bridge_event("e3")
|
|
assert queue.dropped_events == 1
|
|
queue.drain(inner)
|
|
assert inner.events == ["e2", "e3"]
|
|
|
|
|
|
def test_drain_respects_per_tick_budget():
|
|
inner = RecordingSender()
|
|
queue = BoundedReportQueue(max_drain_per_tick=2)
|
|
sender = QueuedReportSender(queue, inner)
|
|
for i in range(5):
|
|
sender.send_bridge_event(f"e{i}")
|
|
queue.drain(inner)
|
|
assert inner.events == ["e0", "e1"]
|
|
queue.drain(inner)
|
|
assert inner.events == ["e0", "e1", "e2", "e3"]
|