46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
from typing import List
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from fastapi import FastAPI
|
||
|
|
from fastapi.exceptions import FastAPIError
|
||
|
|
|
||
|
|
|
||
|
|
class NonPydanticModel:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def test_invalid_response_model_raises():
|
||
|
|
with pytest.raises(FastAPIError):
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
@app.get("/", response_model=NonPydanticModel)
|
||
|
|
def read_root():
|
||
|
|
pass # pragma: nocover
|
||
|
|
|
||
|
|
|
||
|
|
def test_invalid_response_model_sub_type_raises():
|
||
|
|
with pytest.raises(FastAPIError):
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
@app.get("/", response_model=List[NonPydanticModel])
|
||
|
|
def read_root():
|
||
|
|
pass # pragma: nocover
|
||
|
|
|
||
|
|
|
||
|
|
def test_invalid_response_model_in_responses_raises():
|
||
|
|
with pytest.raises(FastAPIError):
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
@app.get("/", responses={"500": {"model": NonPydanticModel}})
|
||
|
|
def read_root():
|
||
|
|
pass # pragma: nocover
|
||
|
|
|
||
|
|
|
||
|
|
def test_invalid_response_model_sub_type_in_responses_raises():
|
||
|
|
with pytest.raises(FastAPIError):
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
@app.get("/", responses={"500": {"model": List[NonPydanticModel]}})
|
||
|
|
def read_root():
|
||
|
|
pass # pragma: nocover
|