wip on the HttpServer instrumentation

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")