mirror of
https://github.com/coredns/coredns.git
synced 2025-10-30 17:53:21 -04:00
test(plugin): improve backend lookup coverage (#7496)
This commit is contained in:
Ville Vesilehto
GitHub
@@ -3,6 +3,7 @@ package plugin
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
@@ -18,14 +19,25 @@ var _ ServiceBackend = &mockBackend{}
|
||||
|
||||
type mockBackend struct {
|
||||
mockServices func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error)
|
||||
mockReverse func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error)
|
||||
mockLookup func(ctx context.Context, state request.Request, name string, typ uint16) (*dns.Msg, error)
|
||||
mockRecords func(ctx context.Context, state request.Request, exact bool) ([]msg.Service, error)
|
||||
minTTL uint32
|
||||
serial uint32
|
||||
}
|
||||
|
||||
func (m *mockBackend) Serial(state request.Request) uint32 {
|
||||
return uint32(time.Now().Unix())
|
||||
if m.serial == 0 {
|
||||
return uint32(time.Now().Unix())
|
||||
}
|
||||
return m.serial
|
||||
}
|
||||
|
||||
func (m *mockBackend) MinTTL(state request.Request) uint32 {
|
||||
return 30
|
||||
if m.minTTL == 0 {
|
||||
return 30
|
||||
}
|
||||
return m.minTTL
|
||||
}
|
||||
|
||||
func (m *mockBackend) Services(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
@@ -33,11 +45,11 @@ func (m *mockBackend) Services(ctx context.Context, state request.Request, exact
|
||||
}
|
||||
|
||||
func (m *mockBackend) Reverse(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return nil, nil
|
||||
return m.mockReverse(ctx, state, exact, opt)
|
||||
}
|
||||
|
||||
func (m *mockBackend) Lookup(ctx context.Context, state request.Request, name string, typ uint16) (*dns.Msg, error) {
|
||||
return nil, nil
|
||||
return m.mockLookup(ctx, state, name, typ)
|
||||
}
|
||||
|
||||
func (m *mockBackend) IsNameError(err error) bool {
|
||||
@@ -45,7 +57,7 @@ func (m *mockBackend) IsNameError(err error) bool {
|
||||
}
|
||||
|
||||
func (m *mockBackend) Records(ctx context.Context, state request.Request, exact bool) ([]msg.Service, error) {
|
||||
return nil, nil
|
||||
return m.mockRecords(ctx, state, exact)
|
||||
}
|
||||
|
||||
func TestNSStateReset(t *testing.T) {
|
||||
@@ -84,3 +96,510 @@ func TestNSStateReset(t *testing.T) {
|
||||
t.Error("Expected no records returned on error")
|
||||
}
|
||||
}
|
||||
|
||||
func TestARecords_Dedup(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{
|
||||
{Host: "1.2.3.4", TTL: 60},
|
||||
{Host: "1.2.3.4", TTL: 60},
|
||||
{Host: "::1", TTL: 60},
|
||||
}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("a.example.org.", dns.TypeA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, tc, err := A(context.Background(), b, "example.org.", state, nil, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if tc {
|
||||
t.Fatal("unexpected truncation")
|
||||
}
|
||||
if len(recs) != 1 {
|
||||
t.Fatalf("expected 1 A record, got %d", len(recs))
|
||||
}
|
||||
if _, ok := recs[0].(*dns.A); !ok {
|
||||
t.Fatalf("expected A record, got %T", recs[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestAAAARecords_Dedup(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{
|
||||
{Host: "::1", TTL: 60},
|
||||
{Host: "::1", TTL: 60},
|
||||
{Host: "1.2.3.4", TTL: 60},
|
||||
}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("aaaa.example.org.", dns.TypeAAAA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, tc, err := AAAA(context.Background(), b, "example.org.", state, nil, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if tc {
|
||||
t.Fatal("unexpected truncation")
|
||||
}
|
||||
if len(recs) != 1 {
|
||||
t.Fatalf("expected 1 AAAA record, got %d", len(recs))
|
||||
}
|
||||
if _, ok := recs[0].(*dns.AAAA); !ok {
|
||||
t.Fatalf("expected AAAA record, got %T", recs[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestTXTWithInternalCNAME(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
switch state.QName() {
|
||||
case "txt.example.org.":
|
||||
return []msg.Service{{Host: "target.example.org.", TTL: 50}}, nil
|
||||
case "target.example.org.":
|
||||
return []msg.Service{{Text: "v=txt1", TTL: 40}}, nil
|
||||
default:
|
||||
return nil, nil
|
||||
}
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("txt.example.org.", dns.TypeTXT)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, tc, err := TXT(context.Background(), b, "example.org.", state, nil, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if tc {
|
||||
t.Fatal("unexpected truncation")
|
||||
}
|
||||
if len(recs) != 2 {
|
||||
t.Fatalf("expected 2 records (CNAME+TXT), got %d", len(recs))
|
||||
}
|
||||
if _, ok := recs[0].(*dns.CNAME); !ok {
|
||||
t.Fatalf("expected first record CNAME, got %T", recs[0])
|
||||
}
|
||||
if _, ok := recs[1].(*dns.TXT); !ok {
|
||||
t.Fatalf("expected second record TXT, got %T", recs[1])
|
||||
}
|
||||
}
|
||||
|
||||
func TestCNAMEHostIsNameAndIpIgnored(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{
|
||||
{Host: "target.example.org.", TTL: 50},
|
||||
{Host: "1.2.3.4", TTL: 50},
|
||||
}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("name.example.org.", dns.TypeCNAME)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, err := CNAME(context.Background(), b, "example.org.", state, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(recs) != 1 {
|
||||
t.Fatalf("expected 1 CNAME, got %d", len(recs))
|
||||
}
|
||||
if _, ok := recs[0].(*dns.CNAME); !ok {
|
||||
t.Fatalf("expected CNAME, got %T", recs[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestCNAMEChainLimitAndLoop(t *testing.T) {
|
||||
// Construct internal CNAME chain longer than maxCnameChainLength and ensure truncation of chain
|
||||
var names []string
|
||||
for i := range maxCnameChainLength + 2 {
|
||||
names = append(names, fmt.Sprintf("c%d.example.org.", i))
|
||||
}
|
||||
chain := map[string]string{}
|
||||
for i := range len(names) - 1 {
|
||||
chain[names[i]] = names[i+1]
|
||||
}
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
if nxt, ok := chain[state.QName()]; ok {
|
||||
return []msg.Service{{Host: nxt, TTL: 10}}, nil
|
||||
}
|
||||
return []msg.Service{}, nil
|
||||
},
|
||||
}
|
||||
// A query should follow the chain only until limit; since no terminal A, result is 0 or just below limit
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion(names[0], dns.TypeA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, _, err := A(context.Background(), b, "example.org.", state, nil, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
// We cannot exceed limit; if any records exist they must be <= maxCnameChainLength
|
||||
if len(recs) > maxCnameChainLength {
|
||||
t.Fatalf("CNAME chain exceeded limit: %d > %d", len(recs), maxCnameChainLength)
|
||||
}
|
||||
|
||||
// Now create a direct loop: qname CNAME qname should be ignored
|
||||
b2 := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{{Host: state.QName(), TTL: 10}}, nil
|
||||
},
|
||||
}
|
||||
req2 := new(dns.Msg)
|
||||
req2.SetQuestion("loop.example.org.", dns.TypeA)
|
||||
state2 := request.Request{Req: req2, W: &test.ResponseWriter{}}
|
||||
recs2, _, err := A(context.Background(), b2, "example.org.", state2, nil, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(recs2) != 0 {
|
||||
t.Fatalf("expected 0 records due to CNAME self-loop, got %d", len(recs2))
|
||||
}
|
||||
}
|
||||
|
||||
func TestAWithExternalCNAMELookupTruncated(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{{Host: "alias.external."}}, nil
|
||||
},
|
||||
mockLookup: func(ctx context.Context, state request.Request, name string, typ uint16) (*dns.Msg, error) {
|
||||
if name != "alias.external." || typ != dns.TypeA {
|
||||
t.Fatalf("unexpected mockLookup: %s %d", name, typ)
|
||||
}
|
||||
m := new(dns.Msg)
|
||||
m.Truncated = true
|
||||
a := &dns.A{Hdr: dns.RR_Header{Name: "alias.external.", Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 10}, A: net.ParseIP("1.2.3.4")}
|
||||
m.Answer = []dns.RR{a}
|
||||
return m, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("cname.example.org.", dns.TypeA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, tc, err := A(context.Background(), b, "example.org.", state, nil, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if !tc {
|
||||
t.Fatal("expected truncation true")
|
||||
}
|
||||
if len(recs) != 2 {
|
||||
t.Fatalf("expected 2 records (CNAME+A), got %d", len(recs))
|
||||
}
|
||||
if _, ok := recs[0].(*dns.CNAME); !ok {
|
||||
t.Fatalf("expected first record CNAME, got %T", recs[0])
|
||||
}
|
||||
if _, ok := recs[1].(*dns.A); !ok {
|
||||
t.Fatalf("expected second record A, got %T", recs[1])
|
||||
}
|
||||
}
|
||||
|
||||
func TestAAAAWithInternalCNAMEChain(t *testing.T) {
|
||||
// Build a short internal chain: n0 -> n1 -> final (AAAA)
|
||||
names := []string{"c0.example.org.", "c1.example.org.", "final.example.org."}
|
||||
chain := map[string]string{names[0]: names[1], names[1]: names[2]}
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
if nxt, ok := chain[state.QName()]; ok {
|
||||
return []msg.Service{{Host: nxt, TTL: 10}}, nil
|
||||
}
|
||||
if state.QName() == names[2] {
|
||||
return []msg.Service{{Host: "::1", TTL: 20}}, nil
|
||||
}
|
||||
return nil, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion(names[0], dns.TypeAAAA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, tc, err := AAAA(context.Background(), b, "example.org.", state, nil, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if tc {
|
||||
t.Fatal("unexpected truncation")
|
||||
}
|
||||
if len(recs) != 3 {
|
||||
t.Fatalf("expected 2 CNAMEs + 1 AAAA, got %d", len(recs))
|
||||
}
|
||||
if _, ok := recs[0].(*dns.CNAME); !ok {
|
||||
t.Fatalf("expected first record CNAME, got %T", recs[0])
|
||||
}
|
||||
if _, ok := recs[1].(*dns.CNAME); !ok {
|
||||
t.Fatalf("expected second record CNAME, got %T", recs[1])
|
||||
}
|
||||
if aaaa, ok := recs[2].(*dns.AAAA); !ok || !aaaa.AAAA.Equal(net.ParseIP("::1")) {
|
||||
t.Fatalf("expected final AAAA with ::1, got %T %v", recs[2], recs[2])
|
||||
}
|
||||
}
|
||||
|
||||
func TestAAAAWithExternalCNAMELookupTruncated(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{{Host: "alias.external."}}, nil
|
||||
},
|
||||
mockLookup: func(ctx context.Context, state request.Request, name string, typ uint16) (*dns.Msg, error) {
|
||||
if name != "alias.external." || typ != dns.TypeAAAA {
|
||||
t.Fatalf("unexpected mockLookup: %s %d", name, typ)
|
||||
}
|
||||
m := new(dns.Msg)
|
||||
m.Truncated = true
|
||||
aaaa := &dns.AAAA{Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeAAAA, Class: dns.ClassINET, Ttl: 10}, AAAA: net.ParseIP("::1")}
|
||||
m.Answer = []dns.RR{aaaa}
|
||||
return m, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("cname6.example.org.", dns.TypeAAAA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, tc, err := AAAA(context.Background(), b, "example.org.", state, nil, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if !tc {
|
||||
t.Fatal("expected truncation true")
|
||||
}
|
||||
if len(recs) != 2 {
|
||||
t.Fatalf("expected 2 records (CNAME+AAAA), got %d", len(recs))
|
||||
}
|
||||
if _, ok := recs[0].(*dns.CNAME); !ok {
|
||||
t.Fatalf("expected first record CNAME, got %T", recs[0])
|
||||
}
|
||||
if _, ok := recs[1].(*dns.AAAA); !ok {
|
||||
t.Fatalf("expected second record AAAA, got %T", recs[1])
|
||||
}
|
||||
}
|
||||
|
||||
func TestPTRDomainOnly(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockReverse: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{
|
||||
{Host: "name.example.org.", TTL: 20},
|
||||
{Host: "1.2.3.4", TTL: 20},
|
||||
}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("4.3.2.1.in-addr.arpa.", dns.TypePTR)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, err := PTR(context.Background(), b, "in-addr.arpa.", state, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(recs) != 1 {
|
||||
t.Fatalf("expected 1 PTR, got %d", len(recs))
|
||||
}
|
||||
if ptr, ok := recs[0].(*dns.PTR); !ok || ptr.Ptr != "name.example.org." {
|
||||
t.Fatalf("unexpected PTR: %T %v", recs[0], recs[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestNSSuccess(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{
|
||||
{Host: "1.2.3.4", TTL: 30, Key: "/skydns/org/example/ns1"},
|
||||
{Host: "::1", TTL: 30, Key: "/skydns/org/example/ns2"},
|
||||
}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("example.org.", dns.TypeNS)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, extra, err := NS(context.Background(), b, "example.org.", state, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(recs) != 2 {
|
||||
t.Fatalf("expected 2 NS records, got %d", len(recs))
|
||||
}
|
||||
if len(extra) != 2 {
|
||||
t.Fatalf("expected 2 extra address records, got %d", len(extra))
|
||||
}
|
||||
}
|
||||
|
||||
func TestSRVAddressesAndExternalLookup(t *testing.T) {
|
||||
// First service is IP host -> produces SRV + extra address; second is CNAME target -> triggers external lookup
|
||||
lookedUp := false
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{
|
||||
{Host: "1.2.3.4", Port: 80, Priority: 10, Weight: 5, TTL: 30, Key: "/skydns/org/example/s1"},
|
||||
{Host: "alias.external.", Port: 80, Priority: 10, Weight: 5, TTL: 30, Key: "/skydns/org/example/s2"},
|
||||
}, nil
|
||||
},
|
||||
mockLookup: func(ctx context.Context, state request.Request, name string, typ uint16) (*dns.Msg, error) {
|
||||
if name == "alias.external." && (typ == dns.TypeA || typ == dns.TypeAAAA) {
|
||||
lookedUp = true
|
||||
m := new(dns.Msg)
|
||||
if typ == dns.TypeA {
|
||||
m.Answer = []dns.RR{&dns.A{Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 10}, A: net.ParseIP("5.6.7.8")}}
|
||||
} else {
|
||||
m.Answer = []dns.RR{&dns.AAAA{Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeAAAA, Class: dns.ClassINET, Ttl: 10}, AAAA: net.ParseIP("::1")}}
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
return &dns.Msg{}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("_sip._tcp.example.org.", dns.TypeSRV)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, extra, err := SRV(context.Background(), b, "example.org.", state, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(recs) != 2 {
|
||||
t.Fatalf("expected 2 SRV records, got %d", len(recs))
|
||||
}
|
||||
if len(extra) == 0 {
|
||||
t.Fatalf("expected extra address records")
|
||||
}
|
||||
if !lookedUp {
|
||||
t.Fatalf("expected external lookup for alias.external.")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMXInternalAndExternalTargets(t *testing.T) {
|
||||
lookedUp := false
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
return []msg.Service{
|
||||
{Host: "1.2.3.4", Mail: true, Priority: 10, TTL: 60, Key: "/skydns/org/example/mx1"},
|
||||
{Host: "alias.external.", Mail: true, Priority: 20, TTL: 60, Key: "/skydns/org/example/mx2"},
|
||||
}, nil
|
||||
},
|
||||
mockLookup: func(ctx context.Context, state request.Request, name string, typ uint16) (*dns.Msg, error) {
|
||||
if name == "alias.external." && (typ == dns.TypeA || typ == dns.TypeAAAA) {
|
||||
lookedUp = true
|
||||
m := new(dns.Msg)
|
||||
if typ == dns.TypeA {
|
||||
m.Answer = []dns.RR{&dns.A{Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 10}, A: net.ParseIP("9.9.9.9")}}
|
||||
} else {
|
||||
m.Answer = []dns.RR{&dns.AAAA{Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeAAAA, Class: dns.ClassINET, Ttl: 10}, AAAA: net.ParseIP("::2")}}
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
return &dns.Msg{}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("example.org.", dns.TypeMX)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, extra, err := MX(context.Background(), b, "example.org.", state, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(recs) != 2 {
|
||||
t.Fatalf("expected 2 MX records, got %d", len(recs))
|
||||
}
|
||||
if len(extra) == 0 {
|
||||
t.Fatalf("expected extra address records for MX")
|
||||
}
|
||||
if !lookedUp {
|
||||
t.Fatalf("expected external lookup for alias.external.")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSOA(t *testing.T) {
|
||||
b := &mockBackend{minTTL: 30, serial: 1234}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("example.org.", dns.TypeSOA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
recs, err := SOA(context.Background(), b, "example.org.", state, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(recs) != 1 {
|
||||
t.Fatalf("expected 1 SOA, got %d", len(recs))
|
||||
}
|
||||
soa, ok := recs[0].(*dns.SOA)
|
||||
if !ok {
|
||||
t.Fatalf("expected SOA, got %T", recs[0])
|
||||
}
|
||||
if soa.Hdr.Ttl != 30 {
|
||||
t.Fatalf("expected ttl 30, got %d", soa.Hdr.Ttl)
|
||||
}
|
||||
if soa.Ns != "ns.dns.example.org." {
|
||||
t.Fatalf("unexpected NS in SOA: %s", soa.Ns)
|
||||
}
|
||||
if soa.Mbox != "hostmaster.example.org." {
|
||||
t.Fatalf("unexpected Mbox in SOA: %s", soa.Mbox)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBackendError(t *testing.T) {
|
||||
b := &mockBackend{}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("example.org.", dns.TypeA)
|
||||
w := &test.ResponseWriter{}
|
||||
state := request.Request{Req: req, W: w}
|
||||
code, err := BackendError(context.Background(), b, "example.org.", dns.RcodeServerFailure, state, fmt.Errorf("err"), Options{})
|
||||
if code != dns.RcodeSuccess {
|
||||
t.Fatalf("expected RcodeSuccess, got %d", code)
|
||||
}
|
||||
if err == nil {
|
||||
t.Fatal("expected error to be returned")
|
||||
}
|
||||
}
|
||||
|
||||
func TestCheckForApex(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
if state.QName() == "apex.dns.example.org." {
|
||||
return []msg.Service{{Host: "1.2.3.4", TTL: 10}}, nil
|
||||
}
|
||||
return []msg.Service{{Host: "::1", TTL: 20}}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("example.org.", dns.TypeA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
services, err := checkForApex(context.Background(), b, "example.org.", state, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(services) != 1 || services[0].Host != "1.2.3.4" {
|
||||
t.Fatalf("expected apex services, got %+v", services)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCheckForApexFallback(t *testing.T) {
|
||||
b := &mockBackend{
|
||||
mockServices: func(ctx context.Context, state request.Request, exact bool, opt Options) ([]msg.Service, error) {
|
||||
if state.QName() == "apex.dns.example.org." {
|
||||
return nil, dns.ErrRcode
|
||||
}
|
||||
return []msg.Service{{Host: "::1", TTL: 20}}, nil
|
||||
},
|
||||
}
|
||||
req := new(dns.Msg)
|
||||
req.SetQuestion("example.org.", dns.TypeA)
|
||||
state := request.Request{Req: req, W: &test.ResponseWriter{}}
|
||||
services, err := checkForApex(context.Background(), b, "example.org.", state, Options{})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(services) != 1 || services[0].Host != "::1" {
|
||||
t.Fatalf("expected fallback services, got %+v", services)
|
||||
}
|
||||
}
|
||||
|
||||
func TestIsDuplicate(t *testing.T) {
|
||||
m := map[item]struct{}{}
|
||||
if isDuplicate(m, "name.", "", 53) {
|
||||
t.Fatal("unexpected duplicate on first insert (port)")
|
||||
}
|
||||
if !isDuplicate(m, "name.", "", 53) {
|
||||
t.Fatal("expected duplicate on second insert (port)")
|
||||
}
|
||||
if isDuplicate(m, "name.", "1.2.3.4", 0) {
|
||||
t.Fatal("unexpected duplicate on first insert (addr)")
|
||||
}
|
||||
if !isDuplicate(m, "name.", "1.2.3.4", 0) {
|
||||
t.Fatal("expected duplicate on second insert (addr)")
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user