1 files changed, 147 insertions, 0 deletions
diff --git a/middleware/etcd/debug_test.go b/middleware/etcd/debug_test.go
new file mode 100644
index 000000000..9e50b6930
--- /dev/null
+++ b/middleware/etcd/debug_test.go
@@ -0,0 +1,147 @@
+package etcd
+
+import (
+	"sort"
+	"testing"
+
+	"github.com/miekg/coredns/middleware"
+	"github.com/miekg/coredns/middleware/etcd/msg"
+	"github.com/miekg/coredns/middleware/test"
+
+	"github.com/miekg/dns"
+)
+
+func TestisDebug(t *testing.T) {
+	if ok := isDebug("o-o.debug.miek.nl."); ok != "miek.nl." {
+		t.Errorf("expected o-o.debug.miek.nl. to be debug")
+	}
+	if ok := isDebug("o-o.Debug.miek.nl."); ok != "miek.nl." {
+		t.Errorf("expected o-o.Debug.miek.nl. to be debug")
+	}
+	if ok := isDebug("i-o.Debug.miek.nl."); ok != "" {
+		t.Errorf("expected i-o.Debug.miek.nl. to be non-debug")
+	}
+	if ok := isDebug("i-o.Debug."); ok != "" {
+		t.Errorf("expected o-o.Debug. to be non-debug")
+	}
+}
+
+func TestDebugLookup(t *testing.T) {
+	for _, serv := range servicesDebug {
+		set(t, etc, serv.Key, 0, serv)
+		defer delete(t, etc, serv.Key)
+	}
+	etc.Debug = true
+	defer func() { etc.Debug = false }()
+	for _, tc := range dnsTestCasesDebug {
+		m := tc.Msg()
+
+		rec := middleware.NewResponseRecorder(&test.ResponseWriter{})
+		_, err := etc.ServeDNS(ctx, rec, m)
+		if err != nil {
+			t.Errorf("expected no error, got %v\n", err)
+			continue
+		}
+		resp := rec.Msg()
+
+		sort.Sort(test.RRSet(resp.Answer))
+		sort.Sort(test.RRSet(resp.Ns))
+		sort.Sort(test.RRSet(resp.Extra))
+
+		if !test.Header(t, tc, resp) {
+			t.Logf("%v\n", resp)
+			continue
+		}
+		if !test.Section(t, tc, test.Answer, resp.Answer) {
+			t.Logf("%v\n", resp)
+		}
+		if !test.Section(t, tc, test.Ns, resp.Ns) {
+			t.Logf("%v\n", resp)
+		}
+		if !test.Section(t, tc, test.Extra, resp.Extra) {
+			t.Logf("%v\n", resp)
+		}
+	}
+}
+
+func TestDebugLookupFalse(t *testing.T) {
+	for _, serv := range servicesDebug {
+		set(t, etc, serv.Key, 0, serv)
+		defer delete(t, etc, serv.Key)
+	}
+	for _, tc := range dnsTestCasesDebugFalse {
+		m := tc.Msg()
+
+		rec := middleware.NewResponseRecorder(&test.ResponseWriter{})
+		_, err := etc.ServeDNS(ctx, rec, m)
+		if err != nil {
+			t.Errorf("expected no error, got %v\n", err)
+			continue
+		}
+		resp := rec.Msg()
+
+		sort.Sort(test.RRSet(resp.Answer))
+		sort.Sort(test.RRSet(resp.Ns))
+		sort.Sort(test.RRSet(resp.Extra))
+
+		if !test.Header(t, tc, resp) {
+			t.Logf("%v\n", resp)
+			continue
+		}
+		if !test.Section(t, tc, test.Answer, resp.Answer) {
+			t.Logf("%v\n", resp)
+		}
+		if !test.Section(t, tc, test.Ns, resp.Ns) {
+			t.Logf("%v\n", resp)
+		}
+		if !test.Section(t, tc, test.Extra, resp.Extra) {
+			t.Logf("%v\n", resp)
+		}
+	}
+}
+
+var servicesDebug = []*msg.Service{
+	{Host: "127.0.0.1", Key: "a.dom.skydns.test."},
+	{Host: "127.0.0.2", Key: "b.sub.dom.skydns.test."},
+}
+
+var dnsTestCasesDebug = []test.Case{
+	{
+		Qname: "o-o.debug.dom.skydns.test.", Qtype: dns.TypeA,
+		Answer: []dns.RR{
+			test.A("dom.skydns.test. 300 IN A 127.0.0.1"),
+			test.A("dom.skydns.test. 300 IN A 127.0.0.2"),
+		},
+		Extra: []dns.RR{
+			test.TXT(`skydns.test.skydns.dom.a.	300	CH	TXT	"127.0.0.1:0(10,0,,false)[0,]"`),
+			test.TXT(`skydns.test.skydns.dom.sub.b.	300	CH	TXT	"127.0.0.2:0(10,0,,false)[0,]"`),
+		},
+	},
+	{
+		Qname: "o-o.debug.dom.skydns.test.", Qtype: dns.TypeTXT,
+		Ns: []dns.RR{
+			test.SOA("skydns.test. 300 IN SOA ns.dns.skydns.test. hostmaster.skydns.test. 1463943291 7200 1800 86400 60"),
+		},
+		Extra: []dns.RR{
+			test.TXT(`skydns.test.skydns.dom.a.	300	CH	TXT	"127.0.0.1:0(10,0,,false)[0,]"`),
+			test.TXT(`skydns.test.skydns.dom.sub.b.	300	CH	TXT	"127.0.0.2:0(10,0,,false)[0,]"`),
+		},
+	},
+}
+
+var dnsTestCasesDebugFalse = []test.Case{
+	{
+		Qname: "o-o.debug.dom.skydns.test.", Qtype: dns.TypeA,
+		Rcode: dns.RcodeNameError,
+		Ns: []dns.RR{
+			test.SOA("skydns.test. 300 IN SOA ns.dns.skydns.test. hostmaster.skydns.test. 1463943291 7200 1800 86400 60"),
+		},
+	},
+	{
+		Qname: "o-o.debug.dom.skydns.test.", Qtype: dns.TypeTXT,
+		Rcode: dns.RcodeNameError,
+		Ns: []dns.RR{
+			test.SOA("skydns.test. 300 IN SOA ns.dns.skydns.test. hostmaster.skydns.test. 1463943291 7200 1800 86400 60"),
+		},
+	},
+}