python - app - Cómo enviar solicitudes con JSON en pruebas unitarias

python flask testing

Question

Tengo código dentro de una aplicación Flask que usa JSON en la solicitud, y puedo obtener el objeto JSON de esta manera:

Request = request.get_json()

Esto ha funcionado bien, sin embargo, estoy tratando de crear pruebas unitarias utilizando el módulo de prueba unitaria de Python y tengo dificultades para encontrar una manera de enviar un JSON con la solicitud.

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

Esto me da:

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

Flask parece tener un argumento JSON donde puede establecer json = dict (foo = ''bar'') dentro de la solicitud de publicación, pero no sé cómo hacerlo con el módulo unittest.

Answer 1

Cambiar la publicación a

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

arreglado.

Gracias a user3012759.

Answer 2

ACTUALIZACIÓN: desde que Flask 1.0 lanzó los métodos flask.testing.FlaskClient acepta el argumento json y el método Response.get_json agregado, vea el example .

para el Frasco 0.x puede usar el recibo a continuación:

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