|
20 | 20 |
|
21 | 21 | import io.opentelemetry.api.metrics.MeterProvider;
|
22 | 22 | import io.opentelemetry.exporter.otlp.http.logs.OtlpHttpLogRecordExporter;
|
23 |
| -import io.opentelemetry.exporter.otlp.http.logs.OtlpHttpLogRecordExporterBuilder; |
24 | 23 | import io.opentelemetry.exporter.otlp.logs.OtlpGrpcLogRecordExporter;
|
25 |
| -import io.opentelemetry.exporter.otlp.logs.OtlpGrpcLogRecordExporterBuilder; |
26 | 24 | import io.opentelemetry.sdk.logs.export.LogRecordExporter;
|
27 | 25 | import okhttp3.HttpUrl;
|
| 26 | +import org.assertj.core.api.InstanceOfAssertFactories; |
28 | 27 | import org.junit.jupiter.api.Test;
|
29 | 28 | import org.junit.jupiter.params.ParameterizedTest;
|
30 | 29 | import org.junit.jupiter.params.provider.ValueSource;
|
|
35 | 34 | import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
36 | 35 | import org.springframework.context.annotation.Bean;
|
37 | 36 | import org.springframework.context.annotation.Configuration;
|
38 |
| -import org.springframework.test.util.ReflectionTestUtils; |
39 | 37 |
|
40 | 38 | import static org.assertj.core.api.Assertions.assertThat;
|
41 | 39 |
|
@@ -162,33 +160,29 @@ void shouldUseGrpcExporterIfTransportIsSetToGrpc() {
|
162 | 160 | }
|
163 | 161 |
|
164 | 162 | @Test
|
165 |
| - @SuppressWarnings("unchecked") |
166 | 163 | void httpShouldUseMeterProviderIfSet() {
|
167 | 164 | this.contextRunner.withUserConfiguration(MeterProviderConfiguration.class)
|
168 | 165 | .withPropertyValues("management.otlp.logging.endpoint=http://localhost:4318/v1/logs")
|
169 | 166 | .run((context) -> {
|
170 | 167 | OtlpHttpLogRecordExporter otlpHttpLogRecordExporter = context.getBean(OtlpHttpLogRecordExporter.class);
|
171 |
| - OtlpHttpLogRecordExporterBuilder builder = otlpHttpLogRecordExporter.toBuilder(); |
172 |
| - Supplier<MeterProvider> meterProviderSupplier = (Supplier<MeterProvider>) ReflectionTestUtils |
173 |
| - .getField(ReflectionTestUtils.getField(builder, "delegate"), "meterProviderSupplier"); |
174 |
| - assertThat(meterProviderSupplier).isNotNull(); |
175 |
| - assertThat(meterProviderSupplier.get()).isSameAs(MeterProviderConfiguration.meterProvider); |
| 168 | + assertThat(otlpHttpLogRecordExporter.toBuilder()) |
| 169 | + .extracting("delegate.meterProviderSupplier", InstanceOfAssertFactories.type(Supplier.class)) |
| 170 | + .satisfies((meterProviderSupplier) -> assertThat(meterProviderSupplier.get()) |
| 171 | + .isSameAs(MeterProviderConfiguration.meterProvider)); |
176 | 172 | });
|
177 | 173 | }
|
178 | 174 |
|
179 | 175 | @Test
|
180 |
| - @SuppressWarnings("unchecked") |
181 | 176 | void grpcShouldUseMeterProviderIfSet() {
|
182 | 177 | this.contextRunner.withUserConfiguration(MeterProviderConfiguration.class)
|
183 | 178 | .withPropertyValues("management.otlp.logging.endpoint=http://localhost:4318/v1/logs",
|
184 | 179 | "management.otlp.logging.transport=grpc")
|
185 | 180 | .run((context) -> {
|
186 | 181 | OtlpGrpcLogRecordExporter otlpGrpcLogRecordExporter = context.getBean(OtlpGrpcLogRecordExporter.class);
|
187 |
| - OtlpGrpcLogRecordExporterBuilder builder = otlpGrpcLogRecordExporter.toBuilder(); |
188 |
| - Supplier<MeterProvider> meterProviderSupplier = (Supplier<MeterProvider>) ReflectionTestUtils |
189 |
| - .getField(ReflectionTestUtils.getField(builder, "delegate"), "meterProviderSupplier"); |
190 |
| - assertThat(meterProviderSupplier).isNotNull(); |
191 |
| - assertThat(meterProviderSupplier.get()).isSameAs(MeterProviderConfiguration.meterProvider); |
| 182 | + assertThat(otlpGrpcLogRecordExporter.toBuilder()) |
| 183 | + .extracting("delegate.meterProviderSupplier", InstanceOfAssertFactories.type(Supplier.class)) |
| 184 | + .satisfies((meterProviderSupplier) -> assertThat(meterProviderSupplier.get()) |
| 185 | + .isSameAs(MeterProviderConfiguration.meterProvider)); |
192 | 186 | });
|
193 | 187 | }
|
194 | 188 |
|
|
0 commit comments