testCloseWhileAggregating() — netty Function Reference

Source Code

codec-base/src/test/java/io/netty/handler/codec/MessageAggregatorTest.java lines 98–120

    @Test
    public void testCloseWhileAggregating() throws Exception {
        ReadCounter counter = new ReadCounter();
        ByteBufHolder first = new TestMessage(Unpooled.copiedBuffer("first", CharsetUtil.US_ASCII));

        MockMessageAggregator agg = spy(MockMessageAggregator.class);
        when(agg.isStartMessage(first)).thenReturn(true);
        when(agg.isLastContentMessage(first)).thenReturn(false);

        final EmbeddedChannel embedded = new EmbeddedChannel(counter, agg);
        embedded.config().setAutoRead(false);

        assertFalse(embedded.writeInbound(first));

        assertEquals(2, counter.value);
        assertThrows(PrematureChannelClosureException.class, new Executable() {
            @Override
            public void execute() {
                embedded.finish();
            }
        });
        assertEquals(0, first.refCnt());
    }