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'})
