]> git.ipfire.org Git - thirdparty/fastapi/fastapi.git/commitdiff
increase test coverage
authorMohammed <barsintod@gmail.com>
Fri, 22 Mar 2019 22:13:09 +0000 (01:13 +0300)
committerMohammed <barsintod@gmail.com>
Fri, 22 Mar 2019 22:13:09 +0000 (01:13 +0300)
tests/test_additional_responses.py

index ddd5b76649d6d1d7801cc008d676bd2cb2dd4d09..aa18dfbc683ca159e4d0beed7bacfd0959123da9 100644 (file)
@@ -1,3 +1,4 @@
+import pytest
 from fastapi import FastAPI
 from fastapi.openapi.models import AdditionalResponse
 from pydantic import BaseModel
@@ -409,6 +410,19 @@ openapi_schema = {
 }
 
 
+def test_uncompatible_response_model_undecorated():
+    app = FastAPI()
+
+    class NotBaseModel:
+        pass
+
+    response_403 = AdditionalResponse(
+        status_code=403, description="Forbidden", models=[NotBaseModel]
+    )
+    with pytest.raises(ValueError):
+        app.add_api_route("/", get_not_decorated, additional_responses=[response_403])
+
+
 def test_openapi_schema():
     response = client.get("/openapi.json")
     assert response.status_code == 200