一、概述和原理
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