67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
import pytest
|
|
from httpx import AsyncClient, ASGITransport
|
|
from main import app
|
|
from app.storage.database import init_db
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def setup_db():
|
|
await init_db()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_health():
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
res = await client.get("/api/health")
|
|
assert res.status_code == 200
|
|
data = res.json()
|
|
assert data["status"] == "healthy"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_route_only():
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
payload = {"prompt": "Write a Python script to benchmark models", "policy": "balanced"}
|
|
res = await client.post("/api/route", json=payload)
|
|
assert res.status_code == 200
|
|
data = res.json()
|
|
assert "decision" in data
|
|
assert "analysis" in data
|
|
assert data["analysis"]["task_type"] == "CODING"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_generate_end_to_end():
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
payload = {"prompt": "Debug this memory leak in python subprocess", "policy": "balanced"}
|
|
res = await client.post("/api/generate", json=payload)
|
|
assert res.status_code == 200
|
|
data = res.json()
|
|
assert "response" in data
|
|
assert "metrics" in data
|
|
assert len(data["response"]["content"]) > 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_models_list():
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
res = await client.get("/api/models")
|
|
assert res.status_code == 200
|
|
models = res.json()
|
|
assert len(models) >= 3
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_analytics():
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
res = await client.get("/api/analytics")
|
|
assert res.status_code == 200
|
|
data = res.json()
|
|
assert "total_requests" in data
|
|
assert "savings" in data
|