init

tests/test_api.py

@@ -0,0 +1,51 @@
+import pytest
+from fastapi import FastAPI
+from fastapi.testclient import TestClient
+from pydantic import ValidationError
+
+from app.speech import SpeechRequest, router
+
+
+@pytest.fixture
+def client():
+    app = FastAPI()
+    app.include_router(router)
+    return TestClient(app)
+
+
+@pytest.mark.parametrize("bad", ["loud", "nope", "x-slower", "1.5"])
+def test_rate_invalid_rejected_by_model(bad):
+    with pytest.raises(ValidationError):
+        SpeechRequest(voice="a", input="текст", rate=bad)
+
+
+@pytest.mark.parametrize("bad", ["loud", "very-high"])
+def test_pitch_invalid_rejected_by_model(bad):
+    with pytest.raises(ValidationError):
+        SpeechRequest(voice="a", input="текст", pitch=bad)
+
+
+@pytest.mark.parametrize("good", ["x-slow", "slow", "medium", "fast", "x-fast"])
+def test_rate_valid_accepted(good):
+    assert SpeechRequest(voice="a", input="текст", rate=good).rate == good
+
+
+@pytest.mark.parametrize("good", ["x-low", "low", "medium", "high", "x-high"])
+def test_pitch_valid_accepted(good):
+    assert SpeechRequest(voice="a", input="текст", pitch=good).pitch == good
+
+
+def test_endpoint_returns_422_for_invalid_rate(client):
+    r = client.post(
+        "/v1/audio/speech",
+        json={"model": "UniosTTS", "voice": "baya", "input": "привет", "rate": "loud"},
+    )
+    assert r.status_code == 422
+
+
+def test_endpoint_returns_422_for_invalid_pitch(client):
+    r = client.post(
+        "/v1/audio/speech",
+        json={"model": "UniosTTS", "voice": "baya", "input": "привет", "pitch": "loud"},
+    )
+    assert r.status_code == 422