testResolveSearchDomainStopOnFirstSuccess() — netty Function Reference

Source Code

resolver-dns/src/test/java/io/netty/resolver/dns/DnsNameResolverTest.java lines 4058–4101

    @ParameterizedTest
    @EnumSource(DnsNameResolverChannelStrategy.class)
    public void testResolveSearchDomainStopOnFirstSuccess(DnsNameResolverChannelStrategy strategy) throws Exception {
        final String addressString = "10.0.0.1";
        final Queue<String> names = new ConcurrentLinkedQueue<String>();
        final TestDnsServer dnsServer2 = new TestDnsServer(new RecordStore() {
            private int called;
            @Override
            public Set<ResourceRecord> getRecords(QuestionRecord question) {
                names.offer(question.getDomainName());
                if (++called == 2) {
                    ResourceRecordModifier rm = new ResourceRecordModifier();
                    rm.setDnsClass(RecordClass.IN);
                    rm.setDnsName(question.getDomainName());
                    rm.setDnsTtl(100);

                    rm.setDnsType(question.getRecordType());
                    rm.put(DnsAttribute.IP_ADDRESS, addressString);
                    return Collections.singleton(rm.getEntry());
                }
                return null;
            }
        });
        dnsServer2.start();

        DnsNameResolver resolver = newResolver(strategy).searchDomains(
                Arrays.asList("search1.netty.io", "search2.netty.io", "search3.netty.io"))
                .ndots(2).nameServerProvider(new SingletonDnsServerAddressStreamProvider(dnsServer2.localAddress()))
                .resolvedAddressTypes(ResolvedAddressTypes.IPV4_ONLY)
                .build();

        byte[] bytes =  NetUtil.createByteArrayFromIpAddressString(addressString);
        InetAddress inetAddress = InetAddress.getByAddress(bytes);
        try {
            final InetAddress addr = resolver.resolve("netty.io").sync().getNow();
            assertEquals(inetAddress, addr);
        } finally {
            resolver.close();
            dnsServer2.stop();
            assertEquals("netty.io.search1.netty.io", names.poll());
            assertEquals("netty.io.search2.netty.io", names.poll());
            assertTrue(names.isEmpty());
        }
    }