Try to fix Cucumber instrumentation tests

DataDog · Sep 12, 2024 · c8f1363 · c8f1363
1 parent c457e5e
commit c8f1363
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/.../main/java/datadog/trace/civisibility/interceptor/CiVisibilityApmProtocolInterceptor.java b/.../main/java/datadog/trace/civisibility/interceptor/CiVisibilityApmProtocolInterceptor.java
@@ -39,12 +39,15 @@ public Collection<? extends MutableSpan> onTraceComplete(
       span.setTag(Tags.TEST_MODULE_ID, (Number) null);
       span.setTag(Tags.TEST_SUITE_ID, (Number) null);
 
-      span.setTag(Tags.RUNTIME_NAME, wellKnownTags.getRuntimeName().toString());
-      span.setTag(Tags.RUNTIME_VENDOR, wellKnownTags.getRuntimeVendor().toString());
-      span.setTag(Tags.RUNTIME_VERSION, wellKnownTags.getRuntimeVersion().toString());
-      span.setTag(Tags.OS_ARCHITECTURE, wellKnownTags.getOsArch().toString());
-      span.setTag(Tags.OS_PLATFORM, wellKnownTags.getOsPlatform().toString());
-      span.setTag(Tags.OS_VERSION, wellKnownTags.getOsVersion().toString());
+      String spanType = span.getSpanType();
+      if (DDSpanTypes.TEST.equals(spanType)) {
+        span.setTag(Tags.RUNTIME_NAME, wellKnownTags.getRuntimeName().toString());
+        span.setTag(Tags.RUNTIME_VENDOR, wellKnownTags.getRuntimeVendor().toString());
+        span.setTag(Tags.RUNTIME_VERSION, wellKnownTags.getRuntimeVersion().toString());
+        span.setTag(Tags.OS_ARCHITECTURE, wellKnownTags.getOsArch().toString());
+        span.setTag(Tags.OS_PLATFORM, wellKnownTags.getOsPlatform().toString());
+        span.setTag(Tags.OS_VERSION, wellKnownTags.getOsVersion().toString());
+      }
     }
     return filteredTrace;
   }