pythonasyncioapiasyncfastapiframeworkjsonjson-schemaopenapiopenapi3pydanticpython-typespython3redocreststarletteswaggerswagger-uiuvicornweb
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.0 KiB
80 lines
2.0 KiB
from typing import Annotated
|
|
|
|
import pytest
|
|
from fastapi import Body, FastAPI
|
|
from fastapi.exceptions import ResponseValidationError
|
|
from fastapi.testclient import TestClient
|
|
from pydantic import v1
|
|
|
|
|
|
class Item(v1.BaseModel):
|
|
name: str
|
|
description: str | None = None
|
|
price: float
|
|
tax: float | None = None
|
|
tags: list = []
|
|
|
|
|
|
class Model(v1.BaseModel):
|
|
name: str
|
|
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
@app.post("/request_body")
|
|
async def request_body(body: Annotated[Item, Body()]):
|
|
return body
|
|
|
|
|
|
@app.get("/response_model", response_model=Model)
|
|
async def response_model():
|
|
return Model(name="valid_model")
|
|
|
|
|
|
@app.get("/response_model__invalid", response_model=Model)
|
|
async def response_model__invalid():
|
|
return 1
|
|
|
|
|
|
@app.get("/response_model_list", response_model=list[Model])
|
|
async def response_model_list():
|
|
return [Model(name="valid_model")]
|
|
|
|
|
|
@app.get("/response_model_list__invalid", response_model=list[Model])
|
|
async def response_model_list__invalid():
|
|
return [1]
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
class TestResponseModel:
|
|
def test_simple__valid(self):
|
|
response = client.get("/response_model")
|
|
assert response.status_code == 200
|
|
assert response.json() == {"name": "valid_model"}
|
|
|
|
def test_simple__invalid(self):
|
|
with pytest.raises(ResponseValidationError):
|
|
client.get("/response_model__invalid")
|
|
|
|
def test_list__valid(self):
|
|
response = client.get("/response_model_list")
|
|
assert response.status_code == 200
|
|
assert response.json() == [{"name": "valid_model"}]
|
|
|
|
def test_list__invalid(self):
|
|
with pytest.raises(ResponseValidationError):
|
|
client.get("/response_model_list__invalid")
|
|
|
|
|
|
class TestRequestBody:
|
|
def test_model__valid(self):
|
|
response = client.post("/request_body", json={"name": "myname", "price": 1.0})
|
|
assert response.status_code == 200, response.text
|
|
|
|
def test_model__invalid(self):
|
|
response = client.post("/request_body", json={"name": "myname"})
|
|
assert response.status_code == 422, response.text
|
|
|