def test_process_request_for_nonstring(mocker):
resp_data = b"{\"status\": \"success\"}"
env = testing.create_environ()
req = Request(env)
req.method = 'POST'
mocker.patch('app.app_middleware.response_process.json_of')
resp = Response()
resp.body = resp_data
cut = ResponseJSONProcessMiddleware()
cut.process_response(req, resp, None, None)
app.app_middleware.response_process.json_of.assert_called_once_with(resp_data)
def test_invalid_process_request(mocker):
req_data = "\"service_id\": \"invoice_extraction1\",\"model_name\": \"invoice_extraction\"," \
"\"image\": \"test_data\"".encode("utf-8")
env = testing.create_environ(headers={
"Content-Type": "application/json",
"Content-Length": str(len(req_data))
})
req = Request(env)
req.method = 'POST'
mocker.patch.object(req.stream, 'read', return_value=req_data)
doc = {
"service_id": "invoice_extraction1",
"model_name": "invoice_extraction",
"image": "test_data"
}
resp = Response()
cut = RequestBodyJSONProcessMiddleware()
with raises(falcon.HTTPBadRequest):
cut.process_request(req, resp)
