-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
15 changed files
with
395 additions
and
41 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,76 @@ | ||
| package org.taymyr.lagom.metrics | ||
|
|
||
| import com.typesafe.config.ConfigFactory | ||
| import io.github.config4k.extract | ||
| import io.kotlintest.matchers.string.shouldBeEmpty | ||
| import io.kotlintest.shouldBe | ||
| import io.kotlintest.shouldNotBe | ||
| import io.kotlintest.shouldThrow | ||
| import io.kotlintest.specs.WordSpec | ||
| import org.taymyr.lagom.metrics.GraphiteReporterType.PICKLE | ||
| import org.taymyr.lagom.metrics.GraphiteReporterType.TCP | ||
| import java.util.concurrent.TimeUnit.MILLISECONDS | ||
| import java.util.concurrent.TimeUnit.SECONDS | ||
|
|
||
| /** | ||
| * @author Sergey Morgunov | ||
| */ | ||
| class ConfigTest : WordSpec({ | ||
|
|
||
| "MetricsConfig" should { | ||
|
|
||
| "be correct for reference.conf" { | ||
| val config = ConfigFactory.defaultReference().extract<MetricsConfig>("taymyr.lagom.metrics") | ||
| config.enableJVM shouldBe false | ||
| config.enableCircuitBreaker shouldBe false | ||
| config.prefix.shouldBeEmpty() | ||
| config.graphiteReporter shouldBe null | ||
| } | ||
|
|
||
| "be able to enable JVM metrics" { | ||
| val config = ConfigFactory.load("default.conf").extract<MetricsConfig>("taymyr.lagom.metrics") | ||
| config.enableJVM shouldBe true | ||
| } | ||
|
|
||
| "be able to enable circuit breakers metrics" { | ||
| val config = ConfigFactory.load("default.conf").extract<MetricsConfig>("taymyr.lagom.metrics") | ||
| config.enableCircuitBreaker shouldBe true | ||
| } | ||
|
|
||
| "be throw exception for incorrect settings graphite reporter" { | ||
| shouldThrow<Exception> { | ||
| ConfigFactory.load("bad_graphite.conf").extract<MetricsConfig>("taymyr.lagom.metrics") | ||
| } | ||
| } | ||
|
|
||
| "be correct for graphite reporter with minimal settings" { | ||
| val config = ConfigFactory.load("min_graphite.conf").extract<MetricsConfig>("taymyr.lagom.metrics") | ||
| config.graphiteReporter shouldNotBe null | ||
| config.graphiteReporter?.let { | ||
| it.type shouldBe PICKLE | ||
| it.host shouldBe "localhost" | ||
| it.port shouldBe 1000 | ||
| it.batchSize shouldBe null | ||
| it.durationUnit shouldBe MILLISECONDS | ||
| it.rateUnit shouldBe SECONDS | ||
| it.period shouldBe 10 | ||
| it.periodUnit shouldBe SECONDS | ||
| } | ||
| } | ||
|
|
||
| "be correct for graphite reporter with full settings" { | ||
| val config = ConfigFactory.load("full_graphite.conf").extract<MetricsConfig>("taymyr.lagom.metrics") | ||
| config.graphiteReporter shouldNotBe null | ||
| config.graphiteReporter?.let { | ||
| it.type shouldBe TCP | ||
| it.host shouldBe "localhost" | ||
| it.port shouldBe 1000 | ||
| it.batchSize shouldBe 1000 | ||
| it.durationUnit shouldBe SECONDS | ||
| it.rateUnit shouldBe MILLISECONDS | ||
| it.period shouldBe 60 | ||
| it.periodUnit shouldBe MILLISECONDS | ||
| } | ||
| } | ||
| } | ||
| }) |
125 changes: 125 additions & 0 deletions
125
src/test/kotlin/org/taymyr/lagom/metrics/MetricsTest.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,125 @@ | ||
| package org.taymyr.lagom.metrics | ||
|
|
||
| import akka.NotUsed | ||
| import akka.stream.javadsl.Source | ||
| import akka.stream.javadsl.Source.from | ||
| import akka.stream.javadsl.Source.maybe | ||
| import com.codahale.metrics.MetricFilter.ALL | ||
| import com.codahale.metrics.MetricRegistry | ||
| import com.codahale.metrics.jvm.GarbageCollectorMetricSet | ||
| import com.codahale.metrics.jvm.JvmAttributeGaugeSet | ||
| import com.codahale.metrics.jvm.MemoryUsageGaugeSet | ||
| import com.codahale.metrics.jvm.ThreadStatesGaugeSet | ||
| import com.lightbend.lagom.internal.server.status.MetricsServiceImpl | ||
| import com.lightbend.lagom.javadsl.api.ServiceCall | ||
| import com.lightbend.lagom.javadsl.api.transport.BadRequest | ||
| import com.lightbend.lagom.javadsl.api.transport.TransportErrorCode.InternalServerError | ||
| import com.lightbend.lagom.javadsl.api.transport.TransportException | ||
| import com.lightbend.lagom.javadsl.server.status.CircuitBreakerStatus | ||
| import com.lightbend.lagom.javadsl.testkit.ServiceTest.TestServer | ||
| import com.lightbend.lagom.javadsl.testkit.ServiceTest.defaultSetup | ||
| import com.lightbend.lagom.javadsl.testkit.ServiceTest.startServer | ||
| import com.nhaarman.mockitokotlin2.atLeast | ||
| import com.nhaarman.mockitokotlin2.clearInvocations | ||
| import com.nhaarman.mockitokotlin2.eq | ||
| import com.nhaarman.mockitokotlin2.isA | ||
| import com.nhaarman.mockitokotlin2.mock | ||
| import com.nhaarman.mockitokotlin2.spy | ||
| import com.nhaarman.mockitokotlin2.verify | ||
| import io.kotlintest.matchers.beInstanceOf | ||
| import io.kotlintest.shouldBe | ||
| import io.kotlintest.shouldThrow | ||
| import io.kotlintest.specs.WordSpec | ||
| import org.taymyr.lagom.metrics.CircuitBreakersMetricSetTest.Companion.closedCircuitBreaker | ||
| import org.taymyr.lagom.metrics.CircuitBreakersMetricSetTest.Companion.halfOpenCircuitBreaker | ||
| import org.taymyr.lagom.metrics.CircuitBreakersMetricSetTest.Companion.openCircuitBreaker | ||
| import play.inject.Bindings.bind | ||
| import java.lang.Thread.sleep | ||
| import java.util.concurrent.CompletableFuture | ||
| import java.util.concurrent.ExecutionException | ||
|
|
||
| /** | ||
| * @author Sergey Morgunov | ||
| */ | ||
| class MetricsTest : WordSpec({ | ||
|
|
||
| val registrySpy = spy(MetricRegistry()) | ||
| val metricsServiceImpl = mock<MetricsServiceImpl> { _ -> | ||
| val cbStatusSource = from(listOf(listOf(openCircuitBreaker, closedCircuitBreaker, halfOpenCircuitBreaker))).concat(maybe()) | ||
| val cbStatusCall = mock<ServiceCall<NotUsed, Source<List<CircuitBreakerStatus>, *>>> { | ||
| on { invoke() }.thenReturn(CompletableFuture.completedFuture(cbStatusSource)) | ||
| } | ||
| on { circuitBreakers() }.thenReturn(cbStatusCall) | ||
| } | ||
| val server: TestServer = startServer(defaultSetup().withCluster(false).configureBuilder { b -> b | ||
| .overrides( | ||
| bind(MetricRegistry::class.java).toInstance(registrySpy), | ||
| bind(MetricsServiceImpl::class.java).toInstance(metricsServiceImpl) | ||
| ) | ||
| }) | ||
|
|
||
| "Initialization Metrics" should { | ||
|
|
||
| "initialize JVM metrics" { | ||
| verify(registrySpy).register(eq("prefix.jvm.attr"), isA<JvmAttributeGaugeSet>()) | ||
| verify(registrySpy).register(eq("prefix.jvm.gc"), isA<GarbageCollectorMetricSet>()) | ||
| verify(registrySpy).register(eq("prefix.jvm.memory"), isA<MemoryUsageGaugeSet>()) | ||
| verify(registrySpy).register(eq("prefix.jvm.threads"), isA<ThreadStatesGaugeSet>()) | ||
| } | ||
|
|
||
| "initialize CircuitBreakers metrics" { | ||
| verify(registrySpy).register(eq("prefix.cb.open"), isA<CircuitBreakersMetricSet>()) | ||
| verify(registrySpy).register(eq("prefix.cb.closed"), isA<CircuitBreakersMetricSet>()) | ||
| verify(registrySpy).register(eq("prefix.cb.half-open"), isA<CircuitBreakersMetricSet>()) | ||
| } | ||
|
|
||
| "initialize Graphite reporter" { | ||
| clearInvocations(registrySpy) | ||
| sleep(1000) // Wait more then reporter period | ||
| verify(registrySpy, atLeast(1)).getGauges(ALL) | ||
| verify(registrySpy, atLeast(1)).getCounters(ALL) | ||
| verify(registrySpy, atLeast(1)).getHistograms(ALL) | ||
| verify(registrySpy, atLeast(1)).getMeters(ALL) | ||
| verify(registrySpy, atLeast(1)).getTimers(ALL) | ||
| } | ||
| } | ||
|
|
||
| "Metrics filter" should { | ||
| "correct register timer and meter for successful request" { | ||
| clearInvocations(registrySpy) | ||
| val testService = server.client(TestService::class.java) | ||
| testService.simpleMethod().invoke().toCompletableFuture().get() | ||
| verify(registrySpy).timer("prefix.routes.all.timer") | ||
| verify(registrySpy).meter("prefix.routes.all.meter") | ||
| verify(registrySpy).timer("prefix.routes.root.foo.bar.GET.timer") | ||
| verify(registrySpy).meter("prefix.routes.root.foo.bar.GET.200.meter") | ||
| } | ||
|
|
||
| "correct register timer and meter for bad request" { | ||
| clearInvocations(registrySpy) | ||
| val testService = server.client(TestService::class.java) | ||
| val badRequest = shouldThrow<ExecutionException> { | ||
| testService.methodWithPathParams(0, "0").invoke("").toCompletableFuture().get() | ||
| } | ||
| badRequest.cause shouldBe beInstanceOf(BadRequest::class) | ||
| verify(registrySpy).timer("prefix.routes.all.timer") | ||
| verify(registrySpy).meter("prefix.routes.all.meter") | ||
| verify(registrySpy).timer("prefix.routes.root.foo._firstId.bar._secondId.POST.timer") | ||
| verify(registrySpy).meter("prefix.routes.root.foo._firstId.bar._secondId.POST.400.meter") | ||
| } | ||
|
|
||
| "correct register timer and meter for unsuccessful request" { | ||
| clearInvocations(registrySpy) | ||
| val testService = server.client(TestService::class.java) | ||
| val internalError = shouldThrow<ExecutionException> { | ||
| testService.methodWithQueryParams(0, 0, 0).invoke().toCompletableFuture().get() | ||
| } | ||
| internalError.cause shouldBe beInstanceOf(TransportException::class) | ||
| (internalError.cause as TransportException).errorCode() shouldBe InternalServerError | ||
| verify(registrySpy).timer("prefix.routes.all.timer") | ||
| verify(registrySpy).meter("prefix.routes.all.meter") | ||
| verify(registrySpy).timer("prefix.routes.root.foo._firstId.bar.pageNo_pageSize.DELETE.timer") | ||
| verify(registrySpy).meter("prefix.routes.root.foo._firstId.bar.pageNo_pageSize.DELETE.500.meter") | ||
| } | ||
| } | ||
| }) |
Oops, something went wrong.