44

我目前正在使用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.

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

4

2 回答 2

81

open也将*argsand **kwargswhich 用作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'});
于 2013-08-16T05:11:13.703 回答
8

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
于 2013-08-16T00:50:02.233 回答