github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-23 02:26:52 +00:00
Code Issues Projects Releases Wiki Activity

Chore: add etp tweak function

Browse Source
This commit is contained in:
mowangdk 2023-03-26 11:44:23 +08:00
parent 2f6bc661dc
commit 23c242770f
1 changed files with 7 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
staging/src/k8s.io/cloud-provider/controllers/service/controller_test.go
View File

@ -73,17 +73,9 @@ func newService(name string, serviceType v1.ServiceType, tweaks ...serviceTweak)
return s return s
} }
func newETPLocalService(name string, serviceType v1.ServiceType) *v1.Service { func tweakAddETP(etpType v1.ServiceExternalTrafficPolicyType) serviceTweak {
return &v1.Service{ return func(s *v1.Service) {
ObjectMeta: metav1.ObjectMeta{ s.Spec.ExternalTrafficPolicy = etpType
Name: name,
Namespace: "default",
UID: "777",
},
Spec: v1.ServiceSpec{
Type: serviceType,
ExternalTrafficPolicy: v1.ServiceExternalTrafficPolicyLocal,
},
} }
} }
@ -588,8 +580,8 @@ func TestNodeChangesForExternalTrafficPolicyLocalServices(t *testing.T) {
syncCallErr bool syncCallErr bool
} }
etpLocalservice1 := newETPLocalService("s0", v1.ServiceTypeLoadBalancer) etpLocalservice1 := newService("s0", v1.ServiceTypeLoadBalancer, tweakAddETP(v1.ServiceExternalTrafficPolicyLocal))
etpLocalservice2 := newETPLocalService("s1", v1.ServiceTypeLoadBalancer) etpLocalservice2 := newService("s1", v1.ServiceTypeLoadBalancer, tweakAddETP(v1.ServiceExternalTrafficPolicyLocal))
service3 := defaultExternalService() service3 := defaultExternalService()
services := []*v1.Service{etpLocalservice1, etpLocalservice2, service3} services := []*v1.Service{etpLocalservice1, etpLocalservice2, service3}
@ -761,8 +753,8 @@ func TestNodeChangesForStableNodeSetEnabled(t *testing.T) {
syncCallErr bool syncCallErr bool
} }
etpLocalservice1 := newETPLocalService("s0", v1.ServiceTypeLoadBalancer) etpLocalservice1 := newService("s0", v1.ServiceTypeLoadBalancer, tweakAddETP(v1.ServiceExternalTrafficPolicyLocal))
etpLocalservice2 := newETPLocalService("s1", v1.ServiceTypeLoadBalancer) etpLocalservice2 := newService("s1", v1.ServiceTypeLoadBalancer, tweakAddETP(v1.ServiceExternalTrafficPolicyLocal))
service3 := defaultExternalService() service3 := defaultExternalService()
services := []*v1.Service{etpLocalservice1, etpLocalservice2, service3} services := []*v1.Service{etpLocalservice1, etpLocalservice2, service3}