mirror of
https://github.com/kubernetes-sigs/external-dns.git
synced 2025-08-06 09:36:58 +02:00
096c68be79
* feat(service): enhance compatibility, process molecule services * ref(service): simplify label detection for molecule servics
427 lines
9.5 KiB
Go
427 lines
9.5 KiB
Go
/*
|
|
Copyright 2017 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 (
|
|
"net"
|
|
"testing"
|
|
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
"k8s.io/client-go/pkg/api/v1"
|
|
|
|
"github.com/kubernetes-incubator/external-dns/endpoint"
|
|
)
|
|
|
|
// Validates that serviceSource is a Source
|
|
var _ Source = &serviceSource{}
|
|
|
|
func TestService(t *testing.T) {
|
|
t.Run("Endpoints", testServiceEndpoints)
|
|
}
|
|
|
|
func TestNewServiceSource(t *testing.T) {
|
|
for _, ti := range []struct {
|
|
title string
|
|
fqdntemplate string
|
|
expectError bool
|
|
}{
|
|
{
|
|
title: "invalid template",
|
|
expectError: true,
|
|
fqdntemplate: "{{.Name",
|
|
},
|
|
{
|
|
title: "valid empty template",
|
|
expectError: false,
|
|
},
|
|
{
|
|
title: "valid template",
|
|
expectError: false,
|
|
fqdntemplate: "{{.Name}}-{{.Namespace}}.ext-dns.test.com",
|
|
},
|
|
} {
|
|
t.Run(ti.title, func(t *testing.T) {
|
|
_, err := NewServiceSource(fake.NewSimpleClientset(), "", ti.fqdntemplate, "")
|
|
if ti.expectError && err == nil {
|
|
t.Error("invalid template should return err")
|
|
}
|
|
if !ti.expectError && err != nil {
|
|
t.Error(err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// testServiceEndpoints tests that various services generate the correct endpoints.
|
|
func testServiceEndpoints(t *testing.T) {
|
|
for _, tc := range []struct {
|
|
title string
|
|
targetNamespace string
|
|
svcNamespace string
|
|
svcName string
|
|
compatibility string
|
|
fqdntemplate string
|
|
labels map[string]string
|
|
annotations map[string]string
|
|
lbs []string
|
|
expected []*endpoint.Endpoint
|
|
}{
|
|
{
|
|
"no annotated services return no endpoints",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{},
|
|
},
|
|
{
|
|
"annotated services return an endpoint with target IP",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "1.2.3.4"},
|
|
},
|
|
},
|
|
{
|
|
"annotated services return an endpoint with target hostname",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{"lb.example.com"}, // Kubernetes omits the trailing dot
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "lb.example.com"},
|
|
},
|
|
},
|
|
{
|
|
"annotated services can omit trailing dot",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org", // Trailing dot is omitted
|
|
},
|
|
[]string{"1.2.3.4", "lb.example.com"}, // Kubernetes omits the trailing dot
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "1.2.3.4"},
|
|
{DNSName: "foo.example.org", Target: "lb.example.com"},
|
|
},
|
|
},
|
|
{
|
|
"our controller type is dns-controller",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
controllerAnnotationKey: controllerAnnotationValue,
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "1.2.3.4"},
|
|
},
|
|
},
|
|
{
|
|
"different controller types are ignored even (with template specified)",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"{{.Name}}.ext-dns.test.com",
|
|
map[string]string{},
|
|
map[string]string{
|
|
controllerAnnotationKey: "some-other-tool",
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{},
|
|
},
|
|
{
|
|
"services are found in target namespace",
|
|
"testing",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "1.2.3.4"},
|
|
},
|
|
},
|
|
{
|
|
"services that are not in target namespace are ignored",
|
|
"testing",
|
|
"other-testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{},
|
|
},
|
|
{
|
|
"services are found in all namespaces",
|
|
"",
|
|
"other-testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "1.2.3.4"},
|
|
},
|
|
},
|
|
{
|
|
"no external entrypoints return no endpoints",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{},
|
|
[]*endpoint.Endpoint{},
|
|
},
|
|
{
|
|
"multiple external entrypoints return multiple endpoints",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4", "8.8.8.8"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "1.2.3.4"},
|
|
{DNSName: "foo.example.org", Target: "8.8.8.8"},
|
|
},
|
|
},
|
|
{
|
|
"services annotated with legacy mate annotations are ignored in default mode",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
"zalando.org/dnsname": "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{},
|
|
},
|
|
{
|
|
"services annotated with legacy mate annotations return an endpoint in compatibility mode",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"mate",
|
|
"",
|
|
map[string]string{},
|
|
map[string]string{
|
|
"zalando.org/dnsname": "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "1.2.3.4"},
|
|
},
|
|
},
|
|
{
|
|
"services annotated with legacy molecule annotations return an endpoint in compatibility mode",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"molecule",
|
|
"",
|
|
map[string]string{
|
|
"dns": "route53",
|
|
},
|
|
map[string]string{
|
|
"domainName": "foo.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.example.org", Target: "1.2.3.4"},
|
|
},
|
|
},
|
|
{
|
|
"not annotated services with set fqdntemplate return an endpoint with target IP",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"{{.Name}}.bar.example.com",
|
|
map[string]string{},
|
|
map[string]string{},
|
|
[]string{"1.2.3.4", "elb.com"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "foo.bar.example.com", Target: "1.2.3.4"},
|
|
{DNSName: "foo.bar.example.com", Target: "elb.com"},
|
|
},
|
|
},
|
|
{
|
|
"not annotated services with unknown tmpl field should not return anything",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"",
|
|
"{{.Calibre}}.bar.example.com",
|
|
map[string]string{},
|
|
map[string]string{},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{},
|
|
},
|
|
{
|
|
"compatibility annotated services with tmpl. compatibility takes precedence",
|
|
"",
|
|
"testing",
|
|
"foo",
|
|
"mate",
|
|
"{{.Name}}.bar.example.com",
|
|
map[string]string{},
|
|
map[string]string{
|
|
"zalando.org/dnsname": "mate.example.org.",
|
|
},
|
|
[]string{"1.2.3.4"},
|
|
[]*endpoint.Endpoint{
|
|
{DNSName: "mate.example.org", Target: "1.2.3.4"},
|
|
},
|
|
},
|
|
} {
|
|
t.Run(tc.title, func(t *testing.T) {
|
|
// Create a Kubernetes testing client
|
|
kubernetes := fake.NewSimpleClientset()
|
|
|
|
// Create a service to test against
|
|
ingresses := []v1.LoadBalancerIngress{}
|
|
for _, lb := range tc.lbs {
|
|
if net.ParseIP(lb) != nil {
|
|
ingresses = append(ingresses, v1.LoadBalancerIngress{IP: lb})
|
|
} else {
|
|
ingresses = append(ingresses, v1.LoadBalancerIngress{Hostname: lb})
|
|
}
|
|
}
|
|
|
|
service := &v1.Service{
|
|
ObjectMeta: v1.ObjectMeta{
|
|
Namespace: tc.svcNamespace,
|
|
Name: tc.svcName,
|
|
Labels: tc.labels,
|
|
Annotations: tc.annotations,
|
|
},
|
|
Status: v1.ServiceStatus{
|
|
LoadBalancer: v1.LoadBalancerStatus{
|
|
Ingress: ingresses,
|
|
},
|
|
},
|
|
}
|
|
|
|
_, err := kubernetes.CoreV1().Services(service.Namespace).Create(service)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Create our object under test and get the endpoints.
|
|
client, _ := NewServiceSource(kubernetes, tc.targetNamespace, tc.fqdntemplate, tc.compatibility)
|
|
|
|
endpoints, err := client.Endpoints()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Validate returned endpoints against desired endpoints.
|
|
validateEndpoints(t, endpoints, tc.expected)
|
|
})
|
|
}
|
|
}
|
|
|
|
func BenchmarkServiceEndpoints(b *testing.B) {
|
|
kubernetes := fake.NewSimpleClientset()
|
|
|
|
service := &v1.Service{
|
|
ObjectMeta: v1.ObjectMeta{
|
|
Namespace: "testing",
|
|
Name: "foo",
|
|
Annotations: map[string]string{
|
|
hostnameAnnotationKey: "foo.example.org.",
|
|
},
|
|
},
|
|
Status: v1.ServiceStatus{
|
|
LoadBalancer: v1.LoadBalancerStatus{
|
|
Ingress: []v1.LoadBalancerIngress{
|
|
{IP: "1.2.3.4"},
|
|
{IP: "8.8.8.8"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
_, err := kubernetes.CoreV1().Services(service.Namespace).Create(service)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
client, _ := NewServiceSource(kubernetes, v1.NamespaceAll, "", "")
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := client.Endpoints()
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|