BAEL-4061: Implementing a Simple HTTP Server with Netty (#9416)
* BAEL-4061: Implementing a Simple HTTP Server with Netty * BAEL-4061: Added live test * BAEL-4061: moved helper class from test to main folder * BAEL-4061: updated tests to follow BDD
This commit is contained in:
Sampada
@@ -0,0 +1,180 @@
|
||||
package com.baeldung.http.server;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
import org.junit.After;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
|
||||
import io.netty.bootstrap.Bootstrap;
|
||||
import io.netty.buffer.Unpooled;
|
||||
import io.netty.channel.Channel;
|
||||
import io.netty.channel.ChannelHandlerContext;
|
||||
import io.netty.channel.ChannelInitializer;
|
||||
import io.netty.channel.ChannelPipeline;
|
||||
import io.netty.channel.EventLoopGroup;
|
||||
import io.netty.channel.SimpleChannelInboundHandler;
|
||||
import io.netty.channel.nio.NioEventLoopGroup;
|
||||
import io.netty.channel.socket.SocketChannel;
|
||||
import io.netty.channel.socket.nio.NioSocketChannel;
|
||||
import io.netty.handler.codec.http.DefaultFullHttpRequest;
|
||||
import io.netty.handler.codec.http.HttpClientCodec;
|
||||
import io.netty.handler.codec.http.HttpContent;
|
||||
import io.netty.handler.codec.http.HttpContentDecompressor;
|
||||
import io.netty.handler.codec.http.HttpHeaderNames;
|
||||
import io.netty.handler.codec.http.HttpHeaderValues;
|
||||
import io.netty.handler.codec.http.HttpMethod;
|
||||
import io.netty.handler.codec.http.HttpObject;
|
||||
import io.netty.handler.codec.http.HttpResponse;
|
||||
import io.netty.handler.codec.http.HttpUtil;
|
||||
import io.netty.handler.codec.http.HttpVersion;
|
||||
import io.netty.handler.codec.http.LastHttpContent;
|
||||
import io.netty.handler.codec.http.cookie.ClientCookieEncoder;
|
||||
import io.netty.handler.codec.http.cookie.DefaultCookie;
|
||||
import io.netty.util.CharsetUtil;
|
||||
|
||||
//Ensure the server class - HttpServer.java is already started before running this test
|
||||
public class HttpServerLiveTest {
|
||||
|
||||
private static final String HOST = "127.0.0.1";
|
||||
private static final int PORT = 8080;
|
||||
private Channel channel;
|
||||
private EventLoopGroup group = new NioEventLoopGroup();
|
||||
ResponseAggregator response = new ResponseAggregator();
|
||||
|
||||
@Before
|
||||
public void setup() throws Exception {
|
||||
Bootstrap b = new Bootstrap();
|
||||
b.group(group)
|
||||
.channel(NioSocketChannel.class)
|
||||
.handler(new ChannelInitializer<SocketChannel>() {
|
||||
|
||||
@Override
|
||||
protected void initChannel(SocketChannel ch) throws Exception {
|
||||
ChannelPipeline p = ch.pipeline();
|
||||
p.addLast(new HttpClientCodec());
|
||||
p.addLast(new HttpContentDecompressor());
|
||||
p.addLast(new SimpleChannelInboundHandler<HttpObject>() {
|
||||
@Override
|
||||
protected void channelRead0(ChannelHandlerContext ctx, HttpObject msg) throws Exception {
|
||||
response = prepareResponse(ctx, msg, response);
|
||||
}
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
channel = b.connect(HOST, PORT)
|
||||
.sync()
|
||||
.channel();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void whenPostSent_thenContentReceivedInUppercase() throws Exception {
|
||||
String body = "Hello World!";
|
||||
|
||||
DefaultFullHttpRequest request = createRequest(body);
|
||||
|
||||
channel.writeAndFlush(request);
|
||||
Thread.sleep(200);
|
||||
|
||||
assertEquals(200, response.getStatus());
|
||||
assertEquals("HTTP/1.1", response.getVersion());
|
||||
|
||||
assertTrue(response.getContent()
|
||||
.contains(body.toUpperCase()));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void whenGetSent_thenCookieReceivedInResponse() throws Exception {
|
||||
DefaultFullHttpRequest request = createRequest(null);
|
||||
|
||||
channel.writeAndFlush(request);
|
||||
Thread.sleep(200);
|
||||
|
||||
assertEquals(200, response.getStatus());
|
||||
assertEquals("HTTP/1.1", response.getVersion());
|
||||
|
||||
Map<String, String> headers = response.getHeaders();
|
||||
String cookies = headers.get("set-cookie");
|
||||
assertTrue(cookies.contains("my-cookie"));
|
||||
}
|
||||
|
||||
@After
|
||||
public void cleanup() throws InterruptedException {
|
||||
channel.closeFuture()
|
||||
.sync();
|
||||
group.shutdownGracefully();
|
||||
}
|
||||
|
||||
private static DefaultFullHttpRequest createRequest(final CharSequence body) throws Exception {
|
||||
DefaultFullHttpRequest request;
|
||||
if (body != null) {
|
||||
request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.POST, "/");
|
||||
request.content()
|
||||
.writeBytes(body.toString()
|
||||
.getBytes(CharsetUtil.UTF_8.name()));
|
||||
request.headers()
|
||||
.set(HttpHeaderNames.CONTENT_TYPE, "application/json");
|
||||
request.headers()
|
||||
.set(HttpHeaderNames.CONTENT_LENGTH, request.content()
|
||||
.readableBytes());
|
||||
} else {
|
||||
request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.GET, "/", Unpooled.EMPTY_BUFFER);
|
||||
request.headers()
|
||||
.set(HttpHeaderNames.COOKIE, ClientCookieEncoder.STRICT.encode(new DefaultCookie("my-cookie", "foo")));
|
||||
}
|
||||
|
||||
request.headers()
|
||||
.set(HttpHeaderNames.HOST, HOST);
|
||||
request.headers()
|
||||
.set(HttpHeaderNames.CONNECTION, HttpHeaderValues.KEEP_ALIVE);
|
||||
|
||||
return request;
|
||||
}
|
||||
|
||||
private static ResponseAggregator prepareResponse(ChannelHandlerContext ctx, HttpObject msg, ResponseAggregator responseAgg) {
|
||||
|
||||
if (msg instanceof HttpResponse) {
|
||||
HttpResponse response = (HttpResponse) msg;
|
||||
|
||||
responseAgg.setStatus(response.status()
|
||||
.code());
|
||||
|
||||
responseAgg.setVersion(response.protocolVersion()
|
||||
.text());
|
||||
|
||||
if (!response.headers()
|
||||
.isEmpty()) {
|
||||
Map<String, String> headers = new HashMap<String, String>();
|
||||
for (CharSequence name : response.headers()
|
||||
.names()) {
|
||||
for (CharSequence value : response.headers()
|
||||
.getAll(name)) {
|
||||
headers.put(name.toString(), value.toString());
|
||||
}
|
||||
}
|
||||
responseAgg.setHeaders(headers);
|
||||
}
|
||||
if (HttpUtil.isTransferEncodingChunked(response)) {
|
||||
responseAgg.setChunked(true);
|
||||
} else {
|
||||
responseAgg.setChunked(false);
|
||||
}
|
||||
}
|
||||
if (msg instanceof HttpContent) {
|
||||
HttpContent content = (HttpContent) msg;
|
||||
String responseData = content.content()
|
||||
.toString(CharsetUtil.UTF_8);
|
||||
|
||||
if (content instanceof LastHttpContent) {
|
||||
responseAgg.setContent(responseData + "} End Of Content");
|
||||
ctx.close();
|
||||
}
|
||||
}
|
||||
return responseAgg;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,52 @@
|
||||
package com.baeldung.http.server;
|
||||
|
||||
import java.util.Map;
|
||||
|
||||
public class ResponseAggregator {
|
||||
int status;
|
||||
String version;
|
||||
Map<String, String> headers;
|
||||
boolean isChunked;
|
||||
String content;
|
||||
|
||||
public int getStatus() {
|
||||
return status;
|
||||
}
|
||||
|
||||
public void setStatus(int status) {
|
||||
this.status = status;
|
||||
}
|
||||
|
||||
public String getVersion() {
|
||||
return version;
|
||||
}
|
||||
|
||||
public void setVersion(String version) {
|
||||
this.version = version;
|
||||
}
|
||||
|
||||
public Map<String, String> getHeaders() {
|
||||
return headers;
|
||||
}
|
||||
|
||||
public void setHeaders(Map<String, String> headers) {
|
||||
this.headers = headers;
|
||||
}
|
||||
|
||||
public boolean isChunked() {
|
||||
return isChunked;
|
||||
}
|
||||
|
||||
public void setChunked(boolean isChunked) {
|
||||
this.isChunked = isChunked;
|
||||
}
|
||||
|
||||
public String getContent() {
|
||||
return content;
|
||||
}
|
||||
|
||||
public void setContent(String content) {
|
||||
this.content = content;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user