Merge pull request #178 from marshmallow-code/set_status_headers_on_reponse_object

Restore set status and headers on response object

Author: lafrech

flask_smorest/response.py

@@ -80,6 +80,8 @@ def wrapper(*args, **kwargs):
 
                 # If return value is a werkzeug BaseResponse, return it
                 if isinstance(result_raw, BaseResponse):
+                    set_status_and_headers_in_response(
+                        result_raw, status, headers)
                     return result_raw
 
                 # Dump result with schema if specified

tests/test_blueprint.py

@@ -8,6 +8,7 @@
 
 import marshmallow as ma
 
+import flask
 from flask.views import MethodView
 
 from flask_smorest import Api, Blueprint, Page
@@ -1118,14 +1119,14 @@ def test_blueprint_response_response_object(self, app, schemas):
         # Schema is ignored when response object is returned
         @blp.response(schemas.DocSchema, code=200)
         def func_response():
-            return {}, 201, {'X-header': 'test'}
+            return flask.jsonify({"test": "test"}), 201, {'X-header': 'test'}
 
         api.register_blueprint(blp)
 
         response = client.get('/test/response')
         assert response.status_code == 201
         assert response.status == '201 CREATED'
-        assert response.json == {}
+        assert response.json == {"test": "test"}
         assert response.headers['X-header'] == 'test'
 
     @pytest.mark.parametrize('decorate', (True, False))