external-dns/source/multisource_test.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 (
	"context"
	"errors"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"sigs.k8s.io/external-dns/endpoint"
	"sigs.k8s.io/external-dns/internal/testutils"
)

func TestMultiSource(t *testing.T) {
	t.Parallel()

	t.Run("Interface", testMultiSourceImplementsSource)
	t.Run("Endpoints", testMultiSourceEndpoints)
	t.Run("EndpointsWithError", testMultiSourceEndpointsWithError)
	t.Run("EndpointsDefaultTargets", testMultiSourceEndpointsDefaultTargets)
}

// testMultiSourceImplementsSource tests that multiSource is a valid Source.
func testMultiSourceImplementsSource(t *testing.T) {
	assert.Implements(t, (*Source)(nil), new(multiSource))
}

// testMultiSourceEndpoints tests merged endpoints from children are returned.
func testMultiSourceEndpoints(t *testing.T) {
	foo := &endpoint.Endpoint{DNSName: "foo", Targets: endpoint.Targets{"8.8.8.8"}}
	bar := &endpoint.Endpoint{DNSName: "bar", Targets: endpoint.Targets{"8.8.4.4"}}

	for _, tc := range []struct {
		title           string
		nestedEndpoints [][]*endpoint.Endpoint
		expected        []*endpoint.Endpoint
	}{
		{
			"no child sources return no endpoints",
			nil,
			[]*endpoint.Endpoint{},
		},
		{
			"single empty child source returns no endpoints",
			[][]*endpoint.Endpoint{{}},
			[]*endpoint.Endpoint{},
		},
		{
			"single non-empty child source returns child's endpoints",
			[][]*endpoint.Endpoint{{foo}},
			[]*endpoint.Endpoint{foo},
		},
		{
			"multiple non-empty child sources returns merged children's endpoints",
			[][]*endpoint.Endpoint{{foo}, {bar}},
			[]*endpoint.Endpoint{foo, bar},
		},
	} {
		tc := tc
		t.Run(tc.title, func(t *testing.T) {
			t.Parallel()

			// Prepare the nested mock sources.
			sources := make([]Source, 0, len(tc.nestedEndpoints))

			// Populate the nested mock sources.
			for _, endpoints := range tc.nestedEndpoints {
				src := new(testutils.MockSource)
				src.On("Endpoints").Return(endpoints, nil)

				sources = append(sources, src)
			}

			// Create our object under test and get the endpoints.
			source := NewMultiSource(sources, nil)

			// Get endpoints from the source.
			endpoints, err := source.Endpoints(context.Background())
			require.NoError(t, err)

			// Validate returned endpoints against desired endpoints.
			validateEndpoints(t, endpoints, tc.expected)

			// Validate that the nested sources were called.
			for _, src := range sources {
				src.(*testutils.MockSource).AssertExpectations(t)
			}
		})
	}
}

// testMultiSourceEndpointsWithError tests that an error by a nested source is bubbled up.
func testMultiSourceEndpointsWithError(t *testing.T) {
	// Create the expected error.
	errSomeError := errors.New("some error")

	// Create a mocked source returning that error.
	src := new(testutils.MockSource)
	src.On("Endpoints").Return(nil, errSomeError)

	// Create our object under test and get the endpoints.
	source := NewMultiSource([]Source{src}, nil)

	// Get endpoints from our source.
	_, err := source.Endpoints(context.Background())
	assert.EqualError(t, err, "some error")

	// Validate that the nested source was called.
	src.AssertExpectations(t)
}

func testMultiSourceEndpointsDefaultTargets(t *testing.T) {
	// Create the expected default targets
	defaultTargetsA := []string{"127.0.0.1", "127.0.0.2"}
	defaultTargetsAAAA := []string{"2001:db8::1"}
	defaultTargetsCName := []string{"foo.example.org"}
	defaultTargets := append(defaultTargetsA, defaultTargetsCName...)
	defaultTargets = append(defaultTargets, defaultTargetsAAAA...)
	labels := endpoint.Labels{"foo": "bar"}

	// Create the expected endpoints
	expectedEndpoints := []*endpoint.Endpoint{
		{DNSName: "foo", Targets: defaultTargetsA, RecordType: "A", Labels: labels},
		{DNSName: "bar", Targets: defaultTargetsA, RecordType: "A", Labels: labels},
		{DNSName: "foo", Targets: defaultTargetsAAAA, RecordType: "AAAA", Labels: labels},
		{DNSName: "bar", Targets: defaultTargetsAAAA, RecordType: "AAAA", Labels: labels},
		{DNSName: "foo", Targets: defaultTargetsCName, RecordType: "CNAME", Labels: labels},
		{DNSName: "bar", Targets: defaultTargetsCName, RecordType: "CNAME", Labels: labels},
	}

	// Create the source endpoints with different targets
	sourceEndpoints := []*endpoint.Endpoint{
		{DNSName: "foo", Targets: endpoint.Targets{"8.8.8.8"}, Labels: labels},
		{DNSName: "bar", Targets: endpoint.Targets{"8.8.4.4"}, Labels: labels},
	}

	// Create a mocked source returning source targets
	src := new(testutils.MockSource)
	src.On("Endpoints").Return(sourceEndpoints, nil)

	// Create our object under test with non-empty defaultTargets and get the endpoints.
	source := NewMultiSource([]Source{src}, defaultTargets)

	// Get endpoints from our source.
	endpoints, err := source.Endpoints(context.Background())
	require.NoError(t, err)

	// Validate returned endpoints against desired endpoints.
	validateEndpoints(t, endpoints, expectedEndpoints)

	// Validate that the nested sources were called.
	src.AssertExpectations(t)
}