policy_test.go

package dnsresolver

import (
	"testing"
	"time"

	"github.com/miekg/dns"
	"github.com/stretchr/testify/assert"
)

func TestDefaultCachePolicy(t *testing.T) {
	cases := []struct {
		q         dns.Question
		answer    []dns.RR
		authority []dns.RR
		want      time.Duration
	}{
		{
			q: dns.Question{Qtype: dns.TypeA, Name: "bbc.co.uk."},
			authority: []dns.RR{
				NS(t, "uk.", 172800, "nsa.nic.uk."),
			},
			want: 172800 * time.Second,
		},
		{
			q: dns.Question{Qtype: dns.TypeA, Name: "bbc.co.uk."},
			authority: []dns.RR{
				NS(t, "bbc.co.uk.", 172800, "dns1.bbc.co.uk."),
			},
			want: 0,
		},
		{
			q: dns.Question{Qtype: dns.TypeA, Name: "bbc.co.uk."},
			authority: []dns.RR{
				NS(t, "uk.", 172800, "nsa.nic.uk."),
				NS(t, "uk.", 172800, "nsb.nic.uk."),
			},
			want: 172800 * time.Second,
		},
		{
			q: dns.Question{Qtype: dns.TypeA, Name: "bbc.co.uk."},
			authority: []dns.RR{
				NS(t, "uk.", 172800, "nsa.nic.uk."),
				NS(t, "co.uk.", 172800, "nsa.nic.uk."),
			},
			want: 0,
		},
		{
			q: dns.Question{Qtype: dns.TypeA, Name: "bbc.co.uk."},
			authority: []dns.RR{
				NS(t, "uk.", 172800, "nsa.nic.uk."),
				NS(t, "bbc.co.uk.", 172800, "dns1.bbc.co.uk."),
			},
			want: 0,
		},
	}

	t.Parallel()

	for _, tc := range cases {
		t.Run("", func(t *testing.T) {
			msg := dns.Msg{}
			msg.Question = append(msg.Question, tc.q)
			msg.Answer = append(msg.Answer, tc.answer...)
			msg.Ns = append(msg.Ns, tc.authority...)

			got := defaultCachePolicy(RecordSet{Raw: msg})
			assert.Equal(t, tc.want, got)
		})
	}
}