def test_remove_task_error(self, mock_redis_cmd):
"""Test whether the lock is released in case of a Redis error and an error TaskRegistryError is thrown"""
mock_redis_cmd.side_effect = RedisError
registry = TaskRegistry(self.conn)
registry.add('mytask', 'git', 'commit', {})
self.assertGreater(registry._rwlock._access_mutex._value, 0)
with self.assertRaises(TaskRegistryError):
registry.remove('mytask')
self.assertGreater(registry._rwlock._access_mutex._value, 0)
def test_remove_not_found(self):
"""Check whether it raises an exception when a task is not found"""
registry = TaskRegistry()
with self.assertRaises(NotFoundError):
registry.remove('mytask')
registry.add('task', 'mock_backend', None, '/tmp/example')
with self.assertRaises(NotFoundError):
registry.remove('mytask')
self.assertEqual(len(registry.tasks), 1)
def test_remove_not_found(self):
"""Check whether it raises an exception when a task is not found"""
registry = TaskRegistry()
with self.assertRaises(NotFoundError):
registry.remove('mytask')
registry.add('task', 'git', "mock_category", {})
with self.assertRaises(NotFoundError):
registry.remove('mytask')
self.assertEqual(len(registry.tasks), 1)
def test_remove_not_found(self):
"""Check whether it raises an exception when a task is not found"""
registry = TaskRegistry()
with self.assertRaises(NotFoundError):
registry.remove('mytask')
registry.add('task', 'mock_backend', None, '/tmp/example')
with self.assertRaises(NotFoundError):
registry.remove('mytask')
self.assertEqual(len(registry.tasks), 1)
def test_remove_task(self):
"""Test to remove a task from the registry"""
args = {'from_date': '1970-01-01', 'component': 'test'}
registry = TaskRegistry()
registry.add('mytask', 'mock_backend', 'mocked_category', args)
registry.add('newtask', 'to_remove', 'mocked_category', None)
registry.add('atask', 'test_backend', 'mocked_category', None)
tasks = registry.tasks
self.assertEqual(len(tasks), 3)
registry.remove('newtask')
tasks = registry.tasks
self.assertEqual(len(tasks), 2)
self.assertEqual(tasks[0].task_id, 'atask')
self.assertEqual(tasks[1].task_id, 'mytask')
def test_remove_task(self):
"""Test to remove a task from the registry"""
args = {'from_date': '1970-01-01', 'component': 'test'}
registry = TaskRegistry()
registry.add('mytask', 'git', 'commit', args)
registry.add('newtask', 'bugzilla', 'issue', None)
registry.add('atask', 'git', 'commit', None)
tasks = registry.tasks
self.assertEqual(len(tasks), 3)
registry.remove('newtask')
tasks = registry.tasks
self.assertEqual(len(tasks), 2)
self.assertEqual(tasks[0].task_id, 'atask')
self.assertEqual(tasks[1].task_id, 'mytask')
def test_remove_task(self):
"""Test to remove a task from the registry"""
args = {
'from_date': '1970-01-01',
'component': 'test'
}
registry = TaskRegistry()
registry.add('mytask', 'mock_backend', 'mocked_category', args)
registry.add('newtask', 'to_remove', 'mocked_category', None)
registry.add('atask', 'test_backend', 'mocked_category', None)
tasks = registry.tasks
self.assertEqual(len(tasks), 3)
registry.remove('newtask')
tasks = registry.tasks
self.assertEqual(len(tasks), 2)
self.assertEqual(tasks[0].task_id, 'atask')
self.assertEqual(tasks[1].task_id, 'mytask')
