Close jaeger sampler in provider test too. (#4113)

This commit is contained in:
Anuraag Agrawal 2022-01-26 08:29:08 +09:00 committed by GitHub
parent 53df4aa8a1
commit f4f61ea2f3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 16 deletions

View File

@ -43,23 +43,21 @@ public class JaegerRemoteSamplerProviderTest {
assertThat(samplerProviders) assertThat(samplerProviders)
.singleElement(type(JaegerRemoteSamplerProvider.class)) .singleElement(type(JaegerRemoteSamplerProvider.class))
.satisfies( .satisfies(
provider -> provider -> {
assertThat(provider.createSampler(mockConfig)) try (JaegerRemoteSampler s =
(JaegerRemoteSampler) provider.createSampler(mockConfig)) {
assertThat(s)
.extracting("sampler", type(Sampler.class)) .extracting("sampler", type(Sampler.class))
.asString() .asString()
.isEqualTo(sampler.toString())) .isEqualTo(sampler.toString());
.satisfies( assertThat(s).extracting("serviceName").isEqualTo("test_service");
provider -> assertThat(s)
assertThat(provider.createSampler(mockConfig))
.extracting("serviceName")
.isEqualTo("test_service"))
.satisfies(
provider ->
assertThat(provider.createSampler(mockConfig))
.extracting("delegate") .extracting("delegate")
.extracting("endpoint") .extracting("endpoint")
.isEqualTo( .isEqualTo(
"http://localhost:9999/jaeger.api_v2.SamplingManager/GetSamplingStrategy")); "http://localhost:9999/jaeger.api_v2.SamplingManager/GetSamplingStrategy");
}
});
} }
@Test @Test
@ -74,9 +72,11 @@ public class JaegerRemoteSamplerProviderTest {
assertThat(samplerProviders) assertThat(samplerProviders)
.singleElement(type(JaegerRemoteSamplerProvider.class)) .singleElement(type(JaegerRemoteSamplerProvider.class))
.satisfies( .satisfies(
provider -> provider -> {
assertThat(provider.createSampler(mockConfig)) try (JaegerRemoteSampler s =
.extracting("serviceName") (JaegerRemoteSampler) provider.createSampler(mockConfig)) {
.isEqualTo("test_service2")); assertThat(s).extracting("serviceName").isEqualTo("test_service2");
}
});
} }
} }