simple test for ActorLogging instrumentation

1 files changed, 34 insertions, 0 deletions

diff --git a/kamon-trace/src/test/scala/kamon/trace/instrumentation/ActorLoggingSpec.scala b/kamon-trace/src/test/scala/kamon/trace/instrumentation/ActorLoggingSpec.scala
new file mode 100644
index 00000000..896faf69
--- /dev/null
+++ b/kamon-trace/src/test/scala/kamon/trace/instrumentation/ActorLoggingSpec.scala
@@ -0,0 +1,34 @@
+package kamon.trace.instrumentation
+
+import akka.testkit.TestKit
+import org.scalatest.{Inspectors, Matchers, WordSpecLike}
+import akka.actor.{Props, ActorLogging, Actor, ActorSystem}
+import akka.event.Logging.{LogEvent}
+import kamon.trace.{ContextAware, TraceContext, Trace}
+
+class ActorLoggingSpec extends TestKit(ActorSystem("actor-logging-spec")) with WordSpecLike with Matchers with Inspectors {
+
+  "the ActorLogging instrumentation" should {
+    "attach the TraceContext (if available) to log events" in {
+      val testTraceContext = Some(TraceContext(Actor.noSender, 1))
+      val loggerActor = system.actorOf(Props[LoggerActor])
+      system.eventStream.subscribe(testActor, classOf[LogEvent])
+
+      Trace.withContext(testTraceContext) {
+        loggerActor ! "info"
+      }
+
+      expectMsgPF() {
+        case event: LogEvent =>
+          val ctxInEvent = event.asInstanceOf[ContextAware].traceContext
+          ctxInEvent should equal(testTraceContext)
+      }
+    }
+  }
+}
+
+class LoggerActor extends Actor with ActorLogging {
+  def receive = {
+    case "info" => log.info("TraceContext => {}", Trace.context())
+  }
+}