|
13 | 13 | import static org.hamcrest.Matchers.not; |
14 | 14 | import static org.hamcrest.Matchers.notNullValue; |
15 | 15 | import static org.hamcrest.Matchers.nullValue; |
16 | | -import static org.junit.jupiter.api.Assertions.assertEquals; |
17 | 16 | import static org.junit.jupiter.api.Assertions.assertThrows; |
18 | | -import static org.junit.jupiter.api.Assertions.assertTrue; |
19 | 17 | import static org.junit.jupiter.params.provider.Arguments.arguments; |
20 | 18 | import static org.mockito.ArgumentMatchers.any; |
21 | 19 | import static org.mockito.ArgumentMatchers.anyBoolean; |
22 | 20 | import static org.mockito.ArgumentMatchers.anyCollection; |
23 | 21 | import static org.mockito.ArgumentMatchers.anyInt; |
24 | 22 | import static org.mockito.ArgumentMatchers.eq; |
25 | | -import static org.mockito.ArgumentMatchers.isA; |
26 | 23 | import static org.mockito.Mockito.doThrow; |
27 | 24 | import static org.mockito.Mockito.lenient; |
28 | 25 | import static org.mockito.Mockito.mock; |
29 | | -import static org.mockito.Mockito.never; |
30 | | -import static org.mockito.Mockito.times; |
31 | 26 | import static org.mockito.Mockito.verify; |
32 | 27 | import static org.mockito.Mockito.verifyNoInteractions; |
33 | 28 | import static org.mockito.Mockito.when; |
34 | 29 | import static org.opensearch.dataprepper.plugins.source.oteltrace.OTelTraceSourceConfig.DEFAULT_PORT; |
35 | 30 | import static org.opensearch.dataprepper.plugins.source.oteltrace.OTelTraceSourceConfig.DEFAULT_REQUEST_TIMEOUT_MS; |
36 | | -import static org.opensearch.dataprepper.plugins.source.oteltrace.OTelTraceSourceConfig.SSL; |
37 | 31 |
|
38 | | -import java.io.ByteArrayOutputStream; |
39 | | -import java.io.IOException; |
40 | | -import java.io.InputStream; |
41 | 32 | import java.nio.charset.StandardCharsets; |
42 | | -import java.nio.file.Files; |
43 | | -import java.nio.file.Path; |
44 | 33 | import java.time.Duration; |
45 | 34 | import java.util.Base64; |
46 | 35 | import java.util.Collection; |
47 | | -import java.util.Collections; |
48 | | -import java.util.HashMap; |
49 | 36 | import java.util.List; |
50 | 37 | import java.util.Map; |
51 | | -import java.util.Optional; |
52 | | -import java.util.StringJoiner; |
53 | 38 | import java.util.UUID; |
54 | 39 | import java.util.concurrent.CompletableFuture; |
55 | | -import java.util.concurrent.ExecutionException; |
56 | 40 | import java.util.concurrent.TimeoutException; |
57 | 41 | import java.util.function.Function; |
58 | 42 | import java.util.stream.Collectors; |
59 | 43 | import java.util.stream.Stream; |
60 | | -import java.util.zip.GZIPOutputStream; |
61 | 44 |
|
62 | | -import org.apache.commons.io.IOUtils; |
63 | 45 | import org.junit.jupiter.api.AfterEach; |
64 | 46 | import org.junit.jupiter.api.BeforeEach; |
65 | 47 | import org.junit.jupiter.api.Test; |
|
72 | 54 | import org.mockito.ArgumentCaptor; |
73 | 55 | import org.mockito.ArgumentMatchers; |
74 | 56 | import org.mockito.Mock; |
75 | | -import org.mockito.MockedStatic; |
76 | | -import org.mockito.Mockito; |
77 | 57 | import org.mockito.junit.jupiter.MockitoExtension; |
78 | 58 | import org.opensearch.dataprepper.GrpcRequestExceptionHandler; |
79 | 59 | import org.opensearch.dataprepper.armeria.authentication.GrpcAuthenticationProvider; |
80 | 60 | import org.opensearch.dataprepper.armeria.authentication.HttpBasicAuthenticationConfig; |
81 | | -import org.opensearch.dataprepper.metrics.MetricNames; |
82 | 61 | import org.opensearch.dataprepper.metrics.MetricsTestUtil; |
83 | 62 | import org.opensearch.dataprepper.metrics.PluginMetrics; |
84 | 63 | import org.opensearch.dataprepper.model.buffer.Buffer; |
|
93 | 72 | import org.opensearch.dataprepper.plugins.certificate.CertificateProvider; |
94 | 73 | import org.opensearch.dataprepper.plugins.certificate.model.Certificate; |
95 | 74 | import org.opensearch.dataprepper.plugins.codec.CompressionOption; |
96 | | -import org.opensearch.dataprepper.plugins.health.HealthGrpcService; |
97 | 75 | import org.opensearch.dataprepper.plugins.source.oteltrace.certificate.CertificateProviderFactory; |
98 | 76 |
|
99 | | -import com.fasterxml.jackson.databind.ObjectMapper; |
100 | | -import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule; |
101 | 77 | import com.google.protobuf.ByteString; |
102 | 78 | import com.google.protobuf.InvalidProtocolBufferException; |
103 | 79 | import com.google.protobuf.util.JsonFormat; |
104 | | -import com.linecorp.armeria.client.ClientFactory; |
105 | 80 | import com.linecorp.armeria.client.Clients; |
106 | 81 | import com.linecorp.armeria.client.WebClient; |
107 | 82 | import com.linecorp.armeria.common.AggregatedHttpResponse; |
108 | 83 | import com.linecorp.armeria.common.HttpData; |
109 | | -import com.linecorp.armeria.common.HttpHeaderNames; |
110 | 84 | import com.linecorp.armeria.common.HttpMethod; |
111 | 85 | import com.linecorp.armeria.common.HttpStatus; |
112 | 86 | import com.linecorp.armeria.common.MediaType; |
113 | 87 | import com.linecorp.armeria.common.RequestHeaders; |
114 | 88 | import com.linecorp.armeria.common.SessionProtocol; |
115 | | -import com.linecorp.armeria.server.HttpService; |
116 | 89 | import com.linecorp.armeria.server.Server; |
117 | 90 | import com.linecorp.armeria.server.ServerBuilder; |
118 | 91 | import com.linecorp.armeria.server.grpc.GrpcService; |
119 | 92 | import com.linecorp.armeria.server.grpc.GrpcServiceBuilder; |
120 | | -import com.linecorp.armeria.server.healthcheck.HealthCheckService; |
121 | 93 |
|
122 | 94 | import io.grpc.BindableService; |
123 | | -import io.grpc.ServerServiceDefinition; |
124 | 95 | import io.grpc.Status; |
125 | 96 | import io.grpc.StatusRuntimeException; |
126 | | -import io.micrometer.core.instrument.Measurement; |
127 | | -import io.micrometer.core.instrument.Statistic; |
128 | 97 | import io.netty.util.AsciiString; |
129 | 98 | import io.opentelemetry.proto.collector.trace.v1.ExportTraceServiceRequest; |
130 | 99 | import io.opentelemetry.proto.collector.trace.v1.ExportTraceServiceResponse; |
|
0 commit comments