fmt

Author: astuyve

datadog_lambda/metric.py

@@ -232,7 +232,13 @@ def submit_batch_item_failures_metric(response, lambda_context):
 
     batch_item_failures = response.get("batchItemFailures")
     if batch_item_failures is not None and isinstance(batch_item_failures, list):
-        lambda_metric("aws.lambda.enhanced.batch_item_failures", len(batch_item_failures), timestamp=None, tags=get_enhanced_metrics_tags(lambda_context), force_async=True)
+        lambda_metric(
+            "aws.lambda.enhanced.batch_item_failures",
+            len(batch_item_failures),
+            timestamp=None,
+            tags=get_enhanced_metrics_tags(lambda_context),
+            force_async=True,
+        )
 
 
 def submit_dynamodb_stream_type_metric(event):

tests/test_metric.py

@@ -356,7 +356,7 @@ def test_submit_batch_item_failures_with_failures(self):
                 3,
                 timestamp=None,
                 tags=["tag1:value1"],
-                force_async=True
+                force_async=True,
             )
 
     def test_submit_batch_item_failures_with_no_failures(self):
@@ -371,7 +371,7 @@ def test_submit_batch_item_failures_with_no_failures(self):
                 0,
                 timestamp=None,
                 tags=["tag1:value1"],
-                force_async=True
+                force_async=True,
             )
 
     def test_submit_batch_item_failures_with_no_field(self):

tests/test_wrapper.py

@@ -903,7 +903,10 @@ def lambda_handler(event, context):
 
 @patch("datadog_lambda.config.Config.trace_enabled", False)
 def test_batch_item_failures_metric():
-    with patch("datadog_lambda.metric.submit_batch_item_failures_metric") as mock_submit:
+    with patch(
+        "datadog_lambda.metric.submit_batch_item_failures_metric"
+    ) as mock_submit:
+
         @wrapper.datadog_lambda_wrapper
         def lambda_handler(event, context):
             return {
@@ -926,7 +929,10 @@ def lambda_handler(event, context):
 
 @patch("datadog_lambda.config.Config.trace_enabled", False)
 def test_batch_item_failures_metric_no_failures():
-    with patch("datadog_lambda.metric.submit_batch_item_failures_metric") as mock_submit:
+    with patch(
+        "datadog_lambda.metric.submit_batch_item_failures_metric"
+    ) as mock_submit:
+
         @wrapper.datadog_lambda_wrapper
         def lambda_handler(event, context):
             return {"batchItemFailures": []}
@@ -939,7 +945,10 @@ def lambda_handler(event, context):
 
 @patch("datadog_lambda.config.Config.trace_enabled", False)
 def test_batch_item_failures_metric_no_response():
-    with patch("datadog_lambda.metric.submit_batch_item_failures_metric") as mock_submit:
+    with patch(
+        "datadog_lambda.metric.submit_batch_item_failures_metric"
+    ) as mock_submit:
+
         @wrapper.datadog_lambda_wrapper
         def lambda_handler(event, context):
             return None