lqs
d20566fe07
Some checks failed
Test / lint (push) Has been cancelled
Test / test (pydantic-v1, 3.10) (push) Has been cancelled
Test / test (pydantic-v1, 3.11) (push) Has been cancelled
Test / test (pydantic-v1, 3.12) (push) Has been cancelled
Test / test (pydantic-v1, 3.8) (push) Has been cancelled
Test / test (pydantic-v1, 3.9) (push) Has been cancelled
Test / test (pydantic-v2, 3.10) (push) Has been cancelled
Test / test (pydantic-v2, 3.11) (push) Has been cancelled
Test / test (pydantic-v2, 3.12) (push) Has been cancelled
Test / test (pydantic-v2, 3.8) (push) Has been cancelled
Test / test (pydantic-v2, 3.9) (push) Has been cancelled
Test / coverage-combine (push) Has been cancelled
Test / check (push) Has been cancelled
Issue Manager / issue-manager (push) Has been cancelled
Label Approved / label-approved (push) Has been cancelled
85 lines
2.0 KiB
Python
85 lines
2.0 KiB
Python
from typing import List, Optional, Union
|
|
|
|
import pytest
|
|
from fastapi import FastAPI
|
|
from fastapi.exceptions import ResponseValidationError
|
|
from fastapi.testclient import TestClient
|
|
from pydantic import BaseModel
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
class Item(BaseModel):
|
|
name: str
|
|
price: Optional[float] = None
|
|
owner_ids: Optional[List[int]] = None
|
|
|
|
|
|
@app.get("/items/invalid", response_model=Item)
|
|
def get_invalid():
|
|
return {"name": "invalid", "price": "foo"}
|
|
|
|
|
|
@app.get("/items/invalidnone", response_model=Item)
|
|
def get_invalid_none():
|
|
return None
|
|
|
|
|
|
@app.get("/items/validnone", response_model=Union[Item, None])
|
|
def get_valid_none(send_none: bool = False):
|
|
if send_none:
|
|
return None
|
|
else:
|
|
return {"name": "invalid", "price": 3.2}
|
|
|
|
|
|
@app.get("/items/innerinvalid", response_model=Item)
|
|
def get_innerinvalid():
|
|
return {"name": "double invalid", "price": "foo", "owner_ids": ["foo", "bar"]}
|
|
|
|
|
|
@app.get("/items/invalidlist", response_model=List[Item])
|
|
def get_invalidlist():
|
|
return [
|
|
{"name": "foo"},
|
|
{"name": "bar", "price": "bar"},
|
|
{"name": "baz", "price": "baz"},
|
|
]
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
def test_invalid():
|
|
with pytest.raises(ResponseValidationError):
|
|
client.get("/items/invalid")
|
|
|
|
|
|
def test_invalid_none():
|
|
with pytest.raises(ResponseValidationError):
|
|
client.get("/items/invalidnone")
|
|
|
|
|
|
def test_valid_none_data():
|
|
response = client.get("/items/validnone")
|
|
data = response.json()
|
|
assert response.status_code == 200
|
|
assert data == {"name": "invalid", "price": 3.2, "owner_ids": None}
|
|
|
|
|
|
def test_valid_none_none():
|
|
response = client.get("/items/validnone", params={"send_none": "true"})
|
|
data = response.json()
|
|
assert response.status_code == 200
|
|
assert data is None
|
|
|
|
|
|
def test_double_invalid():
|
|
with pytest.raises(ResponseValidationError):
|
|
client.get("/items/innerinvalid")
|
|
|
|
|
|
def test_invalid_list():
|
|
with pytest.raises(ResponseValidationError):
|
|
client.get("/items/invalidlist")
|