AskPatternTracingSpec is not leaving the test thread dirty anymore

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)
       }
     }
   }