diff options
author | Ivan Topolnjak <ivantopo@gmail.com> | 2014-01-31 09:01:18 -0300 |
---|---|---|
committer | Ivan Topolnjak <ivantopo@gmail.com> | 2014-01-31 09:01:18 -0300 |
commit | 49c426a635d10182e8a628353dfdf5510c4d9df2 (patch) | |
tree | a9b764090cbbf1c14973d3a3063fa776db53805d /kamon-core/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala | |
parent | a0a57b110a3ee4876797ab51c4758525d166796f (diff) | |
download | Kamon-49c426a635d10182e8a628353dfdf5510c4d9df2.tar.gz Kamon-49c426a635d10182e8a628353dfdf5510c4d9df2.tar.bz2 Kamon-49c426a635d10182e8a628353dfdf5510c4d9df2.zip |
remake of trace context and allow different tracing levels
Diffstat (limited to 'kamon-core/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala')
-rw-r--r-- | kamon-core/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala | 26 |
1 files changed, 12 insertions, 14 deletions
diff --git a/kamon-core/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala b/kamon-core/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala index 9df67391..0387386c 100644 --- a/kamon-core/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala +++ b/kamon-core/src/test/scala/kamon/trace/instrumentation/AskPatternTracingSpec.scala @@ -22,32 +22,30 @@ import akka.event.Logging.Warning import scala.concurrent.duration._ import akka.pattern.ask import akka.util.Timeout -import kamon.trace.{ Trace, ContextAware } +import kamon.trace.{TraceContextAware, TraceRecorder} 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 new TraceContextFixture { + "log a warning with a stack trace and TraceContext taken from the moment the ask was triggered" in { 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.withContext(testTraceContext) { - noReply ? "hello" - Trace.context() - } - - val warn = expectMsgPF() { - case warn: Warning if warn.message.toString.contains("Timeout triggered for ask pattern") ⇒ warn - } - val capturedCtx = warn.asInstanceOf[ContextAware].traceContext + val testTraceContext = TraceRecorder.withNewTraceContext("ask-timeout-warning") { + noReply ? "hello" + TraceRecorder.currentContext + } - capturedCtx should be('defined) - capturedCtx should equal(initialCtx) + val warn = expectMsgPF() { + case warn: Warning if warn.message.toString.contains("Timeout triggered for ask pattern") ⇒ warn } + val capturedCtx = warn.asInstanceOf[TraceContextAware].traceContext + + capturedCtx should be('defined) + capturedCtx should equal(testTraceContext) } } } |