def test_conflict_error(self):
base = BaseController()
base.conflict = MagicMock('aaa')
base.handle_business_errors(ConflictError('conflict'))
base.conflict.assert_called_with()
def test_forbidden_error(self):
base = BaseController()
base.forbidden = MagicMock('aaa')
base.handle_business_errors(PermissionDeniedError())
base.forbidden.assert_called_with()
async def test_authorization_error(self):
base = BaseController()
base.forbidden = MagicMock('aaa')
base.handle_service_errors(AuthorizationError(403, ''))
base.forbidden.assert_called_with()
async def test_not_found_error(self):
base = BaseController()
base.not_found = MagicMock('aaa')
base.handle_service_errors(NotFoundError(404, ''))
base.not_found.assert_called_with()
async def test_authentication_error(self):
base = BaseController()
base.unauthorized = MagicMock('aaa')
base.handle_service_errors(AuthenticationError(401, ''))
base.unauthorized.assert_called_with()
def test_not_found_error(self):
base = BaseController()
base.not_found = MagicMock('aaa')
base.handle_business_errors(ResourceNotFoundError())
base.not_found.assert_called_with()
def test_invalid_data(self):
base = BaseController()
base.bad_request = MagicMock('aaa')
error = {'field': 'invalid'}
base.handle_business_errors(ValidationError(error))
base.bad_request.assert_called_with({'field': 'invalid'})
async def test_unexpected_error(self):
base = BaseController()
base.internal_server_error = MagicMock()
exception = UnexpectedError(500, '')
base.handle_service_errors(exception)
base.internal_server_error.assert_called_with(exception=exception)
def test_get_identity(self):
request = Mock()
request.headers.get.return_value = f'Bearer {self.custom_token}'
contr = BaseController()
identity = contr.get_identity(request)
assert isinstance(identity, RequestScope)
async def test_invalid_data(self):
base = BaseController()
base.bad_request = MagicMock('aaa')
error = {'code': 'some_code', 'reason': {'error': 'something'}}
base.handle_service_errors(InvalidDataError(400, json.dumps(error)))
base.bad_request.assert_called_with({'error': 'something'})
def test_internal_error(self):
base = BaseController()
base.internal_server_error = MagicMock('aaa')
exception = InternalError()
base.handle_business_errors(exception)
base.internal_server_error.assert_called_with(exception=exception)
def test_unauthorized(self, mocked_logger):
""" Default HTTP 401 with errors
"""
base = BaseController()
with pytest.raises(web.HTTPUnauthorized) as expected:
base.unauthorized()
mocked_logger.assert_called_once()
assert expected.value.status == 401
def test_internal_server_error(self, mocked_logger):
""" Default HTTP 500 with errors
"""
base = BaseController()
with pytest.raises(web.HTTPInternalServerError) as expected:
base.internal_server_error(ValueError())
mocked_logger.assert_called_once()
assert expected.value.status == 500
def test_payment_required(self, mocked_logger):
""" Default HTTP 402 with errors
"""
base = BaseController()
with pytest.raises(web.HTTPPaymentRequired) as expected:
base.payment_required(code='aaa', reason='bbb')
mocked_logger.assert_called_once()
assert expected.value.status == 402
def test_forbidden(self, mocked_logger):
""" Default HTTP 403 with errors
"""
base = BaseController()
with pytest.raises(web.HTTPForbidden) as expected:
base.forbidden()
mocked_logger.assert_called_once()
assert expected.value.status == 403
def test_not_found(self, mocked_logger):
""" Default HTTP 404 with errors
"""
base = BaseController()
with pytest.raises(web.HTTPNotFound) as expected:
base.not_found()
mocked_logger.assert_called_once()
assert expected.value.status == 404
def test_conflict(self, mocked_logger):
""" Default HTTP 409 with errors
"""
base = BaseController()
with pytest.raises(web.HTTPConflict) as expected:
base.conflict()
mocked_logger.assert_called_once()
assert expected.value.status == 409
def test_invalid_jwt(self):
request = Mock()
request.headers.get.return_value = 'Bearer aa.aa.aa'
contr = BaseController()
contr.bad_request = MagicMock()
_ = contr.get_identity(request)
contr.bad_request.assert_called_once()
def test_response_ok(self, mocked_resp):
""" Default HTTP 200 with data
"""
base = BaseController()
mocked_resp.return_value = 'abc'
resp = base.response_ok('abc')
assert resp == 'abc'
mocked_resp.assert_called_with('abc')
def test_resp_for_creation(self, mocked_resp):
""" HTTP 201 with new id
"""
base = BaseController()
mocked_resp.return_value = 'abc'
resp = base.resp_for_creation('abc')
assert resp == 'abc'
mocked_resp.assert_called_with({'id': 'abc'}, status=201)
async def test_payment_required_error(self):
base = BaseController()
base.payment_required = MagicMock('aaa')
error = {'code': 'some_code', 'reason': {'error': 'something'}}
base.handle_service_errors(PaymentRequiredError(
402, json.dumps(error)))
base.payment_required.assert_called_with('some_code',
{'error': 'something'})
def test_bad_request(self, mocked_logger):
""" Default HTTP 400 with errors
"""
base = BaseController()
errors = {'error': 'general error'}
with pytest.raises(web.HTTPBadRequest) as expected:
base.bad_request(errors)
mocked_logger.assert_called_once()
resp = json.loads(expected.value.text)
assert resp['reason'] == errors
def test_resp_for_listing(self, mocked_resp):
""" Default HTTP 200 with list
"""
base = BaseController()
paginator = Mock()
paginator.get_info.return_value = 'aaa'
mocked_resp.return_value = 'abc'
resp = base.resp_for_listing(paginator, ['1234'])
assert resp == 'abc'
mocked_resp.assert_called_with({
'items': ['1234'],
'pagination': 'aaa'
})
def test_other_exceptions_error(self):
base = BaseController()
with pytest.raises(ValueError):
base.handle_business_errors(ValueError())