add GaugeMetricSyntax

1 files changed, 25 insertions, 2 deletions

diff --git a/kamon-testkit/src/main/scala/kamon/testkit/MetricInspection.scala b/kamon-testkit/src/main/scala/kamon/testkit/MetricInspection.scala
index d0681fb5..bb6d27f9 100644
--- a/kamon-testkit/src/main/scala/kamon/testkit/MetricInspection.scala
+++ b/kamon-testkit/src/main/scala/kamon/testkit/MetricInspection.scala
@@ -1,8 +1,23 @@
+/* =========================================================================================
+ * Copyright © 2013-2017 the kamon project <http://kamon.io/>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+ * except in compliance with the License. You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the
+ * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+ * either express or implied. See the License for the specific language governing permissions
+ * and limitations under the License.
+ * =========================================================================================
+ */
+
 package kamon.testkit
 
 import kamon.metric._
-import _root_.scala.collection.concurrent.TrieMap
 
+import _root_.scala.collection.concurrent.TrieMap
 
 trait MetricInspection {
 
@@ -12,7 +27,7 @@ trait MetricInspection {
       instrumentsField.setAccessible(true)
 
       val instruments = instrumentsField.get(metric).asInstanceOf[TrieMap[Map[String, String], _]]
-      val instrumentsWithTheTag = instruments.keys.filter(_.keys.find(_ == tag).nonEmpty)
+      val instrumentsWithTheTag = instruments.keys.filter(_.keys.exists(_ == tag))
       instrumentsWithTheTag.map(t => t(tag)).toSeq
     }
   }
@@ -41,5 +56,13 @@ trait MetricInspection {
         case c: LongAdderCounter  => c.snapshot(resetState).value
       }
   }
+
+  implicit class GaugeMetricSyntax(gauge:  Gauge) {
+    def value(resetState: Boolean = true): Long =
+      gauge match {
+        case gm: GaugeMetric    => gm.refine(Map.empty[String, String]).value(resetState)
+        case g: AtomicLongGauge  => g.snapshot().value
+      }
+  }
 }