blob: 53ff8a2495b9247ee17e84befb45bf8309925d69 (
plain) (
tree)
|
|
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)
}
}
}
}
|