Netty 的单元测试,主要是对业务逻辑的 ChannelHandler 做测试(毕竟对 Bootstrap、EventLoop 这些做测试着实没有多大意义),模拟一次入站数据或者出站数据,查看数据流经 ChannelHandler 变成什么样了,以此达到测试的目的。
Netty 的单元测试将Junit4作为测试框架,将 EmbeddedChannel 作为测试通道。基本原理就是:将入站数据或者出站数据写入 EmbeddedChannel 中,然后检查是否有任何东西到达了 ChannelPipeline 的尾端。以这种方式,你便可以知道消息是否流经了 ChannelHandler 以及是否触发了任何 ChannelHandler 的动作,如下图:
EmbeddedChannel 提供了如下方法进行单元测试:
writeInbound(Object... msgs): 将入站消息写到 EmbeddedChannel 中。如果可以通过 readInbound()方法从 EmbeddedChannel 中读取数据,则返回 true。 readInbound() :从 EmbeddedChannel 中读取一个入站消息。任何返回的东西都穿越了整个 ChannelPipeline。如果没有任何可供读取的, 则返回 null。 writeOutbound(Object... msgs): 将出站消息写到EmbeddedChannel中。如果现在可以通过readOutbound()方法从 EmbeddedChannel 中读取到什么东西,则返回 true。 readOutbound(): 从 EmbeddedChannel 中读取一个出站消息。任何返回的东西都穿越了整个 ChannelPipeline。如果没有任何可供读取的,则返回 null。 finish() :将 EmbeddedChannel 标记为完成,并且如果有可被读取的入站数据或者出站数据,则返回 true。这个方法还将会调用 EmbeddedChannel 上的close()方法。
1、将我们要测试的 ChannelHandler 写入 EmbeddedChannel 进行测试。
2、writeInbound(Object... msgs) 将数据写入EmbeddedChannel(模拟接收数据)。
3、ChannelHandler 处理后如果有返回数据,可以通过readInbound() 验证数据结果。如果没有返回数据,可以在 ChannelHandler 业务逻辑中,打印日志,以达到测试目的。
public class DecoderTest {
//1、利用Junit执行单元测试
@Test
public void decoderTest() throws IllegalAccessException {
ByteBuf buf = Unpooled.buffer();
for (int i = 0; i < 9; i++) {
buf.writeByte(i);
}
ByteBuf buf1 = buf.duplicate();
//2、创建EmbeddedChannel,并添加一个Decoder(我们的要测试 ChannelHandler) 其将以3字节帧长度被测试
EmbeddedChannel embeddedChannel = new EmbeddedChannel(new Decoder(3));
//3、将数据写入 EmbeddedChannel
boolean writeInbound = embeddedChannel.writeInbound(buf1.retain());
assertTrue(writeInbound);
//4、标记 Channel 为已完成状态
boolean finish = embeddedChannel.finish();
assertTrue(finish);
//5、读取数据
ByteBuf readInbound = embeddedChannel.readInbound();
ByteBuf readSlice = buf.readSlice(3);
assertEquals(readInbound, readSlice);
readInbound.release();
readInbound = embeddedChannel.readInbound();
readSlice = buf.readSlice(3);
assertEquals(readInbound, readSlice);
readInbound.release();
readInbound = embeddedChannel.readInbound();
readSlice = buf.readSlice(3);
assertEquals(readInbound, readSlice);
readInbound.release();
//是否读取完数据了
assertNull(embeddedChannel.readInbound());
//释放资源
buf.release();
}
}
1、将我们要测试的 ChannelHandler 写入 EmbeddedChannel 进行测试。
2、writeOutbound(Object... msgs) 将数据写入EmbeddedChannel(模拟发送数据)。
3、ChannelHandler 处理后如果有返回数据,可以通过readOutbound() 验证数据结果。如果没有返回数据,可以在 ChannelHandler 业务逻辑中,打印日志,以达到测试目的。
public class EncoderTest {
@Test
public void encoderTest(){
ByteBuf buf = Unpooled.buffer();
for (int i =1; i < 10; i++){
buf.writeInt(i * -1);
}
//1、创建一个EmbeddedChannel 并安装要测试的Encoder
EmbeddedChannel embeddedChannel = new EmbeddedChannel(new Encoder());
//2、写入数据
assertTrue(embeddedChannel.writeOutbound(buf));
assertTrue(embeddedChannel.finish());
//3、读取数据
for (int i = 1; i < 10; i++){
Object o = embeddedChannel.readOutbound();
System.out.println(o);
}
assertNull(embeddedChannel.readOutbound());
}
}
截止到这篇文章,Netty 的基础部分差不多就结束了。不幸的是,公司安排我去研究下 Docker 技术,想在我们项目中使用起来。所以 Netty 的学习就不得不告一段落了~~才翻完《Netty 实战》一半的内容,后面还有编解码器、网络协议、案例研究三个部分,只能先欠下了~
参考资料:《Netty IN ACTION》
演示源代码:https://github.com/JMCuixy/NettyDemo/tree/master/src/main/java/org/netty/demo/unit