Netty学习摘记 —— 单元测试
阅读原文时间:2023年07月08日阅读:2

本文参考

本篇文章是对《Netty In Action》一书第九章"单元测试"的学习摘记,主要内容为使用特殊的 Channel 实现——EmbeddedChannel来测试ChannelHandler

EmbeddedChannel概述

将入站数据或者出站数据写入到 EmbeddedChannel 中,然后检查是否有任何东西到达了 ChannelPipeline 的尾端。以这种方式,你便可以确定消息是否已经被编码或者被解码过了,以及是否触发了任何的ChannelHandler动作

从上表的API描述中我们可以看到,writeInbound()和readInbound()方法、writeOutbound()和readOutbound()方法成对使用,他们的流程是十分简单的

使用EmbeddedChannel测试入站消息

我们设计一个解码器,每次读取固定字节大小的帧

public class FixedLengthFrameDecoder extends ByteToMessageDecoder {

private final int frameLength;

//_指定要生成的帧的长度

_ public FixedLengthFrameDecoder(int frameLength) {

if (frameLength <= 0) {

throw new IllegalArgumentException("frameLength must be a positive integer: " + frameLength);
    }

this.frameLength = frameLength;
  }

  @Override

protected void decode(ChannelHandlerContext ctx, ByteBuf in, List out)

throws Exception {

//_检查是否有足够的字节可以被读取,以生成下一个帧

_ while (in.readableBytes() >= frameLength) {

//_从_ ByteBuf 中读取一个新帧      ByteBuf buf = in.readBytes(frameLength);

//将该帧添加到已被解码的消息列表中      out.add(buf);
    }
  }
}

使用EmbeddedChannel对解码器进行单元测试

@Test
//_第一个测试方法:_testFramesDecoded() public void testFramesDecoded() {

//_创建一个 ByteBuf,并存储 9 字节

  ByteBuf buf = Unpooled._buffer();

for (int i = 0; i < 9; i++) {

    buf.writeByte(i);
  }

  ByteBuf input = buf.duplicate();

//_创建一个__EmbeddedChannel,并添加一个FixedLengthFrameDecoder
  //其将以 3 字节的帧长度被测试

  EmbeddedChannel channel = _new EmbeddedChannel(new FixedLengthFrameDecoder(3));

// write bytes   //_将数据写入__EmbeddedChannel

  assertTrue_(channel.writeInbound(input.retain()));

//_标记 Channel 为已完成状态

_ assertTrue(channel.finish());

// read messages   //_读取所生成的消息,并且验证是否有 3_ _帧(切片),其中每帧(切片)都为 3 字节

_  ByteBuf read = (ByteBuf) channel.readInbound();

assertEquals(buf.readSlice(3), read);

  read.release();

  read = (ByteBuf) channel.readInbound();

assertEquals(buf.readSlice(3), read);

  read.release();

  read = (ByteBuf) channel.readInbound();

assertEquals(buf.readSlice(3), read);

  read.release();

assertNull(channel.readInbound());

  buf.release();
}

该 testFramesDecoded()方法验证了,一个包含 9 个可读字节的 ByteBuf 被解码为 3 个ByteBuf,每个都包含了 3 字节,,通过调用readInbound()方法,从Embedded- Channel中正好读取了3 个帧和一个null

下面测试入站ByteBuf通过两个步骤来写入

@Test
//_第二个测试方法:_testFramesDecoded2() public void testFramesDecoded2() {

  ByteBuf buf = Unpooled.buffer();

for (int i = 0; i < 9; i++) {

    buf.writeByte(i);
  }

  ByteBuf input = buf.duplicate();

  EmbeddedChannel channel = new EmbeddedChannel(

new FixedLengthFrameDecoder(3));

//_返回 false,因为没有一个完整的可供读取的帧

_ assertFalse(channel.writeInbound(input.readBytes(2)));

assertTrue(channel.writeInbound(input.readBytes(7)));

assertTrue(channel.finish());

  ByteBuf read = (ByteBuf) channel.readInbound();

assertEquals(buf.readSlice(3), read);

  read.release();

  read = (ByteBuf) channel.readInbound();

assertEquals(buf.readSlice(3), read);

  read.release();

  read = (ByteBuf) channel.readInbound();

assertEquals(buf.readSlice(3), read);

  read.release();

assertNull(channel.readInbound());

  buf.release();
}

当writeInbound(input.readBytes(2))被调用时,返回了false,这也应证了前面API表中的描述,如果对 readInbound()的后续调用将会返回数据,那么 write- Inbound()方法将会返回true,或者用源码的注释来说,当只有不足3个字节可供读取时,没有生成新的帧添加到已被解码的消息队列中

return true if the write operation did add something to the inbound buffer

使用EmbeddedChannel测试出站消息

和测试入站消息相反,在测试出站消息中,我们用到了编码器,将一种消息格式转换为另一种,我们设计的编码器功能如下:

  • 持有AbsIntegerEncoder的EmbeddedChannel将会以 4 字节的负整数的形式写出 站数据

  • 编码器将从传入的 ByteBuf 中读取每个负整数,并将会调用 Math.abs()方法来获取 其绝对值

  • 编码器将会把每个负整数的绝对值写到ChannelPipeline中

//扩展 MessageToMessageEncoder 以将一个消息编码为另外一种格式 __public class AbsIntegerEncoder extends MessageToMessageEncoder {

  @Override

protected void encode(ChannelHandlerContext channelHandlerContext,

  ByteBuf in, List out) throws Exception {

//_检查是否有足够的字节用来编码

_ while (in.readableBytes() >= 4) {

//_从输入的_ ByteBuf中读取下一个整数,并且计算其绝对值 int value = Math.abs(in.readInt());

//将该整数写入到编码消息的 _Lis__t 中

_      out.add(value);
    }
  }
}

下面是测试代码

@Test
public void testEncoded() {

//_创建一个_ ByteBuf,并且写入 9 个负整数  ByteBuf buf = Unpooled.buffer();

for (int i = 1; i < 10; i++) {

    buf.writeInt(i * -1);
  }

//创建一个__EmbeddedChannel,并安装一个要测试的 AbsIntegerEncoder  EmbeddedChannel channel = new EmbeddedChannel(new AbsIntegerEncoder());

//_写入 ByteBuf,并断言调用 readOutbound()方法将会产生数据

_ assertTrue(channel.writeOutbound(buf));

//_将该_ Channel 标记为已完成状态 assertTrue(channel.finish());

// read bytes   //_读取所产生的消息,并断言它们包含了对应的绝对值

_ for (int i = 1; i < 10; i++) {

assertEquals(i, channel.readOutbound());
  }

assertNull(channel.readOutbound());
}

使用EmbeddedChannel测试异常处理

我们指定最大的帧大小设置为3字节,如果一个帧的大小超出了该限制,那么程序将会丢弃它的字节,并抛出一个 TooLongFrameException。位于 ChannelPipeline 中的其他 ChannelHandler可以选择在exceptionCaught()方法中处理该异常或者忽略它

//扩展 ByteToMessageDecoder以将入站字节解码为消息 __public class FrameChunkDecoder extends ByteToMessageDecoder {

private final int maxFrameSize;

//_指定将要产生的帧的最大允许大小

_ public FrameChunkDecoder(int maxFrameSize) {

this.maxFrameSize = maxFrameSize;
  }

  @Override

protected void decode(ChannelHandlerContext ctx, ByteBuf in, List out)

throws Exception {

int readableBytes = in.readableBytes();

if (readableBytes > maxFrameSize) {

// discard the bytes       //_如果该帧太大,则丢弃它并抛出一个_ TooLongFrameException……      in.clear();

throw new TooLongFrameException();
    }

//_否则,从 ByteBuf 中读取一个新的帧

_    ByteBuf buf = in.readBytes(readableBytes);

//_将该帧添加到解码 读取一个新的帧消息的 List_ _中

_    out.add(buf);
  }
}

下面是测试代码

@Test
public void testFramesDecoded() {

//_创建一个_ ByteBuf,并向它写入 9 字节  ByteBuf buf = Unpooled.buffer();

for (int i = 0; i < 9; i++) {

    buf.writeByte(i);
  }

  ByteBuf input = buf.duplicate();

//_创建一个_ EmbeddedChannel,并向其安装一个帧大小为 3 字节的 FixedLengthFrameDecoder  EmbeddedChannel channel = new EmbeddedChannel(new FrameChunkDecoder(3));

//_向它写入 2 字节,并断言它们将会产生一个新帧

_ assertTrue(channel.writeInbound(input.readBytes(2)));

try {

//_写入一个_ _4 字节大小的帧,并捕获预期的__TooLongFrameException

_    channel.writeInbound(input.readBytes(4));

//_如果上面没有抛出异常,那么就会到达这个断言,并且测试失败

    Assert._fail();
  } catch (TooLongFrameException e) {

// expected exception  }

//_写入剩余的__2字节,并断言将会产生一个有效帧

_ assertTrue(channel.writeInbound(input.readBytes(3)));

//_将该_ Channel 标记为已完成状态 assertTrue(channel.finish());

// Read frames   //_读取产生的消息,并且验证值

_  ByteBuf read = (ByteBuf) channel.readInbound();

assertEquals(buf.readSlice(2), read);

  read.release();

  read = (ByteBuf) channel.readInbound();

assertEquals(buf.skipBytes(4).readSlice(3), read);

  read.release();

  buf.release();
}

注意,如果该类实现了exceptionCaught()方法并处理了该异常,那么它将不会被catch 块所捕获

手机扫一扫

移动阅读更方便

阿里云服务器
腾讯云服务器
七牛云服务器

你可能感兴趣的文章