oc-assistant/app/tests/test_main.py
2024-11-07 16:58:29 -05:00

28 lines
898 B
Python

from fastapi.testclient import TestClient
from app.server import app # Asegúrate de importar tu aplicación FastAPI
# Crea un cliente de prueba
client = TestClient(app)
def test_process_text():
# Define el texto de entrada
input_text = {"text": "Hola, ¿cómo estás?"}
# Realiza una solicitud POST al endpoint
response = client.post("/process_text", json=input_text)
# Verifica que la respuesta tenga un código de estado 200
assert response.status_code == 200
# Verifica que la respuesta contenga la clave 'response'
assert 'response' in response.json()
# Verifica que la respuesta sea una lista
assert isinstance(response.json()['response'], list)
# Aquí puedes agregar más verificaciones
# según lo que esperas en la respuesta
# Por ejemplo, verificar que la lista no esté vacía
assert len(response.json()['response']) > 0