import json import flask def fix_data(data): return data.replace(b'\\"', b'"') def test_errors(problem_app): app_client = problem_app.app.test_client() greeting404 = app_client.get("/v1.0/greeting") # type: flask.Response assert greeting404.content_type == "application/problem+json" assert greeting404.status_code == 404 error404 = flask.json.loads(fix_data(greeting404.data)) assert error404["type"] == "about:blank" assert error404["title"] == "Not Found" assert ( error404["detail"] == "The requested URL was not found on the server. " "If you entered the URL manually please check your spelling and try again." ) assert error404["status"] == 404 assert "instance" not in error404 get_greeting = app_client.get("/v1.0/greeting/jsantos") # type: flask.Response assert get_greeting.content_type == "application/problem+json" assert get_greeting.status_code == 405 error405 = json.loads(get_greeting.data.decode("utf-8", "replace")) assert error405["type"] == "about:blank" assert error405["title"] == "Method Not Allowed" assert error405["status"] == 405 assert "instance" not in error405 get500 = app_client.get("/v1.0/except") # type: flask.Response assert get500.content_type == "application/problem+json" assert get500.status_code == 500 error500 = json.loads(get500.data.decode("utf-8", "replace")) assert error500["type"] == "about:blank" assert error500["title"] == "Internal Server Error" assert ( error500["detail"] == "The server encountered an internal error and was unable to complete your request. " "Either the server is overloaded or there is an error in the application." ) assert error500["status"] == 500 assert "instance" not in error500 get_problem = app_client.get("/v1.0/problem") # type: flask.Response assert get_problem.content_type == "application/problem+json" assert get_problem.status_code == 402 assert get_problem.headers["x-Test-Header"] == "In Test" error_problem = json.loads(get_problem.data.decode("utf-8", "replace")) assert error_problem["type"] == "http://www.example.com/error" assert error_problem["title"] == "Some Error" assert error_problem["detail"] == "Something went wrong somewhere" assert error_problem["status"] == 402 assert error_problem["instance"] == "instance1" get_problem2 = app_client.get("/v1.0/other_problem") # type: flask.Response assert get_problem2.content_type == "application/problem+json" assert get_problem2.status_code == 402 error_problem2 = json.loads(get_problem2.data.decode("utf-8", "replace")) assert error_problem2["type"] == "about:blank" assert error_problem2["title"] == "Some Error" assert error_problem2["detail"] == "Something went wrong somewhere" assert error_problem2["status"] == 402 assert error_problem2["instance"] == "instance1" problematic_json = app_client.get( "/v1.0/json_response_with_undefined_value_to_serialize" ) # type: flask.Response assert problematic_json.status_code == 500 custom_problem = app_client.get("/v1.0/customized_problem_response") assert custom_problem.status_code == 403 problem_body = json.loads(custom_problem.data.decode("utf-8", "replace")) assert "amount" in problem_body assert problem_body["amount"] == 23.0 problem_as_exception = app_client.get("/v1.0/problem_exception_with_extra_args") assert problem_as_exception.status_code == 400 problem_as_exception_body = json.loads( problem_as_exception.data.decode("utf-8", "replace") ) assert "age" in problem_as_exception_body assert problem_as_exception_body["age"] == 30 unsupported_media_type = app_client.post( "/v1.0/post_wrong_content_type", data="", content_type="text/html" ) assert unsupported_media_type.status_code == 415 unsupported_media_type_body = json.loads( unsupported_media_type.data.decode("utf-8", "replace") ) assert unsupported_media_type_body["type"] == "about:blank" assert unsupported_media_type_body["title"] == "Unsupported Media Type" assert ( unsupported_media_type_body["detail"] == "Invalid Content-type (text/html), expected JSON data" ) assert unsupported_media_type_body["status"] == 415