def test_bad_request_other(self): """ Tests a bad request with no classification. """ # Test e = exceptions.BadRequestException({}) code = self.exception_handler.handle_bad_request(e) # Verify self.assertEqual(code, handler.CODE_BAD_REQUEST) self.assertTrue('incorrect' in self.recorder.lines[0])
def test_bad_request_missing_properties(self): """ Tests the missing properties classification of bad request exceptions. """ # Test e = exceptions.BadRequestException({'missing_property_names': ['foo']}) code = self.exception_handler.handle_bad_request(e) # Verify self.assertEqual(code, handler.CODE_BAD_REQUEST) self.assertTrue('not provided' in self.recorder.lines[0]) self.assertTrue('foo' in self.recorder.lines[0])
def test__display_coded_error_single_error(self): """ Test that render_failure_message is called when a coded error is passed in. """ self.prompt.render_failure_message = mock.MagicMock() e = exceptions.BadRequestException( {'error': { 'description': 'test_error' }}) self.exception_handler._display_coded_error(e.extra_data.get('error')) self.prompt.render_failure_message.assert_called_once_with( 'test_error')
def test_bad_request_with_error(self): """ Tests the errors classification of bad request exceptions. """ self.exception_handler._display_coded_error = mock.MagicMock() e = exceptions.BadRequestException( {'error': { 'description': 'test_error' }}) code = self.exception_handler.handle_bad_request(e) self.assertEqual(code, handler.CODE_BAD_REQUEST) self.exception_handler._display_coded_error.assert_called_once_with( {'description': 'test_error'})
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 = []