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