allow enabling/disabling of metrics on a Span

1 files changed, 21 insertions, 0 deletions

diff --git a/kamon-core-tests/src/test/scala/kamon/trace/SpanMetrics.scala b/kamon-core-tests/src/test/scala/kamon/trace/SpanMetrics.scala
index 4425ac92..ea29d92e 100644
--- a/kamon-core-tests/src/test/scala/kamon/trace/SpanMetrics.scala
+++ b/kamon-core-tests/src/test/scala/kamon/trace/SpanMetrics.scala
@@ -42,6 +42,27 @@ class SpanMetrics extends WordSpecLike with Matchers with MetricInspection with
 
     }
 
+    "not be recorded when disableMetrics() is called on the Span" in {
+      val operation = "span-with-disabled-metrics"
+      buildSpan(operation)
+        .start()
+        .disableMetrics()
+        .finish()
+
+      Span.Metrics.ProcessingTime.valuesForTag("operation") shouldNot contain(operation)
+    }
+
+    "be recorded if metrics are enabled by calling enableMetrics() on the Span" in {
+      val operation = "span-with-re-enabled-metrics"
+      buildSpan(operation)
+        .start()
+        .disableMetrics()
+        .enableMetrics()
+        .finish()
+
+      Span.Metrics.ProcessingTime.valuesForTag("operation") should contain(operation)
+    }
+
     "record correctly error latency and count" in {
       val operation = "span-failure"
       val operationTag = "operation" -> operation