TraceCorrelationTest.groovy

import datadog.trace.agent.test.InstrumentationSpecification
import datadog.trace.config.inversion.ConfigHelper
import datadog.trace.api.CorrelationIdentifier

import static datadog.trace.api.config.TraceInstrumentationConfig.TRACE_128_BIT_TRACEID_LOGGING_ENABLED
import static datadog.trace.api.config.TracerConfig.TRACE_128_BIT_TRACEID_GENERATION_ENABLED

class TraceCorrelationTest extends InstrumentationSpecification {

  @Override
  protected void configurePreAgent() {
    super.configurePreAgent()
    // Opt out of strict config validation - test module loads test instrumentations with fake names
    ConfigHelper.get().setConfigInversionStrict(ConfigHelper.StrictnessPolicy.TEST)
  }

  def "access trace correlation only under trace"() {
    when:
    def span = TEST_TRACER.startSpan("test", "myspan")
    def scope = TEST_TRACER.activateManualSpan(span)

    then:
    CorrelationIdentifier.traceId == (span.traceId.toHighOrderLong() == 0 ? span.traceId.toString() : span.traceId.toHexString())
    CorrelationIdentifier.spanId == span.spanId.toString()

    when:
    scope.close()
    span.finish()

    then:
    CorrelationIdentifier.traceId == "0"
    CorrelationIdentifier.spanId == "0"
  }
}
class Trace128bitCorrelationTest extends TraceCorrelationTest {
  @Override
  void configurePreAgent() {
    super.configurePreAgent()
    // Opt out of strict config validation - test module loads test instrumentations with fake names
    ConfigHelper.get().setConfigInversionStrict(ConfigHelper.StrictnessPolicy.TEST)
    injectSysConfig(TRACE_128_BIT_TRACEID_GENERATION_ENABLED, "true")
    injectSysConfig(TRACE_128_BIT_TRACEID_LOGGING_ENABLED, "true")
  }
}