Merge branch 'samplers' of https://github.com/Elyrixia/Kamon into pr/elyrixia-samplers-206

1 files changed, 76 insertions, 0 deletions

diff --git a/kamon-core/src/test/scala/kamon/trace/SamplerSpec.scala b/kamon-core/src/test/scala/kamon/trace/SamplerSpec.scala
new file mode 100644
index 00000000..88cdf116
--- /dev/null
+++ b/kamon-core/src/test/scala/kamon/trace/SamplerSpec.scala
@@ -0,0 +1,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)
+      }
+    }
+  }
+
+}