62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/coredns/coredns/request"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func TestParseRequest(t *testing.T) {
|
|
tests := []struct {
|
|
query string
|
|
expected string // output from r.String()
|
|
}{
|
|
// valid SRV request
|
|
{"_http._tcp.webs.mynamespace.svc.inter.webs.tests.", "http.tcp..webs.mynamespace.svc"},
|
|
// A request of endpoint
|
|
{"1-2-3-4.webs.mynamespace.svc.inter.webs.tests.", "..1-2-3-4.webs.mynamespace.svc"},
|
|
// bare zone
|
|
{"inter.webs.tests.", "....."},
|
|
// bare svc type
|
|
{"svc.inter.webs.tests.", "....."},
|
|
// bare pod type
|
|
{"pod.inter.webs.tests.", "....."},
|
|
// SRV request with empty segments
|
|
{"..webs.mynamespace.svc.inter.webs.tests.", "...webs.mynamespace.svc"},
|
|
}
|
|
for i, tc := range tests {
|
|
m := new(dns.Msg)
|
|
m.SetQuestion(tc.query, dns.TypeA)
|
|
state := request.Request{Zone: zone, Req: m}
|
|
|
|
r, e := parseRequest(state.Name(), state.Zone)
|
|
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 (stringified) recordRequest: %s, got %s", i, tc.expected, rs)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseInvalidRequest(t *testing.T) {
|
|
invalid := []string{
|
|
"webs.mynamespace.pood.inter.webs.test.", // Request must be for pod or svc subdomain.
|
|
"too.long.for.what.I.am.trying.to.pod.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 := parseRequest(state.Name(), state.Zone); e == nil {
|
|
t.Errorf("Test %d: expected error from %s, got none", i, query)
|
|
}
|
|
}
|
|
}
|
|
|
|
const zone = "inter.webs.tests."
|