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

Set read timeout only when testing read-timeout endpoint (#11417)

This commit is contained in:
Lauri Tulmin 2024-05-21 17:40:35 +03:00 committed by GitHub
parent 80b3ee934c
commit cac8d1f497
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 33 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
instrumentation/async-http-client/async-http-client-1.9/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/asynchttpclient/v1_9/AsyncHttpClientTest.java
View File

@ -35,12 +35,24 @@ class AsyncHttpClientTest extends AbstractHttpClientTest<Request> {
private static final int CONNECTION_TIMEOUT_MS = (int) CONNECTION_TIMEOUT.toMillis();
private static final int READ_TIMEOUT_MS = (int) READ_TIMEOUT.toMillis();
private static final AsyncHttpClient client =
new AsyncHttpClient(
new AsyncHttpClientConfig.Builder()
.setConnectTimeout(CONNECTION_TIMEOUT_MS)
.setReadTimeout(READ_TIMEOUT_MS)
.build());
private static final AsyncHttpClient client = buildClient(false);
private static final AsyncHttpClient clientWithReadTimeout = buildClient(true);
private static AsyncHttpClient buildClient(boolean readTimeout) {
AsyncHttpClientConfig.Builder builder =
new AsyncHttpClientConfig.Builder().setConnectTimeout(CONNECTION_TIMEOUT_MS);
if (readTimeout) {
builder.setReadTimeout(READ_TIMEOUT_MS);
}
return new AsyncHttpClient(builder.build());
}
private static AsyncHttpClient getClient(URI uri) {
if (uri.toString().contains("/read-timeout")) {
return clientWithReadTimeout;
}
return client;
}
@Override
public Request buildRequest(String method, URI uri, Map<String, String> headers) {
@ -54,7 +66,7 @@ class AsyncHttpClientTest extends AbstractHttpClientTest<Request> {
@Override
public int sendRequest(Request request, String method, URI uri, Map<String, String> headers)
throws ExecutionException, InterruptedException {
return client.executeRequest(request).get().getStatusCode();
return getClient(uri).executeRequest(request).get().getStatusCode();
}
@Override
@ -64,20 +76,21 @@ class AsyncHttpClientTest extends AbstractHttpClientTest<Request> {
URI uri,
Map<String, String> headers,
HttpClientResult requestResult) {
client.executeRequest(
request,
new AsyncCompletionHandler<Void>() {
@Override
public Void onCompleted(Response response) {
requestResult.complete(response.getStatusCode());
return null;
}
getClient(uri)
.executeRequest(
request,
new AsyncCompletionHandler<Void>() {
@Override
public Void onCompleted(Response response) {
requestResult.complete(response.getStatusCode());
return null;
}
@Override
public void onThrowable(Throwable throwable) {
requestResult.complete(throwable);
}
});
@Override
public void onThrowable(Throwable throwable) {
requestResult.complete(throwable);
}
});
}
@Override