test record processor updates

Author: CLJ2006

recordprocessor/tests/test_audit_table.py

@@ -18,8 +18,6 @@
 with patch.dict("os.environ", MOCK_ENVIRONMENT_DICT):
     from constants import (
         AUDIT_TABLE_NAME,
-        AUDIT_TABLE_FILENAME_GSI,
-        AUDIT_TABLE_QUEUE_NAME_GSI,
         AuditTableKeys,
         FileStatus,
     )
@@ -49,6 +47,7 @@ def tearDown(self):
     @staticmethod
     def get_table_items() -> list:
         """Return all items in the audit table"""
+
         return dynamodb_client.scan(TableName=AUDIT_TABLE_NAME).get("Items", [])
 
     def test_get_next_queued_file_details(self):
@@ -66,14 +65,47 @@ def test_get_next_queued_file_details(self):
         add_entry_to_table(MockFileDetails.ravs_rsv_1, FileStatus.PROCESSED)  # same queue but already processed
         self.assertIsNone(get_next_queued_file_details(queue_to_check))
 
-        # TODO - finish test
         # Test case 3: one queued file in the ravs_rsv queue
-        # add_entry_to_table(MockFileDetails.ravs_rsv_2, file_status=FileStatus.QUEUED)
-        # expected_table_entry = {**MockFileDetails.ravs_rsv_2.audit_table_entry, "status": {"S": FileStatus.QUEUED}}
-        # self.assertEqual(get_next_queued_file_details(queue_to_check), deserialize_dynamodb_types(expected_table_entry))
+        add_entry_to_table(MockFileDetails.ravs_rsv_2, file_status=FileStatus.QUEUED)
+        expected_table_entry = {**MockFileDetails.ravs_rsv_2.audit_table_entry, "status": {"S": FileStatus.QUEUED}}
+        self.assertEqual(get_next_queued_file_details(queue_to_check), deserialize_dynamodb_types(expected_table_entry))
 
         # # Test case 4: multiple queued files in the RAVS_RSV queue
-        # # Note that ravs_rsv files 3 and 4 have later timestamps than file 2, so file 2 remains the first in the queue
-        # add_entry_to_table(MockFileDetails.ravs_rsv_3, file_status=FileStatus.QUEUED)
-        # add_entry_to_table(MockFileDetails.ravs_rsv_4, file_status=FileStatus.QUEUED)
-        # self.assertEqual(get_next_queued_file_details(queue_to_check), deserialize_dynamodb_types(expected_table_entry))
+        # Note that ravs_rsv files 3 and 4 have later timestamps than file 2, so file 2 remains the first in the queue
+        add_entry_to_table(MockFileDetails.ravs_rsv_3, file_status=FileStatus.QUEUED)
+        add_entry_to_table(MockFileDetails.ravs_rsv_4, file_status=FileStatus.QUEUED)
+        self.assertEqual(get_next_queued_file_details(queue_to_check), deserialize_dynamodb_types(expected_table_entry))
+
+    def test_change_audit_table_status_to_processed(self):
+        # Checks audit table is updated correctly
+        queue_to_check = "RAVS_RSV"
+
+        add_entry_to_table(MockFileDetails.rsv_ravs, file_status=FileStatus.QUEUED)
+        table_items = dynamodb_client.scan(TableName=AUDIT_TABLE_NAME).get("Items", [])
+        # print(f"started off queued:{table_items}")
+
+        expected_table_entry = {**MockFileDetails.rsv_ravs.audit_table_entry, "status": {"S": FileStatus.PROCESSED}}
+        file_key = "RSV_Vaccinations_v5_X26_20210730T12000000.csv"
+        message_id = "rsv_ravs_test_id_1"
+        # print(f"file_key_checker: {file_key}")
+        # print(f"message_id: {message_id}")
+        result = change_audit_table_status_to_processed(file_key, message_id)
+        print(result)
+        table_items = dynamodb_client.scan(TableName=AUDIT_TABLE_NAME).get("Items", [])
+        # print(f"change to processed :{table_items}")
+
+        Actual_result = [
+            {
+                "message_id": {"S": "rsv_ravs_test_id_1"},
+                "filename": {"S": "RSV_Vaccinations_v5_X26_20210730T12000000.csv"},
+                "queue_name": {"S": "RAVS_RSV"},
+                "timestamp": {"S": "20211120T12000000"},
+                "status": {"S": "Processed"},
+            }
+        ]
+        Actual_logger = "The status of RSV_Vaccinations_v5_X26_20210730T12000000.csv file, with message id rsv_ravs_test_id_1, was successfully updated to Processed in the audit table"
+        # self.assertEqual(result, deserialize_dynamodb_types(expected_table_entry))
+
+    # Check items in the dynamo table
+    # table_items = dynamodb_client.scan(TableName=AUDIT_TABLE_NAME).get("Items", [])
+    # print(f"table items 4 :{table_items}")