open-telemetry
/
opentelemetry-java
mirror of https://github.com/open-telemetry/opentelemetry-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Retry requests in PrometheusHttpServerTest (#5750)

This commit is contained in:
jack-berg 2023-08-22 12:47:52 -05:00 committed by GitHub
parent 3669e7a2ae
commit 917d75d637
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
exporters/prometheus/src/test/java/io/opentelemetry/exporter/prometheus/PrometheusHttpServerTest.java
View File

@ -13,6 +13,8 @@ import static org.assertj.core.api.Assertions.assertThatThrownBy;
import com.google.common.collect.ImmutableList;
import com.google.common.io.ByteStreams;
import com.linecorp.armeria.client.WebClient;
import com.linecorp.armeria.client.retry.RetryRule;
import com.linecorp.armeria.client.retry.RetryingClient;
import com.linecorp.armeria.common.AggregatedHttpResponse;
import com.linecorp.armeria.common.HttpHeaderNames;
import com.linecorp.armeria.common.HttpMethod;
@ -68,7 +70,10 @@ class PrometheusHttpServerTest {
prometheusServer = PrometheusHttpServer.builder().setHost("localhost").setPort(0).build();
prometheusServer.register(metricProducer);
client = WebClient.of("http://localhost:" + prometheusServer.getAddress().getPort());
client =
WebClient.builder("http://localhost:" + prometheusServer.getAddress().getPort())
.decorator(RetryingClient.newDecorator(RetryRule.failsafe()))
.build();
}
@BeforeEach
@ -180,6 +185,7 @@ class PrometheusHttpServerTest {
void fetchPrometheusCompressed() throws IOException {
WebClient client =
WebClient.builder("http://localhost:" + prometheusServer.getAddress().getPort())
.decorator(RetryingClient.newDecorator(RetryRule.failsafe()))
.addHeader(HttpHeaderNames.ACCEPT_ENCODING, "gzip")
.build();
AggregatedHttpResponse response = client.get("/").aggregate().join();