627687b11f
* mw/kubernetes: remove zone from parseRequest State has the zone info as well, so don't need to have it in parseRequest anymore. * Fix up tests * improve test coverage
79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/coredns/coredns/request"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func TestParseRequest(t *testing.T) {
|
|
k := New([]string{zone})
|
|
|
|
tests := []struct {
|
|
query string
|
|
qtype uint16
|
|
expected string // output from r.String()
|
|
}{
|
|
{
|
|
// valid SRV request
|
|
"_http._tcp.webs.mynamespace.svc.inter.webs.test.", dns.TypeSRV,
|
|
"http.tcp..webs.mynamespace.svc",
|
|
},
|
|
{
|
|
// wildcard acceptance
|
|
"*.any.*.any.svc.inter.webs.test.", dns.TypeSRV,
|
|
"*.any..*.any.svc",
|
|
},
|
|
{
|
|
// A request of endpoint
|
|
"1-2-3-4.webs.mynamespace.svc.inter.webs.test.", dns.TypeA,
|
|
"..1-2-3-4.webs.mynamespace.svc",
|
|
},
|
|
{
|
|
|
|
// 3 segments
|
|
"webs.mynamespace.svc.inter.webs.test.", dns.TypeSRV,
|
|
"*...webs.mynamespace.svc",
|
|
},
|
|
}
|
|
for i, tc := range tests {
|
|
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)
|
|
}
|
|
rs := r.String()
|
|
if rs != tc.expected {
|
|
t.Errorf("Test %d, expected (stringyfied) recordRequest: %s, got %s", i, tc.expected, rs)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseInvalidRequest(t *testing.T) {
|
|
k := New([]string{zone})
|
|
|
|
invalid := map[string]uint16{
|
|
"_http._tcp.webs.mynamespace.svc.inter.webs.test.": dns.TypeA, // A requests cannot have port or protocol
|
|
"_http._pcp.webs.mynamespace.svc.inter.webs.test.": dns.TypeSRV, // SRV protocol must be tcp or udp
|
|
"_http._tcp.ep.webs.ns.svc.inter.webs.test.": dns.TypeSRV, // SRV requests cannot have an endpoint
|
|
"_*._*.webs.mynamespace.svc.inter.webs.test.": dns.TypeSRV, // SRV request with invalid wildcards
|
|
|
|
}
|
|
|
|
for query, qtype := range invalid {
|
|
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)
|
|
}
|
|
}
|
|
}
|
|
|
|
const zone = "intern.webs.tests."
|