add test for rejected preds

Author: fitzworkhub

tests/integration/queries/test_workflow.py

@@ -6,7 +6,7 @@
 
 from indico.client import IndicoClient
 from indico.errors import IndicoError, IndicoInputError
-from indico.filters import SubmissionFilter
+from indico.filters import SubmissionFilter, SubmissionReviewFilter
 from indico.queries import (
     GetSubmission,
     JobStatus,
@@ -323,6 +323,31 @@ def test_workflow_submission_auto_review(
     assert submission.status == "COMPLETE" if force_complete else "PENDING_REVIEW"
 
 
+def test_list_workflow_submission_rejected():
+    client = IndicoClient()
+    wfs = client.call(ListWorkflows(dataset_ids=[org_annotate_dataset.id]))
+    wf = max(wfs, key=lambda w: w.id)
+    wf = client.call(
+        UpdateWorkflowSettings(wf, enable_review=True, enable_auto_review=True)
+    )
+    assert wf.review_enabled and wf.auto_review_enabled
+
+    _file = str(Path(__file__).parents[1]) + "/data/org-sample.pdf"
+
+    sub_ids = client.call(WorkflowSubmission(workflow_id=wf.id, files=[_file]))
+    subs = client.call(WaitForSubmissions(sub_ids, timeout=120))
+    sub = subs[0]
+    assert sub.status == "PENDING_AUTO_REVIEW"
+    job = client.call(
+        SubmitReview(sub.id, rejected=True)
+    )
+    job = client.call(JobStatus(job.id))
+    submissions = client.call(
+        ListSubmissions(filters=SubmissionFilter(reviews=SubmissionReviewFilter(rejected=True)))
+    )
+    assert sub_ids[0] in [s.id for s in submissions]
+    
+
 def _new_dataset_for_updating(client):
     # new dataset
     airline_csv = str(Path(__file__).parents[1]) + "/data/AirlineComplaints.csv"