forked from Rockachopa/Timmy-time-dashboard
31 lines
931 B
Python
31 lines
931 B
Python
|
"""Tests for websocket/handler.py — WebSocket manager."""
|
||
|
|
|
||
|
|
import json
|
||
|
|
|
||
|
|
from websocket.handler import WebSocketManager, WSEvent
|
||
|
|
|
||
|
|
|
||
|
|
def test_ws_event_to_json():
|
||
|
|
event = WSEvent(event="test", data={"key": "val"}, timestamp="2026-01-01T00:00:00Z")
|
||
|
|
j = json.loads(event.to_json())
|
||
|
|
assert j["event"] == "test"
|
||
|
|
assert j["data"]["key"] == "val"
|
||
|
|
|
||
|
|
|
||
|
|
def test_ws_manager_initial_state():
|
||
|
|
mgr = WebSocketManager()
|
||
|
|
assert mgr.connection_count == 0
|
||
|
|
assert mgr.event_history == []
|
||
|
|
|
||
|
|
|
||
|
|
def test_ws_manager_event_history_limit():
|
||
|
|
mgr = WebSocketManager()
|
||
|
|
mgr._max_history = 5
|
||
|
|
for i in range(10):
|
||
|
|
event = WSEvent(event=f"e{i}", data={}, timestamp="t")
|
||
|
|
mgr._event_history.append(event)
|
||
|
|
# Simulate the trim that happens in broadcast
|
||
|
|
if len(mgr._event_history) > mgr._max_history:
|
||
|
|
mgr._event_history = mgr._event_history[-mgr._max_history:]
|
||
|
|
assert len(mgr._event_history) == 5
|