aboutsummaryrefslogtreecommitdiff
path: root/kamon-core-tests/src/test/scala/kamon/metric/TimerSpec.scala
blob: 3fc1e16991b5af167cb12b7771c264cc361f3b83 (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
63
64
65
66
67
68
69
70
71
72
package kamon.metric

import kamon.Kamon
import org.scalatest.{Matchers, WordSpec}


class TimerSpec extends WordSpec with Matchers {
  import TimerTestHelper._

  "a Timer" should {
    "record the duration between calls to .start() and .stop() in the StartedTimer" in {
      val timer = Kamon.timer("timer-spec")
      timer.start().stop()
      timer.start().stop()
      timer.start().stop()

      timer.distribution().count shouldBe(3)
    }

    "ensure that a started timer can only be stopped once" in {
      val timer = Kamon.timer("timer-spec")
      val startedTimer = timer.start()
      startedTimer.stop()
      startedTimer.stop()
      startedTimer.stop()

      timer.distribution().count shouldBe(1)
    }


    "allow to record values and produce distributions as Histograms do" in {
      val timer = Kamon.timer("test-timer")
      timer.record(100)
      timer.record(150, 998)
      timer.record(200)

      val distribution = timer.distribution()
      distribution.min shouldBe(100)
      distribution.max shouldBe(200)
      distribution.count shouldBe(1000)
      distribution.buckets.length shouldBe 3
      distribution.buckets.map(b => (b.value, b.frequency)) should contain.allOf(
        (100 -> 1),
        (150 -> 998),
        (200 -> 1)
      )

      val emptyDistribution = timer.distribution()
      emptyDistribution.min shouldBe(0)
      emptyDistribution.max shouldBe(0)
      emptyDistribution.count shouldBe(0)
      emptyDistribution.buckets.length shouldBe 0
    }
  }
}

object TimerTestHelper {

  implicit class HistogramMetricSyntax(histogram: Histogram) {
    def distribution(resetState: Boolean = true): Distribution = histogram  match {
      case h: AtomicHdrHistogram  => h.snapshot(resetState).distribution
      case h: HdrHistogram        => h.snapshot(resetState).distribution
    }
  }

  implicit class TimerMetricSyntax(metric: TimerMetric) {
    def distribution(resetState: Boolean = true): Distribution =
      metric.refine(Map.empty[String, String]) match {
        case t: TimerImpl     => t.histogram.distribution(resetState)
      }
  }
}