diff options
Diffstat (limited to 'kamon-testkit')
-rw-r--r-- | kamon-testkit/src/main/scala/kamon/testkit/ContextTesting.scala | 3 | ||||
-rw-r--r-- | kamon-testkit/src/main/scala/kamon/testkit/SpanInspection.scala | 13 |
2 files changed, 15 insertions, 1 deletions
diff --git a/kamon-testkit/src/main/scala/kamon/testkit/ContextTesting.scala b/kamon-testkit/src/main/scala/kamon/testkit/ContextTesting.scala index 24e8390a..d36d0df2 100644 --- a/kamon-testkit/src/main/scala/kamon/testkit/ContextTesting.scala +++ b/kamon-testkit/src/main/scala/kamon/testkit/ContextTesting.scala @@ -23,6 +23,9 @@ trait ContextTesting { def contextWithLocal(value: String): Context = Context.of(StringKey, Some(value)) + + + } object ContextTesting extends ContextTesting diff --git a/kamon-testkit/src/main/scala/kamon/testkit/SpanInspection.scala b/kamon-testkit/src/main/scala/kamon/testkit/SpanInspection.scala index fbfdc7c3..36b57f2d 100644 --- a/kamon-testkit/src/main/scala/kamon/testkit/SpanInspection.scala +++ b/kamon-testkit/src/main/scala/kamon/testkit/SpanInspection.scala @@ -19,7 +19,7 @@ import java.time.Instant import kamon.Kamon import kamon.trace.{Span, SpanContext} -import kamon.trace.Span.FinishedSpan +import kamon.trace.Span.{FinishedSpan, TagValue} import scala.reflect.ClassTag import scala.util.Try @@ -52,9 +52,20 @@ object SpanInspection { def spanTags(): Map[String, Span.TagValue] = spanData.tags + def tag(key: String): Option[String] = + spanTag(key).map { + case TagValue.String(string) => string + case TagValue.Number(number) => number.toString + case TagValue.True => "true" + case TagValue.False => "false" + } + def metricTags(): Map[String, String] = getField[Span.Local, Map[String, String]](realSpan, "customMetricTags") + def metricTag(key: String): Option[String] = + metricTags().get(key) + def from(): Instant = getField[Span.Local, Instant](realSpan, "from") |