tests/test_api.py

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"}