feat: Add support for EventBridge Schedule mode

Author: andrewhibbert

main.tf

@@ -204,7 +204,7 @@ resource "aws_scheduler_schedule" "schedule" {
   end_date   = local.schedule_end_date
 
   target {
-    arn      = aws_lambda_function.custodian[each.key].qualified_arn
+    arn      = aws_lambda_function.custodian[each.key].arn
     role_arn = local.schedule_scheduler_role
   }
 

ops/package_lambda_policy.py

@@ -52,7 +52,7 @@
     sys.exit(1)
 
 
-def add_handler_and_config_to_archive(archive, policy_list, exec_options):
+def get_archive(archive, policy_list, exec_options):
     """Add handler template and config to archive.
 
     Args:
@@ -112,7 +112,7 @@ def process_lambda_package(query, policy_list, regions, exec_options, packages):
         Exception: If any step in the packaging process fails
     """
     archive = create_custodian_archive(packages=packages)
-    archive = add_handler_and_config_to_archive(archive, policy_list, exec_options)
+    archive = get_archive(archive, policy_list, exec_options)
     archive.close()
 
     try:

tests/ops/test_package_lambda_policy.py

@@ -23,10 +23,10 @@
 )
 
 
-def test_add_handler_and_config_to_archive_success():
-    """Test successful add_handler_and_config_to_archive."""
+def test_get_archive_success():
+    """Test successful get_archive."""
     from ops.package_lambda_policy import (
-        add_handler_and_config_to_archive,
+        get_archive,
         create_custodian_archive,
     )
 
@@ -35,7 +35,7 @@ def test_add_handler_and_config_to_archive_success():
     policy_list = [SIMPLE_PERIODIC_POLICY_DICT]
     exec_options = EXEC_OPTIONS
 
-    result = add_handler_and_config_to_archive(archive, policy_list, exec_options)
+    result = get_archive(archive, policy_list, exec_options)
     assert result == archive
 
     archive.close()
@@ -55,9 +55,9 @@ def test_add_handler_and_config_to_archive_success():
     archive.remove()
 
 
-def test_add_handler_and_config_to_archive_handler_assertion_error():
-    """Test add_handler_and_config_to_archive with AssertionError on handler template."""
-    from ops.package_lambda_policy import add_handler_and_config_to_archive
+def test_get_archive_handler_assertion_error():
+    """Test get_archive with AssertionError on handler template."""
+    from ops.package_lambda_policy import get_archive
     from unittest.mock import MagicMock
 
     # Create mock archive that raises AssertionError on first add_contents call
@@ -68,12 +68,12 @@ def test_add_handler_and_config_to_archive_handler_assertion_error():
     exec_options = EXEC_OPTIONS
 
     with pytest.raises(RuntimeError):
-        add_handler_and_config_to_archive(mock_archive, policy_list, exec_options)
+        get_archive(mock_archive, policy_list, exec_options)
 
 
-def test_add_handler_and_config_to_archive_config_assertion_error():
-    """Test add_handler_and_config_to_archive with AssertionError on config.json."""
-    from ops.package_lambda_policy import add_handler_and_config_to_archive
+def test_get_archive_config_assertion_error():
+    """Test get_archive with AssertionError on config.json."""
+    from ops.package_lambda_policy import get_archive
     from unittest.mock import MagicMock
 
     # Create mock archive that succeeds first call but fails second
@@ -84,7 +84,7 @@ def test_add_handler_and_config_to_archive_config_assertion_error():
     exec_options = EXEC_OPTIONS
 
     with pytest.raises(RuntimeError):
-        add_handler_and_config_to_archive(mock_archive, policy_list, exec_options)
+        get_archive(mock_archive, policy_list, exec_options)
 
 
 def test_get_custodian_tags_schedule_mode():