def test_bind_no_errors(self, mock_query_manager, mock_unbind): mock_query_manager.return_value.get_group.return_value = {'consumer_ids': ['foo-consumer']} options = {'bar': 'baz'} mock_unbind.return_value = TaskResult(spawned_tasks=[{'task_id': 'foo-request-id'}]) result = consumer_group.unbind('foo_group_id', 'foo_repo_id', 'foo_distributor_id', options) mock_unbind.assert_called_once_with('foo-consumer', 'foo_repo_id', 'foo_distributor_id', options) self.assertEquals(result.spawned_tasks[0], {'task_id': 'foo-request-id'})
def test_bind_with_general_error(self, mock_query_manager, mock_unbind): mock_query_manager.return_value.get_group.return_value = {'consumer_ids': ['foo-consumer']} options = {'bar': 'baz'} side_effect_exception = ValueError() mock_unbind.side_effect = side_effect_exception result = consumer_group.unbind('foo_group_id', 'foo_repo_id', 'foo_distributor_id', options) self.assertTrue(isinstance(result.error, PulpException)) self.assertEquals(result.error.error_code, error_codes.PLP0005) self.assertEquals(result.error.child_exceptions[0], side_effect_exception)
def test_bind_with_general_error(self, mock_query_manager, mock_unbind): mock_query_manager.return_value.get_group.return_value = { 'consumer_ids': ['foo-consumer'] } options = {'bar': 'baz'} side_effect_exception = ValueError() mock_unbind.side_effect = side_effect_exception result = consumer_group.unbind('foo_group_id', 'foo_repo_id', 'foo_distributor_id', options) self.assertTrue(isinstance(result.error, PulpException)) self.assertEquals(result.error.error_code, error_codes.PLP0005) self.assertEquals(result.error.child_exceptions[0], side_effect_exception)
def test_bind_no_errors(self, mock_query_manager, mock_unbind): mock_query_manager.return_value.get_group.return_value = { 'consumer_ids': ['foo-consumer'] } options = {'bar': 'baz'} mock_unbind.return_value = TaskResult( spawned_tasks=[{ 'task_id': 'foo-request-id' }]) result = consumer_group.unbind('foo_group_id', 'foo_repo_id', 'foo_distributor_id', options) mock_unbind.assert_called_once_with('foo-consumer', 'foo_repo_id', 'foo_distributor_id', options) self.assertEquals(result.spawned_tasks[0], {'task_id': 'foo-request-id'})