File size: 2,339 Bytes
829be0a b066638 829be0a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | import httpx
import pytest
from server.main import build_app
pytestmark = pytest.mark.asyncio
async def test_generate_returns_wav_bytes(monkeypatch, fake_classes):
monkeypatch.setattr("server.main._discover_adapter_classes", lambda: fake_classes)
monkeypatch.setattr("server.main.select_device", lambda: "cpu")
app = build_app()
from tests.conftest import lifespan_ctx
transport = httpx.ASGITransport(app=app)
async with lifespan_ctx(app), httpx.AsyncClient(transport=transport, base_url="http://t") as c:
r = await c.post(
"/api/generate",
data={
"text": "hello world",
"model_id": "fake",
"params": "{}",
},
)
assert r.status_code == 200
assert r.headers["content-type"].startswith("audio/wav")
assert r.content == b"FAKEWAV"
assert r.headers["x-seed-used"] == "0"
async def test_generate_unknown_model_404(monkeypatch, fake_classes):
monkeypatch.setattr("server.main._discover_adapter_classes", lambda: fake_classes)
monkeypatch.setattr("server.main.select_device", lambda: "cpu")
app = build_app()
from tests.conftest import lifespan_ctx
transport = httpx.ASGITransport(app=app)
async with lifespan_ctx(app), httpx.AsyncClient(transport=transport, base_url="http://t") as c:
r = await c.post(
"/api/generate",
data={"text": "x", "model_id": "nope", "params": "{}"},
)
assert r.status_code == 404
assert r.json()["error"]["code"] == "model_not_found"
async def test_generate_invalid_reference_returns_400(monkeypatch, fake_classes):
monkeypatch.setattr("server.main._discover_adapter_classes", lambda: fake_classes)
monkeypatch.setattr("server.main.select_device", lambda: "cpu")
app = build_app()
from tests.conftest import lifespan_ctx
transport = httpx.ASGITransport(app=app)
bad = b"not a wav"
async with lifespan_ctx(app), httpx.AsyncClient(transport=transport, base_url="http://t") as c:
r = await c.post(
"/api/generate",
data={"text": "x", "model_id": "fake", "params": "{}"},
files={"reference_wav": ("ref.wav", bad, "audio/wav")},
)
assert r.status_code == 400
assert r.json()["error"]["code"] == "reference_invalid"
|