testPacketsAreDispatchedToCorrectChannel() — netty Function Reference

Source Code

codec-native-quic/src/test/java/io/netty/handler/codec/quic/QuicCodecDispatcherTest.java lines 37–89

    @Test
    public void testPacketsAreDispatchedToCorrectChannel() throws QuicException {
        short localConnectionIdLength = 16;

        AtomicInteger initChannelCalled = new AtomicInteger();
        QuicCodecDispatcher dispatcher = new QuicCodecDispatcher(localConnectionIdLength) {
            @Override
            protected void initChannel(Channel channel, int localConnectionIdLength,
                                       QuicConnectionIdGenerator idGenerator) {
                initChannelCalled.incrementAndGet();
            }
        };

        EmbeddedChannel[] channels = new EmbeddedChannel[8];
        for (int i = 0; i < channels.length; i++) {
            channels[i] = new EmbeddedChannel(dispatcher);
        }

        int numPackets = 0;
        for (int i = 0; i < 100; i++) {
            writePacket(channels, false, localConnectionIdLength);
            numPackets++;
            writePacket(channels, true, localConnectionIdLength);
            numPackets++;
        }

        for (int idx = 0; idx < channels.length; idx++) {
            EmbeddedChannel channel = channels[idx];
            for (;;) {
                DatagramPacket packet = channel.readInbound();
                    if (packet == null) {
                        break;
                    }
                try {
                    boolean hasShortHeader = QuicCodecDispatcher.hasShortHeader(packet.content());
                    ByteBuf id = QuicCodecDispatcher.getDestinationConnectionId(
                            packet.content(), localConnectionIdLength);
                    if (hasShortHeader) {
                        assertNotNull(id);
                        assertEquals(idx, QuicCodecDispatcher.decodeIdx(id));
                    } else {
                        assertNull(id);
                    }
                    numPackets--;
                } finally {
                    packet.release();
                }
            }
            assertFalse(channel.finishAndReleaseAll());
        }
        assertEquals(0, numPackets);
        assertEquals(channels.length, initChannelCalled.get());
    }