aboutsummaryrefslogtreecommitdiff
path: root/kamon-spray/src/test/scala/kamon/spray/SprayServerMetricsSpec.scala
blob: 58bb2885ebae7266453260c66109d2c1ac46b0dd (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package kamon.spray

import akka.testkit.TestProbe
import com.typesafe.config.ConfigFactory
import kamon.testkit.BaseKamonSpec
import org.scalatest.concurrent.{ PatienceConfiguration, ScalaFutures }
import spray.http.{ StatusCodes, HttpResponse, HttpRequest }
import spray.httpx.RequestBuilding

class SprayServerMetricsSpec extends BaseKamonSpec("spray-server-metrics-spec") with RequestBuilding with ScalaFutures
    with PatienceConfiguration with TestServer {

  override lazy val config =
    ConfigFactory.parseString(
      """
        |kamon.metric {
        |  tick-interval = 1 hour
        |}
        |
        |akka.loggers = ["akka.event.slf4j.Slf4jLogger"]
      """.stripMargin)

  "the Spray Server metrics instrumentation" should {
    "record trace metrics for processed requests" in {
      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 = takeSnapshotOf("GET: /record-trace-metrics", "trace")
      snapshot.histogram("elapsed-time").get.numberOfMeasurements should be(15)
    }

    "record http server metrics for all the requests" in {
      val (connection, server) = buildClientConnectionAndServer
      val client = TestProbe()

      // Erase metrics recorder from previous tests.
      takeSnapshotOf("spray-server", "http-server")

      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 = takeSnapshotOf("spray-server", "http-server")
      snapshot.counter("GET: /record-http-metrics_200").get.count should be(10)
      snapshot.counter("GET: /record-http-metrics_400").get.count should be(5)
      snapshot.counter("200").get.count should be(10)
      snapshot.counter("400").get.count should be(5)
    }
  }
}