SslHandlerCoalescingBufferQueueTest Class — netty Architecture

Source Code

handler/src/test/java/io/netty/handler/ssl/SslHandlerCoalescingBufferQueueTest.java lines 34–134

public class SslHandlerCoalescingBufferQueueTest {
    static final Supplier<ByteBuf> BYTEBUF_SUPPLIER = new Supplier<ByteBuf>() {
        @Override
        public ByteBuf get() {
            ByteBuf buf = Unpooled.directBuffer(128);
            buf.writeLong(1);
            buf.writeLong(2);
            return buf;
        }
    };
    static final Function<ByteBuf, ByteBuf> NO_WRAPPER = new Function<ByteBuf, ByteBuf>() {
        @Override
        public ByteBuf apply(ByteBuf byteBuf) {
            return byteBuf;
        }
    };
    static final Function<ByteBuf, ByteBuf> DUPLICATE_WRAPPER = new Function<ByteBuf, ByteBuf>() {
        @Override
        public ByteBuf apply(ByteBuf byteBuf) {
            return byteBuf.duplicate();
        }
    };
    static final Function<ByteBuf, ByteBuf> SLICE_WRAPPER = new Function<ByteBuf, ByteBuf>() {
        @Override
        public ByteBuf apply(ByteBuf byteBuf) {
            return byteBuf.slice();
        }
    };
    static ByteBuffer createNioBuffer() {
        ByteBuffer nioBuffer = ByteBuffer.allocateDirect(128);
        nioBuffer.putLong(1);
        nioBuffer.putLong(2);
        nioBuffer.position(nioBuffer.limit());
        nioBuffer.flip();
        return nioBuffer;
    }

    enum CumulationTestScenario {
        BASIC_NIO_BUFFER(new Supplier() {
            @Override
            public ByteBuf get() {
                return Unpooled.wrappedBuffer(createNioBuffer());
            }
        }, NO_WRAPPER),
        READ_ONLY_AND_DUPLICATE_NIO_BUFFER(new Supplier() {
            @Override
            public ByteBuf get() {
                return Unpooled.wrappedBuffer(createNioBuffer().asReadOnlyBuffer());
            }
        }, DUPLICATE_WRAPPER),
        BASIC_DIRECT_BUFFER(BYTEBUF_SUPPLIER, NO_WRAPPER),
        DUPLICATE_DIRECT_BUFFER(BYTEBUF_SUPPLIER, DUPLICATE_WRAPPER),
        SLICED_DIRECT_BUFFER(BYTEBUF_SUPPLIER, SLICE_WRAPPER);

        final Supplier<ByteBuf> bufferSupplier;
        final Function<ByteBuf, ByteBuf> bufferWrapper;

        CumulationTestScenario(Supplier<ByteBuf> bufferSupplier, Function<ByteBuf, ByteBuf> bufferWrapper) {
            this.bufferSupplier = bufferSupplier;
            this.bufferWrapper = bufferWrapper;
        }
    }

    @ParameterizedTest
    @EnumSource(CumulationTestScenario.class)
    public void testCumulation(CumulationTestScenario testScenario) {
        EmbeddedChannel channel = new EmbeddedChannel();
        SslHandlerCoalescingBufferQueue queue = new SslHandlerCoalescingBufferQueue(channel, 16, false) {
            @Override
            protected int wrapDataSize() {
                return 128;
            }
        };

        ByteBuf original = testScenario.bufferSupplier.get();
        original.writerIndex(8);
        ByteBuf first = testScenario.bufferWrapper.apply(original);
        first.retain();
        queue.add(first);
        ByteBuf second = Unpooled.copyLong(3);
        queue.add(second);