expose the identity provider through the Tracer

1 files changed, 4 insertions, 6 deletions

diff --git a/kamon-core/src/test/scala/kamon/testkit/SpanBuilding.scala b/kamon-core/src/test/scala/kamon/testkit/SpanBuilding.scala
index 29678b34..7a216ecc 100644
--- a/kamon-core/src/test/scala/kamon/testkit/SpanBuilding.scala
+++ b/kamon-core/src/test/scala/kamon/testkit/SpanBuilding.scala
@@ -1,18 +1,16 @@
 package kamon.testkit
 
 import kamon.trace.SpanContext.SamplingDecision
-import kamon.trace.{IdentityProvider, SpanContext, SpanContextCodec}
+import kamon.trace.{IdentityProvider, SpanContext}
 
 trait SpanBuilding {
   private val identityProvider = IdentityProvider.Default()
-  private val extendedB3Codec = SpanContextCodec.ExtendedB3(identityProvider)
 
   def createSpanContext(samplingDecision: SamplingDecision = SamplingDecision.Sample): SpanContext =
     SpanContext(
-      traceID = identityProvider.traceIdentifierGenerator().generate(),
-      spanID = identityProvider.spanIdentifierGenerator().generate(),
-      parentID = identityProvider.spanIdentifierGenerator().generate(),
+      traceID = identityProvider.traceIdGenerator().generate(),
+      spanID = identityProvider.spanIdGenerator().generate(),
+      parentID = identityProvider.spanIdGenerator().generate(),
       samplingDecision = samplingDecision
     )
-
 }