def test_json_favored(self):
data = Person()
data.id = 1
data.name = "tester"
data.email = "*****@*****.**"
response_data = {"id": data.id, "name": data.name, "email": data.email}
app = flask.Flask(__name__)
with app.test_request_context(data=data.SerializeToString(),
method='POST',
content_type="application/x-protobuf",
headers={"Accept": "*/*"}):
@api(json, protobuf(receives=Person))
def view_method():
return flask.request.data_dict
response, status_code, headers = view_method()
self.assertEquals(response.status_code, 200)
self.assertEquals(loads(response.data), response_data)
self.assertEquals(response.mimetype, "application/json")
def test_simple_pb_request(self):
data = Person()
data.id = 1
data.name = "tester"
data.email = "*****@*****.**"
app = flask.Flask(__name__)
with app.test_request_context(
data=data.SerializeToString(),
method='POST',
content_type="application/x-protobuf",
headers={"Accept": "application/x-protobuf"}):
@api(protobuf(receives=Person))
def view_method():
self.assertEqual(flask.request.data_dict, {
'id': 1,
'name': 'tester',
'email': '*****@*****.**'
})
return 200
view_method()
