mw/k8s: use request.Request in parseRequest

1 files changed, 11 insertions, 2 deletions

diff --git a/middleware/kubernetes/parse_test.go b/middleware/kubernetes/parse_test.go
index 164bf295e..010b9e542 100644
--- a/middleware/kubernetes/parse_test.go
+++ b/middleware/kubernetes/parse_test.go
@@ -3,6 +3,7 @@ package kubernetes
 import (
 	"testing"
 
+	"github.com/coredns/coredns/request"
 	"github.com/miekg/dns"
 )
 
@@ -35,7 +36,11 @@ func TestParseRequest(t *testing.T) {
 		},
 	}
 	for i, tc := range tests {
-		r, e := k.parseRequest(tc.query, tc.qtype, zone)
+		m := new(dns.Msg)
+		m.SetQuestion(tc.query, tc.qtype)
+		state := request.Request{Zone: zone, Req: m}
+
+		r, e := k.parseRequest(state)
 		if e != nil {
 			t.Errorf("Test %d, expected no error, got '%v'.", i, e)
 		}
@@ -58,7 +63,11 @@ func TestParseInvalidRequest(t *testing.T) {
 	}
 
 	for query, qtype := range invalid {
-		if _, e := k.parseRequest(query, qtype, zone); e == nil {
+		m := new(dns.Msg)
+		m.SetQuestion(query, qtype)
+		state := request.Request{Zone: zone, Req: m}
+
+		if _, e := k.parseRequest(state); e == nil {
 			t.Errorf("Expected error from %s:%d, got none", query, qtype)
 		}
 	}