Black code formatting

Author: blockmar

aws/logs_monitoring/steps/parsing.py

@@ -82,15 +82,16 @@ def parse_event_type(event):
         return AwsEventType.AWSLOGS
     elif "detail" in event:
         # Check if this is an EventBridge S3 event
-        if event.get("source", "") == "aws.s3" and "Object Created" in event.get("detail-type", ""):
+        if event.get("source", "") == "aws.s3" and "Object Created" in event.get(
+            "detail-type", ""
+        ):
             return AwsEventType.EVENTBRIDGE_S3
         return AwsEventType.EVENTS
     raise Exception("Event type not supported (see #Event supported section)")
 
 
 # Handle S3 event over EventBridge
 def eventbridge_s3_handler(event, context, metadata, cache_layer):
-
     """
     Transform EventBridge S3 event to standard S3 event format.
 
@@ -115,6 +116,7 @@ def eventbridge_s3_handler(event, context, metadata, cache_layer):
         }]
     }
     """
+
     def reformat_eventbridge_s3_event(event):
         return {
             "Records": [

aws/logs_monitoring/tests/test_parsing.py

@@ -228,10 +228,7 @@ def test_parse_non_s3_eventbridge_event_type(self):
             "version": "0",
             "detail-type": "EC2 Instance State-change Notification",
             "source": "aws.ec2",
-            "detail": {
-                "instance-id": "i-1234567890abcdef0",
-                "state": "terminated"
-            },
+            "detail": {"instance-id": "i-1234567890abcdef0", "state": "terminated"},
         }
 
         event_type = parse_event_type(eventbridge_other_event)
@@ -241,12 +238,16 @@ def test_parse_non_s3_eventbridge_event_type(self):
 class TestEventBridgeS3Parsing(unittest.TestCase):
     class Context:
         function_version = "$LATEST"
-        invoked_function_arn = "arn:aws:lambda:us-east-1:123456789012:function:datadog-forwarder"
+        invoked_function_arn = (
+            "arn:aws:lambda:us-east-1:123456789012:function:datadog-forwarder"
+        )
         function_name = "datadog-forwarder"
         memory_limit_in_mb = "128"
 
     @patch("steps.parsing.S3EventHandler")
-    def test_parse_normalizes_eventbridge_s3_event_before_s3_handler(self, mock_s3_handler_cls):
+    def test_parse_normalizes_eventbridge_s3_event_before_s3_handler(
+        self, mock_s3_handler_cls
+    ):
         # Arrange: handler yields one log line; we only care about the input event it received
         mock_s3_handler = mock_s3_handler_cls.return_value
         mock_s3_handler.handle.return_value = iter([{"message": "ok"}])