github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-15 14:53:44 +00:00
Code Issues Projects Releases Wiki Activity

Chore: add app protocol tweak function

Browse Source
This commit is contained in:
mowangdk 2023-04-25 10:58:59 +08:00
parent 06993afdd6
commit 440d5910f7
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -117,6 +117,12 @@ func tweakAddDeletionTimestamp(time time.Time) serviceTweak {
} }
} }
func tweakAddAppProtocol(appProtocol string) serviceTweak {
return func(s *v1.Service) {
s.Spec.Ports[0].AppProtocol = &appProtocol
}
}
// Wrap newService so that you don't have to call default arguments again and again. // Wrap newService so that you don't have to call default arguments again and again.
func defaultExternalService() *v1.Service { func defaultExternalService() *v1.Service {
return newService("external-balancer", v1.ServiceTypeLoadBalancer) return newService("external-balancer", v1.ServiceTypeLoadBalancer)
@ -1440,9 +1446,7 @@ func TestNeedsUpdate(t *testing.T) {
}, { }, {
testName: "If appProtocol is set to a different value", testName: "If appProtocol is set to a different value",
updateFn: func() { updateFn: func() {
protocol := "http" oldSvc = newService("tcp-service", v1.ServiceTypeLoadBalancer, tweakAddPorts(v1.ProtocolTCP, 22), tweakAddAppProtocol("http"))
oldSvc = newService("tcp-service", v1.ServiceTypeLoadBalancer, tweakAddPorts(v1.ProtocolTCP, 22))
oldSvc.Spec.Ports[0].AppProtocol = &protocol
newSvc = oldSvc.DeepCopy() newSvc = oldSvc.DeepCopy()
newProtocol := "tcp" newProtocol := "tcp"
newSvc.Spec.Ports[0].AppProtocol = &newProtocol newSvc.Spec.Ports[0].AppProtocol = &newProtocol