testReregisterOnChannelHandlerContext() — netty Function Reference

Source Code

transport/src/test/java/io/netty/channel/AbstractEventLoopTest.java lines 78–135

    @ParameterizedTest
    @EnumSource(DeregisterMethod.class)
    void testReregisterOnChannelHandlerContext(DeregisterMethod method) throws Exception {
        EventLoopGroup group = newEventLoopGroup();
        AtomicReference<Throwable> throwable = new AtomicReference<>();
        try {
            ServerBootstrap b = new ServerBootstrap();
            ReRegisterHandler reRegisterHandler = new ReRegisterHandler(group, method, throwable);
            b.group(group)
                    .channel(newChannel())
                    .childHandler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        public void initChannel(SocketChannel ch) {
                            ChannelPipeline p = ch.pipeline();
                            p.addLast("logging", new LoggingHandler());
                            p.addLast(reRegisterHandler);
                        }
                    });

            ChannelFuture f = b.bind(0).sync();

            Channel client = new Bootstrap()
                    .group(group)
                    .channel(newSocketChannel())
                    .handler(new ChannelInboundHandlerAdapter() {
                        @Override
                        public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
                            ReferenceCountUtil.release(msg);
                        }

                        @Override
                        public void channelReadComplete(ChannelHandlerContext ctx) throws Exception {
                            super.channelReadComplete(ctx);
                            ctx.close();
                        }

                        @Override
                        public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
                            throwable.set(cause);
                            super.exceptionCaught(ctx, cause);
                            ctx.close();
                        }
                    })
                    .connect(f.channel().localAddress())
                    .sync()
                    .channel();
            client.closeFuture().addListener(ignore -> f.channel().close());
            client.writeAndFlush(Unpooled.copiedBuffer("hello", StandardCharsets.US_ASCII));

            f.channel().closeFuture().sync();
            Throwable caughtThrowable = throwable.get();
            if (caughtThrowable != null) {
                fail(caughtThrowable);
            }
        } finally {
            group.shutdownGracefully();
        }
    }