blob: 53ff8a2495b9247ee17e84befb45bf8309925d69 (
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
|
package kamon.trace
import akka.testkit.TestKit
import akka.actor.ActorSystem
import org.scalatest.{ OptionValues, Matchers, WordSpecLike }
import org.scalatest.concurrent.PatienceConfiguration
class TraceLocalSpec extends TestKit(ActorSystem("trace-local-spec")) with WordSpecLike with Matchers
with PatienceConfiguration with OptionValues {
object SampleTraceLocalKey extends TraceLocal.TraceLocalKey { type ValueType = String }
"the TraceLocal storage" should {
"allow storing and retrieving values" in {
TraceRecorder.withNewTraceContext("store-and-retrieve-trace-local") {
val testString = "Hello World"
TraceLocal.store(SampleTraceLocalKey)(testString)
TraceLocal.retrieve(SampleTraceLocalKey).value should equal(testString)
}
}
"return None when retrieving a non existent key" in {
TraceRecorder.withNewTraceContext("non-existent-key") {
TraceLocal.retrieve(SampleTraceLocalKey) should equal(None)
}
}
"return None when retrieving a key without a current TraceContext" in {
TraceLocal.retrieve(SampleTraceLocalKey) should equal(None)
}
"be attached to the TraceContext when it is propagated" in {
val testString = "Hello World"
val testContext = TraceRecorder.withNewTraceContext("manually-propagated-trace-local") {
TraceLocal.store(SampleTraceLocalKey)(testString)
TraceLocal.retrieve(SampleTraceLocalKey).value should equal(testString)
TraceRecorder.currentContext
}
/** No TraceLocal should be available here */
TraceLocal.retrieve(SampleTraceLocalKey) should equal(None)
TraceRecorder.withTraceContext(testContext) {
TraceLocal.retrieve(SampleTraceLocalKey).value should equal(testString)
}
}
}
}
|