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
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
"""Tests for app.db module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from app.config import Settings
|
|
from app.db import _CONVERSATIONS_DDL, _INTERRUPTS_DDL
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestDbModule:
|
|
@pytest.mark.asyncio
|
|
async def test_create_pool_sets_correct_params(self) -> None:
|
|
settings = Settings(
|
|
database_url="postgresql://user:pass@localhost:5432/testdb",
|
|
anthropic_api_key="key",
|
|
)
|
|
with patch("app.db.AsyncConnectionPool") as MockPool:
|
|
mock_pool = AsyncMock()
|
|
MockPool.return_value = mock_pool
|
|
|
|
from app.db import create_pool
|
|
|
|
await create_pool(settings)
|
|
MockPool.assert_called_once()
|
|
call_kwargs = MockPool.call_args
|
|
assert "postgresql://user:pass@localhost:5432/testdb" in str(call_kwargs)
|
|
mock_pool.open.assert_awaited_once()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_checkpointer_calls_setup(self) -> None:
|
|
mock_pool = AsyncMock()
|
|
with patch("app.db.AsyncPostgresSaver") as MockSaver:
|
|
mock_saver = AsyncMock()
|
|
MockSaver.return_value = mock_saver
|
|
|
|
from app.db import create_checkpointer
|
|
|
|
await create_checkpointer(mock_pool)
|
|
MockSaver.assert_called_once_with(conn=mock_pool)
|
|
mock_saver.setup.assert_awaited_once()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_setup_app_tables_executes_ddl(self) -> None:
|
|
mock_conn = AsyncMock()
|
|
mock_ctx = AsyncMock()
|
|
mock_ctx.__aenter__ = AsyncMock(return_value=mock_conn)
|
|
mock_ctx.__aexit__ = AsyncMock(return_value=None)
|
|
mock_pool = MagicMock()
|
|
mock_pool.connection.return_value = mock_ctx
|
|
|
|
from app.db import setup_app_tables
|
|
|
|
await setup_app_tables(mock_pool)
|
|
assert mock_conn.execute.await_count == 2
|
|
|
|
def test_ddl_statements_valid(self) -> None:
|
|
assert "CREATE TABLE IF NOT EXISTS conversations" in _CONVERSATIONS_DDL
|
|
assert "CREATE TABLE IF NOT EXISTS active_interrupts" in _INTERRUPTS_DDL
|
|
assert "thread_id" in _CONVERSATIONS_DDL
|
|
assert "interrupt_id" in _INTERRUPTS_DDL
|