= kamon-core: add context name and token to MDC

1 files changed, 44 insertions, 0 deletions

diff --git a/kamon-core/src/test/scala/kamon/trace/logging/MdcKeysSupportSpec.scala b/kamon-core/src/test/scala/kamon/trace/logging/MdcKeysSupportSpec.scala
new file mode 100644
index 00000000..6193f028
--- /dev/null
+++ b/kamon-core/src/test/scala/kamon/trace/logging/MdcKeysSupportSpec.scala
@@ -0,0 +1,44 @@
+package kamon.trace.logging
+
+import kamon.testkit.BaseKamonSpec
+import kamon.trace.{ EmptyTraceContext, Tracer }
+import org.slf4j.MDC
+
+class MdcKeysSupportSpec extends BaseKamonSpec("mdc-keys-support-spec") {
+
+  "Running code with MDC support" should {
+    "add nothing to the MDC" when {
+      "the trace context is empty" in {
+        // Given an empty trace context.
+        Tracer.withContext(EmptyTraceContext) {
+          // When some code is executed with MDC support.
+          MdcKeysSupport.withMdc {
+            // Then the MDC should not contain the trace token.
+            Option(MDC.get(MdcKeysSupport.traceTokenKey)) should be(None)
+            // Or name
+            Option(MDC.get(MdcKeysSupport.traceNameKey)) should be(None)
+          }
+        }
+      }
+    }
+    "add the trace token and name to the context" when {
+      "the trace context is not empty" in {
+        // Given a trace context.
+        Tracer.withNewContext("name", Some("token")) {
+          // When some code is executed with MDC support.
+          MdcKeysSupport.withMdc {
+            // Then the MDC should contain the trace token.
+            Option(MDC.get(MdcKeysSupport.traceTokenKey)) should be(Some("token"))
+            // And name
+            Option(MDC.get(MdcKeysSupport.traceNameKey)) should be(Some("name"))
+          }
+
+          // Then after code is executed the MDC should have been cleared.
+          Option(MDC.get(MdcKeysSupport.traceTokenKey)) should be(None)
+          Option(MDC.get(MdcKeysSupport.traceNameKey)) should be(None)
+        }
+      }
+    }
+  }
+
+}