def test_scheduler_clear_queue(self): scheduler = Scheduler() task1 = TestTask1() scheduler.add_task_to_queue(task1) scheduler.clear_queue() self.assertListEqual(scheduler.get_queue(), list())
def test_process_two_tasks(self): scheduler = Scheduler() task1 = TestTask1() task2 = TestTask2() scheduler.clear_queue() task1.clear_preceding_tasks() task2.clear_preceding_tasks() task2.add_preceding_task(task1) scheduler.add_task_to_queue(task1) scheduler.add_task_to_queue(task2) scheduler.run_tasks() self.assertIn(task1, scheduler._result_map) self.assertIn(task2, scheduler._result_map) self.assertGreater(scheduler._result_map[task2], scheduler._result_map[task1])