Unit tests

Author: robertnovac1

filenameprocessor/src/file_name_processor.py

@@ -170,7 +170,7 @@ def lambda_handler(event: dict, context) -> None:  # pylint: disable=unused-argu
     logger.info("Filename processor lambda task completed")
 
 
-if __name__ == "__main__":
+def run_local():
     parser = argparse.ArgumentParser("file_name_processor")
     parser.add_argument("--bucket", required=True, help="Bucket name.", type=str)
     parser.add_argument("--key", required=True, help="Object key.", type=str)
@@ -180,15 +180,15 @@ def lambda_handler(event: dict, context) -> None:  # pylint: disable=unused-argu
         "Records": [
             {
                 "s3": {
-                    "bucket": {
-                        "name": args.bucket
-                    },
-                    "object": {
-                        "key": args.key
-                    }
+                    "bucket": {"name": args.bucket},
+                    "object": {"key": args.key}
                 }
             }
         ]
     }
     print(event)
     print(lambda_handler(event=event, context={}))
+
+
+if __name__ == "__main__":
+    run_local()

filenameprocessor/tests/test_lambda_handler.py

@@ -1,5 +1,5 @@
 """Tests for lambda_handler"""
-
+import sys
 from unittest.mock import patch
 from unittest import TestCase
 from json import loads as json_loads
@@ -154,6 +154,14 @@ def get_audit_table_items():
         """Return all items in the audit table"""
         return dynamodb_client.scan(TableName=AUDIT_TABLE_NAME).get("Items", [])
 
+    def test_lambda_handler_no_file_key_throws_exception(self):
+        """Tests if exception is thrown when file_key is not provided"""
+
+        broken_record = {"Records": [{"s3": {"bucket": {"name": "test"}}}]}
+        with patch("file_name_processor.logger") as mock_logger:
+            lambda_handler(broken_record, None)
+            mock_logger.error.assert_called_once()
+
     def test_lambda_handler_new_file_success_and_first_in_queue(self):
         """
         Tests that for a new file, which passes validation and is the only file processing for the supplier_vaccineType
@@ -528,3 +536,34 @@ def test_successful_processing_from_configs(self):
             "supplier": ravs_rsv_file_details_2.supplier
         }
         self.assertEqual(result, expected_result)
+
+
+class TestMainEntryPoint(TestCase):
+    def test_run_local_constructs_event_and_calls_lambda_handler(self):
+        test_args = [
+            "file_name_processor.py",
+            "--bucket", "test-bucket",
+            "--key", "some/path/file.csv"
+        ]
+
+        expected_event = {
+            "Records": [
+                {
+                    "s3": {
+                        "bucket": {"name": "test-bucket"},
+                        "object": {"key": "some/path/file.csv"}
+                    }
+                }
+            ]
+        }
+
+        with (
+            patch.object(sys, "argv", test_args),
+            patch("file_name_processor.lambda_handler") as mock_lambda_handler,
+            patch("file_name_processor.print") as mock_print
+        ):
+            import file_name_processor
+            file_name_processor.run_local()
+
+            mock_lambda_handler.assert_called_once_with(event=expected_event, context={})
+            mock_print.assert_called()