#3202 Updated code to migrate cognizant and oversight data (#3204)

* #3202 Updated code to migrate cognizant and oversight data

* #3202 Fixed failing tests

backend/census_historical_migration/sac_general_lib/cognizant_oversight.py

@@ -0,0 +1,15 @@
+from ..exception_utils import DataMigrationError
+from ..transforms.xform_string_to_string import string_to_string
+
+
+def cognizant_oversight(audit_header):
+    """Retrieve cognizant oversight information for a given audit header."""
+
+    cognizant = string_to_string(audit_header.COGAGENCY)
+    oversight = string_to_string(audit_header.OVERSIGHTAGENCY)
+    if cognizant and len(cognizant) > 2:
+        raise DataMigrationError(
+            f"Invalid cognizant agency: {cognizant}", "invalid_cognizant"
+        )
+
+    return cognizant, oversight

backend/census_historical_migration/sac_general_lib/sac_creator.py

@@ -3,6 +3,8 @@
 from django.apps import apps
 from django.conf import settings
 
+from .cognizant_oversight import cognizant_oversight
+
 from ..exception_utils import DataMigrationError
 from .general_information import (
     general_information,
@@ -80,6 +82,9 @@ def setup_sac(user, audit_header):
 
     sac.auditee_certification = auditee_certification(audit_header)
     sac.auditor_certification = auditor_certification(audit_header)
+    cognizant, oversight = cognizant_oversight(audit_header)
+    sac.cognizant_agency = cognizant
+    sac.oversight_agency = oversight
     sac.data_source = settings.CENSUS_DATA_SOURCE
 
     if general_info["user_provided_organization_type"] == "tribal":

backend/census_historical_migration/test_sac_creation.py

@@ -56,6 +56,8 @@ def __init__(
             self.TYPEREPORT_SP_FRAMEWORK = "UQAD"
             self.UEI = "ZQGGHJH74DW7"
             self.ZIPCODE = "10451"
+            self.COGAGENCY = "14"
+            self.OVERSIGHTAGENCY = "84"
 
     def _mock_audit_header(self, entity_type, suppression_code=None):
         """Returns a mock audit header with all necessary fields"""