support filtering of events by pattern using eventPattern (#3144)

usmangani1 · web-flow · commit 4595c448280d · 2020-10-21T12:03:41.000+02:00
diff --git a/localstack/services/events/events_starter.py b/localstack/services/events/events_starter.py
@@ -51,6 +51,16 @@ def filter_event_with_target_input_path(target, event):
     return event
 
 
+def filter_event_based_on_event_format(self, rule, event):
+    rule_information = self.events_backend.describe_rule(rule)
+    if rule_information.event_pattern:
+        event_pattern = json.loads(rule_information.event_pattern)
+        for key, value in event_pattern.items():
+            if event.get(key.lower()) and event.get(key.lower()) not in value and event.get(key) != value:
+                return False
+    return True
+
+
 def process_events(event, targets):
     for target in targets:
         arn = target['Arn']
@@ -133,10 +143,6 @@ def events_handler_put_events(self):
 
             rules = EVENT_RULES.get(event_bus, [])
 
-            targets = []
-            for rule in rules:
-                targets.extend(self.events_backend.list_targets_by_rule(rule)['Targets'])
-
             formatted_event = {
                 'version': '0',
                 'id': event_envelope['uuid'],
@@ -148,6 +154,12 @@ def events_handler_put_events(self):
                 'resources': event.get('Resources', []),
                 'detail': json.loads(event.get('Detail')),
             }
+
+            targets = []
+            for rule in rules:
+                if filter_event_based_on_event_format(self, rule, formatted_event):
+                    targets.extend(self.events_backend.list_targets_by_rule(rule)['Targets'])
+
             # process event
             process_events(formatted_event, targets)
 
diff --git a/tests/integration/test_events.py b/tests/integration/test_events.py
@@ -537,12 +537,24 @@ def test_put_events_with_input_path(self):
 
         def get_message(queue_url):
             resp = sqs_client.receive_message(QueueUrl=queue_url)
-            return resp['Messages']
+            return resp.get('Messages')
 
         messages = retry(get_message, retries=3, sleep=1, queue_url=queue_url)
         self.assertEqual(len(messages), 1)
         self.assertEqual(json.loads(messages[0].get('Body')), EVENT_DETAIL)
 
+        self.events_client.put_events(
+            Entries=[{
+                'EventBusName': TEST_EVENT_BUS_NAME,
+                'Source': 'dummySource',
+                'DetailType': TEST_EVENT_PATTERN['DetailType'],
+                'Detail': json.dumps(TEST_EVENT_PATTERN['Detail'])
+            }]
+        )
+
+        messages = retry(get_message, retries=3, sleep=1, queue_url=queue_url)
+        self.assertEqual(messages, None)
+
         # clean up
         sqs_client.delete_queue(QueueUrl=queue_url)
 
