From 76666c541e6db4e5dd49124b6a84cf5a81f28b14 Mon Sep 17 00:00:00 2001 From: constanca Date: Tue, 7 May 2024 14:53:40 +0200 Subject: [PATCH] Fix lint errors Signed-off-by: constanca --- handlers/aws/handler.py | 24 ++++++++++++++++++------ tests/handlers/aws/test_integrations.py | 5 ++++- 2 files changed, 22 insertions(+), 7 deletions(-) diff --git a/handlers/aws/handler.py b/handlers/aws/handler.py index 03289b83..671300f3 100644 --- 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 index c0bf86db..0b75847a 100644 --- 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)