fix __root__

setup.py

@@ -14,7 +14,7 @@
 
 setup(
     name="spectree",
-    version="0.3.15",
+    version="0.3.16",
     author="Keming Yang",
     author_email="kemingy94@gmail.com",
     description=(

spectree/plugins/falcon_plugin.py

@@ -168,7 +168,7 @@ def validate(
             model = resp.find_model(_resp.status[:3])
             if model:
                 try:
-                    model.validate(_resp.media)
+                    model.parse_obj(_resp.media)
                 except ValidationError as err:
                     resp_validation_error = err
                     _resp.status = "500 Internal Service Response Validation Error"

spectree/plugins/flask_plugin.py

@@ -160,7 +160,7 @@ def validate(
             model = resp.find_model(response.status_code)
             if model:
                 try:
-                    model.validate(response.get_json())
+                    model.parse_obj(response.get_json())
                 except ValidationError as err:
                     resp_validation_error = err
                     response = make_response(

spectree/plugins/starlette_plugin.py

@@ -88,7 +88,7 @@ async def validate(
             model = resp.find_model(response.status_code)
             if model:
                 try:
-                    model.validate(json_loads(response.body))
+                    model.parse_obj(json_loads(response.body))
                 except ValidationError as err:
                     resp_validation_error = err
                     response = JSONResponse(err.errors(), 500)

tests/common.py

@@ -1,5 +1,5 @@
 from enum import Enum, IntEnum
-from typing import List
+from typing import Dict, List
 
 from pydantic import BaseModel, Field, root_validator
 
@@ -18,6 +18,10 @@ class JSON(BaseModel):
     limit: int
 
 
+class StrDict(BaseModel):
+    __root__: Dict[str, str]
+
+
 class Resp(BaseModel):
     name: str
     score: List[int]

tests/test_plugin_falcon.py

@@ -6,7 +6,7 @@
 
 from spectree import Response, SpecTree
 
-from .common import JSON, Cookies, Headers, Query, Resp
+from .common import JSON, Cookies, Headers, Query, Resp, StrDict
 
 
 def before_handler(req, resp, err, instance):
@@ -40,6 +40,7 @@ class UserScore:
     def extra_method(self):
         pass
 
+    @api.validate(resp=Response(HTTP_200=StrDict))
     def on_get(self, req, resp, name):
         self.extra_method()
         resp.media = {"name": name}

tests/test_plugin_flask.py

@@ -6,7 +6,7 @@
 
 from spectree import Response, SpecTree
 
-from .common import JSON, Cookies, Headers, Query, Resp
+from .common import JSON, Cookies, Headers, Query, Resp, StrDict
 
 
 def before_handler(req, resp, err, _):
@@ -27,7 +27,7 @@ def api_after_handler(req, resp, err, _):
 
 
 @app.route("/ping")
-@api.validate(headers=Headers, tags=["test", "health"])
+@api.validate(headers=Headers, resp=Response(HTTP_200=StrDict), tags=["test", "health"])
 def ping():
     """summary
     description"""

tests/test_plugin_flask_blueprint.py

@@ -6,7 +6,7 @@
 
 from spectree import Response, SpecTree
 
-from .common import JSON, Cookies, Headers, Query, Resp
+from .common import JSON, Cookies, Headers, Query, Resp, StrDict
 
 
 def before_handler(req, resp, err, _):
@@ -27,7 +27,7 @@ def api_after_handler(req, resp, err, _):
 
 
 @app.route("/ping")
-@api.validate(headers=Headers, tags=["test", "health"])
+@api.validate(headers=Headers, resp=Response(HTTP_200=StrDict), tags=["test", "health"])
 def ping():
     """summary
     description"""

tests/test_plugin_starlette.py

@@ -10,7 +10,7 @@
 
 from spectree import Response, SpecTree
 
-from .common import JSON, Cookies, Headers, Query, Resp
+from .common import JSON, Cookies, Headers, Query, Resp, StrDict
 
 
 def before_handler(req, resp, err, instance):
@@ -32,7 +32,12 @@ def method_handler(req, resp, err, instance):
 class Ping(HTTPEndpoint):
     name = "Ping"
 
-    @api.validate(headers=Headers, tags=["test", "health"], after=method_handler)
+    @api.validate(
+        headers=Headers,
+        resp=Response(HTTP_200=StrDict),
+        tags=["test", "health"],
+        after=method_handler,
+    )
     def get(self, request):
         """summary
         description"""