#1791 Added test cases

backend/audit/cross_validation/test_award_ref_and_references_uniqueness.py

@@ -0,0 +1,88 @@
+import random
+from django.test import TestCase
+from audit.models import SingleAuditChecklist
+from .award_ref_and_references_uniqueness import award_ref_and_references_uniqueness
+from .sac_validation_shape import sac_validation_shape
+from .errors import err_award_ref_repeat_reference
+from model_bakery import baker
+
+
+class AwardRefAndReferencesUniquenessTests(TestCase):
+    AWARD_MIN = 1000
+    AWARD_MAX = 2000
+    REF_MIN = 100
+    REF_MAX = 200
+
+    def _random(self, min, max):
+        return random.randint(min, max)  # nosec
+
+    def _award_reference(self) -> str:
+        return f"AWARD-{self._random(self.AWARD_MIN, self.AWARD_MAX)}"
+
+    def _reference_number(self, ref_num=None) -> str:
+        return (
+            f"2023-{self._random(self.REF_MIN, self.REF_MAX)}"
+            if ref_num is None
+            else f"2023-{ref_num}"
+        )
+
+    def _make_findings_uniform_guidance(
+        self, award_references, reference_numbers
+    ) -> dict:
+        entries = []
+        for award_ref, ref_nums in zip(award_references, reference_numbers):
+            for ref_num in ref_nums:
+                entries.append(
+                    {
+                        "program": {"award_reference": award_ref},
+                        "findings": {"reference_number": ref_num},
+                    }
+                )
+
+        return {
+            "FindingsUniformGuidance": {
+                "auditee_uei": "ABB123456CCC",
+                "findings_uniform_guidance_entries": entries,
+            }
+        }
+
+    def test_no_duplicate_references(self):
+        """
+        Check that no error is returned when there are no duplicate reference numbers.
+        """
+        range_size = self._random(2, 4)
+        sac = baker.make(SingleAuditChecklist)
+        sac.findings_uniform_guidance = self._make_findings_uniform_guidance(
+            [self._award_reference() for _ in range(range_size)],
+            [[self._reference_number(self.REF_MIN + i)] for i in range(range_size)],
+        )
+        errors = award_ref_and_references_uniqueness(sac_validation_shape(sac))
+        self.assertEqual(errors, [])
+
+    def test_duplicate_references_for_award(self):
+        """
+        Check that errors are returned for awards with duplicate reference numbers.
+        """
+        range_size = self._random(2, 4)
+        sac = baker.make(SingleAuditChecklist)
+        sac.findings_uniform_guidance = self._make_findings_uniform_guidance(
+            [self._award_reference() for _ in range(range_size)],
+            [
+                [
+                    self._reference_number(self.REF_MIN),
+                    self._reference_number(self.REF_MIN),
+                    self._reference_number(self.REF_MAX),
+                ]
+            ],
+        )
+        errors = award_ref_and_references_uniqueness(sac_validation_shape(sac))
+        for finding in sac.findings_uniform_guidance["FindingsUniformGuidance"][
+            "findings_uniform_guidance_entries"
+        ]:
+            expected_error = {
+                "error": err_award_ref_repeat_reference(
+                    finding["program"]["award_reference"],
+                    self._reference_number(self.REF_MIN),
+                )
+            }
+            self.assertIn(expected_error, errors)