Flask 和 Werkzeug：使用自定义标头测试发布请求

Posted 2023-02-23

【中文标题】Flask 和 Werkzeug：使用自定义标头测试发布请求

【英文标题】：Flask and Werkzeug: Testing a post request with custom headers

【发布时间】：2013-08-18 07:13:32

【问题描述】：

我目前正在使用来自 http://flask.pocoo.org/docs/testing/ 的建议测试我的应用，但我想在发布请求中添加标头。

我的请求目前是：

self.app.post('/v0/scenes/test/foo', data=dict(image=(StringIO('fake image'), 'image.png')))

但我想在请求中添加一个 content-md5。这可能吗？

我的调查：

Flask 客户端（在 flask/testing.py 中）扩展了 Werkzeug 的客户端，记录在这里： http://werkzeug.pocoo.org/docs/test/

如您所见，post 使用 open。但是open只有：

Parameters: 
 as_tuple – Returns a tuple in the form (environ, result)
 buffered – Set this to True to buffer the application run. This will automatically close the application for you as well.
 follow_redirects – Set this to True if the Client should follow HTTP redirects.

所以它看起来不受支持。不过，我怎样才能让这样的功能发挥作用呢？

【参考方案1】：

Werkzeug 来救援！

from werkzeug.test import EnvironBuilder, run_wsgi_app

builder = EnvironBuilder(path='/v0/scenes/bucket/foo', method='POST', data='image': (StringIO('fake image'), 'image.png'), \
    headers='content-md5': 'some hash')
env = builder.get_environ()

(app_iter, status, headers) = run_wsgi_app(http.app.wsgi_app, env)
status = int(status[:3]) # output will be something like 500 INTERNAL SERVER ERROR

【讨论】：

为什么会引入无关的请求？

【参考方案2】：

open 也将 *args 和 **kwargs 用作 EnvironBuilder 参数。所以你可以在你的第一个帖子请求中添加 headers 参数：

with self.app.test_client() as client:
    client.post('/v0/scenes/test/foo',
                data=dict(image=(StringIO('fake image'), 'image.png')),
                headers='content-md5': 'some hash');