98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
|
"""Tests for timmy_serve/app.py — Serve FastAPI app and endpoints."""
|
||
|
|
|
||
|
|
from unittest.mock import MagicMock, patch
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from fastapi.testclient import TestClient
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def serve_client():
|
||
|
|
"""Create a TestClient for the timmy-serve app."""
|
||
|
|
from timmy_serve.app import create_timmy_serve_app
|
||
|
|
|
||
|
|
app = create_timmy_serve_app(price_sats=100)
|
||
|
|
return TestClient(app)
|
||
|
|
|
||
|
|
|
||
|
|
class TestHealthEndpoint:
|
||
|
|
def test_health_returns_ok(self, serve_client):
|
||
|
|
resp = serve_client.get("/health")
|
||
|
|
assert resp.status_code == 200
|
||
|
|
data = resp.json()
|
||
|
|
assert data["status"] == "healthy"
|
||
|
|
assert data["service"] == "timmy-serve"
|
||
|
|
|
||
|
|
|
||
|
|
class TestServeStatus:
|
||
|
|
def test_status_returns_pricing(self, serve_client):
|
||
|
|
resp = serve_client.get("/serve/status")
|
||
|
|
assert resp.status_code == 200
|
||
|
|
data = resp.json()
|
||
|
|
assert data["price_sats"] == 100
|
||
|
|
assert "total_invoices" in data
|
||
|
|
assert "total_earned_sats" in data
|
||
|
|
|
||
|
|
|
||
|
|
class TestServeChatEndpoint:
|
||
|
|
"""Regression tests for /serve/chat.
|
||
|
|
|
||
|
|
The original implementation declared ``async def serve_chat(request: ChatRequest)``
|
||
|
|
which shadowed FastAPI's ``Request`` object. Calling ``request.headers`` on a
|
||
|
|
Pydantic model raised ``AttributeError``. The fix splits the parameters into
|
||
|
|
``request: Request`` (FastAPI) and ``body: ChatRequest`` (Pydantic).
|
||
|
|
"""
|
||
|
|
|
||
|
|
def test_chat_without_auth_returns_402(self, serve_client):
|
||
|
|
"""Unauthenticated request should get a 402 challenge."""
|
||
|
|
resp = serve_client.post(
|
||
|
|
"/serve/chat",
|
||
|
|
json={"message": "Hello"},
|
||
|
|
)
|
||
|
|
assert resp.status_code == 402
|
||
|
|
data = resp.json()
|
||
|
|
assert data["error"] == "Payment Required"
|
||
|
|
assert "macaroon" in data
|
||
|
|
assert "invoice" in data
|
||
|
|
|
||
|
|
@patch("timmy_serve.app.create_timmy")
|
||
|
|
@patch("timmy_serve.app.verify_l402_token", return_value=True)
|
||
|
|
def test_chat_with_valid_l402_token(self, mock_verify, mock_create, serve_client):
|
||
|
|
"""Authenticated request should reach the chat handler without AttributeError."""
|
||
|
|
mock_agent = MagicMock()
|
||
|
|
mock_result = MagicMock()
|
||
|
|
mock_result.content = "I am Timmy."
|
||
|
|
mock_agent.run.return_value = mock_result
|
||
|
|
mock_create.return_value = mock_agent
|
||
|
|
|
||
|
|
resp = serve_client.post(
|
||
|
|
"/serve/chat",
|
||
|
|
json={"message": "Who are you?"},
|
||
|
|
headers={"Authorization": "L402 fake-macaroon:fake-preimage"},
|
||
|
|
)
|
||
|
|
# The key assertion: we must NOT get a 500 from AttributeError
|
||
|
|
assert resp.status_code == 200
|
||
|
|
data = resp.json()
|
||
|
|
assert data["response"] == "I am Timmy."
|
||
|
|
mock_agent.run.assert_called_once_with("Who are you?", stream=False)
|
||
|
|
|
||
|
|
@patch("timmy_serve.app.create_timmy")
|
||
|
|
@patch("timmy_serve.app.verify_l402_token", return_value=True)
|
||
|
|
def test_chat_reads_auth_header_from_request(
|
||
|
|
self, mock_verify, mock_create, serve_client
|
||
|
|
):
|
||
|
|
"""Ensure auth header is read from the HTTP Request, not the JSON body."""
|
||
|
|
mock_agent = MagicMock()
|
||
|
|
mock_result = MagicMock()
|
||
|
|
mock_result.content = "ok"
|
||
|
|
mock_agent.run.return_value = mock_result
|
||
|
|
mock_create.return_value = mock_agent
|
||
|
|
|
||
|
|
resp = serve_client.post(
|
||
|
|
"/serve/chat",
|
||
|
|
json={"message": "test"},
|
||
|
|
headers={"Authorization": "L402 abc:def"},
|
||
|
|
)
|
||
|
|
assert resp.status_code == 200
|
||
|
|
# Should not raise AttributeError on request.headers
|