testWriteAfterClosedOrShutdown() — netty Function Reference

Source Code

codec-native-quic/src/test/java/io/netty/handler/codec/quic/QuicStreamChannelCloseTest.java lines 191–226

    private static void testWriteAfterClosedOrShutdown(Executor executor, QuicStreamType type,
                                                         boolean halfClose) throws Throwable {
        Channel server = null;
        Channel channel = null;
        try {
            final Promise<Channel> streamPromise = ImmediateEventExecutor.INSTANCE.newPromise();
            server = QuicTestUtils.newServer(executor, new ChannelInboundHandlerAdapter(), new StreamHandler());
            channel = QuicTestUtils.newClient(executor);

            StreamCreationAndTearDownHandler creationHandler =
                    new StreamCreationAndTearDownHandler(type, halfClose, streamPromise);
            QuicChannel quicChannel = QuicTestUtils.newQuicChannelBootstrap(channel)
                    .handler(creationHandler)
                    .streamHandler(new ChannelInboundHandlerAdapter())
                    .remoteAddress(server.localAddress())
                    .connect()
                    .get();

            // ChannelOutputShutdownException should only be used when its a BIDIRECTIONAL channel and half-closure
            // is used.
            Class<? extends Throwable> causeClass =
                    halfClose && type != QuicStreamType.UNIDIRECTIONAL ?
                            ChannelOutputShutdownException.class : ClosedChannelException.class;
            assertInstanceOf(causeClass, streamPromise.await().cause());
            quicChannel.close().sync();

            // Wait till the client was closed
            quicChannel.closeFuture().sync();
            creationHandler.assertState();
        } finally {
            QuicTestUtils.closeIfNotNull(channel);
            QuicTestUtils.closeIfNotNull(server);

            shutdown(executor);
        }
    }