behaviour tests

Author: nhsdevws

ack_backend/tests/test_ack_file_processing_flow.py

@@ -0,0 +1,125 @@
+import unittest
+from unittest.mock import patch, MagicMock, call
+from io import StringIO
+
+import update_ack_file
+from update_ack_file import invoke_filename_lambda
+
+class TestUpdateAckFile(unittest.TestCase):
+    def setUp(self):
+        # Patch all AWS and external dependencies
+        self.s3_client_patcher = patch('update_ack_file.s3_client')
+        self.mock_s3_client = self.s3_client_patcher.start()
+
+        self.logger_patcher = patch('update_ack_file.logger')
+        self.mock_logger = self.logger_patcher.start()
+
+        self.get_row_count_patcher = patch('update_ack_file.get_row_count')
+        self.mock_get_row_count = self.get_row_count_patcher.start()
+
+        self.change_audit_status_patcher = patch('update_ack_file.change_audit_table_status_to_processed')
+        self.mock_change_audit_status = self.change_audit_status_patcher.start()
+
+        self.get_next_queued_file_details_patcher = patch('update_ack_file.get_next_queued_file_details')
+        self.mock_get_next_queued_file_details = self.get_next_queued_file_details_patcher.start()
+
+        self.invoke_filename_lambda_patcher = patch('update_ack_file.invoke_filename_lambda')
+        self.mock_invoke_filename_lambda = self.invoke_filename_lambda_patcher.start()
+
+        self.lambda_client_patcher = patch('update_ack_file.lambda_client')
+        self.mock_lambda_client = self.lambda_client_patcher.start()
+
+    def tearDown(self):
+        self.s3_client_patcher.stop()
+        self.logger_patcher.stop()
+        self.get_row_count_patcher.stop()
+        self.change_audit_status_patcher.stop()
+        self.get_next_queued_file_details_patcher.stop()
+        self.invoke_filename_lambda_patcher.stop()
+        self.lambda_client_patcher.stop()
+        # Setup
+        self.mock_get_row_count.side_effect = [3, 3]
+        self.mock_get_next_queued_file_details.return_value = {"filename": "next_unique.csv", "message_id": "msg-unique-2"}
+        accumulated_csv_content = StringIO("header1|header2\n")
+        ack_data_rows = [
+            {"a": 1, "b": 2, "row": "happy1"},
+            {"a": 3, "b": 4, "row": "happy2"},
+            {"a": 5, "b": 6, "row": "happy3"}
+        ]
+        # Act
+        update_ack_file.upload_ack_file(
+            temp_ack_file_key="TempAck/test_happy_path.csv",
+            message_id="msg-unique-1",
+            supplier_queue="queue-unique-1",
+            accumulated_csv_content=accumulated_csv_content,
+            ack_data_rows=ack_data_rows,
+            archive_ack_file_key="forwardedFile/test_happy_path.csv",
+            file_key="test_happy_path.csv"
+        )
+        # Assert
+        self.mock_s3_client.upload_fileobj.assert_called_once()
+        self.mock_get_next_queued_file_details.assert_called_once_with("queue-unique-1")
+        self.mock_invoke_filename_lambda.assert_called_once_with("next_unique.csv", "msg-unique-2")
+        self.mock_logger.info.assert_called()
+
+    def test_audit_table_updated_correctly(self):
+        """ VED-167 - Test that the audit table has been updated correctly"""
+        # Setup
+        self.mock_get_row_count.side_effect = [3, 3]
+        accumulated_csv_content = StringIO("header1|header2\n")
+        ack_data_rows = [
+            {"a": 1, "b": 2, "row": "audit-test-1"},
+            {"a": 3, "b": 4, "row": "audit-test-2"},
+            {"a": 5, "b": 6, "row": "audit-test-3"}
+        ]
+        message_id = "msg-audit-table"
+        # Act
+        update_ack_file.upload_ack_file(
+            temp_ack_file_key="TempAck/audit_table_test.csv",
+            message_id=message_id,
+            supplier_queue="queue-audit-table",
+            accumulated_csv_content=accumulated_csv_content,
+            ack_data_rows=ack_data_rows,
+            archive_ack_file_key="forwardedFile/audit_table_test.csv",
+            file_key="audit_table_test.csv"
+        )
+        # Assert: Only check audit table update
+        self.mock_change_audit_status.assert_called_once_with("audit_table_test.csv", message_id)
+
+    def test_move_file(self):
+        """ VED-167 test that the file has been moved to the appropriate location """
+        update_ack_file.move_file("bucket-unique", "src/file_unique.csv", "dest/file_unique.csv")
+        self.mock_s3_client.copy_object.assert_called_once_with(
+            Bucket="bucket-unique",
+            CopySource={"Bucket": "bucket-unique", "Key": "src/file_unique.csv"},
+            Key="dest/file_unique.csv"
+        )
+        self.mock_s3_client.delete_object.assert_called_once_with(Bucket="bucket-unique", Key="src/file_unique.csv")
+        self.mock_logger.info.assert_called_with("File moved from %s to %s", "src/file_unique.csv", "dest/file_unique.csv")
+
+    def test_next_queued_file_triggers_lambda(self):
+        """ VED-167 Test that the next queued file details are used to re-invoke the lambda."""
+        # Setup
+        self.mock_get_row_count.side_effect = [3, 3]
+        next_file = "next_for_lambda.csv"
+        next_message_id = "msg-next-lambda"
+        queue_name = "queue-lambda-trigger"
+        self.mock_get_next_queued_file_details.return_value = {"filename": next_file, "message_id": next_message_id}
+        accumulated_csv_content = StringIO("header1|header2\n")
+        ack_data_rows = [
+            {"a": 1, "b": 2, "row": "lambda1"},
+            {"a": 3, "b": 4, "row": "lambda2"},
+            {"a": 5, "b": 6, "row": "lambda3"}
+        ]
+        # Act
+        update_ack_file.upload_ack_file(
+            temp_ack_file_key="TempAck/next_lambda_test.csv",
+            message_id="msg-lambda-trigger",
+            supplier_queue=queue_name,
+            accumulated_csv_content=accumulated_csv_content,
+            ack_data_rows=ack_data_rows,
+            archive_ack_file_key="forwardedFile/next_lambda_test.csv",
+            file_key="next_lambda_test.csv"
+        )
+        # Assert: Check that the next queued file was used to re-invoke the lambda
+        self.mock_get_next_queued_file_details.assert_called_once_with(queue_name)