Http3ServerPushStreamManagerTest Class — netty Architecture

Source Code

codec-http3/src/test/java/io/netty/handler/codec/http3/Http3ServerPushStreamManagerTest.java lines 47–212

public class Http3ServerPushStreamManagerTest {

    private EmbeddedQuicChannel channel;
    private Http3ServerPushStreamManager manager;
    private Http3ServerConnectionHandler connectionHandler;
    private ChannelHandlerContext controlStreamHandlerCtx;
    private EmbeddedQuicStreamChannel localControlStream;

    @BeforeEach
    public void setUp() throws Exception {
        connectionHandler = new Http3ServerConnectionHandler(new Http3RequestStreamInboundHandler() {
            @Override
            protected void channelRead(ChannelHandlerContext ctx, Http3HeadersFrame frame) {
                ReferenceCountUtil.release(frame);
            }

            @Override
            protected void channelRead(ChannelHandlerContext ctx, Http3DataFrame frame) {
                ReferenceCountUtil.release(frame);
            }

            @Override
            protected void channelInputClosed(ChannelHandlerContext ctx) {
                // NOOP
            }
        }, null, null, null, true);
        channel = new EmbeddedQuicChannel(true, connectionHandler);
        localControlStream = (EmbeddedQuicStreamChannel) Http3.getLocalControlStream(channel);
        assertNotNull(localControlStream);
        assertTrue(localControlStream.releaseOutbound()); // settings

        controlStreamHandlerCtx = mock(ChannelHandlerContext.class);
        when(controlStreamHandlerCtx.channel()).thenReturn(localControlStream);
        connectionHandler.localControlStreamHandler.channelRead(controlStreamHandlerCtx,
                new DefaultHttp3SettingsFrame());
        manager = new Http3ServerPushStreamManager(channel);
    }

    @AfterEach
    public void tearDown() {
        assertFalse(localControlStream.finish());
        assertFalse(channel.finish());
    }

    @Test
    public void pushAllowed() throws Exception {
        assertFalse(manager.isPushAllowed());
        sendMaxPushId(1);
        assertTrue(manager.isPushAllowed());
    }

    @Test
    public void reserveWhenPushNotAllowed() {
        assertThrows(IllegalStateException.class, () -> manager.reserveNextPushId());
    }

    @Test
    public void reserveWhenPushAllowed() throws Exception {
        sendMaxPushId(2);
        assertEquals(0, manager.reserveNextPushId());
    }

    @Test
    public void reservesAfterRefreshMaxId() throws Exception {
        sendMaxPushId(0);
        assertEquals(0, manager.reserveNextPushId());
        assertFalse(manager.isPushAllowed());
        sendMaxPushId(1);
        assertEquals(1, manager.reserveNextPushId());
    }

    @Test
    public void pushStreamNoHandler() throws Exception {
        pushStreamCreateAndClose(pushId -> newPushStream(null, pushId));
    }

    @Test
    public void pushStreamWithHandler() throws Exception {
        final PushStreamListener pushStreamHandler = new PushStreamListener();
        pushStreamCreateAndClose(pushId -> newPushStream(pushStreamHandler, pushId));
        assertEquals(1, pushStreamHandler.framesWritten.size());