[PRMP-1084] Refactor document review status handling to use DocumentReviewStatus enum

Author: NogaNHS

tests/bulk-upload/scripts/setup_document_review.py

@@ -2,13 +2,13 @@
 import os
 import uuid
 from datetime import datetime, timedelta
-from enum import StrEnum
 from typing import Any, Dict, List, NamedTuple
 
 import boto3
 from botocore.exceptions import ClientError
 
 from enums.document_review_reason import DocumentReviewReason
+from enums.document_review_status import DocumentReviewStatus
 
 SOURCE_PDF_FILE = "../source_to_copy_from.pdf"
 
@@ -119,7 +119,7 @@ def scenario_1(patient):
             review_id=review_id,
             patient=patient,
             files=files,
-            review_status="PENDING_REVIEW",
+            review_status=DocumentReviewStatus.PENDING_REVIEW,
             review_reason=DocumentReviewReason.GENERAL_ERROR,
             days_ago_uploaded=1,
         )
@@ -141,7 +141,7 @@ def scenario_2(patient):
             review_id=review_id,
             patient=patient,
             files=files,
-            review_status="PENDING_REVIEW",
+            review_status=DocumentReviewStatus.PENDING_REVIEW,
             review_reason=DocumentReviewReason.FILE_COUNT_MISMATCH,
             days_ago_uploaded=2,
         )
@@ -156,7 +156,7 @@ def scenario_3(patient):
             patient=patient,
             review_id=review_id,
             files=files,
-            review_status="APPROVED",
+            review_status=DocumentReviewStatus.APPROVED,
             review_reason=DocumentReviewReason.DEMOGRAPHIC_MISMATCHES,
             days_ago_uploaded=5,
             reviewer="H81109",
@@ -174,7 +174,7 @@ def scenario_4(patient):
             review_id=review_id,
             patient=patient,
             files=files,
-            review_status="REJECTED",
+            review_status=DocumentReviewStatus.REJECTED,
             review_reason=DocumentReviewReason.FILE_NAME_MISMATCH,
             days_ago_uploaded=7,
             reviewer="H81109",
@@ -199,7 +199,7 @@ def scenario_5(patient):
             review_id=review_id,
             patient=patient,
             files=files,
-            review_status="APPROVED",
+            review_status=DocumentReviewStatus.APPROVED,
             review_reason=DocumentReviewReason.DUPLICATE_RECORD,
             days_ago_uploaded=10,
             reviewer="H81109",
@@ -219,7 +219,7 @@ def scenario_6(patient):
             review_id=review_id,
             patient=patient,
             files=files,
-            review_status="PENDING_REVIEW",
+            review_status=DocumentReviewStatus.PENDING_REVIEW,
             review_reason=DocumentReviewReason.UNKNOWN_NHS_NUMBER,
             days_ago_uploaded=3,
         )
@@ -238,7 +238,7 @@ def scenario_7(patient):
             review_id=review_id,
             patient=patient,
             files=files,
-            review_status="NEVER_REVIEWED",
+            review_status=DocumentReviewStatus.NEVER_REVIEWED,
             review_reason=DocumentReviewReason.GENERAL_ERROR,
             review_date=get_timestamp(days_ago=1),
             days_ago_uploaded=15,
@@ -251,7 +251,7 @@ def scenario_7(patient):
             review_id=review_id,
             patient=patient,
             files=files,
-            review_status="PENDING_REVIEW",
+            review_status=DocumentReviewStatus.PENDING_REVIEW,
             review_reason=DocumentReviewReason.GENERAL_ERROR,
             days_ago_uploaded=15,
         )