Flask и 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 (в 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.

Так что похоже, что он не поддерживается. Как я могу заставить такую ​​функцию работать?


python flask werkzeug
person theicfire    schedule 16.08.2013    source источник

Ответы (2)


arrow_upward
77
arrow_downward

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'});
person tbicr    schedule 16.08.2013

arrow_upward
8
arrow_downward

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
person theicfire    schedule 16.08.2013
comment
Зачем посторонний импорт Request? - person Marvin; 09.02.2019