-
Notifications
You must be signed in to change notification settings - Fork 846
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Convert jaxrs-client-1.1 tests to java
- Loading branch information
Showing
2 changed files
with
97 additions
and
90 deletions.
There are no files selected for viewing
90 changes: 0 additions & 90 deletions
90
...umentation/jaxrs-client/jaxrs-client-1.1-testing/src/test/groovy/JaxRsClientV1Test.groovy
This file was deleted.
Oops, something went wrong.
97 changes: 97 additions & 0 deletions
97
...c/test/java/io/opentelemetry/javaagent/instrumentation/jaxrsclient/JaxRsClientV1Test.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,97 @@ | ||
/* | ||
* Copyright The OpenTelemetry Authors | ||
* SPDX-License-Identifier: Apache-2.0 | ||
*/ | ||
|
||
package io.opentelemetry.javaagent.instrumentation.jaxrsclient; | ||
|
||
import com.sun.jersey.api.client.Client; | ||
import com.sun.jersey.api.client.ClientHandlerException; | ||
import com.sun.jersey.api.client.ClientResponse; | ||
import com.sun.jersey.api.client.WebResource; | ||
import com.sun.jersey.api.client.filter.GZIPContentEncodingFilter; | ||
import com.sun.jersey.api.client.filter.LoggingFilter; | ||
import io.opentelemetry.api.common.AttributeKey; | ||
import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension; | ||
import io.opentelemetry.instrumentation.testing.junit.http.AbstractHttpClientTest; | ||
import io.opentelemetry.instrumentation.testing.junit.http.HttpClientInstrumentationExtension; | ||
import io.opentelemetry.instrumentation.testing.junit.http.HttpClientTestOptions; | ||
import io.opentelemetry.instrumentation.testing.junit.http.HttpServerTestOptions; | ||
import io.opentelemetry.semconv.NetworkAttributes; | ||
import java.net.URI; | ||
import java.util.HashSet; | ||
import java.util.Map; | ||
import java.util.Set; | ||
import org.junit.jupiter.api.AfterAll; | ||
import org.junit.jupiter.api.extension.RegisterExtension; | ||
|
||
class JaxRsClientV1Test extends AbstractHttpClientTest<WebResource.Builder> { | ||
@RegisterExtension | ||
static final InstrumentationExtension testing = HttpClientInstrumentationExtension.forAgent(); | ||
|
||
private final Client client = buildClient(false); | ||
private final Client clientWithReadTimeout = buildClient(true); | ||
|
||
private static Client buildClient(boolean readTimeout) { | ||
Client client = Client.create(); | ||
client.setConnectTimeout((int) CONNECTION_TIMEOUT.toMillis()); | ||
if (readTimeout) { | ||
client.setReadTimeout((int) READ_TIMEOUT.toMillis()); | ||
} | ||
// Add filters to ensure spans aren't duplicated. | ||
client.addFilter(new LoggingFilter()); | ||
client.addFilter(new GZIPContentEncodingFilter()); | ||
|
||
return client; | ||
} | ||
|
||
@Override | ||
protected void configure(HttpClientTestOptions.Builder options) { | ||
options.setTestCircularRedirects(false); | ||
options.setTestNonStandardHttpMethod(false); | ||
options.setTestCallback(false); | ||
options.setHttpAttributes( | ||
serverEndpoint -> { | ||
Set<AttributeKey<?>> attributes = | ||
new HashSet<>(HttpServerTestOptions.DEFAULT_HTTP_ATTRIBUTES); | ||
attributes.remove(NetworkAttributes.NETWORK_PROTOCOL_VERSION); | ||
return attributes; | ||
}); | ||
} | ||
|
||
@AfterAll | ||
void tearDown() { | ||
client.destroy(); | ||
clientWithReadTimeout.destroy(); | ||
} | ||
|
||
private Client getClient(URI uri) { | ||
if (uri.toString().contains("/read-timeout")) { | ||
return clientWithReadTimeout; | ||
} | ||
return client; | ||
} | ||
|
||
@Override | ||
public WebResource.Builder buildRequest(String method, URI uri, Map<String, String> headers) { | ||
WebResource.Builder builder = getClient(uri).resource(uri).getRequestBuilder(); | ||
headers.forEach(builder::header); | ||
return builder; | ||
} | ||
|
||
@Override | ||
public int sendRequest( | ||
WebResource.Builder builder, String method, URI uri, Map<String, String> headers) | ||
throws Exception { | ||
String body = "POST".equals(method) || "PUT".equals(method) ? "" : null; | ||
try { | ||
return builder.method(method, ClientResponse.class, body).getStatus(); | ||
} catch (ClientHandlerException exception) { | ||
Throwable cause = exception.getCause(); | ||
if (cause instanceof Exception) { | ||
throw (Exception) cause; | ||
} | ||
throw new IllegalStateException(cause); | ||
} | ||
} | ||
} |