addc comment

Author: michael-zhao459

tests/test_tracing.py

@@ -2545,7 +2545,7 @@ def test_sqs_incorrect_datadog_message_attribute(self, mock_dsm_set_checkpoint):
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
         )
@@ -2570,6 +2570,7 @@ def test_sqs_empty_datadog_message_attribute(self, mock_dsm_set_checkpoint):
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
         )
@@ -2625,7 +2626,7 @@ def test_sqs_invalid_datadog_message_attribute_raises_exception(
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
         )
@@ -2735,7 +2736,7 @@ def test_sns_no_datadog_message_attribute(self, mock_dsm_set_checkpoint):
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sns", "arn:aws:sns:us-east-1:123456789012:test-topic"
         )
@@ -2762,7 +2763,7 @@ def test_sns_incorrect_datadog_message_attribute(self, mock_dsm_set_checkpoint):
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sns", "arn:aws:sns:us-east-1:123456789012:test-topic"
         )
@@ -2789,7 +2790,7 @@ def test_sns_empty_datadog_message_attribute(self, mock_dsm_set_checkpoint):
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sns", "arn:aws:sns:us-east-1:123456789012:test-topic"
         )
@@ -2984,7 +2985,7 @@ def test_sns_to_sqs_no_datadog_message_attribute(self, mock_dsm_set_checkpoint):
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
         )
@@ -3016,7 +3017,7 @@ def test_sns_to_sqs_incorrect_datadog_message_attribute(
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
         )
@@ -3049,7 +3050,7 @@ def test_sns_to_sqs_empty_datadog_message_attribute(self, mock_dsm_set_checkpoin
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
         )
@@ -3112,7 +3113,7 @@ def test_sns_to_sqs_invalid_datadog_message_attribute_raises_exception(
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue"
         )
@@ -3140,7 +3141,7 @@ def test_sns_to_sqs_source_arn_not_found(self, mock_dsm_set_checkpoint):
         extract_context_from_sqs_or_sns_event_or_context(
             event, self.lambda_context, parse_event_source(event)
         )
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(None, "sqs", "")
         self.mock_checkpoint.assert_not_called()
 
@@ -3197,7 +3198,7 @@ def test_kinesis_no_datadog_message_attribute(self, mock_dsm_set_checkpoint):
         }
 
         extract_context_from_kinesis_event(event, self.lambda_context)
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "kinesis", "arn:aws:kinesis:us-east-1:123456789012:stream/test-stream"
         )
@@ -3216,7 +3217,7 @@ def test_kinesis_invalid_datadog_message_attribute_raises_exception(
         }
 
         extract_context_from_kinesis_event(event, self.lambda_context)
-
+        # None indiciates no DSM context propagation
         mock_dsm_set_checkpoint.assert_called_once_with(
             None, "kinesis", "arn:aws:kinesis:us-east-1:123456789012:stream/test-stream"
         )