From b0e00942274d9ccba1e63841eed9745f92370d8a Mon Sep 17 00:00:00 2001 From: Ivan Topolnak Date: Thu, 26 Dec 2013 15:42:54 -0300 Subject: AskPatternTracingSpec is not leaving the test thread dirty anymore --- .../kamon/trace/instrumentation/AskPatternTracingSpec.scala | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'kamon-trace/src') 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) } } } -- cgit v1.2.3