aboutsummaryrefslogtreecommitdiff
path: root/test/erratic_autopath_test.go
blob: 215928b0344c844e21ff2c2d8c354552d4f1809c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package test

import (
	"testing"

	"github.com/miekg/dns"
)

func TestLookupAutoPathErratic(t *testing.T) {
	corefile := `.:0 {
	erratic
	autopath @erratic
	proxy . 8.8.8.8:53
	debug
    }
`
	i, udp, _, err := CoreDNSServerAndPorts(corefile)
	if err != nil {
		t.Fatalf("Could not get CoreDNS serving instance: %s", err)
	}
	defer i.Stop()

	tests := []struct {
		qname          string
		expectedAnswer string
		expectedType   uint16
	}{
		{"google.com.a.example.org.", "google.com.a.example.org.", dns.TypeCNAME},
		{"google.com.", "google.com.", dns.TypeA},
	}

	for i, tc := range tests {
		m := new(dns.Msg)
		// erratic always returns this search path: "a.example.org.", "b.example.org.", "".
		m.SetQuestion(tc.qname, dns.TypeA)

		r, err := dns.Exchange(m, udp)
		if err != nil {
			t.Fatalf("Test %d, failed to sent query: %q", i, err)
		}
		if len(r.Answer) == 0 {
			t.Fatalf("Test %d, answer section should have RRs", i)
		}
		if x := r.Answer[0].Header().Name; x != tc.expectedAnswer {
			t.Fatalf("Test %d, expected answer %s, got %s", i, tc.expectedAnswer, x)
		}
		if x := r.Answer[0].Header().Rrtype; x != tc.expectedType {
			t.Fatalf("Test %d, expected answer type %d, got %d", i, tc.expectedType, x)
		}
	}
}

func TestAutoPathErraticNotLoaded(t *testing.T) {
	corefile := `.:0 {
	autopath @erratic
	proxy . 8.8.8.8:53
	debug
    }
`
	i, err := CoreDNSServer(corefile)
	if err != nil {
		t.Fatalf("Could not get CoreDNS serving instance: %s", err)
	}

	udp, _ := CoreDNSServerPorts(i, 0)
	if udp == "" {
		t.Fatalf("Could not get UDP listening port")
	}
	defer i.Stop()

	m := new(dns.Msg)
	m.SetQuestion("google.com.a.example.org.", dns.TypeA)
	r, err := dns.Exchange(m, udp)
	if err != nil {
		t.Fatalf("Failed to sent query: %q", err)
	}
	if r.Rcode != dns.RcodeNameError {
		t.Fatalf("Expected NXDOMAIN, got %d", r.Rcode)
	}
}