diff --git a/sdk-extensions/jaeger-remote-sampler/src/test/java/io/opentelemetry/sdk/extension/trace/jaeger/sampler/JaegerRemoteSamplerProviderTest.java b/sdk-extensions/jaeger-remote-sampler/src/test/java/io/opentelemetry/sdk/extension/trace/jaeger/sampler/JaegerRemoteSamplerProviderTest.java index 8c923148c8..d4dc82e888 100644 --- a/sdk-extensions/jaeger-remote-sampler/src/test/java/io/opentelemetry/sdk/extension/trace/jaeger/sampler/JaegerRemoteSamplerProviderTest.java +++ b/sdk-extensions/jaeger-remote-sampler/src/test/java/io/opentelemetry/sdk/extension/trace/jaeger/sampler/JaegerRemoteSamplerProviderTest.java @@ -43,23 +43,21 @@ public class JaegerRemoteSamplerProviderTest { assertThat(samplerProviders) .singleElement(type(JaegerRemoteSamplerProvider.class)) .satisfies( - provider -> - assertThat(provider.createSampler(mockConfig)) + provider -> { + try (JaegerRemoteSampler s = + (JaegerRemoteSampler) provider.createSampler(mockConfig)) { + assertThat(s) .extracting("sampler", type(Sampler.class)) .asString() - .isEqualTo(sampler.toString())) - .satisfies( - provider -> - assertThat(provider.createSampler(mockConfig)) - .extracting("serviceName") - .isEqualTo("test_service")) - .satisfies( - provider -> - assertThat(provider.createSampler(mockConfig)) + .isEqualTo(sampler.toString()); + assertThat(s).extracting("serviceName").isEqualTo("test_service"); + assertThat(s) .extracting("delegate") .extracting("endpoint") .isEqualTo( - "http://localhost:9999/jaeger.api_v2.SamplingManager/GetSamplingStrategy")); + "http://localhost:9999/jaeger.api_v2.SamplingManager/GetSamplingStrategy"); + } + }); } @Test @@ -74,9 +72,11 @@ public class JaegerRemoteSamplerProviderTest { assertThat(samplerProviders) .singleElement(type(JaegerRemoteSamplerProvider.class)) .satisfies( - provider -> - assertThat(provider.createSampler(mockConfig)) - .extracting("serviceName") - .isEqualTo("test_service2")); + provider -> { + try (JaegerRemoteSampler s = + (JaegerRemoteSampler) provider.createSampler(mockConfig)) { + assertThat(s).extracting("serviceName").isEqualTo("test_service2"); + } + }); } }