Spaces:
Sleeping
Sleeping
File size: 2,916 Bytes
6822466 | 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 62 63 64 65 66 67 68 69 70 71 | from fastapi.testclient import TestClient
import os
import sys
import pytest
import base64
# Add app to path
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from app.main import app
from app.config import API_KEY
client = TestClient(app)
# Minimal valid MP3 for testing
VALID_MP3_B64 = "SUQzBAAAAAAAI1RTU0UAAAAPAAADTGF2ZjU4Ljc2LjEwMAAAAAAAAAAAAAAA//OEAAAAAAAAAAAAAAAAAAAAAAAASW5mbwAAAA8AAAAEAAABIADAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDB/wAAAP7AAAAAAAAAAABMYW1lMzkuOTUAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA//OEAAAAAAAAAAAAAAAAAAAAAAAASW5mbwAAAA8AAAAEAAABIADAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDB/wAAAP7AAAAAAAAAAABMYW1lMzkuOTUAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
def test_health_check():
response = client.get("/health")
assert response.status_code == 200
assert response.json() == {"status": "healthy", "service": "voice-detection-api"}
def test_auth_failure():
response = client.post("/api/voice-detection", json={})
assert response.status_code == 403
assert response.json()["status"] == "error"
def test_invalid_input_structure():
headers = {"x-api-key": API_KEY}
response = client.post("/api/voice-detection", json={"wrong": "field"}, headers=headers)
assert response.status_code == 400
assert response.json()["status"] == "error"
def test_valid_prediction_flow():
# This uses a mocked/random model since weights might be missing
# But verifies the pipeline runs end-to-end
headers = {"x-api-key": API_KEY}
payload = {
"language": "English",
"audioFormat": "mp3",
"audioBase64": VALID_MP3_B64
}
# We expect either success or a specific decoding error, but NOT 500
response = client.post("/api/voice-detection", json=payload, headers=headers)
# If 500, test fails. If 400 (decoding), it verifies error handling.
# Ideally 200.
if response.status_code == 200:
data = response.json()
assert data["status"] == "success"
assert "confidenceScore" in data
assert "explanation" in data
assert data["classification"] in ["AI_GENERATED", "HUMAN"]
elif response.status_code == 400:
# Acceptable if the minimal MP3 is too short for Librosa but handled gracefully
assert response.json()["status"] == "error"
else:
pytest.fail(f"Unexpected status code: {response.status_code}")
def test_invalid_language():
headers = {"x-api-key": API_KEY}
payload = {
"language": "French", # Not supported
"audioFormat": "mp3",
"audioBase64": VALID_MP3_B64
}
response = client.post("/api/voice-detection", json=payload, headers=headers)
assert response.status_code == 400
assert "Invalid language" in response.json()["message"]
|