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)