Question

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

我的要求目前是：

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

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

我的调查：

Flask Client（在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.

所以看起来它不受支持。但是，我怎么可以使用这样的功能呢？

Answer 1

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'});

Answer 2

Werkzeug救援！

from werkzeug.test import EnvironBuilder, run_wsgi_app
from werkzeug.wrappers import Request

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

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

2 个答案: