summaryrefslogtreecommitdiff
path: root/vendor/k8s.io/kubernetes/test/e2e/apps/statefulset.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/k8s.io/kubernetes/test/e2e/apps/statefulset.go')
-rw-r--r--vendor/k8s.io/kubernetes/test/e2e/apps/statefulset.go185
1 files changed, 185 insertions, 0 deletions
diff --git a/vendor/k8s.io/kubernetes/test/e2e/apps/statefulset.go b/vendor/k8s.io/kubernetes/test/e2e/apps/statefulset.go
index fb82ba005..568c89eec 100644
--- a/vendor/k8s.io/kubernetes/test/e2e/apps/statefulset.go
+++ b/vendor/k8s.io/kubernetes/test/e2e/apps/statefulset.go
@@ -75,6 +75,8 @@ const (
statefulSetTimeout = 10 * time.Minute
// statefulPodTimeout is a timeout for stateful pods to change state
statefulPodTimeout = 5 * time.Minute
+
+ testFinalizer = "example.com/test-finalizer"
)
var httpProbe = &v1.Probe{
@@ -508,6 +510,51 @@ var _ = SIGDescribe("StatefulSet", func() {
})
+ ginkgo.It("should perform canary updates and phased rolling updates of template modifications for partiton1 and delete pod-0 without failing container", func(ctx context.Context) {
+ ginkgo.By("Creating a new StatefulSet without failing container")
+ ss := e2estatefulset.NewStatefulSet("ss2", ns, headlessSvcName, 3, nil, nil, labels)
+ deletingPodForRollingUpdatePartitionTest(ctx, f, c, ns, ss)
+ })
+
+ ginkgo.It("should perform canary updates and phased rolling updates of template modifications for partiton1 and delete pod-0 with failing container", func(ctx context.Context) {
+ ginkgo.By("Creating a new StatefulSet with failing container")
+ ss := e2estatefulset.NewStatefulSet("ss3", ns, headlessSvcName, 3, nil, nil, labels)
+ ss.Spec.Template.Spec.Containers = append(ss.Spec.Template.Spec.Containers, v1.Container{
+ Name: "sleep-exit-with-1",
+ Image: imageutils.GetE2EImage(imageutils.BusyBox),
+ Command: []string{"sh", "-c"},
+ Args: []string{`
+ echo "Running in pod $POD_NAME"
+ _term(){
+ echo "Received SIGTERM signal"
+ if [ "${POD_NAME}" = "ss3-0" ]; then
+ exit 1
+ else
+ exit 0
+ fi
+ }
+ trap _term SIGTERM
+ while true; do
+ echo "Running in infinite loop in $POD_NAME"
+ sleep 1
+ done
+ `,
+ },
+ Env: []v1.EnvVar{
+ {
+ Name: "POD_NAME",
+ ValueFrom: &v1.EnvVarSource{
+ FieldRef: &v1.ObjectFieldSelector{
+ APIVersion: "v1",
+ FieldPath: "metadata.name",
+ },
+ },
+ },
+ },
+ })
+ deletingPodForRollingUpdatePartitionTest(ctx, f, c, ns, ss)
+ })
+
// Do not mark this as Conformance.
// The legacy OnDelete strategy only exists for backward compatibility with pre-v1 APIs.
ginkgo.It("should implement legacy replacement when the update strategy is OnDelete", func(ctx context.Context) {
@@ -1941,6 +1988,144 @@ func rollbackTest(ctx context.Context, c clientset.Interface, ns string, ss *app
}
}
+// This function is used canary updates and phased rolling updates of template modifications for partiton1 and delete pod-0
+func deletingPodForRollingUpdatePartitionTest(ctx context.Context, f *framework.Framework, c clientset.Interface, ns string, ss *appsv1.StatefulSet) {
+ setHTTPProbe(ss)
+ ss.Spec.UpdateStrategy = appsv1.StatefulSetUpdateStrategy{
+ Type: appsv1.RollingUpdateStatefulSetStrategyType,
+ RollingUpdate: func() *appsv1.RollingUpdateStatefulSetStrategy {
+ return &appsv1.RollingUpdateStatefulSetStrategy{
+ Partition: pointer.Int32(1),
+ }
+ }(),
+ }
+ ss, err := c.AppsV1().StatefulSets(ns).Create(ctx, ss, metav1.CreateOptions{})
+ framework.ExpectNoError(err)
+ e2estatefulset.WaitForRunningAndReady(ctx, c, *ss.Spec.Replicas, ss)
+ ss = waitForStatus(ctx, c, ss)
+ currentRevision, updateRevision := ss.Status.CurrentRevision, ss.Status.UpdateRevision
+ gomega.Expect(currentRevision).To(gomega.Equal(updateRevision), fmt.Sprintf("StatefulSet %s/%s created with update revision %s not equal to current revision %s",
+ ss.Namespace, ss.Name, updateRevision, currentRevision))
+ pods := e2estatefulset.GetPodList(ctx, c, ss)
+ for i := range pods.Items {
+ gomega.Expect(pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel]).To(gomega.Equal(currentRevision), fmt.Sprintf("Pod %s/%s revision %s is not equal to currentRevision %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel],
+ currentRevision))
+ }
+
+ ginkgo.By("Adding finalizer for pod-0")
+ pod0name := getStatefulSetPodNameAtIndex(0, ss)
+ pod0, err := c.CoreV1().Pods(ns).Get(ctx, pod0name, metav1.GetOptions{})
+ framework.ExpectNoError(err)
+ pod0.Finalizers = append(pod0.Finalizers, testFinalizer)
+ pod0, err = c.CoreV1().Pods(ss.Namespace).Update(ctx, pod0, metav1.UpdateOptions{})
+ framework.ExpectNoError(err)
+ pods.Items[0] = *pod0
+ defer e2epod.NewPodClient(f).RemoveFinalizer(ctx, pod0.Name, testFinalizer)
+
+ ginkgo.By("Updating image on StatefulSet")
+ newImage := NewWebserverImage
+ oldImage := ss.Spec.Template.Spec.Containers[0].Image
+ ginkgo.By(fmt.Sprintf("Updating stateful set template: update image from %s to %s", oldImage, newImage))
+ gomega.Expect(oldImage).ToNot(gomega.Equal(newImage), "Incorrect test setup: should update to a different image")
+ ss, err = updateStatefulSetWithRetries(ctx, c, ns, ss.Name, func(update *appsv1.StatefulSet) {
+ update.Spec.Template.Spec.Containers[0].Image = newImage
+ })
+ framework.ExpectNoError(err)
+
+ ginkgo.By("Creating a new revision")
+ ss = waitForStatus(ctx, c, ss)
+ currentRevision, updateRevision = ss.Status.CurrentRevision, ss.Status.UpdateRevision
+ gomega.Expect(currentRevision).ToNot(gomega.Equal(updateRevision), "Current revision should not equal update revision during rolling update")
+
+ ginkgo.By("Await for all replicas running, all are updated but pod-0")
+ e2estatefulset.WaitForState(ctx, c, ss, func(set2 *appsv1.StatefulSet, pods2 *v1.PodList) (bool, error) {
+ ss = set2
+ pods = pods2
+ if ss.Status.UpdatedReplicas == *ss.Spec.Replicas-1 && ss.Status.Replicas == *ss.Spec.Replicas && ss.Status.ReadyReplicas == *ss.Spec.Replicas {
+ // rolling updated is not completed, because replica 0 isn't ready
+ return true, nil
+ }
+ return false, nil
+ })
+
+ ginkgo.By("Verify pod images before pod-0 deletion and recreation")
+ for i := range pods.Items {
+ if i < int(*ss.Spec.UpdateStrategy.RollingUpdate.Partition) {
+ gomega.Expect(pods.Items[i].Spec.Containers[0].Image).To(gomega.Equal(oldImage), fmt.Sprintf("Pod %s/%s has image %s not equal to oldimage image %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Spec.Containers[0].Image,
+ oldImage))
+ gomega.Expect(pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel]).To(gomega.Equal(currentRevision), fmt.Sprintf("Pod %s/%s has revision %s not equal to current revision %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel],
+ currentRevision))
+ } else {
+ gomega.Expect(pods.Items[i].Spec.Containers[0].Image).To(gomega.Equal(newImage), fmt.Sprintf("Pod %s/%s has image %s not equal to new image %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Spec.Containers[0].Image,
+ newImage))
+ gomega.Expect(pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel]).To(gomega.Equal(updateRevision), fmt.Sprintf("Pod %s/%s has revision %s not equal to new revision %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel],
+ updateRevision))
+ }
+ }
+
+ ginkgo.By("Deleting the pod-0 so that kubelet terminates it and StatefulSet controller recreates it")
+ deleteStatefulPodAtIndex(ctx, c, 0, ss)
+ ginkgo.By("Await for two replicas to be updated, while the pod-0 is not running")
+ e2estatefulset.WaitForState(ctx, c, ss, func(set2 *appsv1.StatefulSet, pods2 *v1.PodList) (bool, error) {
+ ss = set2
+ pods = pods2
+ return ss.Status.ReadyReplicas == *ss.Spec.Replicas-1, nil
+ })
+
+ ginkgo.By(fmt.Sprintf("Removing finalizer from pod-0 (%v/%v) to allow recreation", pod0.Namespace, pod0.Name))
+ e2epod.NewPodClient(f).RemoveFinalizer(ctx, pod0.Name, testFinalizer)
+
+ ginkgo.By("Await for recreation of pod-0, so that all replicas are running")
+ e2estatefulset.WaitForState(ctx, c, ss, func(set2 *appsv1.StatefulSet, pods2 *v1.PodList) (bool, error) {
+ ss = set2
+ pods = pods2
+ return ss.Status.ReadyReplicas == *ss.Spec.Replicas, nil
+ })
+
+ ginkgo.By("Verify pod images after pod-0 deletion and recreation")
+ pods = e2estatefulset.GetPodList(ctx, c, ss)
+ for i := range pods.Items {
+ if i < int(*ss.Spec.UpdateStrategy.RollingUpdate.Partition) {
+ gomega.Expect(pods.Items[i].Spec.Containers[0].Image).To(gomega.Equal(oldImage), fmt.Sprintf("Pod %s/%s has image %s not equal to current image %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Spec.Containers[0].Image,
+ oldImage))
+ gomega.Expect(pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel]).To(gomega.Equal(currentRevision), fmt.Sprintf("Pod %s/%s has revision %s not equal to current revision %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel],
+ currentRevision))
+ } else {
+ gomega.Expect(pods.Items[i].Spec.Containers[0].Image).To(gomega.Equal(newImage), fmt.Sprintf("Pod %s/%s has image %s not equal to new image %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Spec.Containers[0].Image,
+ newImage))
+ gomega.Expect(pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel]).To(gomega.Equal(updateRevision), fmt.Sprintf("Pod %s/%s has revision %s not equal to new revision %s",
+ pods.Items[i].Namespace,
+ pods.Items[i].Name,
+ pods.Items[i].Labels[appsv1.StatefulSetRevisionLabel],
+ updateRevision))
+ }
+ }
+}
+
// confirmStatefulPodCount asserts that the current number of Pods in ss is count, waiting up to timeout for ss to
// to scale to count.
func confirmStatefulPodCount(ctx context.Context, c clientset.Interface, count int, ss *appsv1.StatefulSet, timeout time.Duration, hard bool) {
generated by cgit v1.2.3 (git 2.52.0) at 2026-08-19 23:52:01 +0000