mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-12-22 21:10:22 +00:00
113 lines
3.8 KiB
Go
113 lines
3.8 KiB
Go
|
/*
|
||
|
Copyright 2017 The Kubernetes Authors.
|
||
|
|
||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
you may not use this file except in compliance with the License.
|
||
|
You may obtain a copy of the License at
|
||
|
|
||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
||
|
Unless required by applicable law or agreed to in writing, software
|
||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
See the License for the specific language governing permissions and
|
||
|
limitations under the License.
|
||
|
*/
|
||
|
|
||
|
package upgrades
|
||
|
|
||
|
import (
|
||
|
"k8s.io/api/core/v1"
|
||
|
"k8s.io/apimachinery/pkg/util/wait"
|
||
|
"k8s.io/kubernetes/test/e2e/framework"
|
||
|
|
||
|
. "github.com/onsi/ginkgo"
|
||
|
)
|
||
|
|
||
|
// ServiceUpgradeTest tests that a service is available before and
|
||
|
// after a cluster upgrade. During a master-only upgrade, it will test
|
||
|
// that a service remains available during the upgrade.
|
||
|
type ServiceUpgradeTest struct {
|
||
|
jig *framework.ServiceTestJig
|
||
|
tcpService *v1.Service
|
||
|
tcpIngressIP string
|
||
|
svcPort int
|
||
|
}
|
||
|
|
||
|
func (ServiceUpgradeTest) Name() string { return "service-upgrade" }
|
||
|
|
||
|
func shouldTestPDBs() bool { return framework.ProviderIs("gce", "gke") }
|
||
|
|
||
|
// Setup creates a service with a load balancer and makes sure it's reachable.
|
||
|
func (t *ServiceUpgradeTest) Setup(f *framework.Framework) {
|
||
|
serviceName := "service-test"
|
||
|
jig := framework.NewServiceTestJig(f.ClientSet, serviceName)
|
||
|
|
||
|
ns := f.Namespace
|
||
|
|
||
|
By("creating a TCP service " + serviceName + " with type=LoadBalancer in namespace " + ns.Name)
|
||
|
tcpService := jig.CreateTCPServiceOrFail(ns.Name, func(s *v1.Service) {
|
||
|
s.Spec.Type = v1.ServiceTypeLoadBalancer
|
||
|
})
|
||
|
tcpService = jig.WaitForLoadBalancerOrFail(ns.Name, tcpService.Name, framework.LoadBalancerCreateTimeoutDefault)
|
||
|
jig.SanityCheckService(tcpService, v1.ServiceTypeLoadBalancer)
|
||
|
|
||
|
// Get info to hit it with
|
||
|
tcpIngressIP := framework.GetIngressPoint(&tcpService.Status.LoadBalancer.Ingress[0])
|
||
|
svcPort := int(tcpService.Spec.Ports[0].Port)
|
||
|
|
||
|
By("creating pod to be part of service " + serviceName)
|
||
|
rc := jig.RunOrFail(ns.Name, jig.AddRCAntiAffinity)
|
||
|
|
||
|
if shouldTestPDBs() {
|
||
|
By("creating a PodDisruptionBudget to cover the ReplicationController")
|
||
|
jig.CreatePDBOrFail(ns.Name, rc)
|
||
|
}
|
||
|
|
||
|
// Hit it once before considering ourselves ready
|
||
|
By("hitting the pod through the service's LoadBalancer")
|
||
|
jig.TestReachableHTTP(tcpIngressIP, svcPort, framework.LoadBalancerLagTimeoutDefault)
|
||
|
|
||
|
t.jig = jig
|
||
|
t.tcpService = tcpService
|
||
|
t.tcpIngressIP = tcpIngressIP
|
||
|
t.svcPort = svcPort
|
||
|
}
|
||
|
|
||
|
// Test runs a connectivity check to the service.
|
||
|
func (t *ServiceUpgradeTest) Test(f *framework.Framework, done <-chan struct{}, upgrade UpgradeType) {
|
||
|
switch upgrade {
|
||
|
case MasterUpgrade:
|
||
|
t.test(f, done, true)
|
||
|
case NodeUpgrade:
|
||
|
// Node upgrades should test during disruption only on GCE/GKE for now.
|
||
|
t.test(f, done, shouldTestPDBs())
|
||
|
default:
|
||
|
t.test(f, done, false)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Teardown cleans up any remaining resources.
|
||
|
func (t *ServiceUpgradeTest) Teardown(f *framework.Framework) {
|
||
|
// rely on the namespace deletion to clean up everything
|
||
|
}
|
||
|
|
||
|
func (t *ServiceUpgradeTest) test(f *framework.Framework, done <-chan struct{}, testDuringDisruption bool) {
|
||
|
if testDuringDisruption {
|
||
|
// Continuous validation
|
||
|
By("continuously hitting the pod through the service's LoadBalancer")
|
||
|
wait.Until(func() {
|
||
|
t.jig.TestReachableHTTP(t.tcpIngressIP, t.svcPort, framework.LoadBalancerLagTimeoutDefault)
|
||
|
}, framework.Poll, done)
|
||
|
} else {
|
||
|
// Block until upgrade is done
|
||
|
By("waiting for upgrade to finish without checking if service remains up")
|
||
|
<-done
|
||
|
}
|
||
|
|
||
|
// Sanity check and hit it once more
|
||
|
By("hitting the pod through the service's LoadBalancer")
|
||
|
t.jig.TestReachableHTTP(t.tcpIngressIP, t.svcPort, framework.LoadBalancerLagTimeoutDefault)
|
||
|
t.jig.SanityCheckService(t.tcpService, v1.ServiceTypeLoadBalancer)
|
||
|
}
|