WIP - XRay test

Author: PerfectSlayer

dd-trace-core/src/test/groovy/datadog/trace/core/propagation/TracingPropagatorTest.groovy

@@ -11,6 +11,7 @@ import datadog.trace.core.test.DDCoreSpecification
 
 import static datadog.trace.api.sampling.PrioritySampling.SAMPLER_KEEP
 import static datadog.trace.api.sampling.PrioritySampling.USER_DROP
+import static datadog.trace.bootstrap.instrumentation.api.AgentPropagation.XRAY_TRACING_CONCERN
 
 class TracingPropagatorTest extends DDCoreSpecification {
   HttpCodec.Injector injector
@@ -136,4 +137,23 @@ class TracingPropagatorTest extends DDCoreSpecification {
     root.finish()
     tracer.close()
   }
+
+  def 'test AWS X-Ray propagator'() {
+    setup:
+    def tracer = tracerBuilder().build()
+    def span = tracer.buildSpan('test', 'operation').start()
+    def propagator = Propagators.forConcerns(XRAY_TRACING_CONCERN)
+    def setter = Mock(CarrierSetter)
+    def carrier = new Object()
+
+    when:
+    propagator.inject(span, carrier, setter)
+
+    then:
+    1 * setter.set(carrier, 'X-Amzn-Trace-Id', _)
+
+    cleanup:
+    span.finish()
+    tracer.close()
+  }
 }