def test_task_update_only_required(self):
t = Task("test_task", [1, 2, 3], "pool", [1], {})
update_dict = {
"wrong": 1,
"error": 2,
}
t.complete(**update_dict)
for key, value in update_dict.items():
with pytest.raises(AttributeError):
assert getattr(t, key) == value
def test_task_update(self):
t = Task("test_task", [1, 2, 3], "pool", [1], {})
update_dict = {
"stdout": 1,
"stderr": 2,
"result": 3,
"traceback": 4,
"status": 5
}
t.complete(**update_dict)
assert t.completed.is_set()
yield from t.completed.wait()
data = t.completed.data
assert data["result"] == update_dict["result"] and data["status"] == update_dict["status"]
for key, value in update_dict.items():
assert getattr(t, key) == value
