Skip to content

Commit fa015b9

Browse files
committed
service_name attr in test
1 parent 8ee8611 commit fa015b9

File tree

1 file changed

+13
-9
lines changed

1 file changed

+13
-9
lines changed

dd-java-agent/instrumentation/spark/src/testFixtures/groovy/datadog/trace/instrumentation/spark/AbstractSparkTest.groovy

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -629,7 +629,7 @@ abstract class AbstractSparkTest extends InstrumentationSpecification {
629629
null | null | null | "(?!.*databricks).*"
630630
}
631631

632-
def "set the proper spark service name"(String ddService, boolean sparkAppNameAsService, String appName, boolean isRunningOnDatabricks, String expectedService) {
632+
def "set the proper spark service name"(String ddService, boolean sparkAppNameAsService, String appName, boolean isRunningOnDatabricks, String expectedService, String expectedServiceNameTag) {
633633
setup:
634634
if (ddService != null) {
635635
injectSysConfig("dd.service", ddService)
@@ -668,35 +668,39 @@ abstract class AbstractSparkTest extends InstrumentationSpecification {
668668
operationName "spark.application"
669669
spanType "spark"
670670
assert span.serviceName ==~ expectedService
671+
assert span.tags["service_name"] == expectedServiceNameTag
671672
}
672673
}
673674
span {
674675
operationName "spark.sql"
675676
spanType "spark"
676677
assert span.serviceName ==~ expectedService
678+
assert span.tags["service_name"] == expectedServiceNameTag
677679
}
678680
span {
679681
operationName "spark.job"
680682
spanType "spark"
681683
assert span.serviceName ==~ expectedService
684+
assert span.tags["service_name"] == expectedServiceNameTag
682685
}
683686
span {
684687
operationName "spark.stage"
685688
spanType "spark"
686689
assert span.serviceName ==~ expectedService
690+
assert span.tags["service_name"] == expectedServiceNameTag
687691
}
688692
}
689693
}
690694

691695
where:
692-
ddService | sparkAppNameAsService | appName | isRunningOnDatabricks | expectedService
693-
"foobar" | true | "some_app" | false | "(?!.*some_app).*"
694-
"spark" | true | "some_app" | false | "some_app"
695-
"hadoop" | true | "some_app" | false | "some_app"
696-
null | true | "some_app" | true | "(?!.*some_app).*"
697-
null | true | "some_app" | false | "some_app"
698-
null | false | "some_app" | false | "(?!.*some_app).*"
699-
null | true | null | false | "(?!.*some_app).*"
696+
ddService | sparkAppNameAsService | appName | isRunningOnDatabricks | expectedService | expectedServiceNameTag
697+
"foobar" | true | "some_app" | false | "(?!.*some_app).*" | null
698+
"spark" | true | "some_app" | false | "some_app" | "some_app"
699+
"hadoop" | true | "some_app" | false | "some_app" | "some_app"
700+
null | true | "some_app" | true | "(?!.*some_app).*" | null
701+
null | true | "some_app" | false | "some_app" | "some_app"
702+
null | false | "some_app" | false | "(?!.*some_app).*" | null
703+
null | true | null | false | "(?!.*some_app).*" | null
700704
}
701705

702706

0 commit comments

Comments
 (0)