mirror of
https://github.com/optim-enterprises-bv/kubernetes.git
synced 2025-11-04 04:08:16 +00:00
483 lines
14 KiB
Go
483 lines
14 KiB
Go
/*
|
|
Copyright 2020 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 endpointslicemirroring
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
discovery "k8s.io/api/discovery/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/util/wait"
|
|
"k8s.io/client-go/informers"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
v1core "k8s.io/client-go/kubernetes/typed/core/v1"
|
|
"k8s.io/client-go/tools/cache"
|
|
"k8s.io/client-go/tools/leaderelection/resourcelock"
|
|
|
|
"k8s.io/klog/v2"
|
|
"k8s.io/klog/v2/ktesting"
|
|
"k8s.io/kubernetes/pkg/controller"
|
|
)
|
|
|
|
// Most of the tests related to EndpointSlice allocation can be found in reconciler_test.go
|
|
// Tests here primarily focus on unique controller functionality before the reconciler begins
|
|
|
|
var alwaysReady = func() bool { return true }
|
|
|
|
type endpointSliceMirroringController struct {
|
|
*Controller
|
|
endpointsStore cache.Store
|
|
endpointSliceStore cache.Store
|
|
serviceStore cache.Store
|
|
}
|
|
|
|
func newController(ctx context.Context, batchPeriod time.Duration) (*fake.Clientset, *endpointSliceMirroringController) {
|
|
client := newClientset()
|
|
informerFactory := informers.NewSharedInformerFactory(client, controller.NoResyncPeriodFunc())
|
|
|
|
esController := NewController(
|
|
ctx,
|
|
informerFactory.Core().V1().Endpoints(),
|
|
informerFactory.Discovery().V1().EndpointSlices(),
|
|
informerFactory.Core().V1().Services(),
|
|
int32(1000),
|
|
client,
|
|
batchPeriod)
|
|
|
|
// The event processing pipeline is normally started via Run() method.
|
|
// However, since we don't start it in unit tests, we explicitly start it here.
|
|
esController.eventBroadcaster.StartLogging(klog.Infof)
|
|
esController.eventBroadcaster.StartRecordingToSink(&v1core.EventSinkImpl{Interface: client.CoreV1().Events("")})
|
|
|
|
esController.endpointsSynced = alwaysReady
|
|
esController.endpointSlicesSynced = alwaysReady
|
|
esController.servicesSynced = alwaysReady
|
|
|
|
return client, &endpointSliceMirroringController{
|
|
esController,
|
|
informerFactory.Core().V1().Endpoints().Informer().GetStore(),
|
|
informerFactory.Discovery().V1().EndpointSlices().Informer().GetStore(),
|
|
informerFactory.Core().V1().Services().Informer().GetStore(),
|
|
}
|
|
}
|
|
|
|
func TestSyncEndpoints(t *testing.T) {
|
|
endpointsName := "testing-sync-endpoints"
|
|
namespace := metav1.NamespaceDefault
|
|
|
|
testCases := []struct {
|
|
testName string
|
|
service *v1.Service
|
|
endpoints *v1.Endpoints
|
|
endpointSlices []*discovery.EndpointSlice
|
|
expectedNumActions int
|
|
expectedNumSlices int
|
|
}{{
|
|
testName: "Endpoints with no addresses",
|
|
service: &v1.Service{},
|
|
endpoints: &v1.Endpoints{
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{},
|
|
expectedNumActions: 0,
|
|
expectedNumSlices: 0,
|
|
}, {
|
|
testName: "Endpoints with skip label true",
|
|
service: &v1.Service{},
|
|
endpoints: &v1.Endpoints{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Labels: map[string]string{discovery.LabelSkipMirror: "true"},
|
|
},
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
Addresses: []v1.EndpointAddress{{IP: "10.0.0.1"}},
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{},
|
|
expectedNumActions: 0,
|
|
expectedNumSlices: 0,
|
|
}, {
|
|
testName: "Endpoints with skip label false",
|
|
service: &v1.Service{},
|
|
endpoints: &v1.Endpoints{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Labels: map[string]string{discovery.LabelSkipMirror: "false"},
|
|
},
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
Addresses: []v1.EndpointAddress{{IP: "10.0.0.1"}},
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{},
|
|
expectedNumActions: 1,
|
|
expectedNumSlices: 1,
|
|
}, {
|
|
testName: "Endpoints with missing Service",
|
|
service: nil,
|
|
endpoints: &v1.Endpoints{
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
Addresses: []v1.EndpointAddress{{IP: "10.0.0.1"}},
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{},
|
|
expectedNumActions: 0,
|
|
expectedNumSlices: 0,
|
|
}, {
|
|
testName: "Endpoints with Service with selector specified",
|
|
service: &v1.Service{
|
|
Spec: v1.ServiceSpec{
|
|
Selector: map[string]string{"foo": "bar"},
|
|
},
|
|
},
|
|
endpoints: &v1.Endpoints{
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
Addresses: []v1.EndpointAddress{{IP: "10.0.0.1"}},
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{},
|
|
expectedNumActions: 0,
|
|
expectedNumSlices: 0,
|
|
}, {
|
|
testName: "Existing EndpointSlices that need to be cleaned up",
|
|
service: &v1.Service{},
|
|
endpoints: &v1.Endpoints{
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: endpointsName + "-1",
|
|
Labels: map[string]string{
|
|
discovery.LabelServiceName: endpointsName,
|
|
discovery.LabelManagedBy: controllerName,
|
|
},
|
|
},
|
|
}},
|
|
expectedNumActions: 1,
|
|
expectedNumSlices: 0,
|
|
}, {
|
|
testName: "Existing EndpointSlices managed by a different controller, no addresses to sync",
|
|
service: &v1.Service{},
|
|
endpoints: &v1.Endpoints{
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: endpointsName + "-1",
|
|
Labels: map[string]string{
|
|
discovery.LabelManagedBy: "something-else",
|
|
},
|
|
},
|
|
}},
|
|
expectedNumActions: 0,
|
|
// This only queries for EndpointSlices managed by this controller.
|
|
expectedNumSlices: 0,
|
|
}, {
|
|
testName: "Endpoints with 1000 addresses",
|
|
service: &v1.Service{},
|
|
endpoints: &v1.Endpoints{
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
Addresses: generateAddresses(1000),
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{},
|
|
expectedNumActions: 1,
|
|
expectedNumSlices: 1,
|
|
}, {
|
|
testName: "Endpoints with 1001 addresses - 1 should not be mirrored",
|
|
service: &v1.Service{},
|
|
endpoints: &v1.Endpoints{
|
|
Subsets: []v1.EndpointSubset{{
|
|
Ports: []v1.EndpointPort{{Port: 80}},
|
|
Addresses: generateAddresses(1001),
|
|
}},
|
|
},
|
|
endpointSlices: []*discovery.EndpointSlice{},
|
|
expectedNumActions: 2, // extra action for creating warning event
|
|
expectedNumSlices: 1,
|
|
}}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.testName, func(t *testing.T) {
|
|
_, ctx := ktesting.NewTestContext(t)
|
|
client, esController := newController(ctx, time.Duration(0))
|
|
tc.endpoints.Name = endpointsName
|
|
tc.endpoints.Namespace = namespace
|
|
esController.endpointsStore.Add(tc.endpoints)
|
|
if tc.service != nil {
|
|
tc.service.Name = endpointsName
|
|
tc.service.Namespace = namespace
|
|
esController.serviceStore.Add(tc.service)
|
|
}
|
|
|
|
for _, epSlice := range tc.endpointSlices {
|
|
epSlice.Namespace = namespace
|
|
esController.endpointSliceStore.Add(epSlice)
|
|
_, err := client.DiscoveryV1().EndpointSlices(namespace).Create(context.TODO(), epSlice, metav1.CreateOptions{})
|
|
if err != nil {
|
|
t.Fatalf("Expected no error creating EndpointSlice, got %v", err)
|
|
}
|
|
}
|
|
|
|
logger, _ := ktesting.NewTestContext(t)
|
|
err := esController.syncEndpoints(logger, fmt.Sprintf("%s/%s", namespace, endpointsName))
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error from syncEndpoints: %v", err)
|
|
}
|
|
|
|
numInitialActions := len(tc.endpointSlices)
|
|
// Wait for the expected event show up in test "Endpoints with 1001 addresses - 1 should not be mirrored"
|
|
err = wait.PollImmediate(time.Millisecond*100, wait.ForeverTestTimeout, func() (done bool, err error) {
|
|
actions := client.Actions()
|
|
numExtraActions := len(actions) - numInitialActions
|
|
if numExtraActions != tc.expectedNumActions {
|
|
t.Logf("Expected %d additional client actions, got %d: %#v. Will retry", tc.expectedNumActions, numExtraActions, actions[numInitialActions:])
|
|
return false, nil
|
|
}
|
|
return true, nil
|
|
})
|
|
if err != nil {
|
|
t.Fatal("Timed out waiting for expected actions")
|
|
}
|
|
|
|
endpointSlices := fetchEndpointSlices(t, client, namespace)
|
|
expectEndpointSlices(t, tc.expectedNumSlices, int(defaultMaxEndpointsPerSubset), *tc.endpoints, endpointSlices)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestShouldMirror(t *testing.T) {
|
|
testCases := []struct {
|
|
testName string
|
|
endpoints *v1.Endpoints
|
|
shouldMirror bool
|
|
}{{
|
|
testName: "Standard Endpoints",
|
|
endpoints: &v1.Endpoints{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test-endpoints",
|
|
},
|
|
},
|
|
shouldMirror: true,
|
|
}, {
|
|
testName: "Endpoints with skip-mirror=true",
|
|
endpoints: &v1.Endpoints{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test-endpoints",
|
|
Labels: map[string]string{
|
|
discovery.LabelSkipMirror: "true",
|
|
},
|
|
},
|
|
},
|
|
shouldMirror: false,
|
|
}, {
|
|
testName: "Endpoints with skip-mirror=invalid",
|
|
endpoints: &v1.Endpoints{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test-endpoints",
|
|
Labels: map[string]string{
|
|
discovery.LabelSkipMirror: "invalid",
|
|
},
|
|
},
|
|
},
|
|
shouldMirror: true,
|
|
}, {
|
|
testName: "Endpoints with leader election annotation",
|
|
endpoints: &v1.Endpoints{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test-endpoints",
|
|
Annotations: map[string]string{
|
|
resourcelock.LeaderElectionRecordAnnotationKey: "",
|
|
},
|
|
},
|
|
},
|
|
shouldMirror: false,
|
|
}}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.testName, func(t *testing.T) {
|
|
_, ctx := ktesting.NewTestContext(t)
|
|
_, c := newController(ctx, time.Duration(0))
|
|
|
|
if tc.endpoints != nil {
|
|
err := c.endpointsStore.Add(tc.endpoints)
|
|
if err != nil {
|
|
t.Fatalf("Error adding Endpoints to store: %v", err)
|
|
}
|
|
}
|
|
|
|
shouldMirror := c.shouldMirror(tc.endpoints)
|
|
|
|
if shouldMirror != tc.shouldMirror {
|
|
t.Errorf("Expected %t to be returned, got %t", tc.shouldMirror, shouldMirror)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestEndpointSlicesMirroredForService(t *testing.T) {
|
|
testCases := []struct {
|
|
testName string
|
|
namespace string
|
|
name string
|
|
endpointSlice *discovery.EndpointSlice
|
|
expectedInList bool
|
|
}{{
|
|
testName: "Service with matching EndpointSlice",
|
|
namespace: "ns1",
|
|
name: "svc1",
|
|
endpointSlice: &discovery.EndpointSlice{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "example-1",
|
|
Namespace: "ns1",
|
|
Labels: map[string]string{
|
|
discovery.LabelServiceName: "svc1",
|
|
discovery.LabelManagedBy: controllerName,
|
|
},
|
|
},
|
|
},
|
|
expectedInList: true,
|
|
}, {
|
|
testName: "Service with EndpointSlice that has different namespace",
|
|
namespace: "ns1",
|
|
name: "svc1",
|
|
endpointSlice: &discovery.EndpointSlice{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "example-1",
|
|
Namespace: "ns2",
|
|
Labels: map[string]string{
|
|
discovery.LabelServiceName: "svc1",
|
|
discovery.LabelManagedBy: controllerName,
|
|
},
|
|
},
|
|
},
|
|
expectedInList: false,
|
|
}, {
|
|
testName: "Service with EndpointSlice that has different service name",
|
|
namespace: "ns1",
|
|
name: "svc1",
|
|
endpointSlice: &discovery.EndpointSlice{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "example-1",
|
|
Namespace: "ns1",
|
|
Labels: map[string]string{
|
|
discovery.LabelServiceName: "svc2",
|
|
discovery.LabelManagedBy: controllerName,
|
|
},
|
|
},
|
|
},
|
|
expectedInList: false,
|
|
}, {
|
|
testName: "Service with EndpointSlice that has different controller name",
|
|
namespace: "ns1",
|
|
name: "svc1",
|
|
endpointSlice: &discovery.EndpointSlice{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "example-1",
|
|
Namespace: "ns1",
|
|
Labels: map[string]string{
|
|
discovery.LabelServiceName: "svc1",
|
|
discovery.LabelManagedBy: controllerName + "foo",
|
|
},
|
|
},
|
|
},
|
|
expectedInList: false,
|
|
}, {
|
|
testName: "Service with EndpointSlice that has missing controller name",
|
|
namespace: "ns1",
|
|
name: "svc1",
|
|
endpointSlice: &discovery.EndpointSlice{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "example-1",
|
|
Namespace: "ns1",
|
|
Labels: map[string]string{
|
|
discovery.LabelServiceName: "svc1",
|
|
},
|
|
},
|
|
},
|
|
expectedInList: false,
|
|
}, {
|
|
testName: "Service with EndpointSlice that has missing service name",
|
|
namespace: "ns1",
|
|
name: "svc1",
|
|
endpointSlice: &discovery.EndpointSlice{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "example-1",
|
|
Namespace: "ns1",
|
|
Labels: map[string]string{
|
|
discovery.LabelManagedBy: controllerName,
|
|
},
|
|
},
|
|
},
|
|
expectedInList: false,
|
|
}}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.testName, func(t *testing.T) {
|
|
_, ctx := ktesting.NewTestContext(t)
|
|
_, c := newController(ctx, time.Duration(0))
|
|
|
|
err := c.endpointSliceStore.Add(tc.endpointSlice)
|
|
if err != nil {
|
|
t.Fatalf("Error adding EndpointSlice to store: %v", err)
|
|
}
|
|
|
|
endpointSlices, err := endpointSlicesMirroredForService(c.endpointSliceLister, tc.namespace, tc.name)
|
|
if err != nil {
|
|
t.Fatalf("Expected no error, got %v", err)
|
|
}
|
|
|
|
if tc.expectedInList {
|
|
if len(endpointSlices) != 1 {
|
|
t.Fatalf("Expected 1 EndpointSlice to be in list, got %d", len(endpointSlices))
|
|
}
|
|
|
|
if endpointSlices[0].Name != tc.endpointSlice.Name {
|
|
t.Fatalf("Expected %s EndpointSlice to be in list, got %s", tc.endpointSlice.Name, endpointSlices[0].Name)
|
|
}
|
|
} else {
|
|
if len(endpointSlices) != 0 {
|
|
t.Fatalf("Expected no EndpointSlices to be in list, got %d", len(endpointSlices))
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func generateAddresses(num int) []v1.EndpointAddress {
|
|
addresses := make([]v1.EndpointAddress, num)
|
|
for i := 0; i < num; i++ {
|
|
part1 := i / 255
|
|
part2 := i % 255
|
|
ip := fmt.Sprintf("10.0.%d.%d", part1, part2)
|
|
addresses[i] = v1.EndpointAddress{IP: ip}
|
|
}
|
|
return addresses
|
|
}
|