Pouzor
e41dbe579c
- Replace BaseHTTPMiddleware with pure ASGI middleware in auth.py to fix the "Unexpected message: http.response.start" crash on SSE streams - Migrate from SseServerTransport to StreamableHTTPSessionManager in main.py - Add _slim_canvas() in tools.py to strip React Flow layout fields from get_canvas responses (60-80% payload reduction) - Update tests: assert canvas slimming, mock session_manager.handle_request in auth tests to avoid uninitialized task group errors
29 lines
827 B
Python
29 lines
827 B
Python
import pytest
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_health_no_key(client):
|
|
resp = await client.get("/health")
|
|
assert resp.status_code == 200
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_missing_api_key(client):
|
|
resp = await client.get("/mcp")
|
|
assert resp.status_code == 401
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_wrong_api_key(client):
|
|
resp = await client.get("/mcp", headers={"X-API-Key": "wrong"})
|
|
assert resp.status_code == 401
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_valid_api_key_passes(client, api_key):
|
|
# Auth passes — mock handle_request so we don't need a live MCP session
|
|
with patch("app.main.session_manager.handle_request", new_callable=AsyncMock):
|
|
resp = await client.get("/mcp", headers={"X-API-Key": api_key})
|
|
assert resp.status_code != 401
|