test extended attributes

Author: Akol125

lambdas/filenameprocessor/tests/test_lambda_handler.py

@@ -195,7 +195,7 @@ def test_lambda_handler_new_file_success_and_first_in_queue(self, mock_get_redis
         mock_redis.hkeys = Mock(return_value=all_vaccine_types_in_this_test_file)
         mock_get_redis_client.return_value = mock_redis
 
-        test_cases = [MockFileDetails.emis_flu, MockFileDetails.ravs_rsv_1]
+        test_cases = [MockFileDetails.emis_flu, MockFileDetails.ravs_rsv_1, MockFileDetails.extended_attributes_file]
         for file_details in test_cases:
             with self.subTest(file_details.name):
                 # Set up the file in the source bucket

lambdas/filenameprocessor/tests/utils_for_tests/values_for_tests.py

@@ -29,18 +29,16 @@ class FileDetails:
 
     def __init__(
         self,
-        supplier: str,
-        vaccine_type: str,
-        spec_version: str = None,
-        organization_code: str = None,
+        supplier: str = None,
+        vaccine_type: str = None,
         ods_code: str = None,
         file_number: int = 1,
+        organization_code: str = None,
     ):
-        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.vaccine_type = vaccine_type.upper() if vaccine_type else None
+        self.ods_code = ods_code.upper() if ods_code else "X8E5B"
+        self.organization_code = organization_code
+        self.supplier = supplier.upper() if supplier else None
         self.queue_name = f"{self.supplier}_{self.vaccine_type}"
 
         self.created_at_formatted_string = f"200{file_number}0101T00000000"
@@ -51,7 +49,7 @@ def __init__(
         file_date_and_time_string = f"20000101T0000000{file_number}"
         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"
+            file_key = f"{extended_attributes_prefix}_v1_5_{organization_code}_{file_date_and_time_string}.csv"
         else:
             file_key = f"{vaccine_type}_Vaccinations_v5_{ods_code}_{file_date_and_time_string}.csv"
 
@@ -95,6 +93,9 @@ class MockFileDetails:
     emis_flu = FileDetails("EMIS", "FLU", "YGM41")
     emis_rsv = FileDetails("EMIS", "RSV", "YGM41")
     ravs_flu = FileDetails("RAVS", "FLU", "X8E5B")
+    extended_attributes_file = FileDetails(
+        vaccine_type="Vaccination_Extended_Attributes", file_number=1, organization_code="RJ123"
+    )
 
 
 MOCK_FILE_HEADERS = (