cloud watch event subscriber, terraform packaging support

aws · Aug 21, 2019 · 2fd2c47 · 2fd2c47
1 parent 56d897e
commit 2fd2c47
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/chalice/package.py b/chalice/package.py
@@ -736,6 +736,17 @@ def _generate_snslambdasubscription(self, resource, template):
                 'source_arn': topic_arn
         }
 
+    def _generate_cloudwatchevent(self, resource, template):
+        # type: (models.CloudWatchEvent, Dict[str, Any]) -> None
+
+        template['resource'].setdefault(
+            'aws_cloudwatch_event_rule', {})[
+                resource.resource_name] = {
+                    'name': resource.resource_name,
+                    'event_pattern': resource.event_pattern
+        }
+        self._cwe_helper(resource, template)
+
     def _generate_scheduledevent(self, resource, template):
         # type: (models.ScheduledEvent, Dict[str, Any]) -> None
 
@@ -745,6 +756,10 @@ def _generate_scheduledevent(self, resource, template):
                     'name': resource.resource_name,
                     'schedule_expression': resource.schedule_expression
         }
+        self._cwe_helper(resource, template)
+
+    def _cwe_helper(self, resource, template):
+        # type: (models.CloudWatchEventBase, Dict[str, Any]) -> None
         template['resource'].setdefault(
             'aws_cloudwatch_event_target', {})[
                 resource.resource_name] = {

diff --git a/tests/unit/test_package.py b/tests/unit/test_package.py
@@ -338,6 +338,30 @@ def test_adds_reserved_concurrency_when_provided(self, sample_app):
         tf_resource = self.get_function(template)
         assert tf_resource['reserved_concurrent_executions'] == 5
 
+    def test_can_generate_cloud_watch_event(self):
+        function = self.lambda_function()
+        event = models.CloudWatchEvent(
+            resource_name='foo-event',
+            rule_name='myrule',
+            event_pattern='{"source": ["aws.ec2"]}',
+            lambda_function=function,
+        )
+        template = self.template_gen.generate(
+            [function, event]
+        )
+        rule = template['resource'][
+            'aws_cloudwatch_event_rule'][event.resource_name]
+        assert rule == {
+            'name': event.resource_name,
+            'event_pattern': event.event_pattern}
+        target = template['resource'][
+            'aws_cloudwatch_event_target'][event.resource_name]
+        assert target == {
+            'target_id': 'foo-event',
+            'rule': '${aws_cloudwatch_event_rule.foo-event.name}',
+            'arn': '${aws_lambda_function.foo.arn}'
+        }
+
     def test_can_generate_scheduled_event(self):
         function = self.lambda_function()
         event = models.ScheduledEvent(