@@ -1793,6 +1793,23 @@ def test_remaps_specific_inferred_span_service_names_from_sqs_event(self):
17931793 self .assertEqual (span2 .get_tag ("operation_name" ), "aws.sqs" )
17941794 self .assertEqual (span2 .service , "different-sqs-url" )
17951795
1796+ def test_create_inferred_span_uses_dd_service_when_set (self ):
1797+ # When DD_SERVICE is set, inferred spans use it as the service,
1798+ # taking precedence over any AWS service representation or mapping.
1799+ event_sample_source = "sqs-string-msg-attribute"
1800+ test_file = event_samples + event_sample_source + ".json"
1801+ with open (test_file , "r" ) as event :
1802+ original_event = json .load (event )
1803+
1804+ ctx = get_mock_context ()
1805+ ctx .aws_request_id = "123"
1806+
1807+ self .set_service_mapping ({"lambda_sqs" : "mapped-name" })
1808+ with patch ("datadog_lambda.config.Config.service" , "my-dd-service" ):
1809+ span = create_inferred_span (original_event , ctx )
1810+ self .assertEqual (span .get_tag ("operation_name" ), "aws.sqs" )
1811+ self .assertEqual (span .service , "my-dd-service" )
1812+
17961813 def test_remaps_all_inferred_span_service_names_from_sns_event (self ):
17971814 self .set_service_mapping ({"lambda_sns" : "new-name" })
17981815 event_sample_source = "sns-string-msg-attribute"
0 commit comments