instrument akka testkit to correctly propagate TraceContexts when replying from a test probe

1 files changed, 3 insertions, 4 deletions

diff --git a/kamon-spray/src/test/scala/kamon/spray/ServerRequestInstrumentationSpec.scala b/kamon-spray/src/test/scala/kamon/spray/ServerRequestInstrumentationSpec.scala
index 9b4acdd9..0edf75e0 100644
--- a/kamon-spray/src/test/scala/kamon/spray/ServerRequestInstrumentationSpec.scala
+++ b/kamon-spray/src/test/scala/kamon/spray/ServerRequestInstrumentationSpec.scala
@@ -33,7 +33,7 @@ import spray.http.HttpHeaders.Host
 import akka.io.{ Tcp, IO }
 import spray.can.Http
 import akka.io.Tcp.Bound
-import kamon.metrics.{TraceMetrics, Metrics}
+import kamon.metrics.{ TraceMetrics, Metrics }
 import kamon.metrics.TraceMetrics.TraceMetricSnapshot
 import kamon.metrics.Subscriptions.TickMetricSnapshot
 
@@ -79,14 +79,13 @@ class ServerRequestInstrumentationSpec extends TestKit(ActorSystem("spec")) with
       client.expectMsgType[HttpResponse]
 
       metricListener.fishForMessage() {
-        case snapshot @ TickMetricSnapshot(_, _, metrics) => metrics.keys.exists(_.name.contains("open-and-finish"))
-        case other => false
+        case snapshot @ TickMetricSnapshot(_, _, metrics) ⇒ metrics.keys.exists(_.name.contains("open-and-finish"))
+        case other                                        ⇒ false
       }
     }
 
   }
 
-
 }
 
 trait TestServer {