forked from jooby-project/jooby
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathHttp2Test.java
More file actions
150 lines (132 loc) · 4.89 KB
/
Copy pathHttp2Test.java
File metadata and controls
150 lines (132 loc) · 4.89 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
package io.jooby;
import static org.junit.jupiter.api.Assertions.assertEquals;
import java.net.InetSocketAddress;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Phaser;
import org.eclipse.jetty.http.HttpFields;
import org.eclipse.jetty.http.HttpURI;
import org.eclipse.jetty.http.HttpVersion;
import org.eclipse.jetty.http.MetaData;
import org.eclipse.jetty.http2.api.Session;
import org.eclipse.jetty.http2.api.Stream;
import org.eclipse.jetty.http2.api.server.ServerSessionListener;
import org.eclipse.jetty.http2.client.HTTP2Client;
import org.eclipse.jetty.http2.frames.DataFrame;
import org.eclipse.jetty.http2.frames.HeadersFrame;
import org.eclipse.jetty.util.Callback;
import org.eclipse.jetty.util.FuturePromise;
import org.eclipse.jetty.util.Jetty;
import org.eclipse.jetty.util.Promise;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import com.google.common.collect.ImmutableMap;
import io.jooby.junit.ServerTest;
import io.jooby.junit.ServerTestRunner;
import okhttp3.MediaType;
import okhttp3.Protocol;
import okhttp3.Request;
import okhttp3.Response;
import okhttp3.ResponseBody;
public class Http2Test {
@ServerTest
public void http2(ServerTestRunner runner) {
runner.define(app -> {
app.setServerOptions(
new ServerOptions()
.setHttp2(true)
.setSecurePort(8443)
);
app.get("/",
ctx -> ImmutableMap.of("secure", ctx.isSecure(), "protocol", ctx.getProtocol(), "scheme",
ctx.getScheme()));
}).ready((http, https) -> {
https.get("/", rsp -> {
assertEquals("{secure=true, protocol=HTTP/2.0, scheme=https}",
rsp.body().string());
});
http.get("/", rsp -> {
assertEquals("{secure=false, protocol=HTTP/1.1, scheme=http}",
rsp.body().string());
});
});
}
@ServerTest
public void http2c(ServerTestRunner runner) {
runner.define(app -> {
app.setServerOptions(
new ServerOptions()
.setHttp2(true)
);
app.get("/",
ctx -> ImmutableMap.of("secure", ctx.isSecure(), "protocol", ctx.getProtocol(), "scheme",
ctx.getScheme()));
}).ready(http -> {
http.get("/", rsp -> {
assertEquals("{secure=false, protocol=HTTP/1.1, scheme=http}",
rsp.body().string());
});
hc2(http, "/", rsp -> {
assertEquals("{secure=false, protocol=HTTP/2.0, scheme=http}",
rsp.body().string());
});
});
}
private void hc2(WebClient http, String path, SneakyThrows.Consumer<Response> consumer)
throws ExecutionException, InterruptedException {
HttpFields requestFields = new HttpFields();
requestFields.put("User-Agent", h2c.getClass().getName() + "/" + Jetty.VERSION);
HttpURI uri = new HttpURI("http://localhost:" + http.getPort() + path);
MetaData.Request metaData = new MetaData.Request("GET", uri, HttpVersion.HTTP_2,
requestFields);
HeadersFrame frame = new HeadersFrame(metaData, null, true);
final Phaser phaser = new Phaser(2);
FuturePromise<org.eclipse.jetty.http2.api.Session> sessionPromise = new FuturePromise<>();
h2c.connect(null, new InetSocketAddress("localhost", http.getPort()),
new ServerSessionListener.Adapter(), sessionPromise);
Session session = sessionPromise.get();
Response.Builder builder = new Response.Builder();
session.newStream(frame, new Promise.Adapter<>(), new Stream.Listener.Adapter() {
@Override
public void onHeaders(final Stream stream, final HeadersFrame frame) {
MetaData.Response md = (MetaData.Response) frame.getMetaData();
StatusCode statusCode = StatusCode.valueOf(md.getStatus());
builder.code(statusCode.value())
.message(statusCode.reason());
md.forEach(f ->
builder.addHeader(f.getName().toLowerCase(), f.getValue().toLowerCase())
);
if (frame.isEndStream()) {
phaser.arrive();
}
}
@Override
public void onData(final Stream stream, final DataFrame frame, final Callback callback) {
byte[] bytes = new byte[frame.getData().remaining()];
frame.getData().get(bytes);
ResponseBody responseBody = ResponseBody.create(bytes, MediaType.parse("text/plain"));
builder.body(responseBody);
callback.succeeded();
if (frame.isEndStream()) {
phaser.arrive();
}
}
});
phaser.awaitAdvanceInterruptibly(phaser.arrive());
consumer.accept(
builder.request(new Request.Builder().url(uri.toString()).method("GET", null)
.build())
.protocol(Protocol.HTTP_2)
.build()
);
}
private static HTTP2Client h2c;
@BeforeAll
public static void init() throws Exception {
h2c = new HTTP2Client();
h2c.start();
}
@AfterAll
public static void close() throws Exception {
h2c.stop();
}
}