Function TestResolver_LookupIP

dns/miekgdns/resolver_test.go:225–296

(t *testing.T)

}

func TestResolver_LookupIP(t *testing.T) {
	logger := log.NewNopLogger()
	period := time.Hour // We don't want the periodic reload running in the test
	tmpDir := t.TempDir()

	t.Run("multiple timeouts", func(t *testing.T) {
		cfgPath := writeResovConf(t, tmpDir, resolvConfContents)
		client := newMockClient()
		client.err["127.0.0.53:53"] = errors.New("timeout 1 in test")
		client.err["127.0.0.54:53"] = errors.New("timeout 2 in test")

		resolver := NewResolverWithClient(cfgPath, logger, period, client)
		t.Cleanup(resolver.Stop)
		_, err := resolver.LookupIPAddr(context.Background(), "cache01.example.com.")

		require.ErrorContains(t, err, "timeout")
	})

	t.Run("one timeout and one success", func(t *testing.T) {
		cfgPath := writeResovConf(t, tmpDir, resolvConfContents)
		client := newMockClient()
		client.err["127.0.0.53:53"] = errors.New("timeout 1 in test")
		client.res["127.0.0.54:53"] = []*dns.Msg{newIPDNSResponse("cache01.example.com.", net.IPv4(10, 0, 0, 1))}

		resolver := NewResolverWithClient(cfgPath, logger, period, client)
		t.Cleanup(resolver.Stop)
		res, err := resolver.LookupIPAddr(context.Background(), "cache01.example.com")

		require.NoError(t, err)
		require.Equal(t, []net.IPAddr{{IP: net.IPv4(10, 0, 0, 1)}}, res)
	})

	t.Run("name error", func(t *testing.T) {
		response1 := new(dns.Msg).SetQuestion("cache01.example.com.", dns.TypeAAAA)
		response1.Rcode = dns.RcodeNameError
		response1.Response = true

		response2 := new(dns.Msg).SetQuestion("cache01.example.com.", dns.TypeA)
		response2.Rcode = dns.RcodeNameError
		response2.Response = true

		cfgPath := writeResovConf(t, tmpDir, resolvConfContents)
		client := newMockClient()
		client.err["127.0.0.53:53"] = errors.New("timeout 1 in test")
		// Include two responses since a failed AAAA lookup will result in an A fallback
		client.res["127.0.0.54:53"] = []*dns.Msg{response1, response2}

		resolver := NewResolverWithClient(cfgPath, logger, period, client)
		t.Cleanup(resolver.Stop)
		res, err := resolver.LookupIPAddr(context.Background(), "cache01.example.com")

		require.NoError(t, err)
		require.Empty(t, res)
	})

	t.Run("one level of CNAME", func(t *testing.T) {
		cfgPath := writeResovConf(t, tmpDir, resolvConfContents)
		client := newMockClient()