diff options
Diffstat (limited to 'kamon-core-tests/src/test/scala/kamon/trace/SpanCustomizerSpec.scala')
-rw-r--r-- | kamon-core-tests/src/test/scala/kamon/trace/SpanCustomizerSpec.scala | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/kamon-core-tests/src/test/scala/kamon/trace/SpanCustomizerSpec.scala b/kamon-core-tests/src/test/scala/kamon/trace/SpanCustomizerSpec.scala new file mode 100644 index 00000000..39bd7cd7 --- /dev/null +++ b/kamon-core-tests/src/test/scala/kamon/trace/SpanCustomizerSpec.scala @@ -0,0 +1,31 @@ +package kamon.trace + +import kamon.Kamon +import kamon.context.Context +import kamon.testkit.SpanInspection +import org.scalatest.{Matchers, WordSpec} + +class SpanCustomizerSpec extends WordSpec with Matchers with SpanInspection { + + "a SpanCustomizer" should { + "default to a Noop implementation when none is in the context" in { + val noopCustomizer = Context.Empty.get(SpanCustomizer.ContextKey) + val spanBuilder = noopCustomizer.customize(Kamon.buildSpan("noop")) + val span = inspect(spanBuilder.start()) + + span.operationName() shouldBe "noop" + span.metricTags() shouldBe empty + span.spanTags() shouldBe empty + } + + "have a simple builder for customizing the operation name" in { + val operationNameCustomizer = SpanCustomizer.forOperationName("myCustomOperationName") + val spanBuilder = operationNameCustomizer.customize(Kamon.buildSpan("noop")) + val span = inspect(spanBuilder.start()) + + span.operationName() shouldBe "myCustomOperationName" + span.metricTags() shouldBe empty + span.spanTags() shouldBe empty + } + } +} |