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

我在 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 参数，您可以在 post 请求中设置 JSON = dict (foo = ‘ bar’) ，但是我不知道如何使用 unittest 模块进行设置。

49908 次浏览
小开
最佳答案

Changing the post to

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

fixed it.

Thanks to user3012759.

小开

Since Flask 1.0 release flask.testing.FlaskClient methods accepts json argument and Response.get_json method added, see pull request

    with app.test_client() as c:
rv = c.post('/api/auth', json={
'username': 'flask', 'password': 'secret'
})
json_data = rv.get_json()

For Flask 0.x compatibility you may use receipt below:

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

    

class Response(BaseResponse):
def get_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

提交答案