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