def test_task_finished_queue(self): # Arrange task = Task(1, "", [""], None, "", "") task.job_id = 1234 task.message_type = TaskMessageType.TASK_PROCESSED connected_task = ConnectedTask(task, "") self.task_manager.in_progress[task.task_id] = connected_task # Act self.task_manager.task_finished(task) # Assert assert self.task_manager.finished_tasks.qsize() == 1 assert self.task_manager.finished_tasks.get() == task
def test_task_finished_in_progress(self): # Arrange finished_task = Task(1, "", [""], None, "", "") task = Task(2, "", [""], None, "", "") self.task_manager.add_new_available_task(finished_task, 1234) self.task_manager.add_new_available_task(task, 1234) self.task_manager.connect_available_tasks(2, "") finished_task.message_type = TaskMessageType.TASK_PROCESSED # Act self.task_manager.task_finished(finished_task) # Assert assert len(self.task_manager.in_progress) == 1 assert self.task_manager.in_progress[task.task_id].task == task assert self.task_manager.finished_tasks.qsize() == 1 assert self.task_manager.finished_tasks.get() == finished_task assert self.task_manager.status_manager.status.num_tasks_done == 1 assert not self.task_manager.status_manager.is_job_done()