diff options
Diffstat (limited to 'kamon-play-2.4.x/src/test/scala/kamon/play/RequestInstrumentationSpec.scala')
-rw-r--r-- | kamon-play-2.4.x/src/test/scala/kamon/play/RequestInstrumentationSpec.scala | 275 |
1 files changed, 275 insertions, 0 deletions
diff --git a/kamon-play-2.4.x/src/test/scala/kamon/play/RequestInstrumentationSpec.scala b/kamon-play-2.4.x/src/test/scala/kamon/play/RequestInstrumentationSpec.scala new file mode 100644 index 00000000..43db7a12 --- /dev/null +++ b/kamon-play-2.4.x/src/test/scala/kamon/play/RequestInstrumentationSpec.scala @@ -0,0 +1,275 @@ +/* ========================================================================================= + * Copyright © 2013-2014 the kamon project <http://kamon.io/> + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file + * except in compliance with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software distributed under the + * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, + * either express or implied. See the License for the specific language governing permissions + * and limitations under the License. + * ========================================================================================= + */ + +package kamon.play + +import javax.inject.Inject + +import kamon.Kamon +import kamon.metric.instrument.CollectionContext +import kamon.play.action.TraceName +import kamon.trace.TraceLocal.HttpContextKey +import kamon.trace.{ Tracer, TraceLocal } +import org.scalatestplus.play._ +import play.api.DefaultGlobal +import play.api.http.{ HttpErrorHandler, Writeable } +import play.api.libs.concurrent.Execution.Implicits.defaultContext +import play.api.libs.ws.WS +import play.api.mvc.Results.Ok +import play.api.mvc._ +import play.api.routing.SimpleRouter +import play.api.test.Helpers._ +import play.api.test._ +import play.core.routing._ +import play.api.http.HttpFilters + +import scala.concurrent.duration._ +import scala.concurrent.{ Await, Future } + +class RequestInstrumentationSpec extends PlaySpec with OneServerPerSuite { + Kamon.start() + System.setProperty("config.file", "./kamon-play/src/test/resources/conf/application.conf") + + override lazy val port: Port = 19002 + val executor = scala.concurrent.ExecutionContext.Implicits.global + + implicit override lazy val app = FakeApplication(withRoutes = { + + case ("GET", "/async") ⇒ + Action.async { + Future { + Ok("Async.async") + }(executor) + } + case ("GET", "/notFound") ⇒ + Action { + Results.NotFound + } + case ("GET", "/error") ⇒ + Action { + throw new Exception("This page generates an error!") + Ok("This page will generate an error!") + } + case ("GET", "/redirect") ⇒ + Action { + Results.Redirect("/redirected", MOVED_PERMANENTLY) + } + case ("GET", "/default") ⇒ + Action { + Ok("default") + } + case ("GET", "/async-renamed") ⇒ + TraceName("renamed-trace") { + Action.async { + Future { + Ok("Async.async") + }(executor) + } + } + case ("GET", "/retrieve") ⇒ + Action { + Ok("retrieve from TraceLocal") + } + }, additionalConfiguration = Map( + ("application.router", "kamon.play.Routes"), + ("play.http.filters", "kamon.play.TestHttpFilters"), + ("play.http.requestHandler", "play.api.http.DefaultHttpRequestHandler"), + ("logger.root", "OFF"), + ("logger.play", "OFF"), + ("logger.application", "OFF"))) + + val traceTokenValue = "kamon-trace-token-test" + val traceTokenHeaderName = "X-Trace-Token" + val expectedToken = Some(traceTokenValue) + val traceTokenHeader = traceTokenHeaderName -> traceTokenValue + val traceLocalStorageValue = "localStorageValue" + val traceLocalStorageKey = "localStorageKey" + val traceLocalStorageHeader = traceLocalStorageKey -> traceLocalStorageValue + + "the Request instrumentation" should { + "respond to the Async Action with X-Trace-Token" in { + val Some(result) = route(FakeRequest(GET, "/async").withHeaders(traceTokenHeader, traceLocalStorageHeader)) + header(traceTokenHeaderName, result) must be(expectedToken) + } + + "respond to the NotFound Action with X-Trace-Token" in { + val Some(result) = route(FakeRequest(GET, "/notFound").withHeaders(traceTokenHeader)) + header(traceTokenHeaderName, result) must be(expectedToken) + } + + "respond to the Default Action with X-Trace-Token" in { + val Some(result) = route(FakeRequest(GET, "/default").withHeaders(traceTokenHeader)) + header(traceTokenHeaderName, result) must be(expectedToken) + } + + "respond to the Redirect Action with X-Trace-Token" in { + val Some(result) = route(FakeRequest(GET, "/redirect").withHeaders(traceTokenHeader)) + header("Location", result) must be(Some("/redirected")) + header(traceTokenHeaderName, result) must be(expectedToken) + } + + "respond to the Async Action with X-Trace-Token and the renamed trace" in { + val result = Await.result(route(FakeRequest(GET, "/async-renamed").withHeaders(traceTokenHeader)).get, 10 seconds) + Tracer.currentContext.name must be("renamed-trace") + Some(result.header.headers(traceTokenHeaderName)) must be(expectedToken) + } + + "propagate the TraceContext and LocalStorage through of filters in the current request" in { + route(FakeRequest(GET, "/retrieve").withHeaders(traceTokenHeader, traceLocalStorageHeader)) + TraceLocal.retrieve(TraceLocalKey).get must be(traceLocalStorageValue) + } + + "response to the getRouted Action and normalise the current TraceContext name" in { + Await.result(WS.url(s"http://localhost:$port/getRouted").get(), 10 seconds) + Kamon.metrics.find("getRouted.get", "trace") must not be empty + } + + "response to the postRouted Action and normalise the current TraceContext name" in { + Await.result(WS.url(s"http://localhost:$port/postRouted").post("content"), 10 seconds) + Kamon.metrics.find("postRouted.post", "trace") must not be empty + } + + "response to the showRouted Action and normalise the current TraceContext name" in { + Await.result(WS.url(s"http://localhost:$port/showRouted/2").get(), 10 seconds) + Kamon.metrics.find("show.some.id.get", "trace") must not be empty + } + + "include HttpContext information for help to diagnose possible errors" in { + Await.result(WS.url(s"http://localhost:$port/getRouted").get(), 10 seconds) + route(FakeRequest(GET, "/default").withHeaders("User-Agent" -> "Fake-Agent")) + + val httpCtx = TraceLocal.retrieve(HttpContextKey).get + httpCtx.agent must be("Fake-Agent") + httpCtx.uri must be("/default") + httpCtx.xforwarded must be("unknown") + } + + "record http server metrics for all processed requests" in { + val collectionContext = CollectionContext(100) + Kamon.metrics.find("play-server", "http-server").get.collect(collectionContext) + + for (repetition ← 1 to 10) { + Await.result(route(FakeRequest(GET, "/default").withHeaders(traceTokenHeader)).get, 10 seconds) + } + + for (repetition ← 1 to 5) { + Await.result(route(FakeRequest(GET, "/notFound").withHeaders(traceTokenHeader)).get, 10 seconds) + } + + for (repetition ← 1 to 5) { + Await.result(routeWithOnError(FakeRequest(GET, "/error").withHeaders(traceTokenHeader)).get, 10 seconds) + } + + val snapshot = Kamon.metrics.find("play-server", "http-server").get.collect(collectionContext) + snapshot.counter("GET: /default_200").get.count must be(10) + snapshot.counter("GET: /notFound_404").get.count must be(5) + snapshot.counter("GET: /error_500").get.count must be(5) + snapshot.counter("200").get.count must be(10) + snapshot.counter("404").get.count must be(5) + snapshot.counter("500").get.count must be(5) + } + } + + def routeWithOnError[T](req: Request[T])(implicit w: Writeable[T]): Option[Future[Result]] = { + route(req).map { result ⇒ + result.recoverWith { + case t: Throwable ⇒ DefaultGlobal.onError(req, t) + } + } + } +} + +object TraceLocalKey extends TraceLocal.TraceLocalKey { + type ValueType = String +} + +class TraceLocalFilter extends Filter { + val traceLocalStorageValue = "localStorageValue" + val traceLocalStorageKey = "localStorageKey" + val traceLocalStorageHeader = traceLocalStorageKey -> traceLocalStorageValue + + override def apply(next: (RequestHeader) ⇒ Future[Result])(header: RequestHeader): Future[Result] = { + Tracer.withContext(Tracer.currentContext) { + + TraceLocal.store(TraceLocalKey)(header.headers.get(traceLocalStorageKey).getOrElse("unknown")) + + next(header).map { + result ⇒ + { + result.withHeaders(traceLocalStorageKey -> TraceLocal.retrieve(TraceLocalKey).get) + } + } + } + } +} + +class TestHttpFilters @Inject() (traceLocalFilter: TraceLocalFilter) extends HttpFilters { + val filters = Seq(traceLocalFilter) +} + +class Routes @Inject() (application: controllers.Application) extends GeneratedRouter with SimpleRouter { + val prefix = "/" + + lazy val defaultPrefix = { + if (prefix.endsWith("/")) "" else "/" + } + + // Gets + private[this] lazy val Application_getRouted = + Route("GET", PathPattern(List(StaticPart(prefix), StaticPart(defaultPrefix), StaticPart("getRouted")))) + + private[this] lazy val Application_show = + Route("GET", PathPattern(List(StaticPart(prefix), StaticPart(defaultPrefix), StaticPart("showRouted/"), DynamicPart("id", """[^/]+""", encodeable = true)))) + + //Posts + private[this] lazy val Application_postRouted = + Route("POST", PathPattern(List(StaticPart(prefix), StaticPart(defaultPrefix), StaticPart("postRouted")))) + + def routes: PartialFunction[RequestHeader, Handler] = { + case Application_getRouted(params) ⇒ call { + createInvoker(application.getRouted, + HandlerDef(this.getClass.getClassLoader, "", "controllers.Application", "getRouted", Nil, "GET", """some comment""", prefix + """getRouted""")).call(application.getRouted) + } + case Application_postRouted(params) ⇒ call { + createInvoker(application.postRouted, + HandlerDef(this.getClass.getClassLoader, "", "controllers.Application", "postRouted", Nil, "POST", """some comment""", prefix + """postRouted""")).call(application.postRouted) + } + case Application_show(params) ⇒ call(params.fromPath[Int]("id", None)) { (id) ⇒ + createInvoker(application.showRouted(id), + HandlerDef(this.getClass.getClassLoader, "", "controllers.Application", "showRouted", Seq(classOf[Int]), "GET", """""", prefix + """show/some/$id<[^/]+>""")).call(application.showRouted(id)) + } + } + + override def errorHandler: HttpErrorHandler = new HttpErrorHandler() { + override def onClientError(request: RequestHeader, statusCode: Int, message: String): Future[Result] = ??? + override def onServerError(request: RequestHeader, exception: Throwable): Future[Result] = ??? + } +} + +object controllers { + import play.api.mvc._ + + class Application extends Controller { + val postRouted = Action { + Ok("invoked postRouted") + } + val getRouted = Action { + Ok("invoked getRouted") + } + def showRouted(id: Int) = Action { + Ok("invoked show with: " + id) + } + } +}
\ No newline at end of file |