forked from Rockachopa/Timmy-time-dashboard
4e11dd2490
Move 97 test files from flat tests/ into 13 subdirectories: tests/dashboard/ (8 files — routes, mobile, mission control) tests/swarm/ (17 files — coordinator, docker, routing, tasks) tests/timmy/ (12 files — agent, backends, CLI, tools) tests/self_coding/ (14 files — git safety, indexer, self-modify) tests/lightning/ (3 files — L402, LND, interface) tests/creative/ (8 files — assembler, director, image/music/video) tests/integrations/ (10 files — chat bridge, telegram, voice, websocket) tests/mcp/ (4 files — bootstrap, discovery, executor) tests/spark/ (3 files — engine, tools, events) tests/hands/ (3 files — registry, oracle, phase5) tests/scripture/ (1 file) tests/infrastructure/ (3 files — router cascade, API) tests/security/ (3 files — XSS, regression) Fix Path(__file__) reference in test_mobile_scenarios.py for new depth. Add __init__.py to all test subdirectories. Tests: 1503 passed, 9 failed (pre-existing), 53 errors (pre-existing) https://claude.ai/code/session_019oMFNvD8uSGSSmBMGkBfQN
32 lines
850 B
Python
32 lines
850 B
Python
"""Tests for ws_manager/handler.py — WebSocket manager."""
|
|
|
|
import json
|
|
|
|
import pytest
|
|
|
|
from ws_manager.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 == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ws_manager_event_history_limit():
|
|
"""History is trimmed to max_history after broadcasts."""
|
|
mgr = WebSocketManager()
|
|
mgr._max_history = 5
|
|
for i in range(10):
|
|
await mgr.broadcast(f"e{i}", {})
|
|
assert len(mgr.event_history) == 5
|
|
assert mgr.event_history[0].event == "e5"
|