def test_register_raise(self):
"""
Test to catch TaskError while registring already present tasks.
"""
with self.assertRaises(TaskError):
register_tasks(**self.tasks)
def setUp(self):
"""
Create two tasks for future unregistration.
"""
# clear dico and add only self names
self.names = 'a', 'b'
for name in self.names:
register_tasks(force=True, **{name: None})
def test_raise(self):
def toto():
pass
_id = path(toto)
register_tasks(force=True, **{_id: 6})
self.assertRaises(TaskError, register_task, toto)
def test_get_registeredtask(self):
"""
Test to get registered task.
"""
_id = 'a'
register_tasks(force=True, **{_id: GetTaskTest})
task = get_task(_id=_id)
self.assertEqual(task, GetTaskTest)
def test_register_force(self):
"""
Test to register existing tasks with force.
"""
self.assertNotIn('d', TASKS_BY_ID)
new_tasks = {'a': 'a', 'c': 'c', 'd': 'd'}
old_tasks = register_tasks(
force=True, **new_tasks
)
for new_task in new_tasks:
self.assertEqual(get_task(new_task), new_tasks[new_task])
self.assertNotIn('b', old_tasks)
self_tasks_wo_b = self.tasks
del self_tasks_wo_b['b']
self.assertEqual(old_tasks, self_tasks_wo_b)
def setUp(self):
"""
"""
# clean task paths
self.tasks = {'a': 1, 'b': 2, 'c': 3}
register_tasks(force=True, **self.tasks)
