Merge pull request #149 from jmcarp/fix-tests

Fix tests

Author: sloria

tests/test_openapi.py

@@ -154,7 +154,7 @@ def path(self, app, spec, function_view):
         return spec._paths['/bands/{band_id}/']
 
     def test_responses(self, schemas, path):
-        response = path['delete']['responses'][204]
+        response = path['delete']['responses']['204']
         assert response['description'] == 'a deleted band'
         assert response['schema'] == {}
 

tests/test_views.py

@@ -271,7 +271,9 @@ def test_schemas_none(self, app, client, models, schemas):
         class ConcreteResource(MethodResource):
             @marshal_with(None, code=204)
             def delete(self, **kwargs):
-                return make_response('', 204)
+                response = make_response('', 204)
+                response.headers = {}
+                return response
 
         app.add_url_rule('/<id>/', view_func=ConcreteResource.as_view('concrete'))
         res = client.delete('/5/')