Bzip2EncoderTest Class — netty Architecture

Source Code

codec-compression/src/test/java/io/netty/handler/codec/compression/Bzip2EncoderTest.java lines 29–57

public class Bzip2EncoderTest extends AbstractEncoderTest {

    @Override
    protected EmbeddedChannel createChannel() {
        return new EmbeddedChannel(new Bzip2Encoder(MIN_BLOCK_SIZE));
    }

    @Override
    protected ByteBuf decompress(ByteBuf compressed, int originalLength) throws Exception {

        byte[] decompressed = new byte[originalLength];
        try (InputStream is = new ByteBufInputStream(compressed, true);
             BZip2CompressorInputStream bzip2Is = new BZip2CompressorInputStream(is)) {

            int remaining = originalLength;
            while (remaining > 0) {
                int read = bzip2Is.read(decompressed, originalLength - remaining, remaining);
                if (read > 0) {
                    remaining -= read;
                } else {
                    break;
                }
            }
            assertEquals(-1, bzip2Is.read());
        }

        return Unpooled.wrappedBuffer(decompressed);
    }
}