diff options
Diffstat (limited to 'kamon-play/src/test')
3 files changed, 117 insertions, 0 deletions
diff --git a/kamon-play/src/test/resources/conf/application.conf b/kamon-play/src/test/resources/conf/application.conf new file mode 100644 index 00000000..f2c654a8 --- /dev/null +++ b/kamon-play/src/test/resources/conf/application.conf @@ -0,0 +1,7 @@ +kamon { + play { + include-trace-token-header = true + trace-token-header-name = "X-UOW" + } +} + diff --git a/kamon-play/src/test/resources/logger.xml b/kamon-play/src/test/resources/logger.xml new file mode 100644 index 00000000..84126e9d --- /dev/null +++ b/kamon-play/src/test/resources/logger.xml @@ -0,0 +1,16 @@ +<configuration scan="true"> + <conversionRule conversionWord="traceToken" converterClass="kamon.trace.logging.LogbackTraceTokenConverter" /> + <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender"> + <encoder> + <pattern>%date{HH:mm:ss.SSS} %-5level [%traceToken][%X{akkaSource}] [%thread] %logger{55} - %msg%n</pattern> + </encoder> + </appender> + + <logger name="play" level="INFO" /> + <logger name="application" level="INFO" /> + + <root level="INFO"> + <appender-ref ref="STDOUT" /> + </root> + +</configuration> diff --git a/kamon-play/src/test/scala/kamon/play/filter/RequestInstrumentationSpec.scala b/kamon-play/src/test/scala/kamon/play/filter/RequestInstrumentationSpec.scala new file mode 100644 index 00000000..65f0219e --- /dev/null +++ b/kamon-play/src/test/scala/kamon/play/filter/RequestInstrumentationSpec.scala @@ -0,0 +1,94 @@ +/* =================================================== + * 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.filter + +import play.api.test._ +import play.api.mvc.{ Results, Action } +import play.api.mvc.Results.Ok +import scala.Some +import scala.concurrent.ExecutionContext.Implicits.global +import scala.concurrent.Future +import org.junit.runner.RunWith +import org.specs2.runner.JUnitRunner +import play.api.mvc.AsyncResult +import play.api.test.FakeApplication + +@RunWith(classOf[JUnitRunner]) +class RequestInstrumentationSpec extends PlaySpecification { + + System.setProperty("config.file", "./kamon-play/src/test/resources/conf/application.conf") + + val appWithRoutes = FakeApplication(withRoutes = { + case ("GET", "/asyncResult") ⇒ + Action { + AsyncResult { + scala.concurrent.Future { + Ok("AsyncResult") + } + } + } + case ("GET", "/async") ⇒ + Action.async { + Future { + Ok("Async.async") + } + } + case ("GET", "/notFound") ⇒ + Action { + Results.NotFound + } + case ("GET", "/redirect") ⇒ + Action { + Results.Redirect("/redirected", MOVED_PERMANENTLY) + } + case ("GET", "/default") ⇒ + Action { + Ok("default") + } + }) + + private val traceTokenValue = "kamon-trace-token-test" + private val traceTokenHeaderName = "X-UOW" + private val expectedToken = Some(traceTokenValue) + private val traceTokenHeader = (traceTokenHeaderName -> traceTokenValue) + + "respond to the asyncResult action with X-Trace-Token" in new WithServer(appWithRoutes) { + val Some(result) = route(FakeRequest(GET, "/asyncResult").withHeaders(traceTokenHeader)) + header(traceTokenHeaderName, result) must equalTo(expectedToken) + } + + "respond to the async action with X-Trace-Token" in new WithServer(appWithRoutes) { + val Some(result) = route(FakeRequest(GET, "/async").withHeaders(traceTokenHeader)) + header(traceTokenHeaderName, result) must equalTo(expectedToken) + } + + "respond to the notFound action with X-Trace-Token" in new WithServer(appWithRoutes) { + val Some(result) = route(FakeRequest(GET, "/notFound").withHeaders(traceTokenHeader)) + header(traceTokenHeaderName, result) must equalTo(expectedToken) + } + + "respond to the default action with X-Trace-Token" in new WithServer(appWithRoutes) { + val Some(result) = route(FakeRequest(GET, "/default").withHeaders(traceTokenHeader)) + header(traceTokenHeaderName, result) must equalTo(expectedToken) + } + + "respond to the redirect action with X-Trace-Token" in new WithServer(appWithRoutes) { + val Some(result) = route(FakeRequest(GET, "/redirect").withHeaders(traceTokenHeader)) + header("Location", result) must equalTo(Some("/redirected")) + header(traceTokenHeaderName, result) must equalTo(expectedToken) + } +}
\ No newline at end of file |