diff options
author | Ivan Topolnak <itopolnak@despegar.com> | 2014-01-30 18:23:33 -0300 |
---|---|---|
committer | Ivan Topolnak <itopolnak@despegar.com> | 2014-01-30 18:23:33 -0300 |
commit | a15e17d2462105ad8b72054be58dc9e8f9dc64ed (patch) | |
tree | 9e8c3d1fa732918dc600538bab24940bd18b6e79 /kamon-core/src/test/scala/akka/testkit | |
parent | e47ed388f6ae147e550c39aa2a3638791ec65b66 (diff) | |
download | Kamon-a15e17d2462105ad8b72054be58dc9e8f9dc64ed.tar.gz Kamon-a15e17d2462105ad8b72054be58dc9e8f9dc64ed.tar.bz2 Kamon-a15e17d2462105ad8b72054be58dc9e8f9dc64ed.zip |
wip
Diffstat (limited to 'kamon-core/src/test/scala/akka/testkit')
-rw-r--r-- | kamon-core/src/test/scala/akka/testkit/TestProbeTracing.scala | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/kamon-core/src/test/scala/akka/testkit/TestProbeTracing.scala b/kamon-core/src/test/scala/akka/testkit/TestProbeTracing.scala index dbbf1246..c681e921 100644 --- a/kamon-core/src/test/scala/akka/testkit/TestProbeTracing.scala +++ b/kamon-core/src/test/scala/akka/testkit/TestProbeTracing.scala @@ -17,7 +17,7 @@ package akka.testkit import org.aspectj.lang.annotation._ -import kamon.trace.{ ContextAware, Trace } +import kamon.trace.{ TraceContextAware, TraceRecorder } import org.aspectj.lang.ProceedingJoinPoint import akka.testkit.TestActor.RealMessage @@ -25,13 +25,13 @@ import akka.testkit.TestActor.RealMessage class TestProbeTracing { @DeclareMixin("akka.testkit.TestActor.RealMessage") - def mixin: ContextAware = ContextAware.default + def mixin: TraceContextAware = TraceContextAware.default @Pointcut("execution(akka.testkit.TestActor.RealMessage.new(..)) && this(ctx)") - def realMessageCreation(ctx: ContextAware): Unit = {} + def realMessageCreation(ctx: TraceContextAware): Unit = {} @After("realMessageCreation(ctx)") - def afterRealMessageCreation(ctx: ContextAware): Unit = { + def afterRealMessageCreation(ctx: TraceContextAware): Unit = { // Necessary to force the initialization of ContextAware at the moment of creation. ctx.traceContext } @@ -42,11 +42,11 @@ class TestProbeTracing { @Around("testProbeReply(testProbe)") def aroundTestProbeReply(pjp: ProceedingJoinPoint, testProbe: TestProbe): Any = { val traceContext = testProbe.lastMessage match { - case msg: RealMessage ⇒ msg.asInstanceOf[ContextAware].traceContext + case msg: RealMessage ⇒ msg.asInstanceOf[TraceContextAware].traceContext case _ ⇒ None } - Trace.withContext(traceContext) { + TraceRecorder.withContext(traceContext) { pjp.proceed } } |