使用FastAPI TestClient测试返回422状态代码

pip install fastapi
pip install pytest

from fastapi import FastAPI

app = FastAPI()

@app.post("/example")
async def example(data: str):
    if not data:
        return {"error": "Missing data"}
    return {"message": "Data received"}

from fastapi.testclient import TestClient
from main import app

client = TestClient(app)

def test_example_missing_data():
    response = client.post("/example", json={})
    assert response.status_code == 422
    assert response.json() == {"error": "Missing data"}

def test_example_valid_data():
    response = client.post("/example", json={"data": "example"})
    assert response.status_code == 200
    assert response.json() == {"message": "Data received"}

pytest test_main.py