diff options
Diffstat (limited to 'kamon-spray/src/test/scala/kamon/spray/SprayServerMetricsSpec.scala')
-rw-r--r-- | kamon-spray/src/test/scala/kamon/spray/SprayServerMetricsSpec.scala | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/kamon-spray/src/test/scala/kamon/spray/SprayServerMetricsSpec.scala b/kamon-spray/src/test/scala/kamon/spray/SprayServerMetricsSpec.scala new file mode 100644 index 00000000..c4b370d7 --- /dev/null +++ b/kamon-spray/src/test/scala/kamon/spray/SprayServerMetricsSpec.scala @@ -0,0 +1,91 @@ +package kamon.spray + +import akka.actor.ActorSystem +import akka.testkit.{ TestProbe, TestKitBase } +import com.typesafe.config.ConfigFactory +import kamon.Kamon +import kamon.http.HttpServerMetrics +import kamon.metric._ +import org.scalatest.concurrent.{ PatienceConfiguration, ScalaFutures } +import org.scalatest.{ Matchers, WordSpecLike } +import spray.http.{ StatusCodes, HttpResponse, HttpRequest } +import spray.httpx.RequestBuilding + +class SprayServerMetricsSpec extends TestKitBase with WordSpecLike with Matchers with RequestBuilding + with ScalaFutures with PatienceConfiguration with TestServer { + + val collectionContext = CollectionContext(100) + + implicit lazy val system: ActorSystem = ActorSystem("spray-server-metrics-spec", ConfigFactory.parseString( + """ + |akka { + | loglevel = ERROR + |} + | + |kamon { + | metrics { + | tick-interval = 1 hour + | + | filters = [ + | { + | trace { + | includes = [ "*" ] + | excludes = [] + | } + | } + | ] + | } + |} + """.stripMargin)) + + "the Spray Server metrics instrumentation" should { + "record trace metrics for requests received" in { + Kamon(Metrics)(system).register(TraceMetrics("GET: /record-trace-metrics"), TraceMetrics.Factory).get.collect(collectionContext) + val (connection, server) = buildClientConnectionAndServer + val client = TestProbe() + + for (repetition ← 1 to 10) { + client.send(connection, Get("/record-trace-metrics")) + server.expectMsgType[HttpRequest] + server.reply(HttpResponse(entity = "ok")) + client.expectMsgType[HttpResponse] + } + + for (repetition ← 1 to 5) { + client.send(connection, Get("/record-trace-metrics")) + server.expectMsgType[HttpRequest] + server.reply(HttpResponse(entity = "bad-request", status = StatusCodes.BadRequest)) + client.expectMsgType[HttpResponse] + } + + val snapshot = Kamon(Metrics)(system).register(TraceMetrics("GET: /record-trace-metrics"), TraceMetrics.Factory).get.collect(collectionContext) + snapshot.elapsedTime.numberOfMeasurements should be(15) + } + + "record http serve metrics for all the requests" in { + Kamon(Metrics)(system).register(HttpServerMetrics, HttpServerMetrics.Factory).get.collect(collectionContext) + val (connection, server) = buildClientConnectionAndServer + val client = TestProbe() + + for (repetition ← 1 to 10) { + client.send(connection, Get("/record-http-metrics")) + server.expectMsgType[HttpRequest] + server.reply(HttpResponse(entity = "ok")) + client.expectMsgType[HttpResponse] + } + + for (repetition ← 1 to 5) { + client.send(connection, Get("/record-http-metrics")) + server.expectMsgType[HttpRequest] + server.reply(HttpResponse(entity = "bad-request", status = StatusCodes.BadRequest)) + client.expectMsgType[HttpResponse] + } + + val snapshot = Kamon(Metrics)(system).register(HttpServerMetrics, HttpServerMetrics.Factory).get.collect(collectionContext) + snapshot.countsPerTraceAndStatusCode("GET: /record-http-metrics")("200").count should be(10) + snapshot.countsPerTraceAndStatusCode("GET: /record-http-metrics")("400").count should be(5) + snapshot.countsPerStatusCode("200").count should be(10) + snapshot.countsPerStatusCode("400").count should be(5) + } + } +} |