Fix lint errors

Signed-off-by: constanca <[email protected]>
elastic · May 7, 2024 · 76666c5 · 76666c5
1 parent 13fe3ec
commit 76666c5
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 7 deletions.
diff --git a/handlers/aws/handler.py b/handlers/aws/handler.py
@@ -229,8 +229,12 @@ def lambda_handler(lambda_event: dict[str, Any], lambda_context: context_.Contex
         composite_shipper.flush()
         shared_logger.info(
             "lambda processed all the events",
-            extra={"sent_events": sent_events, "empty_events": empty_events, "skipped_events": skipped_events,
-                   "error_events": error_events},
+            extra={
+                "sent_events": sent_events,
+                "empty_events": empty_events,
+                "skipped_events": skipped_events,
+                "error_events": error_events,
+            },
         )
 
     if trigger_type == "kinesis-data-stream":
@@ -330,8 +334,12 @@ def lambda_handler(lambda_event: dict[str, Any], lambda_context: context_.Contex
         composite_shipper.flush()
         shared_logger.info(
             "lambda processed all the events",
-            extra={"sent_events": sent_events, "empty_events": empty_events, "skipped_events": skipped_events,
-                   "error_events": error_events},
+            extra={
+                "sent_events": sent_events,
+                "empty_events": empty_events,
+                "skipped_events": skipped_events,
+                "error_events": error_events,
+            },
         )
 
     if trigger_type == "s3-sqs" or trigger_type == "sqs":
@@ -553,8 +561,12 @@ def handle_timeout(
 
         shared_logger.info(
             "lambda processed all the events",
-            extra={"sent_events": sent_events, "empty_events": empty_events, "skipped_events": skipped_events,
-                   "error_events": error_events},
+            extra={
+                "sent_events": sent_events,
+                "empty_events": empty_events,
+                "skipped_events": skipped_events,
+                "error_events": error_events,
+            },
         )
 
     return "completed"
diff --git a/tests/handlers/aws/test_integrations.py b/tests/handlers/aws/test_integrations.py
@@ -3247,7 +3247,10 @@ def test_sqs_no_input_defined(self) -> None:
         assert len(messages_sqs) == 3
         assert len(replayed_messages) == 3
         for i, message in enumerate(replayed_messages):
-            assert messages_sqs[i]["eventSourceARN"] == message["messageAttributes"]["originalEventSourceARN"]["stringValue"]
+            assert (
+                messages_sqs[i]["eventSourceARN"]
+                == message["messageAttributes"]["originalEventSourceARN"]["stringValue"]
+            )
 
     def test_sqs_last_ending_offset_reset(self) -> None:
         assert isinstance(self.logstash, LogstashContainer)