7e56cc74e5
* middleware/kubernetes: pull TXT out of parseRequest Put the TXT handling one layer higher and remove it from parseRequest. Also rename the podsvc field in there to podOrSvc. Now that it isn't used anymore for TXT record (dns-version) that was put in there. We can make this a boolean (in a future PR). Make parseRequest get an optional Zone that is from state.Zone and use that instead of its own code. Removed some tests and other smaller cleanups. Fixes #836 * add this reverse * another check * readd * Rename to kPod and kService for some clarity
119 lines
3 KiB
Go
119 lines
3 KiB
Go
package kubernetes
|
|
|
|
import "testing"
|
|
import "net"
|
|
|
|
import "k8s.io/client-go/1.5/pkg/api"
|
|
import "github.com/miekg/dns"
|
|
|
|
func TestRecordForNS(t *testing.T) {
|
|
k := Kubernetes{Zones: []string{"inter.webs.test."}}
|
|
corednsRecord.Hdr.Name = "coredns.kube-system."
|
|
corednsRecord.A = net.IP("1.2.3.4")
|
|
r, _ := k.parseRequest("inter.webs.test.", dns.TypeNS, "inter.webs.test.")
|
|
|
|
expected := "/coredns/test/webs/inter/kube-system/coredns"
|
|
svc := k.recordsForNS(r)
|
|
if svc.Key != expected {
|
|
t.Errorf("Expected result '%v'. Instead got result '%v'.", expected, svc.Key)
|
|
}
|
|
}
|
|
|
|
func TestDefaultNSMsg(t *testing.T) {
|
|
k := Kubernetes{Zones: []string{"inter.webs.test."}}
|
|
corednsRecord.Hdr.Name = "coredns.kube-system."
|
|
corednsRecord.A = net.IP("1.2.3.4")
|
|
r, _ := k.parseRequest("ns.dns.inter.webs.test.", dns.TypeA, "inter.webs.test.")
|
|
|
|
expected := "/coredns/test/webs/inter/dns/ns"
|
|
svc := k.defaultNSMsg(r)
|
|
if svc.Key != expected {
|
|
t.Errorf("Expected result '%v'. Instead got result '%v'.", expected, svc.Key)
|
|
}
|
|
}
|
|
|
|
func TestIsDefaultNS(t *testing.T) {
|
|
k := Kubernetes{Zones: []string{"inter.webs.test."}}
|
|
r, _ := k.parseRequest("ns.dns.inter.webs.test", dns.TypeA, "inter.webs.test.")
|
|
|
|
var name string
|
|
var expected bool
|
|
|
|
name = "ns.dns.inter.webs.test."
|
|
expected = true
|
|
if isDefaultNS(name, r) != expected {
|
|
t.Errorf("Expected IsDefaultNS('%v') to be '%v'.", name, expected)
|
|
}
|
|
name = "ns.dns.blah.inter.webs.test"
|
|
expected = false
|
|
if isDefaultNS(name, r) != expected {
|
|
t.Errorf("Expected IsDefaultNS('%v') to be '%v'.", name, expected)
|
|
}
|
|
}
|
|
|
|
type APIConnTest struct{}
|
|
|
|
func (APIConnTest) Run() { return }
|
|
func (APIConnTest) Stop() error { return nil }
|
|
func (APIConnTest) PodIndex(string) []interface{} { return nil }
|
|
|
|
func (APIConnTest) ServiceList() []*api.Service {
|
|
svc := api.Service{
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "dns-service",
|
|
Namespace: "kube-system",
|
|
},
|
|
Spec: api.ServiceSpec{
|
|
ClusterIP: "10.0.0.111",
|
|
},
|
|
}
|
|
|
|
return []*api.Service{&svc}
|
|
|
|
}
|
|
|
|
func (APIConnTest) EndpointsList() api.EndpointsList {
|
|
return api.EndpointsList{
|
|
Items: []api.Endpoints{
|
|
{
|
|
Subsets: []api.EndpointSubset{
|
|
{
|
|
Addresses: []api.EndpointAddress{
|
|
{
|
|
IP: "172.0.40.10",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
ObjectMeta: api.ObjectMeta{
|
|
Name: "dns-service",
|
|
Namespace: "kube-system",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (APIConnTest) GetNodeByName(name string) (api.Node, error) { return api.Node{}, nil }
|
|
|
|
func TestDoCoreDNSRecord(t *testing.T) {
|
|
|
|
corednsRecord = dns.A{}
|
|
k := Kubernetes{Zones: []string{"inter.webs.test"}}
|
|
|
|
k.interfaceAddrsFunc = func() net.IP { return net.ParseIP("172.0.40.10") }
|
|
|
|
k.APIConn = &APIConnTest{}
|
|
|
|
cdr := k.coreDNSRecord()
|
|
|
|
expected := "10.0.0.111"
|
|
|
|
if cdr.A.String() != expected {
|
|
t.Errorf("Expected A to be '%v', got '%v'", expected, cdr.A.String())
|
|
}
|
|
expected = "dns-service.kube-system.svc."
|
|
if cdr.Hdr.Name != expected {
|
|
t.Errorf("Expected Hdr.Name to be '%v', got '%v'", expected, cdr.Hdr.Name)
|
|
}
|
|
}
|