Exemplo n.º 1
0
    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())
Exemplo n.º 2
0
    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])