From 4d55f9038886d9e29d1457a2cc8ffbbc007ee05b Mon Sep 17 00:00:00 2001
From: Miek Gieben <miek@miek.nl>
Date: Tue, 18 Oct 2016 12:05:19 +0100
Subject: [PATCH] middleware/etcd: more tests (#342)

Add a test which enables caching and debug queries and make sure
the debug query does not overwrite the cache.
---
 test/etcd_cache_debug_test.go | 92 +++++++++++++++++++++++++++++++++++
 test/etcd_test.go             |  2 +-
 2 files changed, 93 insertions(+), 1 deletion(-)
 create mode 100644 test/etcd_cache_debug_test.go

diff --git a/test/etcd_cache_debug_test.go b/test/etcd_cache_debug_test.go
new file mode 100644
index 000000000..5343ad620
--- /dev/null
+++ b/test/etcd_cache_debug_test.go
@@ -0,0 +1,92 @@
+// +build etcd
+
+package test
+
+import (
+	"io/ioutil"
+	"log"
+	"testing"
+
+	"github.com/miekg/coredns/middleware/etcd/msg"
+	"github.com/miekg/coredns/middleware/proxy"
+	"github.com/miekg/coredns/middleware/test"
+	"github.com/miekg/coredns/request"
+
+	"github.com/miekg/dns"
+	"golang.org/x/net/context"
+)
+
+// uses some stuff from etcd_tests.go
+
+func TestEtcdCacheAndDebug(t *testing.T) {
+	corefile := `.:0 {
+    etcd skydns.test {
+        path /skydns
+	debug
+    }
+    cache skydns.test
+}`
+
+	ex, err := CoreDNSServer(corefile)
+	if err != nil {
+		t.Fatalf("Could not get CoreDNS serving instance: %s", err)
+	}
+
+	udp, _ := CoreDNSServerPorts(ex, 0)
+	if udp == "" {
+		t.Fatalf("Could not get UDP listening port")
+	}
+	defer ex.Stop()
+
+	etc := etcdMiddleware()
+	log.SetOutput(ioutil.Discard)
+
+	var ctx = context.TODO()
+	for _, serv := range servicesCacheTest {
+		set(ctx, t, etc, serv.Key, 0, serv)
+		defer delete(ctx, t, etc, serv.Key)
+	}
+
+	p := proxy.New([]string{udp})
+	state := request.Request{W: &test.ResponseWriter{}, Req: new(dns.Msg)}
+
+	resp, err := p.Lookup(state, "b.example.skydns.test.", dns.TypeA)
+	if err != nil {
+		t.Errorf("Expected to receive reply, but didn't: %s", err)
+	}
+	checkResponse(t, resp)
+
+	resp, err = p.Lookup(state, "o-o.debug.b.example.skydns.test.", dns.TypeA)
+	if err != nil {
+		t.Errorf("Expected to receive reply, but didn't: %s", err)
+	}
+	checkResponse(t, resp)
+	if len(resp.Extra) != 1 {
+		t.Errorf("Expected one RR in additional section, got: %d", len(resp.Extra))
+	}
+
+	resp, err = p.Lookup(state, "b.example.skydns.test.", dns.TypeA)
+	if err != nil {
+		t.Errorf("Expected to receive reply, but didn't: %s", err)
+	}
+	checkResponse(t, resp)
+	if len(resp.Extra) != 0 {
+		t.Errorf("Expected no RRs in additional section, got: %d", len(resp.Extra))
+	}
+}
+
+func checkResponse(t *testing.T, resp *dns.Msg) {
+	if len(resp.Answer) == 0 {
+		t.Fatal("Expected to at least one RR in the answer section, got none")
+	}
+	if resp.Answer[0].Header().Rrtype != dns.TypeA {
+		t.Errorf("Expected RR to A, got: %d", resp.Answer[0].Header().Rrtype)
+	}
+	if resp.Answer[0].(*dns.A).A.String() != "127.0.0.1" {
+		t.Errorf("Expected 127.0.0.1, got: %d", resp.Answer[0].(*dns.A).A.String())
+	}
+}
+
+var servicesCacheTest = []*msg.Service{
+	{Host: "127.0.0.1", Port: 666, Key: "b.example.skydns.test."},
+}
diff --git a/test/etcd_test.go b/test/etcd_test.go
index 265162a83..a21fa6296 100644
--- a/test/etcd_test.go
+++ b/test/etcd_test.go
@@ -26,7 +26,7 @@ func etcdMiddleware() *etcd.Etcd {
 	}
 	cli, _ := etcdc.New(etcdCfg)
 	client := etcdc.NewKeysAPI(cli)
-	return &etcd.Etcd{Client: client}
+	return &etcd.Etcd{Client: client, PathPrefix: "/skydns"}
 }
 
 // This test starts two coredns servers (and needs etcd). Configure a stubzones in both (that will loop) and