From 7956b37f62dc8213a168bddf38c42003d2e806be Mon Sep 17 00:00:00 2001 From: Dan Winship Date: Thu, 20 Mar 2025 08:48:35 -0400 Subject: [PATCH] Add test for `trafficDistribution: PreferSameZone` (Identical to the PreferClose test.) --- test/e2e/network/traffic_distribution.go | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/test/e2e/network/traffic_distribution.go b/test/e2e/network/traffic_distribution.go index 6713242ad75..e6259c82644 100644 --- a/test/e2e/network/traffic_distribution.go +++ b/test/e2e/network/traffic_distribution.go @@ -29,6 +29,7 @@ import ( "k8s.io/apimachinery/pkg/util/intstr" "k8s.io/apimachinery/pkg/util/sets" clientset "k8s.io/client-go/kubernetes" + "k8s.io/kubernetes/pkg/features" "k8s.io/kubernetes/test/e2e/framework" e2enode "k8s.io/kubernetes/test/e2e/framework/node" e2epod "k8s.io/kubernetes/test/e2e/framework/pod" @@ -260,10 +261,17 @@ var _ = common.SIGDescribe("Traffic Distribution", func() { // Main test specifications. //////////////////////////////////////////////////////////////////////////// - ginkgo.It("should route traffic to an endpoint in the same zone when using PreferClose", func(ctx context.Context) { + framework.It("should route traffic to an endpoint in the same zone when using PreferClose", func(ctx context.Context) { clientPods, serverPods := allocateClientsAndServers(ctx) svc := createService(ctx, v1.ServiceTrafficDistributionPreferClose) createPods(ctx, svc, clientPods, serverPods) checkTrafficDistribution(ctx, clientPods) }) + + framework.It("should route traffic to an endpoint in the same zone when using PreferSameZone", framework.WithFeatureGate(features.PreferSameTrafficDistribution), func(ctx context.Context) { + clientPods, serverPods := allocateClientsAndServers(ctx) + svc := createService(ctx, v1.ServiceTrafficDistributionPreferSameZone) + createPods(ctx, svc, clientPods, serverPods) + checkTrafficDistribution(ctx, clientPods) + }) })