test for extended attributes

Author: Akol125

lambdas/filenameprocessor/src/file_name_processor.py

@@ -77,7 +77,7 @@ def handle_record(record) -> dict:
         message_id = str(uuid4())
         s3_response = get_s3_client().get_object(Bucket=bucket_name, Key=file_key)
         created_at_formatted_string, expiry_timestamp = get_creation_and_expiry_times(s3_response)
-        if file_key.startswith(tuple(EXTENDED_ATTRIBUTES_PREFIXES)):
+        if file_key.startswith(EXTENDED_ATTRIBUTES_PREFIXES):
             pass
         else:
             vaccine_type, supplier = validate_file_key(file_key)

lambdas/filenameprocessor/tests/utils_for_tests/values_for_tests.py

@@ -27,9 +27,19 @@ class FileDetails:
     vaccine type.
     """
 
-    def __init__(self, supplier: str, vaccine_type: str, ods_code: str, file_number: int = 1):
+    def __init__(
+        self,
+        supplier: str,
+        vaccine_type: str,
+        spec_version: str = None,
+        organization_code: str = None,
+        ods_code: str = None,
+        file_number: int = 1,
+    ):
         self.vaccine_type = vaccine_type.upper()
         self.ods_code = ods_code.upper()
+        self.spec_version = spec_version
+        self.orgnization_code = organization_code
         self.supplier = supplier.upper()
         self.queue_name = f"{self.supplier}_{self.vaccine_type}"
 
@@ -39,7 +49,14 @@ def __init__(self, supplier: str, vaccine_type: str, ods_code: str, file_number:
         self.name = f"{self.vaccine_type}/ {self.supplier} file"
 
         file_date_and_time_string = f"20000101T0000000{file_number}"
-        self.file_key = f"{vaccine_type}_Vaccinations_v5_{ods_code}_{file_date_and_time_string}.csv"
+        extended_attributes_prefix = "Vaccination_Extended_Attributes"
+        if self.vaccine_type.startswith(extended_attributes_prefix):
+            file_key = f"{extended_attributes_prefix}_{self.spec_version}_{self.organization_code}_{file_date_and_time_string}.csv"
+        else:
+            file_key = f"{vaccine_type}_Vaccinations_v5_{ods_code}_{file_date_and_time_string}.csv"
+
+        self.file_key = file_key
+
         self.ack_file_key = f"ack/{self.file_key[:-4]}_InfAck_{self.created_at_formatted_string}.csv"
 
         self.permissions_list = [f"{self.vaccine_type}_FULL"]