fix lint

Author: michael-zhao459

datadog_lambda/tracing.py

@@ -231,7 +231,7 @@ def create_sns_event(message):
     }
 
 
-def extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_checkpoint_if_enabled(
+def extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_ckpt_if_enabled(
     event, lambda_context
 ):
     """
@@ -638,7 +638,7 @@ def extract_dd_trace_context(
             event, lambda_context, event_source, decode_authorizer_context
         )
     elif event_source.equals(EventTypes.SNS) or event_source.equals(EventTypes.SQS):
-        context = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_checkpoint_if_enabled(
+        context = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_ckpt_if_enabled(
             event, lambda_context
         )
     elif event_source.equals(EventTypes.EVENTBRIDGE):

tests/test_tracing.py

@@ -45,7 +45,7 @@
     _dsm_set_checkpoint,
     _create_carrier_get,
     extract_context_from_kinesis_event_and_set_dsm_checkpoint_if_enabled,
-    extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_checkpoint_if_enabled,
+    extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_ckpt_if_enabled,
 )
 
 from tests.utils import get_mock_context
@@ -2564,7 +2564,7 @@ def test_sqs_event_with_datadog_message_attributes(
         mock_context = Context(trace_id=12345, span_id=67890, sampling_priority=1)
         mock_extract.return_value = mock_context
 
-        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_checkpoint_if_enabled(
+        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_ckpt_if_enabled(
             event, self.lambda_context
         )
 
@@ -2597,7 +2597,7 @@ def test_sqs_event_with_binary_datadog_message_attributes(
         mock_context = Context(trace_id=12345, span_id=67890, sampling_priority=1)
         mock_extract.return_value = mock_context
 
-        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_checkpoint_if_enabled(
+        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_ckpt_if_enabled(
             event, self.lambda_context
         )
 
@@ -2632,7 +2632,7 @@ def test_sns_event_with_datadog_message_attributes(
         mock_context = Context(trace_id=12345, span_id=67890, sampling_priority=1)
         mock_extract.return_value = mock_context
 
-        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_checkpoint_if_enabled(
+        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_ckpt_if_enabled(
             event, self.lambda_context
         )
 
@@ -2665,7 +2665,7 @@ def test_sqs_event_determines_is_sqs_true_when_event_source_arn_present(
         mock_context = Context(trace_id=12345, span_id=67890, sampling_priority=1)
         mock_extract.return_value = mock_context
 
-        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_checkpoint_if_enabled(
+        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_ckpt_if_enabled(
             event, self.lambda_context
         )
 
@@ -2706,7 +2706,7 @@ def test_sns_to_sqs_event_detection_and_processing(
         mock_context = Context(trace_id=12345, span_id=67890, sampling_priority=1)
         mock_extract.return_value = mock_context
 
-        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_checkpoint_if_enabled(
+        result = extract_context_from_sqs_or_sns_event_or_context_and_set_dsm_ckpt_if_enabled(
             event, self.lambda_context
         )