Skip to content

Commit d728eb4

Browse files
committed
test(network): replace RCs with Deployments in util function jig.Run
See kubernetes#119021
1 parent 8ccba7f commit d728eb4

File tree

4 files changed

+52
-46
lines changed

4 files changed

+52
-46
lines changed

test/e2e/framework/service/jig.go

Lines changed: 38 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ import (
2020
"context"
2121
"errors"
2222
"fmt"
23+
appsv1 "k8s.io/api/apps/v1"
24+
e2edeployment "k8s.io/kubernetes/test/e2e/framework/deployment"
2325
"math/rand"
2426
"net"
2527
"strconv"
@@ -655,23 +657,25 @@ func (j *TestJig) waitForCondition(ctx context.Context, timeout time.Duration, m
655657
return service, nil
656658
}
657659

658-
// newRCTemplate returns the default v1.ReplicationController object for
659-
// this j, but does not actually create the RC. The default RC has the same
660+
// newDeploymentTemplate returns the default appsv1.Deployment object for
661+
// this j, but does not actually create the Deployment. The default Deployment has the same
660662
// name as the j and runs the "netexec" container.
661-
func (j *TestJig) newRCTemplate() *v1.ReplicationController {
663+
func (j *TestJig) newDeploymentTemplate() *appsv1.Deployment {
662664
var replicas int32 = 1
663665
var grace int64 = 3 // so we don't race with kube-proxy when scaling up/down
664666

665-
rc := &v1.ReplicationController{
667+
deployment := &appsv1.Deployment{
666668
ObjectMeta: metav1.ObjectMeta{
667669
Namespace: j.Namespace,
668670
Name: j.Name,
669671
Labels: j.Labels,
670672
},
671-
Spec: v1.ReplicationControllerSpec{
673+
Spec: appsv1.DeploymentSpec{
672674
Replicas: &replicas,
673-
Selector: j.Labels,
674-
Template: &v1.PodTemplateSpec{
675+
Selector: &metav1.LabelSelector{
676+
MatchLabels: j.Labels,
677+
},
678+
Template: v1.PodTemplateSpec{
675679
ObjectMeta: metav1.ObjectMeta{
676680
Labels: j.Labels,
677681
},
@@ -697,32 +701,32 @@ func (j *TestJig) newRCTemplate() *v1.ReplicationController {
697701
},
698702
},
699703
}
700-
return rc
704+
return deployment
701705
}
702706

703-
// AddRCAntiAffinity adds AntiAffinity to the given ReplicationController.
704-
func (j *TestJig) AddRCAntiAffinity(rc *v1.ReplicationController) {
707+
// AddDeploymentAntiAffinity adds AntiAffinity to the given Deployment.
708+
func (j *TestJig) AddDeploymentAntiAffinity(deployment *appsv1.Deployment) {
705709
var replicas int32 = 2
706710

707-
rc.Spec.Replicas = &replicas
708-
if rc.Spec.Template.Spec.Affinity == nil {
709-
rc.Spec.Template.Spec.Affinity = &v1.Affinity{}
711+
deployment.Spec.Replicas = &replicas
712+
if deployment.Spec.Template.Spec.Affinity == nil {
713+
deployment.Spec.Template.Spec.Affinity = &v1.Affinity{}
710714
}
711-
if rc.Spec.Template.Spec.Affinity.PodAntiAffinity == nil {
712-
rc.Spec.Template.Spec.Affinity.PodAntiAffinity = &v1.PodAntiAffinity{}
715+
if deployment.Spec.Template.Spec.Affinity.PodAntiAffinity == nil {
716+
deployment.Spec.Template.Spec.Affinity.PodAntiAffinity = &v1.PodAntiAffinity{}
713717
}
714-
rc.Spec.Template.Spec.Affinity.PodAntiAffinity.RequiredDuringSchedulingIgnoredDuringExecution = append(
715-
rc.Spec.Template.Spec.Affinity.PodAntiAffinity.RequiredDuringSchedulingIgnoredDuringExecution,
718+
deployment.Spec.Template.Spec.Affinity.PodAntiAffinity.RequiredDuringSchedulingIgnoredDuringExecution = append(
719+
deployment.Spec.Template.Spec.Affinity.PodAntiAffinity.RequiredDuringSchedulingIgnoredDuringExecution,
716720
v1.PodAffinityTerm{
717721
LabelSelector: &metav1.LabelSelector{MatchLabels: j.Labels},
718722
Namespaces: nil,
719723
TopologyKey: "kubernetes.io/hostname",
720724
})
721725
}
722726

723-
// CreatePDB returns a PodDisruptionBudget for the given ReplicationController, or returns an error if a PodDisruptionBudget isn't ready
724-
func (j *TestJig) CreatePDB(ctx context.Context, rc *v1.ReplicationController) (*policyv1.PodDisruptionBudget, error) {
725-
pdb := j.newPDBTemplate(rc)
727+
// CreatePDB returns a PodDisruptionBudget for the given Deployment, or returns an error if a PodDisruptionBudget isn't ready
728+
func (j *TestJig) CreatePDB(ctx context.Context, deployment *appsv1.Deployment) (*policyv1.PodDisruptionBudget, error) {
729+
pdb := j.newPDBTemplate(deployment)
726730
newPdb, err := j.Client.PolicyV1().PodDisruptionBudgets(j.Namespace).Create(ctx, pdb, metav1.CreateOptions{})
727731
if err != nil {
728732
return nil, fmt.Errorf("failed to create PDB %q %v", pdb.Name, err)
@@ -736,8 +740,8 @@ func (j *TestJig) CreatePDB(ctx context.Context, rc *v1.ReplicationController) (
736740

737741
// newPDBTemplate returns the default policyv1.PodDisruptionBudget object for
738742
// this j, but does not actually create the PDB. The default PDB specifies a
739-
// MinAvailable of N-1 and matches the pods created by the RC.
740-
func (j *TestJig) newPDBTemplate(rc *v1.ReplicationController) *policyv1.PodDisruptionBudget {
743+
// MinAvailable of N-1 and matches the pods created by the Deployment.
744+
func (j *TestJig) newPDBTemplate(rc *appsv1.Deployment) *policyv1.PodDisruptionBudget {
741745
minAvailable := intstr.FromInt32(*rc.Spec.Replicas - 1)
742746

743747
pdb := &policyv1.PodDisruptionBudget{
@@ -755,25 +759,25 @@ func (j *TestJig) newPDBTemplate(rc *v1.ReplicationController) *policyv1.PodDisr
755759
return pdb
756760
}
757761

758-
// Run creates a ReplicationController and Pod(s) and waits for the
759-
// Pod(s) to be running. Callers can provide a function to tweak the RC object
762+
// Run creates a Deployment and Pod(s) and waits for the
763+
// Pod(s) to be running. Callers can provide a function to tweak the Deployment object
760764
// before it is created.
761-
func (j *TestJig) Run(ctx context.Context, tweak func(rc *v1.ReplicationController)) (*v1.ReplicationController, error) {
762-
rc := j.newRCTemplate()
765+
func (j *TestJig) Run(ctx context.Context, tweak func(rc *appsv1.Deployment)) (*appsv1.Deployment, error) {
766+
deployment := j.newDeploymentTemplate()
763767
if tweak != nil {
764-
tweak(rc)
768+
tweak(deployment)
765769
}
766-
result, err := j.Client.CoreV1().ReplicationControllers(j.Namespace).Create(ctx, rc, metav1.CreateOptions{})
770+
771+
result, err := j.Client.AppsV1().Deployments(j.Namespace).Create(ctx, deployment, metav1.CreateOptions{})
767772
if err != nil {
768-
return nil, fmt.Errorf("failed to create RC %q: %w", rc.Name, err)
773+
return nil, fmt.Errorf("failed to create Deployment %q: %w", deployment.Name, err)
769774
}
770-
pods, err := j.waitForPodsCreated(ctx, int(*(rc.Spec.Replicas)))
775+
776+
err = e2edeployment.WaitForDeploymentComplete(j.Client, result)
771777
if err != nil {
772-
return nil, fmt.Errorf("failed to create pods: %w", err)
773-
}
774-
if err := j.waitForPodsReady(ctx, pods); err != nil {
775-
return nil, fmt.Errorf("failed waiting for pods to be running: %w", err)
778+
return nil, fmt.Errorf("failed waiting for Deployment %q: %w", deployment.Name, err)
776779
}
780+
777781
return result, nil
778782
}
779783

test/e2e/network/loadbalancer.go

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@ import (
4747
e2enode "k8s.io/kubernetes/test/e2e/framework/node"
4848
e2epod "k8s.io/kubernetes/test/e2e/framework/pod"
4949
e2eoutput "k8s.io/kubernetes/test/e2e/framework/pod/output"
50-
e2erc "k8s.io/kubernetes/test/e2e/framework/rc"
5150
e2eservice "k8s.io/kubernetes/test/e2e/framework/service"
5251
e2eskipper "k8s.io/kubernetes/test/e2e/framework/skipper"
5352
"k8s.io/kubernetes/test/e2e/network/common"
@@ -1115,10 +1114,10 @@ var _ = common.SIGDescribe("LoadBalancers ExternalTrafficPolicy: Local", feature
11151114
endpointNodeName := nodes.Items[i].Name
11161115

11171116
ginkgo.By("creating a pod to be part of the service " + serviceName + " on node " + endpointNodeName)
1118-
_, err = jig.Run(ctx, func(rc *v1.ReplicationController) {
1119-
rc.Name = serviceName
1117+
_, err = jig.Run(ctx, func(deployment *appsv1.Deployment) {
1118+
deployment.Name = serviceName
11201119
if endpointNodeName != "" {
1121-
rc.Spec.Template.Spec.NodeName = endpointNodeName
1120+
deployment.Spec.Template.Spec.NodeName = endpointNodeName
11221121
}
11231122
})
11241123
framework.ExpectNoError(err)
@@ -1146,7 +1145,9 @@ var _ = common.SIGDescribe("LoadBalancers ExternalTrafficPolicy: Local", feature
11461145
threshold)
11471146
framework.ExpectNoError(err)
11481147
}
1149-
framework.ExpectNoError(e2erc.DeleteRCAndWaitForGC(ctx, f.ClientSet, namespace, serviceName))
1148+
1149+
err = f.ClientSet.AppsV1().Deployments(namespace).Delete(ctx, serviceName, metav1.DeleteOptions{})
1150+
framework.ExpectNoError(err)
11501151
}
11511152
})
11521153

@@ -1172,9 +1173,9 @@ var _ = common.SIGDescribe("LoadBalancers ExternalTrafficPolicy: Local", feature
11721173
framework.Logf("ingress is %s:%d", ingress, svcPort)
11731174

11741175
ginkgo.By("creating endpoints on multiple nodes")
1175-
_, err = jig.Run(ctx, func(rc *v1.ReplicationController) {
1176-
rc.Spec.Replicas = ptr.To[int32](2)
1177-
rc.Spec.Template.Spec.Affinity = &v1.Affinity{
1176+
_, err = jig.Run(ctx, func(deployment *appsv1.Deployment) {
1177+
deployment.Spec.Replicas = ptr.To[int32](2)
1178+
deployment.Spec.Template.Spec.Affinity = &v1.Affinity{
11781179
PodAntiAffinity: &v1.PodAntiAffinity{
11791180
RequiredDuringSchedulingIgnoredDuringExecution: []v1.PodAffinityTerm{
11801181
{

test/e2e/upgrades/network/services.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,11 +62,11 @@ func (t *ServiceUpgradeTest) Setup(ctx context.Context, f *framework.Framework)
6262
svcPort := int(tcpService.Spec.Ports[0].Port)
6363

6464
ginkgo.By("creating pod to be part of service " + serviceName)
65-
rc, err := jig.Run(ctx, jig.AddRCAntiAffinity)
65+
deployment, err := jig.Run(ctx, jig.AddDeploymentAntiAffinity)
6666
framework.ExpectNoError(err)
6767

68-
ginkgo.By("creating a PodDisruptionBudget to cover the ReplicationController")
69-
_, err = jig.CreatePDB(ctx, rc)
68+
ginkgo.By("creating a PodDisruptionBudget to cover the Deployment")
69+
_, err = jig.CreatePDB(ctx, deployment)
7070
framework.ExpectNoError(err)
7171

7272
// Hit it once before considering ourselves ready

test/e2e/windows/service.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ package windows
1919
import (
2020
"context"
2121
"fmt"
22+
appsv1 "k8s.io/api/apps/v1"
2223
"net"
2324
"strconv"
2425

@@ -66,7 +67,7 @@ var _ = sigDescribe("Services", skipUnlessWindows(func() {
6667

6768
ginkgo.By("creating Pod to be part of service " + serviceName)
6869
// tweak the Jig to use windows...
69-
windowsNodeSelectorTweak := func(rc *v1.ReplicationController) {
70+
windowsNodeSelectorTweak := func(rc *appsv1.Deployment) {
7071
rc.Spec.Template.Spec.NodeSelector = map[string]string{
7172
"kubernetes.io/os": "windows",
7273
}

0 commit comments

Comments
 (0)