1 files changed, 72 insertions, 0 deletions
diff --git a/kamon-core/src/test/scala/kamon/metric/instrument/GaugeSpec.scala b/kamon-core/src/test/scala/kamon/metric/instrument/GaugeSpec.scala
new file mode 100644
index 00000000..9192d999
--- /dev/null
+++ b/kamon-core/src/test/scala/kamon/metric/instrument/GaugeSpec.scala
@@ -0,0 +1,72 @@
+package kamon.metric.instrument
+
+import java.util.concurrent.atomic.AtomicLong
+
+import akka.actor.ActorSystem
+import com.typesafe.config.ConfigFactory
+import kamon.Kamon
+import kamon.metric.{ Metrics, Scale, CollectionContext }
+import org.scalatest.{ Matchers, WordSpecLike }
+import scala.concurrent.duration._
+
+class GaugeSpec extends WordSpecLike with Matchers {
+  implicit val system = ActorSystem("gauge-spec", ConfigFactory.parseString(
+    """
+      |kamon.metrics {
+      |  flush-interval = 1 hour
+      |  default-collection-context-buffer-size = 10
+      |  precision {
+      |    default-gauge-precision {
+      |      refresh-interval = 100 milliseconds
+      |      highest-trackable-value = 999999999
+      |      significant-value-digits = 2
+      |    }
+      |  }
+      |}
+    """.stripMargin))
+
+  "a Gauge" should {
+    "automatically record the current value using the configured refresh-interval" in {
+      val numberOfValuesRecorded = new AtomicLong(0)
+      val gauge = Gauge.fromDefaultConfig(system) { () ⇒ numberOfValuesRecorded.addAndGet(1) }
+
+      Thread.sleep(1.second.toMillis)
+      numberOfValuesRecorded.get() should be(10L +- 1L)
+      gauge.cleanup
+    }
+
+    "stop automatically recording after a call to cleanup" in {
+      val numberOfValuesRecorded = new AtomicLong(0)
+      val gauge = Gauge.fromDefaultConfig(system) { () ⇒ numberOfValuesRecorded.addAndGet(1) }
+
+      Thread.sleep(1.second.toMillis)
+      gauge.cleanup
+      numberOfValuesRecorded.get() should be(10L +- 1L)
+      Thread.sleep(1.second.toMillis)
+      numberOfValuesRecorded.get() should be(10L +- 1L)
+    }
+
+    "produce a Histogram snapshot including all the recorded values" in {
+      val numberOfValuesRecorded = new AtomicLong(0)
+      val gauge = Gauge.fromDefaultConfig(system) { () ⇒ numberOfValuesRecorded.addAndGet(1) }
+
+      Thread.sleep(1.second.toMillis)
+      gauge.cleanup
+      val snapshot = gauge.collect(Kamon(Metrics).buildDefaultCollectionContext)
+
+      snapshot.numberOfMeasurements should be(10L +- 1L)
+      snapshot.min should be(1)
+      snapshot.max should be(10L +- 1L)
+    }
+
+    "not record the current value when doing a collection" in {
+      val numberOfValuesRecorded = new AtomicLong(0)
+      val gauge = Gauge(Histogram.Precision.Normal, 10000L, Scale.Unit, 1 hour, system)(() ⇒ numberOfValuesRecorded.addAndGet(1))
+
+      val snapshot = gauge.collect(Kamon(Metrics).buildDefaultCollectionContext)
+
+      snapshot.numberOfMeasurements should be(0)
+      numberOfValuesRecorded.get() should be(0)
+    }
+  }
+}