aboutsummaryrefslogtreecommitdiff
path: root/kamon-core/src/test/scala/kamon/metric/instrument/GaugeSpec.scala
blob: bd39652cb68da1403d93a7cbaa091bf526ed6578 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package kamon.metric.instrument

import java.util.concurrent.atomic.AtomicLong
import kamon.metric.instrument.Histogram.DynamicRange
import kamon.testkit.BaseKamonSpec
import scala.concurrent.duration._

class GaugeSpec extends BaseKamonSpec("gauge-spec") {

  "a Gauge" should {
    "automatically record the current value using the configured refresh-interval" in new GaugeFixture {
      val (numberOfValuesRecorded, gauge) = createGauge()
      Thread.sleep(1.second.toMillis)

      numberOfValuesRecorded.get() should be(10L +- 1L)
      gauge.cleanup
    }

    "stop automatically recording after a call to cleanup" in new GaugeFixture {
      val (numberOfValuesRecorded, gauge) = createGauge()
      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 new GaugeFixture {
      val (numberOfValuesRecorded, gauge) = createGauge()

      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 new GaugeFixture {
      val (numberOfValuesRecorded, gauge) = createGauge(10 seconds)

      val snapshot = gauge.collect(kamon.metrics.buildDefaultCollectionContext)
      snapshot.numberOfMeasurements should be(0)
      numberOfValuesRecorded.get() should be(0)
    }
  }

  trait GaugeFixture {
    def createGauge(refreshInterval: FiniteDuration = 100 millis): (AtomicLong, Gauge) = {
      val recordedValuesCounter = new AtomicLong(0)
      val gauge = Gauge(DynamicRange(1, 100, 2), refreshInterval, kamon.metrics.settings.refreshScheduler, {
        ()  recordedValuesCounter.addAndGet(1)
      })

      (recordedValuesCounter, gauge)
    }

  }
}