traefik/provider/kubernetes/builder_endpoint_test.go

package kubernetes

import (
	"testing"

	"github.com/stretchr/testify/assert"
	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/types"
)

func buildEndpoint(opts ...func(*corev1.Endpoints)) *corev1.Endpoints {
	e := &corev1.Endpoints{}
	for _, opt := range opts {
		opt(e)
	}
	return e
}

func eNamespace(value string) func(*corev1.Endpoints) {
	return func(i *corev1.Endpoints) {
		i.Namespace = value
	}
}

func eName(value string) func(*corev1.Endpoints) {
	return func(i *corev1.Endpoints) {
		i.Name = value
	}
}

func eUID(value types.UID) func(*corev1.Endpoints) {
	return func(i *corev1.Endpoints) {
		i.UID = value
	}
}

func subset(opts ...func(*corev1.EndpointSubset)) func(*corev1.Endpoints) {
	return func(e *corev1.Endpoints) {
		s := &corev1.EndpointSubset{}
		for _, opt := range opts {
			opt(s)
		}
		e.Subsets = append(e.Subsets, *s)
	}
}

func eAddresses(opts ...func(*corev1.EndpointAddress)) func(*corev1.EndpointSubset) {
	return func(subset *corev1.EndpointSubset) {
		for _, opt := range opts {
			a := &corev1.EndpointAddress{}
			opt(a)
			subset.Addresses = append(subset.Addresses, *a)
		}
	}
}

func eAddress(ip string) func(*corev1.EndpointAddress) {
	return func(address *corev1.EndpointAddress) {
		address.IP = ip
	}
}

func eAddressWithTargetRef(targetRef, ip string) func(*corev1.EndpointAddress) {
	return func(address *corev1.EndpointAddress) {
		address.TargetRef = &corev1.ObjectReference{Name: targetRef}
		address.IP = ip
	}
}

func ePorts(opts ...func(port *corev1.EndpointPort)) func(*corev1.EndpointSubset) {
	return func(spec *corev1.EndpointSubset) {
		for _, opt := range opts {
			p := &corev1.EndpointPort{}
			opt(p)
			spec.Ports = append(spec.Ports, *p)
		}
	}
}

func ePort(port int32, name string) func(*corev1.EndpointPort) {
	return func(sp *corev1.EndpointPort) {
		sp.Port = port
		sp.Name = name
	}
}

// Test

func TestBuildEndpoint(t *testing.T) {
	actual := buildEndpoint(
		eNamespace("testing"),
		eName("service3"),
		eUID("3"),
		subset(
			eAddresses(eAddress("10.15.0.1")),
			ePorts(
				ePort(8080, "http"),
				ePort(8443, "https"),
			),
		),
		subset(
			eAddresses(eAddress("10.15.0.2")),
			ePorts(
				ePort(9080, "http"),
				ePort(9443, "https"),
			),
		),
	)

	assert.EqualValues(t, sampleEndpoint1(), actual)
}

func sampleEndpoint1() *corev1.Endpoints {
	return &corev1.Endpoints{
		ObjectMeta: metav1.ObjectMeta{
			Name:      "service3",
			UID:       "3",
			Namespace: "testing",
		},
		Subsets: []corev1.EndpointSubset{
			{
				Addresses: []corev1.EndpointAddress{
					{
						IP: "10.15.0.1",
					},
				},
				Ports: []corev1.EndpointPort{
					{
						Name: "http",
						Port: 8080,
					},
					{
						Name: "https",
						Port: 8443,
					},
				},
			},
			{
				Addresses: []corev1.EndpointAddress{
					{
						IP: "10.15.0.2",
					},
				},
				Ports: []corev1.EndpointPort{
					{
						Name: "http",
						Port: 9080,
					},
					{
						Name: "https",
						Port: 9443,
					},
				},
			},
		},
	}
}