Merge pull request #64 from decaz/fix-schema-many

Fix wrapping of data parsed by schema with `many=True`

Author: sloria

flask_apispec/wrapper.py

@@ -35,7 +35,10 @@ def call_view(self, *args, **kwargs):
             for option in annotation.options:
                 schema = utils.resolve_instance(option['args'])
                 parsed = parser.parse(schema, locations=option['kwargs']['locations'])
-                kwargs.update(parsed)
+                if getattr(schema, 'many', False):
+                    args += tuple(parsed)
+                else:
+                    kwargs.update(parsed)
         return self.func(*args, **kwargs)
 
     def marshal_result(self, unpacked, status_code):

tests/test_views.py

@@ -1,5 +1,7 @@
 # -*- coding: utf-8 -*-
 
+import json
+
 from flask import make_response
 from marshmallow import fields, Schema
 
@@ -31,6 +33,18 @@ def view(**kwargs):
         res = client.get('/', {'name': 'freddie'})
         assert res.json == {'name': 'freddie'}
 
+    def test_use_kwargs_schema_many(self, app, client):
+        class ArgSchema(Schema):
+            name = fields.Str()
+
+        @app.route('/', methods=('POST',))
+        @use_kwargs(ArgSchema(many=True), locations=('json',))
+        def view(*args):
+            return list(args)
+        data = [{'name': 'freddie'}, {'name': 'john'}]
+        res = client.post('/', json.dumps(data), content_type='application/json')
+        assert res.json == data
+
     def test_use_kwargs_multiple(self, app, client):
         @app.route('/')
         @use_kwargs({'name': fields.Str()})