test enable and disable ServiceCIDR feature

2024-05-20 20:35:00 +00:00
parent f36975b193
commit 101722249d
1 changed files with 142 additions and 0 deletions
--- a/test/integration/servicecidr/feature_enable_disable.go
+++ b/test/integration/servicecidr/feature_enable_disable.go
@@ -0,0 +1,142 @@
+/*
+Copyright 2024 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 servicecidr
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	v1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	clientset "k8s.io/client-go/kubernetes"
+	kubeapiservertesting "k8s.io/kubernetes/cmd/kube-apiserver/app/testing"
+	"k8s.io/kubernetes/pkg/features"
+	"k8s.io/kubernetes/test/integration/framework"
+)
+
+func TestEnableDisableServiceCIDR(t *testing.T) {
+
+	svc := func(i int) *v1.Service {
+		return &v1.Service{
+			ObjectMeta: metav1.ObjectMeta{
+				Name: fmt.Sprintf("svc-%v", i),
+			},
+			Spec: v1.ServiceSpec{
+				Type: v1.ServiceTypeClusterIP,
+				Ports: []v1.ServicePort{
+					{Port: 80},
+				},
+			},
+		}
+	}
+	// start etcd instance
+	etcdOptions := framework.SharedEtcd()
+	// apiserver with the feature disabled
+	apiServerOptions := kubeapiservertesting.NewDefaultTestServerOptions()
+	s1 := kubeapiservertesting.StartTestServerOrDie(t, apiServerOptions,
+		[]string{
+			"--runtime-config=networking.k8s.io/v1alpha1=false",
+			"--service-cluster-ip-range=10.0.0.0/24",
+			"--disable-admission-plugins=ServiceAccount",
+			fmt.Sprintf("--feature-gates=%s=false", features.MultiCIDRServiceAllocator)},
+		etcdOptions)
+
+	client1, err := clientset.NewForConfig(s1.ClientConfig)
+	if err != nil {
+		t.Fatalf("Unexpected error: %v", err)
+	}
+
+	ns := framework.CreateNamespaceOrDie(client1, "test-enable-disable-service-cidr", t)
+	// make 2 services , there will be 3 services counting the kubernetes.default
+	for i := 0; i < 2; i++ {
+		if _, err := client1.CoreV1().Services(ns.Name).Create(context.TODO(), svc(i), metav1.CreateOptions{}); err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	services, err := client1.CoreV1().Services("").List(context.TODO(), metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(services.Items) != 3 {
+		t.Fatalf("expected 3 Services got %d", len(services.Items))
+	}
+
+	// shutdown s1
+	s1.TearDownFn()
+
+	// apiserver with the feature enabled
+	s2 := kubeapiservertesting.StartTestServerOrDie(t, apiServerOptions,
+		[]string{
+			"--runtime-config=networking.k8s.io/v1alpha1=true",
+			"--service-cluster-ip-range=10.0.0.0/24",
+			"--disable-admission-plugins=ServiceAccount",
+			fmt.Sprintf("--feature-gates=%s=true", features.MultiCIDRServiceAllocator)},
+		etcdOptions)
+
+	client2, err := clientset.NewForConfig(s2.ClientConfig)
+	if err != nil {
+		t.Fatalf("Unexpected error: %v", err)
+	}
+
+	// make 2 services , there will be 5 services now
+	for i := 2; i < 5; i++ {
+		if _, err := client2.CoreV1().Services(ns.Name).Create(context.TODO(), svc(i), metav1.CreateOptions{}); err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	services, err = client2.CoreV1().Services("").List(context.TODO(), metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(services.Items) != 5 {
+		t.Fatalf("expected 5 Services got %d", len(services.Items))
+	}
+
+	// shutdown apiserver with the feature enabled
+	s2.TearDownFn()
+	// start an apiserver without the deature enabled
+	s3 := kubeapiservertesting.StartTestServerOrDie(t, apiServerOptions,
+		[]string{
+			"--runtime-config=networking.k8s.io/v1alpha1=false",
+			"--service-cluster-ip-range=10.0.0.0/24",
+			"--disable-admission-plugins=ServiceAccount",
+			fmt.Sprintf("--feature-gates=%s=false", features.MultiCIDRServiceAllocator)},
+		etcdOptions)
+
+	client3, err := clientset.NewForConfig(s3.ClientConfig)
+	if err != nil {
+		t.Fatalf("Unexpected error: %v", err)
+	}
+	// make 2 services , there will be 7 services now
+	for i := 5; i < 7; i++ {
+		if _, err := client2.CoreV1().Services(ns.Name).Create(context.TODO(), svc(i), metav1.CreateOptions{}); err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	services, err = client3.CoreV1().Services("").List(context.TODO(), metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(services.Items) != 7 {
+		t.Fatalf("expected 5 Services got %d", len(services.Items))
+	}
+
+}