format again

Author: Czechh

tests/test_tracing.py

@@ -88,7 +88,7 @@ def test_with_incomplete_datadog_trace_headers(self):
         lambda_ctx = get_mock_context()
         ctx = extract_dd_trace_context(
             {"headers": {TraceHeader.TRACE_ID: "123", TraceHeader.PARENT_ID: "321"}},
-            lambda_ctx
+            lambda_ctx,
         )
         self.assertDictEqual(
             ctx,
@@ -118,7 +118,7 @@ def test_with_complete_datadog_trace_headers(self):
                     TraceHeader.SAMPLING_PRIORITY: "1",
                 }
             },
-            lambda_ctx
+            lambda_ctx,
         )
         self.assertDictEqual(
             ctx,
@@ -157,21 +157,23 @@ def test_with_sqs_distributed_datadog_trace_data(self):
                         "ApproximateReceiveCount": "1",
                         "SentTimestamp": "1545082649183",
                         "SenderId": "AIDAIENQZJOLO23YVJ4VO",
-                        "ApproximateFirstReceiveTimestamp": "1545082649185"
+                        "ApproximateFirstReceiveTimestamp": "1545082649185",
                     },
                     "messageAttributes": {
                         "_datadog": {
-                            "StringValue": json.dumps({
-                                TraceHeader.TRACE_ID: "123",
-                                TraceHeader.PARENT_ID: "321",
-                                TraceHeader.SAMPLING_PRIORITY: "1",
-                            })
+                            "StringValue": json.dumps(
+                                {
+                                    TraceHeader.TRACE_ID: "123",
+                                    TraceHeader.PARENT_ID: "321",
+                                    TraceHeader.SAMPLING_PRIORITY: "1",
+                                }
+                            )
                         }
                     },
                     "md5OfBody": "e4e68fb7bd0e697a0ae8f1bb342846b3",
                     "eventSource": "aws:sqs",
                     "eventSourceARN": "arn:aws:sqs:us-east-2:123456789012:my-queue",
-                    "awsRegion": "us-east-2"
+                    "awsRegion": "us-east-2",
                 }
             ]
         }
@@ -202,19 +204,18 @@ def test_with_sqs_distributed_datadog_trace_data(self):
         )
 
     def test_with_client_context_datadog_trace_data(self):
-        lambda_ctx = get_mock_context(client_context={
-            "custom": {
-                "_datadog": {
-                    TraceHeader.TRACE_ID: "666",
-                    TraceHeader.PARENT_ID: "777",
-                    TraceHeader.SAMPLING_PRIORITY: "1",
+        lambda_ctx = get_mock_context(
+            client_context={
+                "custom": {
+                    "_datadog": {
+                        TraceHeader.TRACE_ID: "666",
+                        TraceHeader.PARENT_ID: "777",
+                        TraceHeader.SAMPLING_PRIORITY: "1",
+                    }
                 }
             }
-        })
-        ctx = extract_dd_trace_context(
-            { "test": "foo" },
-            lambda_ctx
         )
+        ctx = extract_dd_trace_context({}, lambda_ctx)
         self.assertDictEqual(
             ctx,
             {
@@ -250,7 +251,7 @@ def test_with_complete_datadog_trace_headers_with_mixed_casing(self):
                     "X-Datadog-Sampling-Priority": "1",
                 }
             },
-            lambda_ctx
+            lambda_ctx,
         )
         self.assertDictEqual(
             get_dd_trace_context(),

tests/test_wrapper.py

@@ -15,7 +15,7 @@ def get_mock_context(
     memory_limit_in_mb="256",
     invoked_function_arn="arn:aws:lambda:us-west-1:123457598159:function:python-layer-test:1",
     function_version="1",
-    client_context={}
+    client_context={},
 ):
     lambda_context = MagicMock()
     lambda_context.aws_request_id = aws_request_id
@@ -98,7 +98,9 @@ def lambda_handler(event, context):
             [call("test.metric", 100, timestamp=None, tags=ANY)]
         )
         self.mock_wrapper_lambda_stats.flush.assert_called()
-        self.mock_extract_dd_trace_context.assert_called_with(lambda_event, lambda_context)
+        self.mock_extract_dd_trace_context.assert_called_with(
+            lambda_event, lambda_context
+        )
         self.mock_set_correlation_ids.assert_called()
         self.mock_inject_correlation_ids.assert_called()
         self.mock_patch_all.assert_called()