Skip to content

Add abstract client tests #246

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 7 commits into from
Feb 3, 2021
Merged

Add abstract client tests #246

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
f84f83d
adding AbstractHttpClientTest and an echo target in TestHttpServer
shashank11p Jan 29, 2021
029e2dd
updating AbstractHttpClientTest
shashank11p Feb 1, 2021
d775706
updating AbstractHttpClientTest
shashank11p Feb 1, 2021
b7190b2
updating okHttp and apache http client tests
shashank11p Feb 2, 2021
7cbcb85
review comments, refactoring
shashank11p Feb 2, 2021
a405879
review comments and removing apache async http client changes
shashank11p Feb 2, 2021
e163156
review comments
shashank11p Feb 3, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
208 changes: 30 additions & 178 deletions ...instrumentation/hypertrace/apachehttpclient/v4_0/ApacheHttpClientInstrumentationTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,204 +16,56 @@

package io.opentelemetry.javaagent.instrumentation.hypertrace.apachehttpclient.v4_0;

import io.opentelemetry.api.trace.Span;
import io.opentelemetry.sdk.trace.data.SpanData;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.URI;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeoutException;
import org.apache.http.HttpEntity;
import java.util.Map;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.protocol.HTTP;
import org.hypertrace.agent.core.instrumentation.HypertraceSemanticAttributes;
import org.hypertrace.agent.testing.AbstractInstrumenterTest;
import org.hypertrace.agent.testing.TestHttpServer;
import org.hypertrace.agent.testing.TestHttpServer.GetJsonHandler;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.hypertrace.agent.testing.AbstractHttpClientTest;

public class ApacheHttpClientInstrumentationTest extends AbstractInstrumenterTest {

private static final String JSON = "{\"id\":1,\"name\":\"John\"}";
private static final TestHttpServer testHttpServer = new TestHttpServer();
public class ApacheHttpClientInstrumentationTest extends AbstractHttpClientTest {

private final HttpClient client = new DefaultHttpClient();

@BeforeAll
public static void startServer() throws Exception {
testHttpServer.start();
}

@AfterAll
public static void closeServer() throws Exception {
testHttpServer.close();
public ApacheHttpClientInstrumentationTest() {
super(true);
}

@Test
public void getJson() throws IOException, TimeoutException, InterruptedException {
HttpGet getRequest = new HttpGet();
getRequest.addHeader("foo", "bar");
getRequest.setURI(
URI.create(String.format("http://localhost:%d/get_json", testHttpServer.port())));
HttpResponse response = client.execute(getRequest);
Assertions.assertEquals(200, response.getStatusLine().getStatusCode());
InputStream inputStream = response.getEntity().getContent();
Assertions.assertEquals(GetJsonHandler.RESPONSE_BODY, readInputStream(inputStream));

Assertions.assertEquals(false, Span.current().isRecording());

TEST_WRITER.waitForTraces(1);
List<List<SpanData>> traces = TEST_WRITER.getTraces();
Assertions.assertEquals(1, traces.size());
Assertions.assertEquals(2, traces.get(0).size());
SpanData clientSpan = traces.get(0).get(0);

Assertions.assertEquals(
"test-value",
clientSpan
.getAttributes()
.get(HypertraceSemanticAttributes.httpResponseHeader("test-response-header")));
Assertions.assertEquals(
"bar",
clientSpan.getAttributes().get(HypertraceSemanticAttributes.httpRequestHeader("foo")));
Assertions.assertNull(
clientSpan.getAttributes().get(HypertraceSemanticAttributes.HTTP_REQUEST_BODY));
SpanData responseBodySpan = traces.get(0).get(1);
Assertions.assertEquals(
GetJsonHandler.RESPONSE_BODY,
responseBodySpan.getAttributes().get(HypertraceSemanticAttributes.HTTP_RESPONSE_BODY));
}

@Test
public void postUrlEncoded() throws IOException, TimeoutException, InterruptedException {
List<NameValuePair> nvps = new ArrayList<>();
nvps.add(new BasicNameValuePair("key1", "value1"));
nvps.add(new BasicNameValuePair("key2", "value2"));

HttpPost postRequest = new HttpPost();
postRequest.setEntity(new UrlEncodedFormEntity(nvps, HTTP.UTF_8));
postRequest.setURI(
URI.create(String.format("http://localhost:%d/post", testHttpServer.port())));
HttpResponse response = client.execute(postRequest);
Assertions.assertEquals(204, response.getStatusLine().getStatusCode());

TEST_WRITER.waitForTraces(1);
List<List<SpanData>> traces = TEST_WRITER.getTraces();
Assertions.assertEquals(1, traces.size());
Assertions.assertEquals(1, traces.get(0).size());
SpanData clientSpan = traces.get(0).get(0);

Assertions.assertEquals(
"test-value",
clientSpan
.getAttributes()
.get(HypertraceSemanticAttributes.httpResponseHeader("test-response-header")));
Assertions.assertEquals(
"key1=value1&key2=value2",
clientSpan.getAttributes().get(HypertraceSemanticAttributes.HTTP_REQUEST_BODY));
Assertions.assertNull(
clientSpan.getAttributes().get(HypertraceSemanticAttributes.HTTP_RESPONSE_BODY));
}

@Test
public void postJson() throws IOException, TimeoutException, InterruptedException {
StringEntity entity = new StringEntity(JSON);
entity.setContentType("application/json");
postJsonEntity(entity);
}
@Override
public Response doPostRequest(
String uri, Map<String, String> headers, String body, String contentType) throws IOException {

@Test
public void postJsonNonRepeatableEntity()
throws IOException, TimeoutException, InterruptedException {
StringEntity entity = new NonRepeatableStringEntity(JSON);
entity.setContentType("application/json");
postJsonEntity(entity);
}

public void postJsonEntity(HttpEntity entity)
throws TimeoutException, InterruptedException, IOException {
HttpPost postRequest = new HttpPost();
postRequest.setEntity(entity);
postRequest.setHeader("Content-type", "application/json");
postRequest.setURI(
URI.create(String.format("http://localhost:%d/post", testHttpServer.port())));
HttpResponse response = client.execute(postRequest);
Assertions.assertEquals(204, response.getStatusLine().getStatusCode());

TEST_WRITER.waitForTraces(1);
List<List<SpanData>> traces = TEST_WRITER.getTraces();
Assertions.assertEquals(1, traces.size());
Assertions.assertEquals(1, traces.get(0).size());
SpanData clientSpan = traces.get(0).get(0);

Assertions.assertEquals(
"test-value",
clientSpan
.getAttributes()
.get(HypertraceSemanticAttributes.httpResponseHeader("test-response-header")));
Assertions.assertEquals(
readInputStream(entity.getContent()),
clientSpan.getAttributes().get(HypertraceSemanticAttributes.HTTP_REQUEST_BODY));
Assertions.assertNull(
clientSpan.getAttributes().get(HypertraceSemanticAttributes.HTTP_RESPONSE_BODY));
}

@Test
public void getContent_throws_exception() throws IOException {
HttpClient client = new DefaultHttpClient();
HttpGet getRequest = new HttpGet();
getRequest.setURI(
URI.create(String.format("http://localhost:%d/get_json", testHttpServer.port())));
HttpResponse response = client.execute(getRequest);
HttpEntity entity = response.getEntity();
Assertions.assertNotNull(entity.getContent());
try {
entity.getContent();
} catch (Exception ex) {
Assertions.assertEquals(IllegalStateException.class, ex.getClass());
}
}

private static String readInputStream(InputStream inputStream) throws IOException {
StringBuilder textBuilder = new StringBuilder();

try (BufferedReader reader =
new BufferedReader(
new InputStreamReader(inputStream, Charset.forName(StandardCharsets.UTF_8.name())))) {
int c;
while ((c = reader.read()) != -1) {
textBuilder.append((char) c);
}
HttpPost request = new HttpPost();
for (String key : headers.keySet()) {
request.addHeader(key, headers.get(key));
}
return textBuilder.toString();
request.setURI(URI.create(uri));
StringEntity entity = new StringEntity(body);
entity.setContentType(contentType);
request.setEntity(entity);
request.addHeader("Content-type", contentType);
HttpResponse response = client.execute(request);
InputStream inputStream = response.getEntity().getContent();
return new Response(readInputStream(inputStream), response.getStatusLine().getStatusCode());
}

class NonRepeatableStringEntity extends StringEntity {

public NonRepeatableStringEntity(String s) throws UnsupportedEncodingException {
super(s);
@Override
public Response doGetRequest(String uri, Map<String, String> headers) throws IOException {
HttpGet request = new HttpGet();
for (String key : headers.keySet()) {
request.addHeader(key, headers.get(key));
}

@Override
public boolean isRepeatable() {
return false;
request.setURI(URI.create(uri));
HttpResponse response = client.execute(request);
if (response.getEntity() == null || response.getEntity().getContentLength() <= 0) {
return new Response(null, response.getStatusLine().getStatusCode());
}
InputStream inputStream = response.getEntity().getContent();
return new Response(readInputStream(inputStream), response.getStatusLine().getStatusCode());
}
}
Loading