python - flask 测试客户端 : Testing DELETE request with data

Question

我正在尝试使用来自 http://flask.pocoo.org/docs/testing/ 的建议来测试我的 Flask 应用程序，但我无法弄清楚如何使用表单数据测试 DELETE 方法。

我的删除方法看起来像这样:

from flask.ext.restful import Resource, reqparse 
...
def delete(self):
    self.reqparse.add_argument('arg1', type=str, required=True, location='form')
    args = self.reqparse.parse_args()
    ...

我想测试一下:

def setUp(self):
    self.app = myApp.app.test_client()

def test_delete(self):
    rv = self.app.delete('MyEndPoint', data={'arg1', 'val'})

但它不起作用。我还在 werkzeug.test 中查看了 EnvironBuilder 的源代码，但仍然不知道如何传递数据。

Answer 1

我刚遇到同样的问题，这主要是因为 Werkzeug 的测试方法目前不支持设置 DELETE 请求的 content_type。

代码here展示了 Werkzeug 如何获取内容类型:

def _get_content_type(self):
    ct = self.headers.get('Content-Type')
    if ct is None and not self._input_stream:
        if self.method in ('POST', 'PUT', 'PATCH'):
            if self._files:
                return 'multipart/form-data'
            return 'application/x-www-form-urlencoded'
        return None
    return ct

如果没有 content_type，那么表单数据永远不会从 environ 中进入请求，因此您的 Flask 服务器实际上不会发送数据.

最终，这是 Werkzeug 的一个错误，因为您可以制作一个使用 DELETE 方法并包含表单数据的 curl 请求。我已经向 Werkzeug 存储库提交了一个拉取请求来解决这个问题。欢迎在 github 上发表意见:https://github.com/mitsuhiko/werkzeug/pull/620

更新:同时要实际解决您的问题，您可以通过在请求中明确说明内容类型来解决这个问题，如下所示:

def test_delete(self):
    rv = self.app.delete('MyEndPoint',
                         data={'arg1', 'val'},
                         headers={'Content-Type': 'application/x-www-form-urlencoded'})

再次更新:我提交的拉取请求已经过审查、完善和合并，并将包含在 Werkzeug 的 0.10 版本中，希望这不再是问题:)