github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-12 05:21:58 +00:00
Code Issues Projects Releases Wiki Activity

Chore: add load balancer tweak function

Browse Source
This commit is contained in:
mowangdk 2023-03-26 12:19:50 +08:00
parent 1c465c51e8
commit 28db34a231
1 changed files with 8 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -99,6 +99,12 @@ func tweakAddPorts(protocol v1.Protocol, targetPort int) serviceTweak {
}
}
func tweakAddLBClass(loadBalancerClass *string) serviceTweak {
return func(s *v1.Service) {
s.Spec.LoadBalancerClass = loadBalancerClass
}
}
// Wrap newService so that you don't have to call default arguments again and again.
func defaultExternalService() *v1.Service {
return newService("external-balancer", v1.ServiceTypeLoadBalancer)
@ -201,16 +207,7 @@ func TestSyncLoadBalancerIfNeeded(t *testing.T) {
},
{
desc: "service specifies loadBalancerClass",
service: &v1.Service{
ObjectMeta: metav1.ObjectMeta{
Name: "with-external-balancer",
Namespace: "default",
},
Spec: v1.ServiceSpec{
Type: v1.ServiceTypeLoadBalancer,
LoadBalancerClass: utilpointer.StringPtr("custom-loadbalancer"),
},
},
service: newService("with-external-balancer", v1.ServiceTypeLoadBalancer, tweakAddLBClass(utilpointer.String("custom-loadbalancer"))),
expectOp: deleteLoadBalancer,
expectCreateAttempt: false,
expectPatchStatus: false,
@ -218,20 +215,7 @@ func TestSyncLoadBalancerIfNeeded(t *testing.T) {
},
{
desc: "service doesn't specify loadBalancerClass",
service: &v1.Service{
ObjectMeta: metav1.ObjectMeta{
Name: "with-external-balancer",
Namespace: "default",
},
Spec: v1.ServiceSpec{
Ports: []v1.ServicePort{{
Port: 80,
Protocol: v1.ProtocolSCTP,
}},
Type: v1.ServiceTypeLoadBalancer,
LoadBalancerClass: nil,
},
},
service: newService("with-external-balancer", v1.ServiceTypeLoadBalancer, tweakAddLBClass(nil), tweakAddPorts(v1.ProtocolSCTP, 0)),
expectOp: ensureLoadBalancer,
expectCreateAttempt: true,
expectPatchStatus: true,