1 files changed, 80 insertions, 0 deletions
diff --git a/kamon-statsd/src/test/scala/kamon/statsd/SimpleMetricKeyGeneratorSpec.scala b/kamon-statsd/src/test/scala/kamon/statsd/SimpleMetricKeyGeneratorSpec.scala
new file mode 100644
index 00000000..ed3fae5b
--- /dev/null
+++ b/kamon-statsd/src/test/scala/kamon/statsd/SimpleMetricKeyGeneratorSpec.scala
@@ -0,0 +1,80 @@
+package kamon.statsd
+
+import com.typesafe.config.ConfigFactory
+import kamon.metric.{ MetricGroupCategory, MetricGroupIdentity, MetricIdentity }
+import org.scalatest.{ Matchers, WordSpec }
+
+class SimpleMetricKeyGeneratorSpec extends WordSpec with Matchers {
+
+  val defaultConfiguration = ConfigFactory.parseString(
+    """
+      |kamon.statsd.simple-metric-key-generator {
+      |  application = kamon
+      |  hostname-override = none
+      |  include-hostname = true
+      |  metric-name-normalization-strategy = normalize
+      |}
+    """.stripMargin)
+
+  "the StatsDMetricSender" should {
+    "generate metric names that follow the application.host.entity.entity-name.metric-name pattern by default" in {
+      implicit val metricKeyGenerator = new SimpleMetricKeyGenerator(defaultConfiguration) {
+        override def hostName: String = "localhost"
+      }
+
+      buildMetricKey("actor", "/user/example", "processing-time") should be("kamon.localhost.actor._user_example.processing-time")
+      buildMetricKey("trace", "POST: /kamon/example", "elapsed-time") should be("kamon.localhost.trace.POST-_kamon_example.elapsed-time")
+    }
+
+    "allow to override the hostname" in {
+      val hostOverrideConfig = ConfigFactory.parseString("kamon.statsd.simple-metric-key-generator.hostname-override = kamon-host")
+      implicit val metricKeyGenerator = new SimpleMetricKeyGenerator(hostOverrideConfig.withFallback(defaultConfiguration)) {
+        override def hostName: String = "localhost"
+      }
+
+      buildMetricKey("actor", "/user/example", "processing-time") should be("kamon.kamon-host.actor._user_example.processing-time")
+      buildMetricKey("trace", "POST: /kamon/example", "elapsed-time") should be("kamon.kamon-host.trace.POST-_kamon_example.elapsed-time")
+    }
+
+    "removes host name when attribute 'include-hostname' is set to false" in {
+      val hostOverrideConfig = ConfigFactory.parseString("kamon.statsd.simple-metric-key-generator.include-hostname = false")
+      implicit val metricKeyGenerator = new SimpleMetricKeyGenerator(hostOverrideConfig.withFallback(defaultConfiguration)) {
+        override def hostName: String = "localhost"
+      }
+
+      buildMetricKey("actor", "/user/example", "processing-time") should be("kamon.actor._user_example.processing-time")
+      buildMetricKey("trace", "POST: /kamon/example", "elapsed-time") should be("kamon.trace.POST-_kamon_example.elapsed-time")
+    }
+
+    "remove spaces, colons and replace '/' with '_' when the normalization strategy is 'normalize'" in {
+      val hostOverrideConfig = ConfigFactory.parseString("kamon.statsd.simple-metric-key-generator.metric-name-normalization-strategy = normalize")
+      implicit val metricKeyGenerator = new SimpleMetricKeyGenerator(hostOverrideConfig.withFallback(defaultConfiguration)) {
+        override def hostName: String = "localhost.local"
+      }
+
+      buildMetricKey("actor", "/user/example", "processing-time") should be("kamon.localhost_local.actor._user_example.processing-time")
+      buildMetricKey("trace", "POST: /kamon/example", "elapsed-time") should be("kamon.localhost_local.trace.POST-_kamon_example.elapsed-time")
+    }
+
+    "percent-encode special characters in the group name and hostname when the normalization strategy is 'normalize'" in {
+      val hostOverrideConfig = ConfigFactory.parseString("kamon.statsd.simple-metric-key-generator.metric-name-normalization-strategy = percent-encode")
+      implicit val metricKeyGenerator = new SimpleMetricKeyGenerator(hostOverrideConfig.withFallback(defaultConfiguration)) {
+        override def hostName: String = "localhost.local"
+      }
+
+      buildMetricKey("actor", "/user/example", "processing-time") should be("kamon.localhost%2Elocal.actor.%2Fuser%2Fexample.processing-time")
+      buildMetricKey("trace", "POST: /kamon/example", "elapsed-time") should be("kamon.localhost%2Elocal.trace.POST%3A%20%2Fkamon%2Fexample.elapsed-time")
+    }
+  }
+
+  def buildMetricKey(categoryName: String, entityName: String, metricName: String)(implicit metricKeyGenerator: SimpleMetricKeyGenerator): String = {
+    val metricIdentity = new MetricIdentity { val name: String = metricName }
+    val groupIdentity = new MetricGroupIdentity {
+      val name: String = entityName
+      val category: MetricGroupCategory = new MetricGroupCategory {
+        val name: String = categoryName
+      }
+    }
+    metricKeyGenerator.generateKey(groupIdentity, metricIdentity)
+  }
+}