testResponseFeedbackStream() — netty Function Reference

Source Code

resolver-dns/src/test/java/io/netty/resolver/dns/DnsNameResolverTest.java lines 4246–4324

    @ParameterizedTest
    @EnumSource(DnsNameResolverChannelStrategy.class)
    public void testResponseFeedbackStream(DnsNameResolverChannelStrategy strategy) {
        final AtomicBoolean successCalled = new AtomicBoolean();
        final AtomicBoolean failureCalled = new AtomicBoolean();
        final AtomicBoolean returnSuccess = new AtomicBoolean(false);
        final DnsNameResolver resolver = newResolver(strategy, true, new DnsServerAddressStreamProvider() {
            @Override
            public DnsServerAddressStream nameServerAddressStream(String hostname) {
                return new DnsServerResponseFeedbackAddressStream() {
                    @Override
                    public void feedbackSuccess(InetSocketAddress address, long queryResponseTimeNanos) {
                        assertThat(queryResponseTimeNanos).isGreaterThanOrEqualTo(0L);
                        successCalled.set(true);
                    }

                    @Override
                    public void feedbackFailure(InetSocketAddress address,
                                                Throwable failureCause,
                                                long queryResponseTimeNanos) {
                        assertThat(queryResponseTimeNanos).isGreaterThanOrEqualTo(0L);
                        assertNotNull(failureCause);
                        failureCalled.set(true);
                    }

                    @Override
                    public InetSocketAddress next() {
                        if (returnSuccess.get()) {
                            return dnsServer.localAddress();
                        }
                        try {
                            return new InetSocketAddress(InetAddress.getByAddress("foo.com",
                                    new byte[] {(byte) 169, (byte) 254, 12, 34 }), 53);
                        } catch (UnknownHostException e) {
                            throw new Error(e);
                        }
                    }

                    @Override
                    public int size() {
                        return 1;
                    }

                    @Override
                    public DnsServerAddressStream duplicate() {
                        return this;
                    }
                };
            }
        }).build();
        try {
            // setup call to be successful and verify
            returnSuccess.set(true);
            resolver.resolve("google.com").syncUninterruptibly().getNow();
            assertTrue(successCalled.get());
            assertFalse(failureCalled.get());

            // reset state for next query
            successCalled.set(false);
            failureCalled.set(false);

            // setup call to fail and verify
            returnSuccess.set(false);
            try {
                resolver.resolve("yahoo.com").syncUninterruptibly().getNow();
                fail();
            } catch (Exception e) {
                // expected
                assertInstanceOf(UnknownHostException.class, e);
            } finally {
                assertFalse(successCalled.get());
                assertTrue(failureCalled.get());
            }
        } finally {
            if (resolver != null) {
                resolver.close();
            }
        }
    }