Skip to content

Commit 65e6d57

Browse files
committed
Test cases are updated to use JUnit 5
1 parent 37032d2 commit 65e6d57

File tree

3 files changed

+40
-16
lines changed

3 files changed

+40
-16
lines changed

instrumentation/activej-http-6.0/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/activejhttp/ActivejHttpServerTest.java

+28-7
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@
99
import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.CAPTURE_HEADERS;
1010
import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.ERROR;
1111
import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.EXCEPTION;
12+
import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.ID_PARAMETER_NAME;
13+
import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.INDEXED_CHILD;
1214
import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.NOT_FOUND;
1315
import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.QUERY_PARAM;
1416
import static io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint.REDIRECT;
@@ -31,12 +33,11 @@
3133
import io.opentelemetry.instrumentation.testing.junit.http.HttpServerTestOptions;
3234
import io.opentelemetry.instrumentation.testing.junit.http.ServerEndpoint;
3335
import io.opentelemetry.testing.internal.armeria.internal.shaded.guava.collect.ImmutableSet;
34-
import org.junit.ClassRule;
3536
import org.junit.jupiter.api.extension.RegisterExtension;
3637

3738
public class ActivejHttpServerTest extends AbstractHttpServerTest<HttpServer> {
3839

39-
@ClassRule public static final EventloopRule eventloopRule = new EventloopRule();
40+
@RegisterExtension static final EventloopExtension eventloopExtension = new EventloopExtension();
4041

4142
@RegisterExtension
4243
static final InstrumentationExtension testing = HttpServerInstrumentationExtension.forAgent();
@@ -58,6 +59,30 @@ protected HttpServer setupServer() throws Exception {
5859
controller(CAPTURE_HEADERS, () -> httpResponse);
5960
return httpResponse.toPromise();
6061
};
62+
AsyncServlet indexChildAsyncServlet =
63+
request -> {
64+
HttpResponse httpResponse =
65+
HttpResponse.builder()
66+
.withBody(INDEXED_CHILD.getBody())
67+
.withCode(INDEXED_CHILD.getStatus())
68+
.build();
69+
INDEXED_CHILD.collectSpanAttributes(
70+
id ->
71+
id.equals(ID_PARAMETER_NAME)
72+
? request.getQueryParameter(ID_PARAMETER_NAME)
73+
: null);
74+
controller(
75+
INDEXED_CHILD,
76+
() -> {
77+
INDEXED_CHILD.collectSpanAttributes(
78+
id ->
79+
id.equals(ID_PARAMETER_NAME)
80+
? request.getQueryParameter(ID_PARAMETER_NAME)
81+
: null);
82+
return httpResponse;
83+
});
84+
return httpResponse.toPromise();
85+
};
6186

6287
RoutingServlet routingServlet =
6388
RoutingServlet.builder(eventloop)
@@ -68,6 +93,7 @@ protected HttpServer setupServer() throws Exception {
6893
.with(GET, EXCEPTION.getPath(), request -> prepareResponse(EXCEPTION))
6994
.with(GET, REDIRECT.getPath(), request -> prepareResponse(REDIRECT))
7095
.with(GET, CAPTURE_HEADERS.getPath(), captureHttpHeadersAsyncServlet)
96+
.with(GET, INDEXED_CHILD.getPath(), indexChildAsyncServlet)
7197
.build();
7298

7399
HttpServer server = HttpServer.builder(eventloop, routingServlet).withListenPort(port).build();
@@ -84,11 +110,6 @@ protected void stopServer(HttpServer server) throws Exception {
84110
thread.join();
85111
}
86112

87-
@Override
88-
protected void assertHighConcurrency(int count) {
89-
//
90-
}
91-
92113
@Override
93114
protected void configure(HttpServerTestOptions options) {
94115
options.setTestHttpPipelining(false);

instrumentation/activej-http-6.0/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/activejhttp/ActivejRoutingServletTest.java

+2-3
Original file line numberDiff line numberDiff line change
@@ -26,13 +26,12 @@
2626
import io.opentelemetry.api.trace.SpanKind;
2727
import io.opentelemetry.instrumentation.testing.junit.AgentInstrumentationExtension;
2828
import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension;
29-
import org.junit.ClassRule;
3029
import org.junit.jupiter.api.Test;
3130
import org.junit.jupiter.api.extension.RegisterExtension;
3231

33-
public class ActivejRoutingServletTest {
32+
class ActivejRoutingServletTest {
3433

35-
@ClassRule public static final EventloopRule eventloopRule = new EventloopRule();
34+
@RegisterExtension static final EventloopExtension eventloopExtension = new EventloopExtension();
3635

3736
@RegisterExtension
3837
static final InstrumentationExtension testing = AgentInstrumentationExtension.create();
+10-6
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@
99

1010
import io.activej.eventloop.Eventloop;
1111
import io.activej.reactor.Reactor;
12-
import org.junit.rules.TestRule;
13-
import org.junit.runner.Description;
14-
import org.junit.runners.model.Statement;
12+
import org.junit.jupiter.api.extension.AfterAllCallback;
13+
import org.junit.jupiter.api.extension.BeforeEachCallback;
14+
import org.junit.jupiter.api.extension.ExtensionContext;
1515

16-
public final class EventloopRule implements TestRule {
16+
public class EventloopExtension implements BeforeEachCallback, AfterAllCallback {
1717

1818
static {
1919
createEventloop();
@@ -24,11 +24,15 @@ private static void createEventloop() {
2424
}
2525

2626
@Override
27-
public Statement apply(Statement base, Description description) {
27+
public void beforeEach(ExtensionContext context) {
2828
Reactor currentReactor = Reactor.getCurrentReactor();
2929
if (!(currentReactor instanceof Eventloop) || !currentReactor.inReactorThread()) {
3030
createEventloop();
3131
}
32-
return base;
32+
}
33+
34+
@Override
35+
public void afterAll(ExtensionContext context) {
36+
//
3337
}
3438
}

0 commit comments

Comments
 (0)