web-dev-qa-db-ja.com

Flask and Werkzeug：カスタムヘッダーを使用した投稿リクエストのテスト

現在、 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のClientを拡張し、ここに文書化されています： 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.

そのため、サポートされていないようです。しかし、どのようにしてそのような機能を動作させることができますか？

pythonflaskwerkzeug
33
theicfire

openも*argsおよび**kwargsEnvironBuilder引数として使用します。したがって、最初の投稿リクエストに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'});
61
tbicr

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
6
theicfire