def my_example(event, context):
try:
response = example_service(event)
except HttpBaseException as e:
return e.get_response()
return HttpResponse.success(response)
def __init__(self, status_code, body=None):
super().__init__(HttpResponse.build(status_code), body)
def __init__(self, body=None):
super().__init__(HttpResponse.error(), body)
def __init__(self, body=None):
super().__init__(HttpResponse.unprocessable(), body)
def __init__(self, body=None):
super().__init__(HttpResponse.forbidden(), body)
def __init__(self, body=None):
super().__init__(HttpResponse.unauthorized(), body)
def __init__(self, body=None):
super().__init__(HttpResponse.bad_request(), body)
def __init__(self, body=None):
super().__init__(HttpResponse.not_found(), body)
def test_get_response():
response = exceptions.HttpBaseException(
HttpResponse.success()).get_response()
assert response == {'statusCode': 200, 'body': None}
def test_get_response_error():
response = exceptions.HttpBaseException(
HttpResponse.error()).get_response()
assert response.get('statusCode') == 500
def test_get_response_unauthorized():
response = exceptions.HttpBaseException(
HttpResponse.unauthorized()).get_response()
assert response.get('statusCode') == 401
def test_get_response_bad_request():
response = exceptions.HttpBaseException(
HttpResponse.bad_request()).get_response()
assert response.get('statusCode') == 400
def test_get_response_with_body():
response = exceptions.HttpBaseException(HttpResponse.success(),
body={
'message': 'ok'
}).get_response()
assert response == {'statusCode': 200, 'body': '{"message": "ok"}'}