coredns/plugin/kubernetes/ns_test.go

package kubernetes

import (
	"testing"

	meta "k8s.io/apimachinery/pkg/apis/meta/v1"
	api "k8s.io/client-go/pkg/api/v1"
)

type APIConnTest struct{}

func (APIConnTest) Run()                       { return }
func (APIConnTest) Stop() error                { return nil }
func (APIConnTest) PodIndex(string) []*api.Pod { return nil }

func (APIConnTest) ServiceList() []*api.Service {
	svcs := []*api.Service{
		{
			ObjectMeta: meta.ObjectMeta{
				Name:      "dns-service",
				Namespace: "kube-system",
			},
			Spec: api.ServiceSpec{
				ClusterIP: "10.0.0.111",
			},
		},
	}
	return svcs
}

func (APIConnTest) EndpointsList() []*api.Endpoints {
	eps := []*api.Endpoints{
		{
			Subsets: []api.EndpointSubset{
				{
					Addresses: []api.EndpointAddress{
						{
							IP: "127.0.0.1",
						},
					},
				},
			},
			ObjectMeta: meta.ObjectMeta{
				Name:      "dns-service",
				Namespace: "kube-system",
			},
		},
	}
	return eps
}

func (APIConnTest) GetNodeByName(name string) (*api.Node, error) { return &api.Node{}, nil }

func TestNsAddr(t *testing.T) {

	k := New([]string{"inter.webs.test."})
	k.APIConn = &APIConnTest{}

	cdr := k.nsAddr()
	expected := "10.0.0.111"

	if cdr.A.String() != expected {
		t.Errorf("Expected A to be %q, got %q", expected, cdr.A.String())
	}
	expected = "dns-service.kube-system.svc."
	if cdr.Hdr.Name != expected {
		t.Errorf("Expected Hdr.Name to be %q, got %q", expected, cdr.Hdr.Name)
	}
}