mirror of
https://github.com/smittix/intercept.git
synced 2026-04-24 22:59:59 -07:00
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
"""Tests for SSE fanout queue behavior."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import queue
|
|
import time
|
|
import uuid
|
|
|
|
import pytest
|
|
|
|
from utils.sse import subscribe_fanout_queue
|
|
|
|
|
|
def _channel_key(prefix: str) -> str:
|
|
return f"{prefix}-{uuid.uuid4()}"
|
|
|
|
|
|
def test_fanout_drains_source_queue_without_subscribers() -> None:
|
|
"""Queued messages should be dropped while no SSE clients are connected."""
|
|
source = queue.Queue()
|
|
channel_key = _channel_key("sse-idle")
|
|
|
|
# Start fanout distributor, then remove the only subscriber.
|
|
_, unsubscribe = subscribe_fanout_queue(source, channel_key=channel_key, source_timeout=0.01)
|
|
unsubscribe()
|
|
|
|
source.put({"type": "aprs", "callsign": "N0CALL"})
|
|
time.sleep(0.05)
|
|
|
|
assert source.qsize() == 0
|
|
|
|
|
|
def test_fanout_does_not_replay_stale_message_after_re_subscribe() -> None:
|
|
"""A message queued while disconnected should not be replayed on reconnect."""
|
|
source = queue.Queue()
|
|
channel_key = _channel_key("sse-resub")
|
|
|
|
_, unsubscribe = subscribe_fanout_queue(source, channel_key=channel_key, source_timeout=0.01)
|
|
unsubscribe()
|
|
|
|
source.put({"type": "aprs", "callsign": "K1ABC"})
|
|
|
|
subscriber, unsubscribe2 = subscribe_fanout_queue(
|
|
source,
|
|
channel_key=channel_key,
|
|
source_timeout=0.01,
|
|
)
|
|
try:
|
|
with pytest.raises(queue.Empty):
|
|
subscriber.get(timeout=0.1)
|
|
live = {"type": "aprs", "callsign": "LIVE01"}
|
|
source.put(live)
|
|
got = subscriber.get(timeout=0.25)
|
|
finally:
|
|
unsubscribe2()
|
|
|
|
assert got == live
|