@@ -2614,40 +2614,6 @@ def test_sns_event_with_datadog_message_attributes(
26142614 )
26152615 self .assertEqual (result , mock_context )
26162616
2617- @patch ("datadog_lambda.tracing._dsm_set_checkpoint" )
2618- @patch ("datadog_lambda.tracing.propagator.extract" )
2619- def test_sqs_event_determines_is_sqs_true_when_event_source_arn_present (
2620- self , mock_extract , mock_dsm_set_checkpoint
2621- ):
2622- """Test that is_sqs = True when eventSource is SQS"""
2623- dd_data = {DSM_PROPAGATION_KEY_BASE_64 : "12345" }
2624- dd_json_data = json .dumps (dd_data )
2625-
2626- event = {
2627- "Records" : [
2628- {
2629- "eventSourceARN" : "arn:aws:sqs:us-east-1:123456789012:test-queue" ,
2630- "messageAttributes" : {
2631- "_datadog" : {"dataType" : "String" , "stringValue" : dd_json_data }
2632- },
2633- "eventSource" : "aws:sqs" ,
2634- }
2635- ]
2636- }
2637-
2638- mock_context = Context (trace_id = 12345 , span_id = 67890 , sampling_priority = 1 )
2639- mock_extract .return_value = mock_context
2640-
2641- result = extract_context_from_sqs_or_sns_event_or_context (
2642- event , self .lambda_context , parse_event_source (event )
2643- )
2644-
2645- mock_extract .assert_called_once_with (dd_data )
2646- mock_dsm_set_checkpoint .assert_called_once_with (
2647- dd_data , "sqs" , "arn:aws:sqs:us-east-1:123456789012:test-queue"
2648- )
2649- self .assertEqual (result , mock_context )
2650-
26512617 @patch ("datadog_lambda.tracing._dsm_set_checkpoint" )
26522618 @patch ("datadog_lambda.tracing.propagator.extract" )
26532619 def test_sns_to_sqs_event_detection_and_processing (
0 commit comments