Comment envoyer des requêtes avec des JSON dans des tests unitaires

J'ai du code dans une application Flask qui utilise des JSON dans la demande, et je peux obtenir l'objet JSON comme suit:

Request = request.get_json()

Cela fonctionne bien, mais j'essaie de créer des tests unitaires à l'aide du module unittest de Python et j'ai du mal à trouver un moyen d'envoyer un JSON avec la demande.

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

Cela me donne:

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

Flask semble avoir un argument JSON où vous pouvez définir json = dict (foo = 'bar') dans la demande de publication, mais je ne sais pas comment faire cela avec le module unittest.

Changer le message en

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

réparé.

Merci à user3012759.

MISE À JOUR: Puisque les flask.testing.FlaskClientméthodes publiées par Flask 1.0 acceptent l' jsonargument et la Response.get_jsonméthode ajoutés, voir l' exemple .

pour Flask 0.x, vous pouvez utiliser le reçu ci-dessous:

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