From 8e7f64134c6e2ef48926bf63943fa3338636a62c Mon Sep 17 00:00:00 2001 From: dxspasojevic Date: Sat, 18 Jun 2016 12:37:01 +1000 Subject: = kamon-core: add context name and token to MDC --- .../kamon/trace/logging/MdcKeysSupportSpec.scala | 44 ++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 kamon-core/src/test/scala/kamon/trace/logging/MdcKeysSupportSpec.scala (limited to 'kamon-core/src/test/scala/kamon') 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) + } + } + } + } + +} -- cgit v1.2.3