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"))
