testCnameWithAAndAdditionalsAndAuthorities() — netty Function Reference

Source Code

resolver-dns/src/test/java/io/netty/resolver/dns/DnsNameResolverTest.java lines 4326–4396

    @ParameterizedTest
    @EnumSource(DnsNameResolverChannelStrategy.class)
    public void testCnameWithAAndAdditionalsAndAuthorities(DnsNameResolverChannelStrategy strategy) throws Exception {
        final String hostname = "test.netty.io";
        final String cname = "cname.netty.io";

        final List<String> nameServers = new ArrayList<String>();

        for (int i = 0; i < 13; i++) {
            nameServers.add("ns" + i + ".foo.bar");
        }

        TestDnsServer server = new TestDnsServer(new RecordStore() {
            @Override
            public Set<ResourceRecord> getRecords(QuestionRecord questionRecord) {
                ResourceRecordModifier rm = new ResourceRecordModifier();
                rm.setDnsClass(RecordClass.IN);
                rm.setDnsName(hostname);
                rm.setDnsTtl(10000);
                rm.setDnsType(RecordType.CNAME);
                rm.put(DnsAttribute.DOMAIN_NAME, cname);

                Set<ResourceRecord> records = new LinkedHashSet<ResourceRecord>();
                records.add(rm.getEntry());
                records.add(newARecord(cname, "10.0.0.2"));
                return records;
            }
        }) {
            @Override
            protected DnsMessage filterMessage(DnsMessage message) {
                for (QuestionRecord record: message.getQuestionRecords()) {
                    if (record.getDomainName().equals(hostname)) {
                        // Let's add some extra records.
                        message.getAuthorityRecords().clear();
                        message.getAdditionalRecords().clear();

                        for (String nameserver: nameServers) {
                            message.getAuthorityRecords().add(TestDnsServer.newNsRecord(".", nameserver));
                            message.getAdditionalRecords().add(newAddressRecord(nameserver, RecordType.A, "10.0.0.1"));
                            message.getAdditionalRecords().add(newAddressRecord(nameserver, RecordType.AAAA, "::1"));
                        }

                        return message;
                    }
                }
                return message;
            }
        };
        server.start();
        EventLoopGroup group = new MultiThreadIoEventLoopGroup(1, NioIoHandler.newFactory());

        SingletonDnsServerAddressStreamProvider provider =
                new SingletonDnsServerAddressStreamProvider(server.localAddress());
        final DnsNameResolver resolver = new DnsNameResolver(
                group.next(), new ReflectiveChannelFactory<DatagramChannel>(NioDatagramChannel.class), null, false,
                NoopDnsCache.INSTANCE, NoopDnsCnameCache.INSTANCE, NoopAuthoritativeDnsServerCache.INSTANCE, null,
                NoopDnsQueryLifecycleObserverFactory.INSTANCE, 2000, ResolvedAddressTypes.IPV4_ONLY,
                true, 8, true, 4096,
                false, HostsFileEntriesResolver.DEFAULT,
                provider, new ThreadLocalNameServerAddressStream(provider),
                new String [] { "k8se-apps.svc.cluster.local, svc.cluster.local, cluster.local" }, 1,
                true, false, 0, strategy);
        try {
            InetAddress address = resolver.resolve(hostname).sync().getNow();
            assertArrayEquals(new byte[] { 10, 0, 0, 2 }, address.getAddress());
        } finally {
            resolver.close();
            group.shutdownGracefully(0, 0, TimeUnit.SECONDS);
            server.stop();
        }
    }