|
import json |
|
from fastapi.testclient import TestClient |
|
from megabots import bot, create_api |
|
|
|
qnabot = bot("qna-over-docs", index="./examples/files") |
|
app = create_api(qnabot) |
|
|
|
client = TestClient(app) |
|
|
|
|
|
def test_successful_response(): |
|
response = client.get("/v1/ask/What is your name?") |
|
assert response.status_code == 200 |
|
assert "answer" in response.json() |
|
assert isinstance(response.json()["answer"], str) |
|
|
|
|
|
def test_missing_question_parameter(): |
|
response = client.get("/v1/ask/") |
|
assert response.status_code == 404 |
|
assert response.json() == {"detail": "Not Found"} |
|
|