kamon-core/src/test/scala/kamon/trace/SamplerSpec.scala


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
73
74
75
76

package kamon.trace

import kamon.testkit.BaseKamonSpec
import kamon.util.NanoInterval

class SamplerSpec extends BaseKamonSpec("sampler-spec") {

  "the Sampler" should {
    "work as intended" when {
      "using all mode" in {
        val sampler = SampleAll

        sampler.shouldTrace should be(true)

        sampler.shouldReport(NanoInterval.default) should be(true)
      }

      "using random mode" in {
        val sampler = new RandomSampler(100)

        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(true)

        sampler.shouldReport(NanoInterval.default) should be(true)
      }

      "using ordered mode" in {
        var sampler = new OrderedSampler(1)

        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(true)

        sampler = new OrderedSampler(2)

        sampler.shouldTrace should be(false)
        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(false)
        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(false)
        sampler.shouldTrace should be(true)

        sampler.shouldReport(NanoInterval.default) should be(true)
      }

      "using threshold mode" in {
        val sampler = new ThresholdSampler(new NanoInterval(10000000L))

        sampler.shouldTrace should be(true)

        sampler.shouldReport(new NanoInterval(5000000L)) should be(false)
        sampler.shouldReport(new NanoInterval(10000000L)) should be(true)
        sampler.shouldReport(new NanoInterval(15000000L)) should be(true)
        sampler.shouldReport(new NanoInterval(0L)) should be(false)
      }

      "using clock mode" in {
        val sampler = new ClockSampler(new NanoInterval(10000000L))

        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(false)
        Thread.sleep(1L)
        sampler.shouldTrace should be(false)
        Thread.sleep(10L)
        sampler.shouldTrace should be(true)
        sampler.shouldTrace should be(false)

        sampler.shouldReport(NanoInterval.default) should be(true)
      }
    }
  }

}