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)