def test_apache_server_error(self):
"""
Tests handling the case where Apache raised an exception.
"""
# Test
msg = 'Test Message'
e = exceptions.ApacheServerException(msg)
code = self.exception_handler.handle_apache_error(e)
# Verify
self.assertEqual(code, handler.CODE_APACHE_SERVER_EXCEPTION)
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
def _handle_exceptions(self, response_code, response_body):
code_class_mappings = {400 : exceptions.BadRequestException,
401 : exceptions.PermissionsException,
404 : exceptions.NotFoundException,
409 : exceptions.ConflictException}
if response_code not in code_class_mappings:
# Apache errors are simply strings as compared to Pulp's dicts,
# so differentiate based on that so we don't get a parse error
if isinstance(response_body, basestring):
raise exceptions.ApacheServerException(response_body)
else:
raise exceptions.PulpServerException(response_body)
else:
raise code_class_mappings[response_code](response_body)
def test_handle_exception(self):
"""
Tests the high level call that branches based on exception type for all types.
"""
# For each exception type, check that the proper code is returned and
# that a failure message has been output. For simplicity in those tests,
# reset the tags after each run.
code = self.exception_handler.handle_exception(
exceptions.BadRequestException({}))
self.assertEqual(code, handler.CODE_BAD_REQUEST)
self.assertEqual(3, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
exceptions.ConflictException({}))
self.assertEqual(code, handler.CODE_CONFLICT)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
exceptions.ConnectionException({}))
self.assertEqual(code, handler.CODE_CONNECTION_EXCEPTION)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
exceptions.NotFoundException({'resources': {
'repo_id': 'foo'
}}))
self.assertEqual(code, handler.CODE_NOT_FOUND)
self.assertEqual(2, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
exceptions.PermissionsException({}))
self.assertEqual(code, handler.CODE_PERMISSIONS_EXCEPTION)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
exceptions.PulpServerException({}))
self.assertEqual(code, handler.CODE_PULP_SERVER_EXCEPTION)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
InvalidConfig('Test Message'))
self.assertEqual(code, handler.CODE_INVALID_CONFIG)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
WrongHost('expected', 'actual'))
self.assertEqual(code, handler.CODE_WRONG_HOST)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
exceptions.ApacheServerException('Test Message'))
self.assertEqual(code, handler.CODE_APACHE_SERVER_EXCEPTION)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(gaierror())
self.assertEqual(code, handler.CODE_UNKNOWN_HOST)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(socket_error())
self.assertEqual(code, handler.CODE_SOCKET_ERROR)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
code = self.exception_handler.handle_exception(
exceptions.ClientCertificateExpiredException(CERT_FILENAME))
self.assertEqual(code, handler.CODE_PERMISSIONS_EXCEPTION)
self.assertEqual([TAG_FAILURE, TAG_PARAGRAPH],
self.prompt.get_write_tags())
self.prompt.tags = []
code = self.exception_handler.handle_exception(Exception({}))
self.assertEqual(code, handler.CODE_UNEXPECTED)
self.assertEqual(1, len(self.prompt.tags))
self.assertEqual(TAG_FAILURE, self.prompt.get_write_tags()[0])
self.prompt.tags = []
