def test_remove_task(self): """Check whether the removal of tasks is properly handled""" task_1 = "arthur.task-1" task_2 = "arthur.task-2" task_params = { "backend": "backend", "category": "category", "backend_params": { "a": "a", "b": "b" } } app = Arthur(self.conn, async_mode=False) app.add_task(task_1, task_params['backend'], task_params['category'], task_params['backend_params']) app.add_task(task_2, task_params['backend'], task_params['category'], task_params['backend_params']) tasks = len(app._tasks.tasks) self.assertEqual(tasks, 2) self.assertTrue(app.remove_task(task_1)) self.assertTrue(app.remove_task(task_2)) tasks = len(app._tasks.tasks) self.assertEqual(tasks, 0)
def test_remove_non_existing_task(self): """Check whether the removal of non existing tasks is properly handled""" app = Arthur(self.conn, async_mode=False) self.assertFalse(app.remove_task("task-x"))