def test_queue_complete_task(self):
q = MultiLockPriorityPoolQueue()
t1 = Task("test_task", [1, 2, 3], "pool", [1], {})
q.put(t1)
task = q.get("pool")
assert len(q._locks) == 3
q.complete(str(task.id), {"stdout": "", "stderr": "", "result": "", "status": "success"})
assert len(q._locks) == 0
def test_queue_complete_missing_task(self):
q = MultiLockPriorityPoolQueue()
with pytest.raises(LookupError):
q.complete(str(uuid.uuid4()), {})
