DefaultDnsRecordEncoderTest Class — netty Architecture

Source Code

codec-dns/src/test/java/io/netty/handler/codec/dns/DefaultDnsRecordEncoderTest.java lines 30–187

public class DefaultDnsRecordEncoderTest {

    @Test
    public void testEncodePtr() throws Exception {
        DefaultDnsRecordEncoder encoder = new DefaultDnsRecordEncoder();
        DnsPtrRecord ptrRecord = new DefaultDnsPtrRecord("131.186.250.142.in-addr.arpa.",
                DnsRecord.CLASS_IN, 80, "fra24s07-in-f3.1e100.net.");
        ByteBuf out = Unpooled.buffer();
        ByteBuf expectedBuf = Unpooled.buffer();
        expectedBuf.writeBytes(
                new byte[] {
                        3, '1', '3', '1',
                        3, '1', '8', '6',
                        3, '2', '5', '0',
                        3, '1', '4', '2',
                        7, 'i', 'n', '-', 'a', 'd', 'd', 'r',
                        4, 'a', 'r', 'p', 'a',
                        0
        });
        expectedBuf.writeShort(DnsRecordType.PTR.intValue());
        expectedBuf.writeShort(DnsRecord.CLASS_IN);
        expectedBuf.writeInt(80);
        byte[] hostname = new byte[] {
                14, 'f', 'r', 'a', '2', '4', 's', '0', '7', '-', 'i', 'n', '-', 'f', '3',
                5, '1', 'e', '1', '0', '0', 3, 'n', 'e', 't',
                0
        };
        expectedBuf.writeShort(hostname.length);
        expectedBuf.writeBytes(hostname);
        try {
            encoder.encodeRecord(ptrRecord, out);
            assertEquals(expectedBuf, out);
        } finally {
            out.release();
            expectedBuf.release();
        }
    }

    @Test
    public void testEncodeName() throws Exception {
        testEncodeName(new byte[] { 5, 'n', 'e', 't', 't', 'y', 2, 'i', 'o', 0 }, "netty.io.");
    }

    @Test
    public void testEncodeNameWithoutTerminator() throws Exception {
        testEncodeName(new byte[] { 5, 'n', 'e', 't', 't', 'y', 2, 'i', 'o', 0 }, "netty.io");
    }

    @Test
    public void testEncodeNameWithExtraTerminator() throws Exception {
        testEncodeName(new byte[] { 5, 'n', 'e', 't', 't', 'y', 2, 'i', 'o', 0 }, "netty.io..");
    }

    // Test for https://github.com/netty/netty/issues/5014
    @Test
    public void testEncodeEmptyName() throws Exception {
        testEncodeName(new byte[] { 0 }, StringUtil.EMPTY_STRING);
    }

    @Test
    public void testEncodeRootName() throws Exception {
        testEncodeName(new byte[] { 0 }, ".");
    }

    private static void testEncodeName(byte[] expected, String name) throws Exception {
        DefaultDnsRecordEncoder encoder = new DefaultDnsRecordEncoder();
        ByteBuf out = Unpooled.buffer();
        ByteBuf expectedBuf = Unpooled.wrappedBuffer(expected);
        try {
            encoder.encodeName(name, out);
            assertEquals(expectedBuf, out);
        } finally {
            out.release();
            expectedBuf.release();
        }
    }

    @Test
    public void testOptEcsRecordIpv4() throws Exception {
        testOptEcsRecordIp(SocketUtils.addressByName("1.2.3.4"));
        testOptEcsRecordIp(SocketUtils.addressByName("1.2.3.255"));