mirror of
https://github.com/kubernetes-sigs/external-dns.git
synced 2025-08-05 17:16:59 +02:00
381 lines
11 KiB
Go
381 lines
11 KiB
Go
/*
|
|
Copyright 2021 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package source
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
fakeDynamic "k8s.io/client-go/dynamic/fake"
|
|
fakeKube "k8s.io/client-go/kubernetes/fake"
|
|
"sigs.k8s.io/external-dns/endpoint"
|
|
)
|
|
|
|
// This is a compile-time validation that kongTCPIngressSource is a Source.
|
|
var _ Source = &kongTCPIngressSource{}
|
|
|
|
const defaultKongNamespace = "kong"
|
|
|
|
func TestKongTCPIngressEndpoints(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
for _, ti := range []struct {
|
|
title string
|
|
tcpProxy TCPIngress
|
|
ignoreHostnameAnnotation bool
|
|
expected []*endpoint.Endpoint
|
|
}{
|
|
{
|
|
title: "TCPIngress with hostname annotation",
|
|
tcpProxy: TCPIngress{
|
|
TypeMeta: metav1.TypeMeta{
|
|
APIVersion: kongGroupdVersionResource.GroupVersion().String(),
|
|
Kind: "TCPIngress",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "tcp-ingress-annotation",
|
|
Namespace: defaultKongNamespace,
|
|
Annotations: map[string]string{
|
|
"external-dns.alpha.kubernetes.io/hostname": "a.example.com",
|
|
"kubernetes.io/ingress.class": "kong",
|
|
},
|
|
},
|
|
Spec: tcpIngressSpec{
|
|
Rules: []tcpIngressRule{
|
|
{
|
|
Port: 30000,
|
|
},
|
|
{
|
|
Port: 30001,
|
|
},
|
|
},
|
|
},
|
|
Status: tcpIngressStatus{
|
|
LoadBalancer: corev1.LoadBalancerStatus{
|
|
Ingress: []corev1.LoadBalancerIngress{
|
|
{
|
|
Hostname: "a691234567a314e71861a4303f06a3bd-1291189659.us-east-1.elb.amazonaws.com",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: []*endpoint.Endpoint{
|
|
{
|
|
DNSName: "a.example.com",
|
|
Targets: []string{"a691234567a314e71861a4303f06a3bd-1291189659.us-east-1.elb.amazonaws.com"},
|
|
RecordType: endpoint.RecordTypeCNAME,
|
|
RecordTTL: 0,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-annotation",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
title: "TCPIngress using SNI",
|
|
tcpProxy: TCPIngress{
|
|
TypeMeta: metav1.TypeMeta{
|
|
APIVersion: kongGroupdVersionResource.GroupVersion().String(),
|
|
Kind: "TCPIngress",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "tcp-ingress-sni",
|
|
Namespace: defaultKongNamespace,
|
|
Annotations: map[string]string{
|
|
"kubernetes.io/ingress.class": "kong",
|
|
},
|
|
},
|
|
Spec: tcpIngressSpec{
|
|
Rules: []tcpIngressRule{
|
|
{
|
|
Port: 30002,
|
|
Host: "b.example.com",
|
|
},
|
|
{
|
|
Port: 30003,
|
|
Host: "c.example.com",
|
|
},
|
|
},
|
|
},
|
|
Status: tcpIngressStatus{
|
|
LoadBalancer: corev1.LoadBalancerStatus{
|
|
Ingress: []corev1.LoadBalancerIngress{
|
|
{
|
|
Hostname: "a123456769a314e71861a4303f06a3bd-1291189659.us-east-1.elb.amazonaws.com",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: []*endpoint.Endpoint{
|
|
{
|
|
DNSName: "b.example.com",
|
|
Targets: []string{"a123456769a314e71861a4303f06a3bd-1291189659.us-east-1.elb.amazonaws.com"},
|
|
RecordType: endpoint.RecordTypeCNAME,
|
|
RecordTTL: 0,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-sni",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
{
|
|
DNSName: "c.example.com",
|
|
Targets: []string{"a123456769a314e71861a4303f06a3bd-1291189659.us-east-1.elb.amazonaws.com"},
|
|
RecordType: endpoint.RecordTypeCNAME,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-sni",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
title: "TCPIngress with hostname annotation and using SNI",
|
|
tcpProxy: TCPIngress{
|
|
TypeMeta: metav1.TypeMeta{
|
|
APIVersion: kongGroupdVersionResource.GroupVersion().String(),
|
|
Kind: "TCPIngress",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "tcp-ingress-both",
|
|
Namespace: defaultKongNamespace,
|
|
Annotations: map[string]string{
|
|
"external-dns.alpha.kubernetes.io/hostname": "d.example.com",
|
|
"kubernetes.io/ingress.class": "kong",
|
|
},
|
|
},
|
|
Spec: tcpIngressSpec{
|
|
Rules: []tcpIngressRule{
|
|
{
|
|
Port: 30004,
|
|
Host: "e.example.com",
|
|
},
|
|
{
|
|
Port: 30005,
|
|
Host: "f.example.com",
|
|
},
|
|
},
|
|
},
|
|
Status: tcpIngressStatus{
|
|
LoadBalancer: corev1.LoadBalancerStatus{
|
|
Ingress: []corev1.LoadBalancerIngress{
|
|
{
|
|
Hostname: "a12e71861a4303f063456769a314a3bd-1291189659.us-east-1.elb.amazonaws.com",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: []*endpoint.Endpoint{
|
|
{
|
|
DNSName: "d.example.com",
|
|
Targets: []string{"a12e71861a4303f063456769a314a3bd-1291189659.us-east-1.elb.amazonaws.com"},
|
|
RecordType: endpoint.RecordTypeCNAME,
|
|
RecordTTL: 0,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-both",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
{
|
|
DNSName: "e.example.com",
|
|
Targets: []string{"a12e71861a4303f063456769a314a3bd-1291189659.us-east-1.elb.amazonaws.com"},
|
|
RecordType: endpoint.RecordTypeCNAME,
|
|
RecordTTL: 0,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-both",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
{
|
|
DNSName: "f.example.com",
|
|
Targets: []string{"a12e71861a4303f063456769a314a3bd-1291189659.us-east-1.elb.amazonaws.com"},
|
|
RecordType: endpoint.RecordTypeCNAME,
|
|
RecordTTL: 0,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-both",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
title: "TCPIngress ignoring hostname annotation",
|
|
tcpProxy: TCPIngress{
|
|
TypeMeta: metav1.TypeMeta{
|
|
APIVersion: kongGroupdVersionResource.GroupVersion().String(),
|
|
Kind: "TCPIngress",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "tcp-ingress-both",
|
|
Namespace: defaultKongNamespace,
|
|
Annotations: map[string]string{
|
|
"external-dns.alpha.kubernetes.io/hostname": "d.example.com",
|
|
"kubernetes.io/ingress.class": "kong",
|
|
},
|
|
},
|
|
Spec: tcpIngressSpec{
|
|
Rules: []tcpIngressRule{
|
|
{
|
|
Port: 30004,
|
|
Host: "e.example.com",
|
|
},
|
|
{
|
|
Port: 30005,
|
|
Host: "f.example.com",
|
|
},
|
|
},
|
|
},
|
|
Status: tcpIngressStatus{
|
|
LoadBalancer: corev1.LoadBalancerStatus{
|
|
Ingress: []corev1.LoadBalancerIngress{
|
|
{
|
|
Hostname: "a12e71861a4303f063456769a314a3bd-1291189659.us-east-1.elb.amazonaws.com",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
ignoreHostnameAnnotation: true,
|
|
expected: []*endpoint.Endpoint{
|
|
{
|
|
DNSName: "e.example.com",
|
|
Targets: []string{"a12e71861a4303f063456769a314a3bd-1291189659.us-east-1.elb.amazonaws.com"},
|
|
RecordType: endpoint.RecordTypeCNAME,
|
|
RecordTTL: 0,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-both",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
{
|
|
DNSName: "f.example.com",
|
|
Targets: []string{"a12e71861a4303f063456769a314a3bd-1291189659.us-east-1.elb.amazonaws.com"},
|
|
RecordType: endpoint.RecordTypeCNAME,
|
|
RecordTTL: 0,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-both",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
title: "TCPIngress with target annotation",
|
|
tcpProxy: TCPIngress{
|
|
TypeMeta: metav1.TypeMeta{
|
|
APIVersion: kongGroupdVersionResource.GroupVersion().String(),
|
|
Kind: "TCPIngress",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "tcp-ingress-sni",
|
|
Namespace: defaultKongNamespace,
|
|
Annotations: map[string]string{
|
|
"kubernetes.io/ingress.class": "kong",
|
|
"external-dns.alpha.kubernetes.io/target": "203.2.45.7",
|
|
},
|
|
},
|
|
Spec: tcpIngressSpec{
|
|
Rules: []tcpIngressRule{
|
|
{
|
|
Port: 30002,
|
|
Host: "b.example.com",
|
|
},
|
|
{
|
|
Port: 30003,
|
|
Host: "c.example.com",
|
|
},
|
|
},
|
|
},
|
|
Status: tcpIngressStatus{
|
|
LoadBalancer: corev1.LoadBalancerStatus{
|
|
Ingress: []corev1.LoadBalancerIngress{
|
|
{
|
|
Hostname: "a123456769a314e71861a4303f06a3bd-1291189659.us-east-1.elb.amazonaws.com",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expected: []*endpoint.Endpoint{
|
|
{
|
|
DNSName: "b.example.com",
|
|
Targets: []string{"203.2.45.7"},
|
|
RecordType: endpoint.RecordTypeA,
|
|
RecordTTL: 0,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-sni",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
{
|
|
DNSName: "c.example.com",
|
|
Targets: []string{"203.2.45.7"},
|
|
RecordType: endpoint.RecordTypeA,
|
|
Labels: endpoint.Labels{
|
|
"resource": "tcpingress/kong/tcp-ingress-sni",
|
|
},
|
|
ProviderSpecific: endpoint.ProviderSpecific{},
|
|
},
|
|
},
|
|
},
|
|
} {
|
|
ti := ti
|
|
t.Run(ti.title, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
fakeKubernetesClient := fakeKube.NewSimpleClientset()
|
|
scheme := runtime.NewScheme()
|
|
scheme.AddKnownTypes(kongGroupdVersionResource.GroupVersion(), &TCPIngress{}, &TCPIngressList{})
|
|
fakeDynamicClient := fakeDynamic.NewSimpleDynamicClient(scheme)
|
|
|
|
tcpi := unstructured.Unstructured{}
|
|
|
|
tcpIngressAsJSON, err := json.Marshal(ti.tcpProxy)
|
|
assert.NoError(t, err)
|
|
|
|
assert.NoError(t, tcpi.UnmarshalJSON(tcpIngressAsJSON))
|
|
|
|
// Create proxy resources
|
|
_, err = fakeDynamicClient.Resource(kongGroupdVersionResource).Namespace(defaultKongNamespace).Create(context.Background(), &tcpi, metav1.CreateOptions{})
|
|
assert.NoError(t, err)
|
|
|
|
source, err := NewKongTCPIngressSource(context.TODO(), fakeDynamicClient, fakeKubernetesClient, defaultKongNamespace, "kubernetes.io/ingress.class=kong", ti.ignoreHostnameAnnotation)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, source)
|
|
|
|
count := &unstructured.UnstructuredList{}
|
|
for len(count.Items) < 1 {
|
|
count, _ = fakeDynamicClient.Resource(kongGroupdVersionResource).Namespace(defaultKongNamespace).List(context.Background(), metav1.ListOptions{})
|
|
}
|
|
|
|
endpoints, err := source.Endpoints(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Len(t, endpoints, len(ti.expected))
|
|
assert.Equal(t, ti.expected, endpoints)
|
|
})
|
|
}
|
|
}
|