diff options
author | Jakob Odersky <jakob@driver.xyz> | 2017-10-04 13:48:50 -0700 |
---|---|---|
committer | Jakob Odersky <jakob@driver.xyz> | 2017-10-04 13:49:26 -0700 |
commit | 586ed28262f1609a2338e01236548716e1d2c264 (patch) | |
tree | 994b593c217718b0a07c009bbe24cb122406c686 /src/test/scala/TracingDirectivesSpec.scala | |
parent | df2c41dac7d29dea49950700b0146229947fdf65 (diff) | |
download | tracing-586ed28262f1609a2338e01236548716e1d2c264.tar.gz tracing-586ed28262f1609a2338e01236548716e1d2c264.tar.bz2 tracing-586ed28262f1609a2338e01236548716e1d2c264.zip |
Add closing functionality to tracers
Diffstat (limited to 'src/test/scala/TracingDirectivesSpec.scala')
-rw-r--r-- | src/test/scala/TracingDirectivesSpec.scala | 26 |
1 files changed, 11 insertions, 15 deletions
diff --git a/src/test/scala/TracingDirectivesSpec.scala b/src/test/scala/TracingDirectivesSpec.scala index ebe7858..783f59b 100644 --- a/src/test/scala/TracingDirectivesSpec.scala +++ b/src/test/scala/TracingDirectivesSpec.scala @@ -16,13 +16,7 @@ class TracingDirectivesSpec with BeforeAndAfterAll with ScalatestRouteTest { - implicit val tracer = new GoogleTracer( - "driverinc-sandbox", - Paths.get( - system.settings.config.getString("tracing.google.service-account-file")) - ) - - val route: Route = trace(tracer, "example.org") { + def route(tracer: Tracer): Route = trace(tracer, "example.org") { pathPrefix("1") { trace(tracer, "test-sub-trace-1") { Thread.sleep(2) @@ -43,18 +37,20 @@ class TracingDirectivesSpec } } - "Tracer" should "submit" in { - for (i <- 0 until 100) { - Get(s"https://example.org/${i % 2 + 1}") ~> route ~> check { + "Google Tracer" should "submit" in { + val tracer = new GoogleTracer( + "driverinc-sandbox", + Paths.get( + system.settings.config.getString("tracing.google.service-account-file")) + ) + + val futures: Seq[Assertion] = for (i <- 0 until 100) yield { + Get(s"https://example.org/${i % 2 + 1}") ~> route(tracer) ~> check { assert(responseAs[String] == "ok") } } - } - override def afterAll() = { - tracer.queue.complete() - Await.ready(tracer.queue.watchCompletion(), Duration.Inf) - super.afterAll() + Await.ready(tracer.close(), 30.seconds) } } |