118ef3dfa0
Stop looking at the qtype in parseRequest and make k.Namespace a map. Fallout from this is that pkg/strings as it is not used anymore. Also add a few helper functions to make unexposed namespaces easier to see in the code. Add wildcard tests to the middleware tests.
70 lines
1.6 KiB
Go
70 lines
1.6 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",
|
|
},
|
|
}
|
|
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 := []string{
|
|
"webs.mynamespace.pood.inter.webs.test.", // Request must be for pod or svc subdomain.
|
|
"too.long.for.what.I.am.trying.to.do.inter.webs.tests.", // Too long.
|
|
}
|
|
|
|
for i, query := range invalid {
|
|
m := new(dns.Msg)
|
|
m.SetQuestion(query, dns.TypeA)
|
|
state := request.Request{Zone: zone, Req: m}
|
|
|
|
if _, e := k.parseRequest(state); e == nil {
|
|
t.Errorf("Test %d: expected error from %s, got none", i, query)
|
|
}
|
|
}
|
|
}
|
|
|
|
const zone = "intern.webs.tests."
|