60f483e98b
While running make encountered the following error:
```
$ docker run -i -t --rm -v $PWD:/v --net=host -w /v golang:1.12 make
...
...
** presubmit/test-lowercase
plugin/kubernetes/metadata_test.go: t.Errorf("case %d expected metadata %v and got %v", i, tc.Md, md)
** presubmit/test-lowercase: please start with an upper case letter when using t.Error*()
Makefile:62: recipe for target 'presubmit' failed
make: *** [presubmit] Error 1
```
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
126 lines
3 KiB
Go
126 lines
3 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/coredns/coredns/plugin/metadata"
|
|
"github.com/coredns/coredns/plugin/test"
|
|
"github.com/coredns/coredns/request"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
var metadataCases = []struct {
|
|
Qname string
|
|
Qtype uint16
|
|
RemoteIP string
|
|
Md map[string]string
|
|
}{
|
|
{
|
|
Qname: "foo.bar.notapod.cluster.local.", Qtype: dns.TypeA,
|
|
Md: map[string]string{
|
|
"kubernetes/parse-error": "invalid query name",
|
|
},
|
|
},
|
|
{
|
|
Qname: "10-240-0-1.podns.pod.cluster.local.", Qtype: dns.TypeA,
|
|
Md: map[string]string{
|
|
"kubernetes/endpoint": "",
|
|
"kubernetes/kind": "pod",
|
|
"kubernetes/namespace": "podns",
|
|
"kubernetes/port-name": "*",
|
|
"kubernetes/protocol": "*",
|
|
"kubernetes/service": "10-240-0-1",
|
|
"kubernetes/client-namespace": "podns",
|
|
"kubernetes/client-pod-name": "foo",
|
|
},
|
|
},
|
|
{
|
|
Qname: "s.ns.svc.cluster.local.", Qtype: dns.TypeA,
|
|
Md: map[string]string{
|
|
"kubernetes/endpoint": "",
|
|
"kubernetes/kind": "svc",
|
|
"kubernetes/namespace": "ns",
|
|
"kubernetes/port-name": "*",
|
|
"kubernetes/protocol": "*",
|
|
"kubernetes/service": "s",
|
|
"kubernetes/client-namespace": "podns",
|
|
"kubernetes/client-pod-name": "foo",
|
|
},
|
|
},
|
|
{
|
|
Qname: "s.ns.svc.cluster.local.", Qtype: dns.TypeA,
|
|
RemoteIP: "10.10.10.10",
|
|
Md: map[string]string{
|
|
"kubernetes/endpoint": "",
|
|
"kubernetes/kind": "svc",
|
|
"kubernetes/namespace": "ns",
|
|
"kubernetes/port-name": "*",
|
|
"kubernetes/protocol": "*",
|
|
"kubernetes/service": "s",
|
|
},
|
|
},
|
|
{
|
|
Qname: "_http._tcp.s.ns.svc.cluster.local.", Qtype: dns.TypeSRV,
|
|
RemoteIP: "10.10.10.10",
|
|
Md: map[string]string{
|
|
"kubernetes/endpoint": "",
|
|
"kubernetes/kind": "svc",
|
|
"kubernetes/namespace": "ns",
|
|
"kubernetes/port-name": "http",
|
|
"kubernetes/protocol": "tcp",
|
|
"kubernetes/service": "s",
|
|
},
|
|
},
|
|
{
|
|
Qname: "ep.s.ns.svc.cluster.local.", Qtype: dns.TypeA,
|
|
RemoteIP: "10.10.10.10",
|
|
Md: map[string]string{
|
|
"kubernetes/endpoint": "ep",
|
|
"kubernetes/kind": "svc",
|
|
"kubernetes/namespace": "ns",
|
|
"kubernetes/port-name": "*",
|
|
"kubernetes/protocol": "*",
|
|
"kubernetes/service": "s",
|
|
},
|
|
},
|
|
}
|
|
|
|
func mapsDiffer(a, b map[string]string) bool {
|
|
if len(a) != len(b) {
|
|
return true
|
|
}
|
|
|
|
for k, va := range a {
|
|
vb, ok := b[k]
|
|
if !ok || va != vb {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func TestMetadata(t *testing.T) {
|
|
k := New([]string{"cluster.local."})
|
|
k.APIConn = &APIConnServeTest{}
|
|
|
|
for i, tc := range metadataCases {
|
|
ctx := metadata.ContextWithMetadata(context.Background())
|
|
state := request.Request{
|
|
Req: &dns.Msg{Question: []dns.Question{{Name: tc.Qname, Qtype: tc.Qtype}}},
|
|
Zone: "cluster.local.",
|
|
W: &test.ResponseWriter{RemoteIP: tc.RemoteIP},
|
|
}
|
|
|
|
k.Metadata(ctx, state)
|
|
|
|
md := make(map[string]string)
|
|
for _, l := range metadata.Labels(ctx) {
|
|
md[l] = metadata.ValueFunc(ctx, l)()
|
|
}
|
|
if mapsDiffer(tc.Md, md) {
|
|
t.Errorf("Case %d expected metadata %v and got %v", i, tc.Md, md)
|
|
}
|
|
}
|
|
}
|