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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
package kamon.trace
import kamon.testkit.{MetricInspection, Reconfigure, TestSpanReporter}
import kamon.util.Registration
import kamon.Kamon
import kamon.trace.Span.{Annotation, TagValue}
import org.scalatest.concurrent.Eventually
import org.scalatest.{BeforeAndAfterAll, Matchers, OptionValues, WordSpec}
import org.scalatest.time.SpanSugar._
class LocalSpanSpec extends WordSpec with Matchers with BeforeAndAfterAll with Eventually with OptionValues
with Reconfigure with MetricInspection {
"a real span" when {
"sampled and finished" should {
"be sent to the Span reporters" in {
Kamon.buildSpan("test-span")
.withSpanTag("test", "value")
.withStartTimestamp(100)
.start()
.finish(200)
eventually(timeout(2 seconds)) {
val finishedSpan = reporter.nextSpan().value
finishedSpan.operationName shouldBe("test-span")
finishedSpan.startTimestampMicros shouldBe 100
finishedSpan.endTimestampMicros shouldBe 200
finishedSpan.tags should contain("test" -> TagValue.String("value"))
}
}
"pass all the tags, annotations and baggage to the FinishedSpan instance when started and finished" in {
Kamon.buildSpan("full-span")
.withSpanTag("builder-string-tag", "value")
.withSpanTag("builder-boolean-tag-true", true)
.withSpanTag("builder-boolean-tag-false", false)
.withSpanTag("builder-number-tag", 42)
.withStartTimestamp(100)
.start()
.addSpanTag("span-string-tag", "value")
.addSpanTag("span-boolean-tag-true", true)
.addSpanTag("span-boolean-tag-false", false)
.addSpanTag("span-number-tag", 42)
.annotate("simple-annotation")
.annotate("regular-annotation", Map("data" -> "something"))
.annotate(4200, "custom-annotation-1", Map("custom" -> "yes-1"))
.annotate(Annotation(4201, "custom-annotation-2", Map("custom" -> "yes-2")))
.setOperationName("fully-populated-span")
.finish(200)
eventually(timeout(2 seconds)) {
val finishedSpan = reporter.nextSpan().value
finishedSpan.operationName shouldBe ("fully-populated-span")
finishedSpan.startTimestampMicros shouldBe 100
finishedSpan.endTimestampMicros shouldBe 200
finishedSpan.tags should contain allOf(
"builder-string-tag" -> TagValue.String("value"),
"builder-boolean-tag-true" -> TagValue.True,
"builder-boolean-tag-false" -> TagValue.False,
"builder-number-tag" -> TagValue.Number(42),
"span-string-tag" -> TagValue.String("value"),
"span-boolean-tag-true" -> TagValue.True,
"span-boolean-tag-false" -> TagValue.False,
"span-number-tag" -> TagValue.Number(42)
)
finishedSpan.annotations.length shouldBe (4)
val annotations = finishedSpan.annotations.groupBy(_.name)
annotations.keys should contain allOf(
"simple-annotation",
"regular-annotation",
"custom-annotation-1",
"custom-annotation-2"
)
val customAnnotationOne = annotations("custom-annotation-1").head
customAnnotationOne.timestampMicros shouldBe (4200)
customAnnotationOne.fields shouldBe (Map("custom" -> "yes-1"))
val customAnnotationTwo = annotations("custom-annotation-2").head
customAnnotationTwo.timestampMicros shouldBe (4201)
customAnnotationTwo.fields shouldBe (Map("custom" -> "yes-2"))
}
}
}
}
@volatile var registration: Registration = _
val reporter = new TestSpanReporter()
override protected def beforeAll(): Unit = {
enableFastSpanFlushing()
sampleAlways()
registration = Kamon.addReporter(reporter)
}
override protected def afterAll(): Unit = {
registration.cancel()
}
}
|
