Question

我在Flask应用程序中有代码，在请求中使用JSON，我可以像这样获取JSON对象：

Request = request.get_json()

这一直运行良好，但我正在尝试使用Python的unittest模块创建单元测试，而我很难找到一种方法来发送带有请求的JSON。

response=self.app.post('/test_function', 
                       data=json.dumps(dict(foo = 'bar')))

这给了我：

>>> request.get_data()
'{"foo": "bar"}'
>>> request.get_json()
None

Flask似乎有一个JSON参数，您可以在帖子请求中设置json = dict（foo =＆＃39; bar＆＃39;），但我不知道如何使用unittest模块执行此操作

Answer 1

将帖子更改为

response=self.app.post('/test_function', 
                       data=json.dumps(dict(foo='bar')),
                       content_type='application/json')

修好了。

感谢user3012759。

Answer 2

更新：由于Flask 1.0发布的flask.testing.FlaskClient方法接受了json参数，并添加了Response.get_json方法，请参阅example。

对于Flask 0.x，您可以使用以下收据：

from flask import Flask, Response as BaseResponse, json
from flask.testing import FlaskClient
from werkzeug.utils import cached_property


class Response(BaseResponse):
    @cached_property
    def json(self):
        return json.loads(self.data)


class TestClient(FlaskClient):
    def open(self, *args, **kwargs):
        if 'json' in kwargs:
            kwargs['data'] = json.dumps(kwargs.pop('json'))
            kwargs['content_type'] = 'application/json'
        return super(TestClient, self).open(*args, **kwargs)


app = Flask(__name__)
app.response_class = Response
app.test_client_class = TestClient
app.testing = True

如何在单元测试中使用JSON发送请求

2 个答案: