def test_tasks_finished_some_failed(self):
# Arrange
tasks = [
Task(1, "", [""], None, "", ""),
Task(2, "", [""], None, "", ""),
Task(3, "", [""], None, "", ""),
Task(4, "", [""], None, "", "")
]
tasks[0].message_type = TaskMessageType.TASK_PROCESSED
tasks[1].message_type = TaskMessageType.TASK_FAILED
tasks[2].message_type = TaskMessageType.TASK_FAILED
tasks[3].message_type = TaskMessageType.TASK_PROCESSED
for task in tasks:
task.job_id = 1234
connected_task = ConnectedTask(task, "")
self.task_manager.in_progress[task.task_id] = connected_task
# Act
self.task_manager.tasks_finished(tasks)
# Assert
assert self.task_manager.finished_tasks.qsize() == 2
assert self.task_manager.status_manager.status.num_tasks_done == 2
assert not self.task_manager.status_manager.is_job_done()
assert len(self.task_manager.in_progress) == 0
assert self.task_manager.available_tasks.qsize() == 2
assert self.task_manager.available_tasks.get() == tasks[1]
assert self.task_manager.available_tasks.get() == tasks[2]
def test_connect_available_task_no_available_tasks(self):
# Arrange
connected_task = ConnectedTask(Task(1, "", [""], None, "", ""), "")
self.task_manager.in_progress[
connected_task.task.task_id] = connected_task
# Act & Assert
with pytest.raises(NoMoreAvailableTasks):
assert self.task_manager.connect_available_task("")
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_connect_available_tasks(self):
# Arrange
new_task_1 = Task(1, "", [""], None, "", "")
new_task_2 = Task(2, "", [""], None, "", "")
self.task_manager.add_new_available_task(new_task_1, 1234)
self.task_manager.add_new_available_task(new_task_2, 1234)
connection_id = "conn_1"
expected_connected_task_1 = ConnectedTask(new_task_1, connection_id)
expected_connected_task_2 = ConnectedTask(new_task_2, connection_id)
# Act
tasks = self.task_manager.connect_available_tasks(3, connection_id)
# Assert
assert new_task_1 in tasks
assert new_task_2 in tasks
assert len(self.task_manager.in_progress) == 2
assert self.task_manager.in_progress[
new_task_1.task_id].task == expected_connected_task_1.task
assert self.task_manager.in_progress[
new_task_2.task_id].task == expected_connected_task_2.task
assert self.task_manager.in_progress[
new_task_1.task_id].connection_id == connection_id
assert self.task_manager.in_progress[
new_task_2.task_id].connection_id == connection_id
def test_connect_available_task(self):
# Arrange
new_task = Task(1, "", [""], None, "", "")
self.task_manager.add_new_available_task(new_task, 1234)
connection_id = "conn_1"
expected_connected_task = ConnectedTask(new_task, connection_id)
# Act
task = self.task_manager.connect_available_task(connection_id)
# Assert
assert new_task == task
assert len(self.task_manager.in_progress) == 1
assert self.task_manager.in_progress.get(
new_task.task_id).task == expected_connected_task.task
assert self.task_manager.in_progress.get(
new_task.task_id
).connection_id == expected_connected_task.connection_id
def test_flush_finished_tasks(self):
# Arrange
tasks = [
Task(1, "", [""], None, "", ""),
Task(2, "", [""], None, "", ""),
Task(3, "", [""], None, "", "")
]
for task in tasks:
task.set_job(1234)
task.message_type = TaskMessageType.TASK_PROCESSED
connected_task = ConnectedTask(task, "")
self.task_manager.in_progress[task.task_id] = connected_task
finished_tasks = tasks.copy()
self.task_manager.tasks_finished(finished_tasks)
# Act
flushed = self.task_manager.flush_finished_tasks()
# Assert
assert flushed == tasks
def test_tasks_finished(self):
# Arrange
finished_tasks = [
Task(1, "", [""], None, "", ""),
Task(2, "", [""], None, "", ""),
Task(3, "", [""], None, "", "")
]
for task in finished_tasks:
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.tasks_finished(finished_tasks)
# Assert
assert self.task_manager.finished_tasks.qsize() == 3
assert self.task_manager.status_manager.status.num_tasks_done == 3
assert self.task_manager.status_manager.is_job_done()
def test_connect_available_tasks_less_than_available(self):
# Arrange
new_task_1 = Task(1, "", [""], None, "", "")
new_task_2 = Task(2, "", [""], None, "", "")
self.task_manager.add_new_available_task(new_task_1, 1234)
self.task_manager.add_new_available_task(new_task_2, 1234)
connection_id = "conn_1"
expected_connected_task_1 = ConnectedTask(new_task_1, connection_id)
# Act
tasks = self.task_manager.connect_available_tasks(1, connection_id)
# Assert
assert new_task_1 in tasks
assert new_task_2 not in tasks
assert len(self.task_manager.in_progress) == 1
assert self.task_manager.in_progress.get(
new_task_1.task_id).task == expected_connected_task_1.task
assert self.task_manager.in_progress.get(
new_task_1.task_id
).connection_id == expected_connected_task_1.connection_id