mw/k8s: use request.Request in parseRequest

1 files changed, 10 insertions, 3 deletions

diff --git a/middleware/kubernetes/federation_test.go b/middleware/kubernetes/federation_test.go
index be5069c26..410d2a54e 100644
--- a/middleware/kubernetes/federation_test.go
+++ b/middleware/kubernetes/federation_test.go
@@ -6,6 +6,7 @@ import (
 	"testing"
 
 	"github.com/coredns/coredns/middleware/etcd/msg"
+	"github.com/coredns/coredns/request"
 	"github.com/miekg/dns"
 	"k8s.io/client-go/1.5/pkg/api"
 )
@@ -90,12 +91,18 @@ func TestFederationCNAMERecord(t *testing.T) {
 	k.APIConn = apiConnFedTest{}
 	k.interfaceAddrsFunc = func() net.IP { return net.ParseIP("10.9.8.7") }
 
-	r, _ := k.parseRequest("s1.ns.fed.svc.inter.webs.", dns.TypeA, "inter.webs.")
+	m := new(dns.Msg)
+	state := request.Request{Zone: "inter.webs.", Req: m}
+
+	m.SetQuestion("s1.ns.fed.svc.inter.webs.", dns.TypeA)
+	r, _ := k.parseRequest(state)
 	testFederationCNAMERecord(t, k, r, msg.Service{Key: "/coredns/webs/inter/svc/fed/ns/s1", Host: "s1.ns.fed.svc.fd-az.fd-r.era.tion.com"})
 
-	r, _ = k.parseRequest("ep1.s1.ns.fed.svc.inter.webs.", dns.TypeA, "inter.webs.")
+	m.SetQuestion("ep1.s1.ns.fed.svc.inter.webs.", dns.TypeA)
+	r, _ = k.parseRequest(state)
 	testFederationCNAMERecord(t, k, r, msg.Service{Key: "/coredns/webs/inter/svc/fed/ns/s1/ep1", Host: "ep1.s1.ns.fed.svc.fd-az.fd-r.era.tion.com"})
 
-	r, _ = k.parseRequest("ep1.s1.ns.foo.svc.inter.webs.", dns.TypeA, "inter.webs.")
+	m.SetQuestion("ep1.s1.ns.foo.svc.inter.webs.", dns.TypeA)
+	r, _ = k.parseRequest(state)
 	testFederationCNAMERecord(t, k, r, msg.Service{Key: "", Host: ""})
 }