style: apply ruff formatting to DurableContext test files

Author: oyiz-michael

tests/functional/logger/required_dependencies/test_logger_durable_context.py

@@ -66,7 +66,7 @@ def handler(event, context):
 
     # THEN lambda contextual info from the unwrapped context should be in the logs
     log = capture_logging_output(stdout)
-    
+
     expected_logger_context_keys = (
         "function_name",
         "function_memory_size",
@@ -75,7 +75,7 @@ def handler(event, context):
     )
     for key in expected_logger_context_keys:
         assert key in log
-    
+
     # Verify the actual values match the embedded lambda_context
     assert log["function_name"] == durable_context.lambda_context.function_name
     assert log["function_memory_size"] == durable_context.lambda_context.memory_limit_in_mb
@@ -88,7 +88,7 @@ def test_inject_lambda_context_with_durable_context_log_event(durable_context, s
     """Test that inject_lambda_context with log_event=True works with DurableContext."""
     # GIVEN Logger is initialized
     logger = Logger(service=service_name, stream=stdout)
-    
+
     test_event = {"test_key": "test_value"}
 
     # WHEN a lambda function is decorated with log_event=True and receives DurableContext
@@ -101,7 +101,7 @@ def handler(event, context):
     # THEN both the event and lambda contextual info should be logged
     logs = capture_multiple_logging_statements_output(stdout)
     assert len(logs) >= 2  # At least event log and info log
-    
+
     # First log should be the event
     assert logs[0]["message"] == test_event
 
@@ -121,11 +121,11 @@ def handler(event, context):
 
     # THEN the custom key should be cleared and lambda context should be present
     log = capture_logging_output(stdout)
-    
+
     # Lambda context fields should be present
     assert "function_name" in log
     assert log["function_name"] == durable_context.lambda_context.function_name
-    
+
     # Custom key should not be present (cleared)
     assert "custom_key" not in log or log.get("custom_key") != "initial_value"
 
@@ -144,7 +144,7 @@ def handler(event, context):
 
     # THEN lambda contextual info should be in the logs
     log = capture_logging_output(stdout)
-    
+
     expected_logger_context_keys = (
         "function_name",
         "function_memory_size",
@@ -153,6 +153,6 @@ def handler(event, context):
     )
     for key in expected_logger_context_keys:
         assert key in log
-    
+
     assert log["function_name"] == lambda_context.function_name
     assert log["message"] == "Hello from standard lambda"

tests/functional/metrics/required_dependencies/test_metrics_durable_context.py

@@ -7,10 +7,10 @@
 import pytest
 
 from aws_lambda_powertools import Metrics
-from aws_lambda_powertools.utilities.typing import DurableContextProtocol
 
 # Reset cold start flag before each test
 from aws_lambda_powertools.metrics.provider import cold_start
+from aws_lambda_powertools.utilities.typing import DurableContextProtocol
 
 
 def capture_metrics_output(capsys):
@@ -66,22 +66,22 @@ def lambda_handler(evt, context):
 
     # THEN metrics should be emitted successfully
     output = capture_metrics_output(capsys)
-    
+
     assert output["test_metric"] == [1.0]
     assert output["service"] == service
 
 
 def test_log_metrics_capture_cold_start_with_durable_context(capsys, namespace, service):
     """Test that capture_cold_start_metric works with DurableContext."""
     reset_cold_start_flag()
-    
+
     # GIVEN Metrics is initialized
     my_metrics = Metrics(service=service, namespace=namespace)
-    
+
     # Create a DurableContext with embedded Lambda context
     LambdaContext = namedtuple("LambdaContext", "function_name")
     lambda_ctx = LambdaContext("durable_test_function")
-    
+
     durable_ctx = Mock(spec=DurableContextProtocol)
     durable_ctx.lambda_context = lambda_ctx
     durable_ctx.state = Mock(operations=[{"id": "op1"}])
@@ -95,27 +95,25 @@ def lambda_handler(evt, context):
 
     # THEN ColdStart metric should be captured with the function name from unwrapped context
     outputs = capture_metrics_output_multiple_emf_objects(capsys)
-    
+
     # Cold start is in a separate EMF blob
     cold_start_output = outputs[0]
     assert cold_start_output["ColdStart"] == [1.0]
     assert cold_start_output["function_name"] == "durable_test_function"
     assert cold_start_output["service"] == service
 
 
-def test_log_metrics_capture_cold_start_with_durable_context_explicit_function_name(
-    capsys, namespace, service
-):
+def test_log_metrics_capture_cold_start_with_durable_context_explicit_function_name(capsys, namespace, service):
     """Test capture_cold_start_metric with explicit function_name and DurableContext."""
     reset_cold_start_flag()
-    
+
     # GIVEN Metrics is initialized with explicit function_name
     my_metrics = Metrics(service=service, namespace=namespace, function_name="explicit_function")
-    
+
     # Create a DurableContext
     LambdaContext = namedtuple("LambdaContext", "function_name")
     lambda_ctx = LambdaContext("context_function")
-    
+
     durable_ctx = Mock(spec=DurableContextProtocol)
     durable_ctx.lambda_context = lambda_ctx
     durable_ctx.state = Mock(operations=[{"id": "op1"}])
@@ -129,7 +127,7 @@ def lambda_handler(evt, context):
 
     # THEN explicit function_name should take priority
     output = capture_metrics_output(capsys)
-    
+
     assert output.get("function_name") == "explicit_function"
 
 
@@ -147,18 +145,18 @@ def lambda_handler(evt, context):
 
     # THEN metrics should be emitted successfully
     output = capture_metrics_output(capsys)
-    
+
     assert output["regression_test"] == [42.0]
     assert output["service"] == service
 
 
 def test_log_metrics_capture_cold_start_standard_context_still_works(capsys, namespace, service):
     """Test that capture_cold_start_metric with standard context still works (regression test)."""
     reset_cold_start_flag()
-    
+
     # GIVEN Metrics is initialized
     my_metrics = Metrics(service=service, namespace=namespace)
-    
+
     LambdaContext = namedtuple("LambdaContext", "function_name")
     standard_context = LambdaContext("standard_function")
 
@@ -171,6 +169,6 @@ def lambda_handler(evt, context):
 
     # THEN ColdStart metric should be captured
     output = capture_metrics_output(capsys)
-    
+
     assert "ColdStart" in output or output.get("ColdStart") == [1.0]
     assert output.get("function_name") == "standard_function"