Skip to content

Commit e2cdc94

Browse files
committed
test: add inferred-span coverage for DD_SERVICE precedence
Verify create_inferred_span uses DD_SERVICE as the service (over an explicit service mapping) when DD_SERVICE is set.
1 parent d21fa18 commit e2cdc94

1 file changed

Lines changed: 17 additions & 0 deletions

File tree

tests/test_tracing.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)