diff --git a/indico_toolkit/types/workflow_object.py b/indico_toolkit/types/workflow_object.py index bd4fa30..57e193d 100644 --- a/indico_toolkit/types/workflow_object.py +++ b/indico_toolkit/types/workflow_object.py @@ -126,7 +126,7 @@ def review_notes(self) -> int: @property def review_rejected(self) -> int: - return self.result["review_rejected"] + return self.result.get("review_rejected", False) @property def admin_review(self) -> bool: diff --git a/tests/types/test_worflow_result.py b/tests/types/test_worflow_result.py index fe9e8c8..252ff5a 100644 --- a/tests/types/test_worflow_result.py +++ b/tests/types/test_worflow_result.py @@ -48,3 +48,14 @@ def test_classification_predictions(): "model_v1", ) assert isinstance(wf_result.get_predictions, Classification) + + +def test_review_rejected_defaults_false_when_missing(): + wf_result = WorkflowResult( + { + "submission_id": 12, + "results": {"document": {"results": {"model_v1": {"pre_review": []}}}}, + }, + "model_v1", + ) + assert wf_result.review_rejected is False