33488fd634
Backend: - FastAPI WebSocket /ws endpoint with streaming via LangGraph astream - LangGraph Supervisor connecting 3 mock agents (order_lookup, order_actions, fallback) - YAML Agent Registry with Pydantic validation and immutable configs - PostgresSaver checkpoint persistence via langgraph-checkpoint-postgres - Session TTL with 30-min sliding window and interrupt extension - LLM provider abstraction (Anthropic/OpenAI/Google) - Token usage + cost tracking callback handler - Input validation: message size cap, thread_id format, content length - Security: no hardcoded defaults, startup API key validation, no input reflection Frontend: - React 19 + TypeScript + Vite chat UI - WebSocket hook with reconnect + exponential backoff - Streaming token display with agent attribution - Interrupt approval/reject UI for write operations - Collapsible tool call viewer Testing: - 87 unit tests, 87% coverage (exceeds 80% requirement) - Ruff lint + format clean Infrastructure: - Docker Compose (PostgreSQL 16 + backend) - pyproject.toml with full dependency management
71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
"""Tests for app.session_manager module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import time
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from app.session_manager import SessionManager
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestSessionManager:
|
|
def test_new_session_not_expired(self, session_manager: SessionManager) -> None:
|
|
session_manager.touch("thread-1")
|
|
assert not session_manager.is_expired("thread-1")
|
|
|
|
def test_unknown_session_is_expired(self, session_manager: SessionManager) -> None:
|
|
assert session_manager.is_expired("unknown")
|
|
|
|
def test_session_expires_after_ttl(self) -> None:
|
|
mgr = SessionManager(session_ttl_seconds=1)
|
|
mgr.touch("t1")
|
|
with patch("app.session_manager.time") as mock_time:
|
|
mock_time.time.return_value = time.time() + 2
|
|
assert mgr.is_expired("t1")
|
|
|
|
def test_touch_resets_ttl(self) -> None:
|
|
mgr = SessionManager(session_ttl_seconds=5)
|
|
mgr.touch("t1")
|
|
initial_state = mgr.get_state("t1")
|
|
# Touch again after some time
|
|
with patch("app.session_manager.time") as mock_time:
|
|
mock_time.time.return_value = time.time() + 3
|
|
mgr.touch("t1")
|
|
new_state = mgr.get_state("t1")
|
|
assert new_state.last_activity > initial_state.last_activity
|
|
|
|
def test_interrupt_suspends_expiration(self) -> None:
|
|
mgr = SessionManager(session_ttl_seconds=1)
|
|
mgr.touch("t1")
|
|
mgr.extend_for_interrupt("t1")
|
|
with patch("app.session_manager.time") as mock_time:
|
|
mock_time.time.return_value = time.time() + 100
|
|
assert not mgr.is_expired("t1")
|
|
|
|
def test_resolve_interrupt_resumes_ttl(self) -> None:
|
|
mgr = SessionManager(session_ttl_seconds=1)
|
|
mgr.touch("t1")
|
|
mgr.extend_for_interrupt("t1")
|
|
mgr.resolve_interrupt("t1")
|
|
state = mgr.get_state("t1")
|
|
assert not state.has_pending_interrupt
|
|
|
|
def test_extend_for_nonexistent_creates_session(self) -> None:
|
|
mgr = SessionManager()
|
|
mgr.extend_for_interrupt("new-thread")
|
|
state = mgr.get_state("new-thread")
|
|
assert state is not None
|
|
|
|
def test_remove_session(self, session_manager: SessionManager) -> None:
|
|
session_manager.touch("t1")
|
|
session_manager.remove("t1")
|
|
assert session_manager.get_state("t1") is None
|
|
|
|
def test_session_state_is_immutable(self, session_manager: SessionManager) -> None:
|
|
state = session_manager.touch("t1")
|
|
with pytest.raises(Exception):
|
|
state.thread_id = "new"
|