diff options
author | 2017-08-11 12:12:21 +0100 | |
---|---|---|
committer | 2017-08-11 12:12:21 +0100 | |
commit | 6e91408f61bdd78aef8f8e368410793334eb35fc (patch) | |
tree | 153cc070363281a302add2194cfb7756f4639f4d /middleware/kubernetes/parse_test.go | |
parent | b5d2a82ed742f80016868f492f76e86b74645919 (diff) | |
download | coredns-6e91408f61bdd78aef8f8e368410793334eb35fc.tar.gz coredns-6e91408f61bdd78aef8f8e368410793334eb35fc.tar.zst coredns-6e91408f61bdd78aef8f8e368410793334eb35fc.zip |
mw/k8s: use request.Request in parseRequest
Diffstat (limited to 'middleware/kubernetes/parse_test.go')
-rw-r--r-- | middleware/kubernetes/parse_test.go | 13 |
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) } } |