diff options
-rw-r--r-- | kamon-trace/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/kamon-trace/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala b/kamon-trace/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala index a3e43aee..9df67391 100644 --- a/kamon-trace/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala +++ b/kamon-trace/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala @@ -28,15 +28,17 @@ import org.scalatest.OptionValues._ class AskPatternTracingSpec extends TestKit(ActorSystem("ask-pattern-tracing-spec")) with WordSpecLike with Matchers { "the AskPatternTracing" should { - "log a warning with a stack trace and TraceContext taken from the moment the ask was triggered" in { + "log a warning with a stack trace and TraceContext taken from the moment the ask was triggered" in new TraceContextFixture { implicit val ec = system.dispatcher implicit val timeout = Timeout(10 milliseconds) val noReply = system.actorOf(Props[NoReply]) system.eventStream.subscribe(testActor, classOf[Warning]) within(500 milliseconds) { - val initialCtx = Trace.start("ask-test") - noReply ? "hello" + val initialCtx = Trace.withContext(testTraceContext) { + noReply ? "hello" + Trace.context() + } val warn = expectMsgPF() { case warn: Warning if warn.message.toString.contains("Timeout triggered for ask pattern") ⇒ warn @@ -44,7 +46,7 @@ class AskPatternTracingSpec extends TestKit(ActorSystem("ask-pattern-tracing-spe val capturedCtx = warn.asInstanceOf[ContextAware].traceContext capturedCtx should be('defined) - capturedCtx.value should equal(initialCtx) + capturedCtx should equal(initialCtx) } } } |