! all: introduced support for metric tags.

1 files changed, 6 insertions, 1 deletions

diff --git a/kamon-core/src/test/scala/kamon/testkit/BaseKamonSpec.scala b/kamon-core/src/test/scala/kamon/testkit/BaseKamonSpec.scala
index f3d809bf..e7b18770 100644
--- a/kamon-core/src/test/scala/kamon/testkit/BaseKamonSpec.scala
+++ b/kamon-core/src/test/scala/kamon/testkit/BaseKamonSpec.scala
@@ -20,7 +20,7 @@ import akka.testkit.{ ImplicitSender, TestKitBase }
 import akka.actor.ActorSystem
 import com.typesafe.config.{ Config, ConfigFactory }
 import kamon.Kamon
-import kamon.metric.{ SubscriptionsDispatcher, EntitySnapshot, MetricsImpl }
+import kamon.metric.{ Entity, SubscriptionsDispatcher, EntitySnapshot, MetricsImpl }
 import kamon.trace.TraceContext
 import kamon.util.LazyActorRef
 import org.scalatest.{ BeforeAndAfterAll, Matchers, WordSpecLike }
@@ -48,6 +48,11 @@ abstract class BaseKamonSpec(actorSystemName: String) extends TestKitBase with W
     recorder.collect(collectionContext)
   }
 
+  def takeSnapshotOf(name: String, category: String, tags: Map[String, String]): EntitySnapshot = {
+    val recorder = Kamon.metrics.find(Entity(name, category, tags)).get
+    recorder.collect(collectionContext)
+  }
+
   def flushSubscriptions(): Unit = {
     val subscriptionsField = Kamon.metrics.getClass.getDeclaredField("_subscriptions")
     subscriptionsField.setAccessible(true)