add ContextTesting utilities

1 files changed, 11 insertions, 0 deletions

diff --git a/kamon-testkit/src/main/scala/kamon/testkit/ContextTesting.scala b/kamon-testkit/src/main/scala/kamon/testkit/ContextTesting.scala
new file mode 100644
index 00000000..507f17ad
--- /dev/null
+++ b/kamon-testkit/src/main/scala/kamon/testkit/ContextTesting.scala
@@ -0,0 +1,11 @@
+package kamon.testkit
+
+import kamon.context.{Context, Key}
+
+trait ContextTesting {
+  val TestLocalKey = Key.local[Option[String]]("test-local-key", None)
+  val TestBroadcastKey = Key.broadcast[Option[String]]("test-local-key", None)
+
+  def contextWithLocal(value: String): Context =
+    Context.create(TestLocalKey, Some(value))
+}