aboutsummaryrefslogtreecommitdiff
path: root/kamon-testkit/src/main/scala/testkit/TestProbeInstrumentation.scala
diff options
context:
space:
mode:
Diffstat (limited to 'kamon-testkit/src/main/scala/testkit/TestProbeInstrumentation.scala')
-rw-r--r--kamon-testkit/src/main/scala/testkit/TestProbeInstrumentation.scala4
1 files changed, 2 insertions, 2 deletions
diff --git a/kamon-testkit/src/main/scala/testkit/TestProbeInstrumentation.scala b/kamon-testkit/src/main/scala/testkit/TestProbeInstrumentation.scala
index de867035..825cc718 100644
--- a/kamon-testkit/src/main/scala/testkit/TestProbeInstrumentation.scala
+++ b/kamon-testkit/src/main/scala/testkit/TestProbeInstrumentation.scala
@@ -17,7 +17,7 @@
package akka.testkit
import org.aspectj.lang.annotation._
-import kamon.trace.{ TraceContextAware, TraceRecorder }
+import kamon.trace.{ EmptyTraceContext, TraceContextAware, TraceRecorder }
import org.aspectj.lang.ProceedingJoinPoint
import akka.testkit.TestActor.RealMessage
@@ -43,7 +43,7 @@ class TestProbeInstrumentation {
def aroundTestProbeReply(pjp: ProceedingJoinPoint, testProbe: TestProbe): Any = {
val traceContext = testProbe.lastMessage match {
case msg: RealMessage ⇒ msg.asInstanceOf[TraceContextAware].traceContext
- case _ ⇒ None
+ case _ ⇒ EmptyTraceContext
}
TraceRecorder.withTraceContext(traceContext) {