Support distributed traces in tests

Author: nikita-tkachenko-datadog

dd-trace-core/src/main/java/datadog/trace/civisibility/interceptor/CiVisibilityTraceInterceptor.java

@@ -1,14 +1,12 @@
 package datadog.trace.civisibility.interceptor;
 
-import datadog.trace.api.DDSpanTypes;
 import datadog.trace.api.DDTags;
 import datadog.trace.api.interceptor.AbstractTraceInterceptor;
 import datadog.trace.api.interceptor.MutableSpan;
 import datadog.trace.bootstrap.instrumentation.api.UTF8BytesString;
 import datadog.trace.core.DDSpan;
 import datadog.trace.core.DDTraceCoreInfo;
 import java.util.Collection;
-import java.util.Collections;
 
 public class CiVisibilityTraceInterceptor extends AbstractTraceInterceptor {
 
@@ -29,20 +27,6 @@ public Collection<? extends MutableSpan> onTraceComplete(
     }
 
     final DDSpan firstSpan = (DDSpan) trace.iterator().next();
-    final DDSpan localRootSpan = firstSpan.getLocalRootSpan();
-
-    final DDSpan spanToCheck = null == localRootSpan ? firstSpan : localRootSpan;
-
-    // If root span is not a CI visibility span, we drop the full trace.
-    CharSequence type = spanToCheck.getType(); // Don't null pointer if there is no type
-    if (type == null
-        || (!DDSpanTypes.TEST.contentEquals(type)
-            && !DDSpanTypes.TEST_SUITE_END.contentEquals(type)
-            && !DDSpanTypes.TEST_MODULE_END.contentEquals(type)
-            && !DDSpanTypes.TEST_SESSION_END.contentEquals(type))) {
-      return Collections.emptyList();
-    }
-
     // If the trace belongs to a "test", we need to set the origin to `ciapp-test` and the
     // `library_version` tag for all spans.
     firstSpan.context().setOrigin(CIAPP_TEST_ORIGIN);

dd-trace-core/src/test/groovy/datadog/trace/civisibility/interceptor/CiVisibilityTraceInterceptorTest.groovy

@@ -16,14 +16,6 @@ class CiVisibilityTraceInterceptorTest extends DDCoreSpecification {
     tracer?.close()
   }
 
-  def "discard a trace that does not come from a test"() {
-    tracer.addTraceInterceptor(CiVisibilityTraceInterceptor.INSTANCE)
-    tracer.buildSpan("sample-span").start().finish()
-
-    expect:
-    writer.size() == 0
-  }
-
   def "add ciapp origin and tracer version to spans of type #spanType"() {
     setup:
     tracer.addTraceInterceptor(CiVisibilityTraceInterceptor.INSTANCE)