DatagramMulticastTest Class — netty Architecture

Source Code

testsuite/src/main/java/io/netty/testsuite/transport/socket/DatagramMulticastTest.java lines 52–249

public class DatagramMulticastTest extends AbstractDatagramTest {

    @Test
    public void testMulticast(TestInfo testInfo) throws Throwable {
        run(testInfo, new Runner<Bootstrap, Bootstrap>() {
            @Override
            public void run(Bootstrap bootstrap, Bootstrap bootstrap2) throws Throwable {
                testMulticast(bootstrap, bootstrap2);
            }
        });
    }

    public void testMulticast(Bootstrap sb, Bootstrap cb) throws Throwable {
        NetworkInterface iface = multicastNetworkInterface();
        assumeTrue(iface != null, "No NetworkInterface found that supports multicast and " +
                             socketInternetProtocalFamily());

        MulticastTestHandler mhandler = new MulticastTestHandler();

        sb.handler(new SimpleChannelInboundHandler<Object>() {
            @Override
            public void channelRead0(ChannelHandlerContext ctx, Object msg) throws Exception {
                // Nothing will be sent.
            }

            @Override
            public boolean isSharable() {
                return true;
            }
        });

        cb.handler(mhandler);

        sb.option(ChannelOption.IP_MULTICAST_IF, iface);
        sb.option(ChannelOption.SO_REUSEADDR, true);

        cb.option(ChannelOption.IP_MULTICAST_IF, iface);
        cb.option(ChannelOption.SO_REUSEADDR, true);

        DatagramChannel sc = null;

        int attempts = 5;
        ChannelFuture clientFuture;
        do {
            if (sc != null) {
                sc.close().sync();
            }
            sc = (DatagramChannel) sb.bind(newSocketAddress(iface)).sync().channel();
            if (sc instanceof OioDatagramChannel) {
                // skip the test for OIO, as it fails because of
                // No route to host which makes no sense.
                // Maybe a JDK bug ?
                sc.close().awaitUninterruptibly();
                return;
            }
            assertEquals(iface, sc.config().getNetworkInterface());
            assertInterfaceAddress(iface, sc.config().getInterface());

            InetSocketAddress addr = sc.localAddress();
            cb.localAddress(addr.getPort());
            clientFuture = cb.bind().await();
        } while (!clientFuture.isSuccess() && --attempts > 0);
        DatagramChannel cc = (DatagramChannel) clientFuture.sync().channel();
        assertEquals(iface, cc.config().getNetworkInterface());
        assertInterfaceAddress(iface, cc.config().getInterface());

        InetSocketAddress groupAddress = SocketUtils.socketAddress(groupAddress(), sc.localAddress().getPort());

        cc.joinGroup(groupAddress, iface).sync();

        sc.writeAndFlush(new DatagramPacket(Unpooled.copyInt(1), groupAddress)).sync();
        assertTrue(mhandler.await());

        // leave the group
        cc.leaveGroup(groupAddress, iface).sync();

        // sleep a second to make sure we left the group
        Thread.sleep(1000);

        // we should not receive a message anymore as we left the group before
        sc.writeAndFlush(new DatagramPacket(Unpooled.copyInt(1), groupAddress)).sync();