/* Copyright 2022 The Kruise 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 e2e import ( "context" "fmt" "sort" "strings" "time" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" appsv1alpha1 "github.com/openkruise/kruise-api/apps/v1alpha1" appsv1beta1 "github.com/openkruise/kruise-api/apps/v1beta1" "github.com/openkruise/rollouts/api/v1beta1" "github.com/openkruise/rollouts/pkg/util" apps "k8s.io/api/apps/v1" v1 "k8s.io/api/core/v1" netv1 "k8s.io/api/networking/v1" "k8s.io/apimachinery/pkg/api/errors" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured" "k8s.io/apimachinery/pkg/types" "k8s.io/apimachinery/pkg/util/intstr" "k8s.io/client-go/util/retry" "k8s.io/klog/v2" utilpointer "k8s.io/utils/pointer" "sigs.k8s.io/controller-runtime/pkg/client" // "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured" // "k8s.io/apimachinery/pkg/util/intstr" // gatewayv1beta1 "sigs.k8s.io/gateway-api/apis/v1beta1" // "github.com/openkruise/rollouts/api/v1alpha1" // "k8s.io/apimachinery/pkg/api/errors" ) var _ = SIGDescribe("Rollout v1beta1", func() { var namespace string DumpAllResources := func() { rollout := &v1beta1.RolloutList{} _ = k8sClient.List(context.TODO(), rollout, client.InNamespace(namespace)) fmt.Println(util.DumpJSON(rollout)) batch := &v1beta1.BatchReleaseList{} _ = k8sClient.List(context.TODO(), batch, client.InNamespace(namespace)) fmt.Println(util.DumpJSON(batch)) deploy := &apps.DeploymentList{} _ = k8sClient.List(context.TODO(), deploy, client.InNamespace(namespace)) fmt.Println(util.DumpJSON(deploy)) rs := &apps.ReplicaSetList{} _ = k8sClient.List(context.TODO(), rs, client.InNamespace(namespace)) fmt.Println(util.DumpJSON(rs)) cloneSet := &appsv1alpha1.CloneSetList{} _ = k8sClient.List(context.TODO(), cloneSet, client.InNamespace(namespace)) fmt.Println(util.DumpJSON(cloneSet)) sts := &apps.StatefulSetList{} _ = k8sClient.List(context.TODO(), sts, client.InNamespace(namespace)) fmt.Println(util.DumpJSON(sts)) asts := &appsv1beta1.StatefulSetList{} _ = k8sClient.List(context.TODO(), asts, client.InNamespace(namespace)) fmt.Println(util.DumpJSON(asts)) } CreateObject := func(object client.Object, options ...client.CreateOption) { object.SetNamespace(namespace) Expect(k8sClient.Create(context.TODO(), object)).NotTo(HaveOccurred()) } GetObject := func(name string, object client.Object) error { key := types.NamespacedName{Namespace: namespace, Name: name} return k8sClient.Get(context.TODO(), key, object) } getRolloutCondition := func(status v1beta1.RolloutStatus, condType v1beta1.RolloutConditionType) *v1beta1.RolloutCondition { for i := range status.Conditions { c := status.Conditions[i] if c.Type == condType { return &c } } return nil } UpdateDeployment := func(object *apps.Deployment) *apps.Deployment { var clone *apps.Deployment Expect(retry.RetryOnConflict(retry.DefaultRetry, func() error { clone = &apps.Deployment{} err := GetObject(object.Name, clone) if err != nil { return err } clone.Spec.Replicas = utilpointer.Int32(*object.Spec.Replicas) clone.Spec.Template = *object.Spec.Template.DeepCopy() clone.Labels = mergeMap(clone.Labels, object.Labels) clone.Annotations = mergeMap(clone.Annotations, object.Annotations) clone.Spec.Paused = object.Spec.Paused return k8sClient.Update(context.TODO(), clone) })).NotTo(HaveOccurred()) return clone } UpdateCloneSet := func(object *appsv1alpha1.CloneSet) *appsv1alpha1.CloneSet { var clone *appsv1alpha1.CloneSet Expect(retry.RetryOnConflict(retry.DefaultRetry, func() error { clone = &appsv1alpha1.CloneSet{} err := GetObject(object.Name, clone) if err != nil { return err } clone.Spec.Replicas = utilpointer.Int32(*object.Spec.Replicas) clone.Spec.Template = *object.Spec.Template.DeepCopy() clone.Labels = mergeMap(clone.Labels, object.Labels) clone.Annotations = mergeMap(clone.Annotations, object.Annotations) return k8sClient.Update(context.TODO(), clone) })).NotTo(HaveOccurred()) return clone } // UpdateDaemonSet := func(object *appsv1alpha1.DaemonSet) *appsv1alpha1.DaemonSet { // var daemon *appsv1alpha1.DaemonSet // Expect(retry.RetryOnConflict(retry.DefaultRetry, func() error { // daemon = &appsv1alpha1.DaemonSet{} // err := GetObject(object.Name, daemon) // if err != nil { // return err // } // // daemon.Spec.Replicas = utilpointer.Int32(*object.Spec.Replicas) // daemon.Spec.Template = *object.Spec.Template.DeepCopy() // daemon.Spec.UpdateStrategy = *object.Spec.UpdateStrategy.DeepCopy() // daemon.Labels = mergeMap(daemon.Labels, object.Labels) // daemon.Annotations = mergeMap(daemon.Annotations, object.Annotations) // return k8sClient.Update(context.TODO(), daemon) // })).NotTo(HaveOccurred()) // return daemon // } UpdateNativeStatefulSet := func(object *apps.StatefulSet) *apps.StatefulSet { var clone *apps.StatefulSet Expect(retry.RetryOnConflict(retry.DefaultRetry, func() error { clone = &apps.StatefulSet{} err := GetObject(object.Name, clone) if err != nil { return err } clone.Spec.Replicas = utilpointer.Int32(*object.Spec.Replicas) clone.Spec.Template = *object.Spec.Template.DeepCopy() clone.Labels = mergeMap(clone.Labels, object.Labels) clone.Annotations = mergeMap(clone.Annotations, object.Annotations) return k8sClient.Update(context.TODO(), clone) })).NotTo(HaveOccurred()) return clone } UpdateAdvancedStatefulSet := func(object *appsv1beta1.StatefulSet) *appsv1beta1.StatefulSet { var clone *appsv1beta1.StatefulSet Expect(retry.RetryOnConflict(retry.DefaultRetry, func() error { clone = &appsv1beta1.StatefulSet{} err := GetObject(object.Name, clone) if err != nil { return err } clone.Spec.Replicas = utilpointer.Int32(*object.Spec.Replicas) clone.Spec.Template = *object.Spec.Template.DeepCopy() clone.Labels = mergeMap(clone.Labels, object.Labels) clone.Annotations = mergeMap(clone.Annotations, object.Annotations) return k8sClient.Update(context.TODO(), clone) })).NotTo(HaveOccurred()) return clone } UpdateRollout := func(object *v1beta1.Rollout) *v1beta1.Rollout { var clone *v1beta1.Rollout Expect(retry.RetryOnConflict(retry.DefaultRetry, func() error { clone = &v1beta1.Rollout{} err := GetObject(object.Name, clone) if err != nil { return err } clone.Spec = *object.Spec.DeepCopy() return k8sClient.Update(context.TODO(), clone) })).NotTo(HaveOccurred()) return clone } ResumeRolloutCanary := func(name string) { Eventually(func() bool { clone := &v1beta1.Rollout{} Expect(GetObject(name, clone)).NotTo(HaveOccurred()) if clone.Status.CanaryStatus.CurrentStepState != v1beta1.CanaryStepStatePaused { fmt.Println("resume rollout success, and CurrentStepState", util.DumpJSON(clone.Status)) return true } body := fmt.Sprintf(`{"status":{"canaryStatus":{"currentStepState":"%s"}}}`, v1beta1.CanaryStepStateReady) Expect(k8sClient.Status().Patch(context.TODO(), clone, client.RawPatch(types.MergePatchType, []byte(body)))).NotTo(HaveOccurred()) return false }, 10*time.Second, time.Millisecond*500).Should(BeTrue()) } RolloutJumpCanaryStep := func(name string, target int) { Eventually(func() bool { clone := &v1beta1.Rollout{} Expect(GetObject(name, clone)).NotTo(HaveOccurred()) if clone.Status.CanaryStatus.CurrentStepState != v1beta1.CanaryStepStatePaused { fmt.Println("Jump successfully, and current status ", util.DumpJSON(clone.Status)) return true } body := fmt.Sprintf(`{"status":{"canaryStatus":{"nextStepIndex":%d}}}`, target) Expect(k8sClient.Status().Patch(context.TODO(), clone, client.RawPatch(types.MergePatchType, []byte(body)))).NotTo(HaveOccurred()) return false }, 10*time.Second, time.Second).Should(BeTrue()) } // RolloutJumpBlueGreenStep := func(name string, target int) { // Eventually(func() bool { // clone := &v1alpha1.Rollout{} // Expect(GetObject(name, clone)).NotTo(HaveOccurred()) // if clone.Status.CanaryStatus.CurrentStepState !=v1beta1.CanaryStepStatePaused { // fmt.Println("Jump successfully, and current status ", util.DumpJSON(clone.Status)) // return true // } // body := fmt.Sprintf(`{"status":{"blueGreenStatus":{"nextStepIndex":"%d"}}}`, target) // Expect(k8sClient.Status().Patch(context.TODO(), clone, client.RawPatch(types.MergePatchType, []byte(body)))).NotTo(HaveOccurred()) // return false // }, 10*time.Second, time.Second).Should(BeTrue()) // } WaitDeploymentAllPodsReady := func(deployment *apps.Deployment) { Eventually(func() bool { clone := &apps.Deployment{} Expect(GetObject(deployment.Name, clone)).NotTo(HaveOccurred()) return clone.Status.ObservedGeneration == clone.Generation && *clone.Spec.Replicas == clone.Status.UpdatedReplicas && *clone.Spec.Replicas == clone.Status.ReadyReplicas && *clone.Spec.Replicas == clone.Status.Replicas }, 5*time.Minute, time.Second).Should(BeTrue()) } WaitCloneSetAllPodsReady := func(cloneset *appsv1alpha1.CloneSet) { Eventually(func() bool { clone := &appsv1alpha1.CloneSet{} Expect(GetObject(cloneset.Name, clone)).NotTo(HaveOccurred()) return clone.Status.ObservedGeneration == clone.Generation && *clone.Spec.Replicas == clone.Status.UpdatedReplicas && *clone.Spec.Replicas == clone.Status.ReadyReplicas && *clone.Spec.Replicas == clone.Status.Replicas }, 5*time.Minute, time.Second).Should(BeTrue()) } WaitNativeStatefulSetPodsReady := func(statefulset *apps.StatefulSet) { Eventually(func() bool { set := &apps.StatefulSet{} Expect(GetObject(statefulset.Name, set)).NotTo(HaveOccurred()) return set.Status.ObservedGeneration == set.Generation && *set.Spec.Replicas == set.Status.UpdatedReplicas && *set.Spec.Replicas == set.Status.ReadyReplicas && *set.Spec.Replicas == set.Status.Replicas }, 20*time.Minute, 3*time.Second).Should(BeTrue()) } WaitAdvancedStatefulSetPodsReady := func(statefulset *appsv1beta1.StatefulSet) { Eventually(func() bool { set := &appsv1beta1.StatefulSet{} Expect(GetObject(statefulset.Name, set)).NotTo(HaveOccurred()) return set.Status.ObservedGeneration == set.Generation && *set.Spec.Replicas == set.Status.UpdatedReplicas && *set.Spec.Replicas == set.Status.ReadyReplicas && *set.Spec.Replicas == set.Status.Replicas }, 20*time.Minute, 3*time.Second).Should(BeTrue()) } // WaitDaemonSetAllPodsReady := func(daemonset *appsv1alpha1.DaemonSet) { // Eventually(func() bool { // daemon := &appsv1alpha1.DaemonSet{} // Expect(GetObject(daemonset.Name, daemon)).NotTo(HaveOccurred()) // klog.Infof("DaemonSet updateStrategy(%s) Generation(%d) ObservedGeneration(%d) DesiredNumberScheduled(%d) UpdatedNumberScheduled(%d) NumberReady(%d)", // util.DumpJSON(daemon.Spec.UpdateStrategy), daemon.Generation, daemon.Status.ObservedGeneration, daemon.Status.DesiredNumberScheduled, daemon.Status.UpdatedNumberScheduled, daemon.Status.NumberReady) // return daemon.Status.ObservedGeneration == daemon.Generation && daemon.Status.DesiredNumberScheduled == daemon.Status.UpdatedNumberScheduled && daemon.Status.DesiredNumberScheduled == daemon.Status.NumberReady // }, 5*time.Minute, time.Second).Should(BeTrue()) // } // WaitDeploymentReplicas := func(deployment *apps.Deployment) { // Eventually(func() bool { // clone := &apps.Deployment{} // Expect(GetObject(deployment.Name, clone)).NotTo(HaveOccurred()) // return clone.Status.ObservedGeneration == clone.Generation && // *clone.Spec.Replicas == clone.Status.ReadyReplicas && *clone.Spec.Replicas == clone.Status.Replicas // }, 10*time.Minute, time.Second).Should(BeTrue()) // } WaitRolloutCanaryStepPaused := func(name string, stepIndex int32) { start := time.Now() Eventually(func() bool { if start.Add(time.Minute * 5).Before(time.Now()) { DumpAllResources() Expect(true).Should(BeFalse()) } clone := &v1beta1.Rollout{} Expect(GetObject(name, clone)).NotTo(HaveOccurred()) if clone.Status.CanaryStatus == nil { return false } klog.Infof("current step:%v target step:%v current step state %v", clone.Status.CanaryStatus.CurrentStepIndex, stepIndex, clone.Status.CanaryStatus.CurrentStepState) return clone.Status.CanaryStatus.CurrentStepIndex == stepIndex && clone.Status.CanaryStatus.CurrentStepState == v1beta1.CanaryStepStatePaused }, 20*time.Minute, time.Second).Should(BeTrue()) } WaitRolloutStatusPhase := func(name string, phase v1beta1.RolloutPhase) { Eventually(func() bool { clone := &v1beta1.Rollout{} Expect(GetObject(name, clone)).NotTo(HaveOccurred()) return clone.Status.Phase == phase }, 20*time.Minute, time.Second).Should(BeTrue()) } WaitRolloutWorkloadGeneration := func(name string, generation int64) { Eventually(func() bool { clone := &v1beta1.Rollout{} Expect(GetObject(name, clone)).NotTo(HaveOccurred()) return clone.Status.CanaryStatus.ObservedWorkloadGeneration == generation }, time.Minute, time.Second).Should(BeTrue()) } WaitRolloutNotFound := func(name string) { Eventually(func() bool { clone := &v1beta1.Rollout{} err := GetObject(name, clone) if err == nil { return false } else if errors.IsNotFound(err) { return true } else { Expect(err).NotTo(HaveOccurred()) return false } }, 5*time.Minute, time.Second).Should(BeTrue()) } GetCanaryDeployment := func(stable *apps.Deployment) (*apps.Deployment, error) { canaryList := &apps.DeploymentList{} selector, _ := metav1.LabelSelectorAsSelector(&metav1.LabelSelector{MatchLabels: map[string]string{util.CanaryDeploymentLabel: stable.Name}}) err := k8sClient.List(context.TODO(), canaryList, &client.ListOptions{Namespace: stable.Namespace, LabelSelector: selector}) if err != nil { return nil, err } else if len(canaryList.Items) == 0 { return nil, nil } sort.Slice(canaryList.Items, func(i, j int) bool { return canaryList.Items[j].CreationTimestamp.Before(&canaryList.Items[i].CreationTimestamp) }) return &canaryList.Items[0], nil } ListPods := func(namespace string, labelSelector *metav1.LabelSelector) ([]*v1.Pod, error) { appList := &v1.PodList{} selector, _ := metav1.LabelSelectorAsSelector(labelSelector) err := k8sClient.List(context.TODO(), appList, &client.ListOptions{Namespace: namespace, LabelSelector: selector}) if err != nil { return nil, err } apps := make([]*v1.Pod, 0) for i := range appList.Items { pod := &appList.Items[i] if pod.DeletionTimestamp.IsZero() { apps = append(apps, pod) } } return apps, nil } CheckPodBatchLabel := func(namespace string, labelSelector *metav1.LabelSelector, rolloutID, batchID string, expected int) { pods, err := ListPods(namespace, labelSelector) Expect(err).NotTo(HaveOccurred()) count := 0 for _, pod := range pods { if pod.Labels[v1beta1.RolloutIDLabel] == rolloutID && pod.Labels[v1beta1.RolloutBatchIDLabel] == batchID { count++ } } Expect(count).Should(BeNumerically("==", expected)) } ListReplicaSet := func(d *apps.Deployment) []*apps.ReplicaSet { var rss []*apps.ReplicaSet rsLister := &apps.ReplicaSetList{} selectorOpt, _ := metav1.LabelSelectorAsSelector(d.Spec.Selector) err := k8sClient.List(context.TODO(), rsLister, &client.ListOptions{LabelSelector: selectorOpt, Namespace: d.Namespace}) Expect(err).NotTo(HaveOccurred()) for i := range rsLister.Items { rs := &rsLister.Items[i] if !rs.DeletionTimestamp.IsZero() { continue } rss = append(rss, rs) } return rss } GetStableRSRevision := func(d *apps.Deployment) string { rss := ListReplicaSet(d) _, stable := util.FindCanaryAndStableReplicaSet(rss, d) if stable != nil { return stable.Labels[apps.DefaultDeploymentUniqueLabelKey] } return "" } GetCanaryRSRevision := func(d *apps.Deployment) string { rss := ListReplicaSet(d) canary, _ := util.FindCanaryAndStableReplicaSet(rss, d) if canary != nil { return canary.Labels[apps.DefaultDeploymentUniqueLabelKey] } return "" } CheckIngressRestored := func(sname string) { // stable service service := &v1.Service{} Expect(GetObject(sname, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(BeEmpty()) //canary service cService := &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).To(HaveOccurred()) // canary ingress cIngress := &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).To(HaveOccurred()) } type trafficContext struct { stableRevision string canaryRevision string service *v1.Service selectorKey string } CheckIngressConfigured := func(ctx *trafficContext, step *v1beta1.CanaryStep) { selectorKey := ctx.selectorKey if selectorKey == "" { selectorKey = apps.DefaultDeploymentUniqueLabelKey } sname := ctx.service.Name // stable service service := &v1.Service{} Expect(GetObject(sname, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[selectorKey]).Should(Equal(ctx.stableRevision)) //canary service cService := &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[selectorKey]).Should(Equal(ctx.canaryRevision)) // canary ingress cIngress := &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) if step.Traffic != nil { Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*step.Traffic))) } if step.Matches != nil { By("Fatal: Unimplemented to check match for ingress!") Expect(false).Should(BeTrue()) } } BeforeEach(func() { namespace = randomNamespaceName("rollout") ns := v1.Namespace{ ObjectMeta: metav1.ObjectMeta{ Name: namespace, }, } Expect(k8sClient.Create(context.TODO(), &ns)).Should(SatisfyAny(BeNil())) }) AfterEach(func() { By("[TEST] Clean up resources after an integration test") _ = k8sClient.DeleteAllOf(context.TODO(), &apps.Deployment{}, client.InNamespace(namespace)) _ = k8sClient.DeleteAllOf(context.TODO(), &appsv1alpha1.CloneSet{}, client.InNamespace(namespace)) _ = k8sClient.DeleteAllOf(context.TODO(), &v1beta1.BatchRelease{}, client.InNamespace(namespace)) _ = k8sClient.DeleteAllOf(context.TODO(), &v1beta1.Rollout{}, client.InNamespace(namespace)) _ = k8sClient.DeleteAllOf(context.TODO(), &v1.Service{}, client.InNamespace(namespace)) _ = k8sClient.DeleteAllOf(context.TODO(), &netv1.Ingress{}, client.InNamespace(namespace)) Expect(k8sClient.Delete(context.TODO(), &v1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: namespace}}, client.PropagationPolicy(metav1.DeletePropagationForeground))).Should(Succeed()) time.Sleep(time.Second * 3) }) KruiseDescribe("Step Jump", func() { // step1-> 2-> 3-> 4-> 3-(TrafficChange)-> 3-> 2-> 1-> 5 It("V1->V2: Deployment, Canary, patch nextStepIndex to jump", func() { finder := util.NewControllerFinder(k8sClient) By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_canary_base.yaml", rollout)).ToNot(HaveOccurred()) CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &apps.Deployment{} Expect(ReadYamlToObject("./test_data/rollout/deployment.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitDeploymentAllPodsReady(workload) rss, err := finder.GetReplicaSetsForDeployment(workload) Expect(err).NotTo(HaveOccurred()) Expect(len(rss)).Should(BeNumerically("==", 1)) stableRevision := rss[0].Labels[apps.DefaultDeploymentUniqueLabelKey] // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) By("Update deployment image from(version1) -> to(version2)") time.Sleep(time.Second * 3) // wait step 1 complete By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 2)) // canary workload cWorkload, err := GetCanaryDeployment(workload) Expect(err).NotTo(HaveOccurred()) crss, err := finder.GetReplicaSetsForDeployment(cWorkload) Expect(err).NotTo(HaveOccurred()) Expect(len(crss)).Should(BeNumerically("==", 1)) Expect(cWorkload.Status.AvailableReplicas).Should(BeNumerically("==", 1)) // workload Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 0)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) // wait step 2 complete By("wait step(2) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 3)) // canary workload cWorkload, err = GetCanaryDeployment(workload) Expect(err).NotTo(HaveOccurred()) Expect(cWorkload.Status.AvailableReplicas).Should(BeNumerically("==", 2)) // workload Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 0)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) // wait step 3 complete By("wait step(3) pause") ResumeRolloutCanary(rollout.Name) // rollout WaitRolloutCanaryStepPaused(rollout.Name, 3) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) // canary workload cWorkload, err = GetCanaryDeployment(workload) Expect(err).NotTo(HaveOccurred()) Expect(cWorkload.Status.AvailableReplicas).Should(BeNumerically("==", 3)) // workload Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 0)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) // wait step 4 complete By("wait step(4) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 4) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 5)) // canary workload cWorkload, err = GetCanaryDeployment(workload) Expect(err).NotTo(HaveOccurred()) canaryRevision := crss[0].Labels[apps.DefaultDeploymentUniqueLabelKey] Expect(cWorkload.Status.AvailableReplicas).Should(BeNumerically("==", 4)) // workload Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 0)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) // stable service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(stableRevision)) // canary service cService := &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress := &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[3].Traffic))) // Jump to step 3 By("Jump to step 3") RolloutJumpCanaryStep(rollout.Name, 3) WaitRolloutCanaryStepPaused(rollout.Name, 3) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) // canary workload (won't scale down indeed) cWorkload, err = GetCanaryDeployment(workload) Expect(err).NotTo(HaveOccurred()) canaryRevision = crss[0].Labels[apps.DefaultDeploymentUniqueLabelKey] Expect(cWorkload.Status.AvailableReplicas).Should(BeNumerically("==", 4)) // workload Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 0)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) // canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[2].Traffic))) // Change traffic of current step, which shouldn't cause jump By("Change traffic of step 3") Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // update rollout step configuration rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("21%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("41%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "40%"}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, }, { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("61%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, }, { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("81%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "80%"}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, }, { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("100%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, }, } rollout = UpdateRollout(rollout) By("update rollout configuration, and wait rollout re-run current step(3)") time.Sleep(time.Second * 3) WaitRolloutCanaryStepPaused(rollout.Name, 3) // batch release batch := &v1beta1.BatchRelease{} Expect(GetObject(rollout.Name, batch)).NotTo(HaveOccurred()) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) // canary workload (won't scale down indeed) cWorkload, err = GetCanaryDeployment(workload) Expect(err).NotTo(HaveOccurred()) canaryRevision = crss[0].Labels[apps.DefaultDeploymentUniqueLabelKey] Expect(cWorkload.Status.AvailableReplicas).Should(BeNumerically("==", 4)) // workload Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 0)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) // canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[2].Traffic))) // Jump to step 2 By("Jump to step 2") RolloutJumpCanaryStep(rollout.Name, 2) WaitRolloutCanaryStepPaused(rollout.Name, 2) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 3)) // canary workload (won't scale down indeed) cWorkload, err = GetCanaryDeployment(workload) Expect(err).NotTo(HaveOccurred()) canaryRevision = crss[0].Labels[apps.DefaultDeploymentUniqueLabelKey] Expect(cWorkload.Status.AvailableReplicas).Should(BeNumerically("==", 4)) // workload Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 0)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) // canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[1].Traffic))) // Jump to step 1 By("Jump to step 1") RolloutJumpCanaryStep(rollout.Name, 1) WaitRolloutCanaryStepPaused(rollout.Name, 1) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 2)) // canary workload (won't scale down indeed) cWorkload, err = GetCanaryDeployment(workload) Expect(err).NotTo(HaveOccurred()) canaryRevision = crss[0].Labels[apps.DefaultDeploymentUniqueLabelKey] Expect(cWorkload.Status.AvailableReplicas).Should(BeNumerically("==", 4)) // workload Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 0)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) // canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[0].Traffic))) // Jump to step 5 By("Jump to step 5") RolloutJumpCanaryStep(rollout.Name, 5) // wait rollout complete WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) klog.Infof("rollout(%s) completed, and check", namespace) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", -1)) // check service & ingress & deployment // ingress Expect(GetObject(ingress.Name, ingress)).NotTo(HaveOccurred()) cIngress = &netv1.Ingress{} Expect(GetObject(fmt.Sprintf("%s-canary", ingress.Name), cIngress)).To(HaveOccurred()) // service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal("")) cService = &v1.Service{} Expect(GetObject(fmt.Sprintf("%s-canary", service.Name), cService)).To(HaveOccurred()) // deployment Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Spec.Paused).Should(BeFalse()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.Replicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version2")) } } // check progressing succeed Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) // step1-> 2-> 3-> 4-> 3-(TrafficChange)-> 3-> 2-> 1-> 5 It("V1->V2: Deployment, Partition, patch nextStepIndex to jump", func() { finder := util.NewControllerFinder(k8sClient) By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &apps.Deployment{} Expect(ReadYamlToObject("./test_data/rollout/deployment.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitDeploymentAllPodsReady(workload) rss, err := finder.GetReplicaSetsForDeployment(workload) Expect(err).NotTo(HaveOccurred()) Expect(len(rss)).Should(BeNumerically("==", 1)) // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) By("Update deployment image from(version1) -> to(version2)") time.Sleep(time.Second * 3) // wait step 1 complete By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) stableRevision := GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) strategy := util.GetDeploymentStrategy(workload) extraStatus := util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress // stable service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(stableRevision)) //canary service cService := &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress := &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[0].Traffic))) // wait step 2 complete By("wait step(2) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) stableRevision = GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 2)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 2)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // if network configuration has restored CheckIngressRestored(service.Name) // wait step 3 complete By("wait step(3) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 3) stableRevision = GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 3)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // if network configuration has restored CheckIngressRestored(service.Name) // wait step 4 complete By("wait step(4) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 4) stableRevision = GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 4)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 5)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // if network configuration has restored CheckIngressRestored(service.Name) // Jump to step 3 By("Jump to step 3") RolloutJumpCanaryStep(rollout.Name, 3) WaitRolloutCanaryStepPaused(rollout.Name, 3) stableRevision = GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) // won't scale down Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // if network configuration has restored CheckIngressRestored(service.Name) // Change traffic of current step, which shouldn't cause jump By("Change traffic of step 3") Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // update rollout step configuration rollout.Spec.Strategy.Canary.Steps[2] = v1beta1.CanaryStep{ TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("60%"), }, // change to format of integer, so that traffic can be set Replicas: &intstr.IntOrString{Type: intstr.Int, IntVal: 3}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, } rollout = UpdateRollout(rollout) By("update rollout configuration, and wait rollout re-run current step(3)") time.Sleep(time.Second * 3) WaitRolloutCanaryStepPaused(rollout.Name, 3) // batch release batch := &v1beta1.BatchRelease{} Expect(GetObject(rollout.Name, batch)).NotTo(HaveOccurred()) stableRevision = GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) // won't scale down Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) canaryRevision = rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress // stable service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(stableRevision)) //canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[2].Traffic))) // Jump to step 2 By("Jump to step 2") RolloutJumpCanaryStep(rollout.Name, 2) WaitRolloutCanaryStepPaused(rollout.Name, 2) stableRevision = GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // if network configuration has restored CheckIngressRestored(service.Name) // Jump to step 1 By("Jump to step 1") RolloutJumpCanaryStep(rollout.Name, 1) WaitRolloutCanaryStepPaused(rollout.Name, 1) stableRevision = GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) canaryRevision = rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress // stable service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(stableRevision)) //canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[0].Traffic))) // Jump to step 5 By("Jump to step 5") RolloutJumpCanaryStep(rollout.Name, 5) // wait rollout complete WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhase(v1beta1.RolloutPhaseHealthy)) klog.Infof("rollout(%s) completed, and check", namespace) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", -1)) // if network configuration has restored CheckIngressRestored(service.Name) // deployment Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Spec.Paused).Should(BeFalse()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.Replicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version2")) } } // check progressing succeed Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) // step1-> 2-> 3-> 4-> 3-(TrafficChange)-> 3-> 2-> 1-> 5 It("V1->V2: CloneSet, Partition, patch nextStepIndex to jump", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitCloneSetAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision[strings.LastIndex(workload.Status.CurrentRevision, "-")+1:])) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") time.Sleep(time.Second * 3) // wait step 1 complete By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress // stable service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(stableRevision)) //canary service cService := &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress := &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[0].Traffic))) // wait step 2 complete By("wait step(2) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 2)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 2)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // if network configuration has restored CheckIngressRestored(service.Name) // wait step 3 complete By("wait step(3) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 3) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 3)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // if network configuration has restored CheckIngressRestored(service.Name) // Change traffic of current step, which shouldn't cause jump By("Change traffic of step 3") Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // update rollout step configuration rollout.Spec.Strategy.Canary.Steps[2] = v1beta1.CanaryStep{ TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("60%"), }, // change to format of integer, so that traffic can be set Replicas: &intstr.IntOrString{Type: intstr.Int, IntVal: 3}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, } rollout = UpdateRollout(rollout) By("update rollout configuration, and wait rollout re-run current step(3)") time.Sleep(time.Second * 3) WaitRolloutCanaryStepPaused(rollout.Name, 3) // batch release batch := &v1beta1.BatchRelease{} Expect(GetObject(rollout.Name, batch)).NotTo(HaveOccurred()) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 3)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) canaryRevision = rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress // stable service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(stableRevision)) //canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[2].Traffic))) // wait step 4 complete By("wait step(4) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 4) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 4)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 5)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // if network configuration has restored CheckIngressRestored(service.Name) // Jump to step 3 By("Jump to step 3") RolloutJumpCanaryStep(rollout.Name, 3) WaitRolloutCanaryStepPaused(rollout.Name, 3) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) canaryRevision = rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 4)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress // stable service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(stableRevision)) //canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[2].Traffic))) // Jump to step 2 By("Jump to step 2") RolloutJumpCanaryStep(rollout.Name, 2) WaitRolloutCanaryStepPaused(rollout.Name, 2) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 3)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check if network configuration has restored CheckIngressRestored(service.Name) // Jump to step 1 By("Jump to step 1") RolloutJumpCanaryStep(rollout.Name, 1) WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 4)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 4)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) canaryRevision = rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress // stable service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(stableRevision)) //canary service cService = &v1.Service{} Expect(GetObject(service.Name+"-canary", cService)).NotTo(HaveOccurred()) Expect(cService.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal(canaryRevision)) // canary ingress cIngress = &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[0].Traffic))) // Jump to step 5 By("Jump to step 5") RolloutJumpCanaryStep(rollout.Name, 5) // wait rollout complete WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhase(v1beta1.RolloutPhaseHealthy)) klog.Infof("rollout(%s) completed, and check", namespace) // rollout Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.NextStepIndex).Should(BeNumerically("==", -1)) // check if network configuration has restored CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.Replicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version2")) } } // check progressing succeed Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) }) KruiseDescribe("CloneSet canary rollout with Ingress", func() { It("CloneSet V1->V2: Percentage, 20%,60% Succeeded", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, } rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitCloneSetAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision[strings.LastIndex(workload.Status.CurrentRevision, "-")+1:])) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress CheckIngressConfigured(&trafficContext{ stableRevision: stableRevision, canaryRevision: canaryRevision, service: service, }, &rollout.Spec.Strategy.Canary.Steps[0]) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("resume rollout, and wait next step(2)") WaitRolloutCanaryStepPaused(rollout.Name, 2) // check stable, canary service & ingress CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 3)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) // resume rollout ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitCloneSetAllPodsReady(workload) By("rollout completed, and check") // check if network configuration has restored CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 5)) Expect(workload.Spec.UpdateStrategy.Partition.IntVal).Should(BeNumerically("==", 0)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevision)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version2")) } } time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) //Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(canaryRevision)) // scale up replicas 5 -> 6 workload.Spec.Replicas = utilpointer.Int32(6) UpdateCloneSet(workload) By("Update cloneSet replicas from(5) -> to(6)") time.Sleep(time.Second * 2) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) It("V1->V2: Percentage, 20%, and rollback(v1)", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitCloneSetAllPodsReady(workload) // check rollout status // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision[strings.LastIndex(workload.Status.CurrentRevision, "-")+1:])) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Image = "echoserver:failed" workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete time.Sleep(time.Second * 20) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 0)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.CurrentStepState).Should(Equal(v1beta1.CanaryStepStateUpgrade)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout canary ResumeRolloutCanary(rollout.Name) time.Sleep(time.Second * 15) // rollback -> v1 newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version1"}) workload.Spec.Template.Spec.Containers[0].Image = "cilium/echoserver:latest" workload.Spec.Template.Spec.Containers[0].Env = newEnvs workload.Spec.Template.Spec.Containers[0].ImagePullPolicy = v1.PullIfNotPresent UpdateCloneSet(workload) By("Rollback deployment env NODE_NAME from(version2) -> to(version1)") time.Sleep(time.Second * 2) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitCloneSetAllPodsReady(workload) By("rollout completed, and check") // check progressing canceled Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) Expect(string(cond.Status)).Should(Equal("False")) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 5)) Expect(workload.Spec.UpdateStrategy.Partition.IntVal).Should(BeNumerically("==", 0)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(stableRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(stableRevision)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version1")) } } }) It("Cloneset V1->V2: Percentage, 20%,40% and continuous release v3", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitCloneSetAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision[strings.LastIndex(workload.Status.CurrentRevision, "-")+1:])) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) canaryRevisionV1 := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout canary ResumeRolloutCanary(rollout.Name) time.Sleep(time.Second * 15) // v1 -> v2 -> v3, continuous release newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version3"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) By("Update cloneSet env NODE_NAME from(version2) -> to(version3)") time.Sleep(time.Second * 10) // wait step 0 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) //Expect(rollout.Status.CanaryStatus.CanaryReplicas).Should(BeNumerically("==", 1)) //Expect(rollout.Status.CanaryStatus.CanaryReadyReplicas).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).ShouldNot(Equal(canaryRevisionV1)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) canaryRevisionV2 := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) // check stable, canary service & ingress CheckIngressConfigured(&trafficContext{ stableRevision: stableRevision, canaryRevision: canaryRevisionV2, service: service, }, &rollout.Spec.Strategy.Canary.Steps[0]) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("check rollout canary status success, resume rollout, and wait rollout canary complete") WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitCloneSetAllPodsReady(workload) By("rollout completed, and check") // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 5)) Expect(workload.Spec.UpdateStrategy.Partition.IntVal).Should(BeNumerically("==", 0)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevisionV2)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevisionV2)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version3")) } } time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) It("V1->V2: disable quickly rollback policy without traffic routing", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.TrafficRoutings = nil addAnnotation(rollout, v1beta1.RollbackInBatchAnnotation, "true") CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // workload workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitCloneSetAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision[strings.LastIndex(workload.Status.CurrentRevision, "-")+1:])) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // v1 -> v2 -> v1, continuous release By("Update cloneSet env NODE_NAME from(version2) -> to(version1)") // resume rollout canary ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 3) newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version1"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) // make sure CloneSet is rolling back in batch By("Wait step 1 paused") WaitRolloutCanaryStepPaused(rollout.Name, 1) By("Wait step 2 paused") ResumeRolloutCanary(rollout.Name) By("check rollout canary status success, resume rollout, and wait rollout canary complete") WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitCloneSetAllPodsReady(workload) By("rollout completed, and check") Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 5)) Expect(workload.Spec.UpdateStrategy.Partition.IntVal).Should(BeNumerically("==", 0)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version1")) } } time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) It("CloneSet V1->V2: Percentage, 20%,40%,60%,80%,100%, no traffic, Succeeded", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.TrafficRoutings = nil CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) workload.Spec.UpdateStrategy.Type = appsv1alpha1.InPlaceOnlyCloneSetUpdateStrategyType workload.Spec.UpdateStrategy.MaxUnavailable = &intstr.IntOrString{ Type: intstr.Int, IntVal: 1, } workload.Spec.UpdateStrategy.MaxSurge = nil CreateObject(workload) WaitCloneSetAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision[strings.LastIndex(workload.Status.CurrentRevision, "-")+1:])) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action //newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Image = "cilium/echoserver:1.10.2" UpdateCloneSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision[strings.LastIndex(workload.Status.UpdateRevision, "-")+1:])) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitCloneSetAllPodsReady(workload) By("rollout completed, and check") // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.UpdatedReadyReplicas).Should(BeNumerically("==", 5)) Expect(workload.Spec.UpdateStrategy.Partition.IntVal).Should(BeNumerically("==", 0)) Expect(workload.Spec.UpdateStrategy.Paused).Should(BeFalse()) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevision)) time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) //Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(canaryRevision)) // scale up replicas 5 -> 6 workload.Spec.Replicas = utilpointer.Int32(6) UpdateCloneSet(workload) By("Update cloneSet replicas from(5) -> to(6)") time.Sleep(time.Second * 2) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) }) KruiseDescribe("Advanced Deployment canary rollout with Ingress", func() { It("advanced deployment rolling with traffic case", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, } rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "Deployment", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &apps.Deployment{} Expect(ReadYamlToObject("./test_data/rollout/deployment.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitDeploymentAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) stableRevision := GetStableRSRevision(workload) By(stableRevision) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) strategy := util.GetDeploymentStrategy(workload) extraStatus := util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(strategy.Paused).Should(BeFalse()) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(util.ComputeHash(&workload.Spec.Template, nil))) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(GetCanaryRSRevision(workload))) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress CheckIngressConfigured(&trafficContext{ stableRevision: stableRevision, canaryRevision: canaryRevision, service: service, }, &rollout.Spec.Strategy.Canary.Steps[0]) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("resume rollout, and wait next step(2)") WaitRolloutCanaryStepPaused(rollout.Name, 2) // check stable, canary service & ingress CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 3)) Expect(strategy.Paused).Should(BeFalse()) // resume rollout ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitDeploymentAllPodsReady(workload) By("rollout completed, and check") // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.AvailableReplicas).Should(BeNumerically("==", 5)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version2")) } } time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) //Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(canaryRevision)) // scale up replicas 5 -> 6 workload.Spec.Replicas = utilpointer.Int32(6) UpdateDeployment(workload) By("Update cloneSet replicas from(5) -> to(6)") time.Sleep(time.Second * 2) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) It("advanced deployment continuous rolling case", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.Strategy.Canary.TrafficRoutings = nil rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{Duration: utilpointer.Int32(0)}, }, } rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "Deployment", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") workload := &apps.Deployment{} Expect(ReadYamlToObject("./test_data/rollout/deployment.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitDeploymentAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) By("check rollout status & paused success") // v1 -> v2, start rollout action By("update workload env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) stableRevision := GetStableRSRevision(workload) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) strategy := util.GetDeploymentStrategy(workload) extraStatus := util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(strategy.Paused).Should(BeFalse()) By("check workload status & paused success") // resume rollout canary ResumeRolloutCanary(rollout.Name) By("resume rollout, and wait next step(2)") WaitRolloutCanaryStepPaused(rollout.Name, 2) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 3)) Expect(strategy.Paused).Should(BeFalse()) By("update workload env NODE_NAME from(version2) -> to(version3)") newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version3"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) WaitRolloutCanaryStepPaused(rollout.Name, 1) stableRevision = workload.Labels[v1beta1.DeploymentStableRevisionLabel] Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(strategy.Paused).Should(BeFalse()) By("check workload status & paused success") // resume rollout canary ResumeRolloutCanary(rollout.Name) By("resume rollout, and wait next step(2)") WaitRolloutCanaryStepPaused(rollout.Name, 2) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 3)) Expect(strategy.Paused).Should(BeFalse()) }) It("advanced deployment rollback case", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.Strategy.Canary.TrafficRoutings = nil rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{Duration: utilpointer.Int32(0)}, }, } rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "Deployment", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") workload := &apps.Deployment{} Expect(ReadYamlToObject("./test_data/rollout/deployment.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitDeploymentAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) By("check rollout status & paused success") // v1 -> v2, start rollout action By("update cloneSet env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) stableRevision := GetStableRSRevision(workload) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) strategy := util.GetDeploymentStrategy(workload) extraStatus := util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(strategy.Paused).Should(BeFalse()) By("check workload status & paused success") // resume rollout canary ResumeRolloutCanary(rollout.Name) By("resume rollout, and wait next step(2)") WaitRolloutCanaryStepPaused(rollout.Name, 2) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) strategy = util.GetDeploymentStrategy(workload) extraStatus = util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 3)) Expect(strategy.Paused).Should(BeFalse()) By("update workload env NODE_NAME from(version2) -> to(version1)") newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version1"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitDeploymentAllPodsReady(workload) }) It("advanced deployment delete rollout case", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.Strategy.Canary.TrafficRoutings = nil rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{Duration: utilpointer.Int32(0)}, }, } rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "Deployment", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") workload := &apps.Deployment{} Expect(ReadYamlToObject("./test_data/rollout/deployment.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitDeploymentAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) By("check rollout status & paused success") // v1 -> v2, start rollout action By("update workload env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) stableRevision := GetStableRSRevision(workload) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) strategy := util.GetDeploymentStrategy(workload) extraStatus := util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(strategy.Paused).Should(BeFalse()) By("check workload status & paused success") By("delete rollout and check deployment") _ = k8sClient.Delete(context.TODO(), rollout) WaitRolloutNotFound(rollout.Name) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Spec.Strategy.Type).Should(Equal(apps.RollingUpdateDeploymentStrategyType)) Expect(workload.Spec.Paused).Should(BeFalse()) WaitDeploymentAllPodsReady(workload) }) It("advanced deployment scaling case", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.Strategy.Canary.TrafficRoutings = nil rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{Duration: utilpointer.Int32(0)}, }, } rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "Deployment", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") workload := &apps.Deployment{} Expect(ReadYamlToObject("./test_data/rollout/deployment.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitDeploymentAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) By("check rollout status & paused success") // v1 -> v2, start rollout action By("update workload env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) stableRevision := GetStableRSRevision(workload) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) strategy := util.GetDeploymentStrategy(workload) extraStatus := util.GetDeploymentExtraStatus(workload) Expect(extraStatus.UpdatedReadyReplicas).Should(BeNumerically("==", 1)) Expect(strategy.Paused).Should(BeFalse()) By("check workload status & paused success") By("scale up workload from 5 to 10, and check") workload.Spec.Replicas = utilpointer.Int32(10) UpdateDeployment(workload) Eventually(func() bool { object := &v1beta1.Rollout{} Expect(GetObject(rollout.Name, object)).NotTo(HaveOccurred()) return object.Status.CanaryStatus.CanaryReadyReplicas == 2 }, 5*time.Minute, time.Second).Should(BeTrue()) By("scale down workload from 10 to 5, and check") workload.Spec.Replicas = utilpointer.Int32(5) UpdateDeployment(workload) Eventually(func() bool { object := &v1beta1.Rollout{} Expect(GetObject(rollout.Name, object)).NotTo(HaveOccurred()) return object.Status.CanaryStatus.CanaryReadyReplicas == 1 }, 5*time.Minute, time.Second).Should(BeTrue()) By("rolling deployment to be completed") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) ResumeRolloutCanary(rollout.Name) WaitDeploymentAllPodsReady(workload) }) }) KruiseDescribe("Others", func() { It("Patch batch id to pods: normal case", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1beta1", Kind: "StatefulSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.TrafficRoutings = nil CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // headless-service headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1beta1.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/advanced_statefulset.yaml", workload)).ToNot(HaveOccurred()) workload.Labels[v1beta1.RolloutIDLabel] = "1" CreateObject(workload) WaitAdvancedStatefulSetPodsReady(workload) By("Update statefulset env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateAdvancedStatefulSet(workload) // wait step 1 complete By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("resume rollout, and wait next step(2)") WaitRolloutCanaryStepPaused(rollout.Name, 2) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 1) // resume rollout ResumeRolloutCanary(rollout.Name) By("check rollout canary status success, resume rollout, and wait rollout canary complete") WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitAdvancedStatefulSetPodsReady(workload) // check batch id after rollout By("rollout completed, and check pod batch label") CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "3", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "4", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "5", 1) }) It("patch batch id to pods: scaling case", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "40%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{ Duration: utilpointer.Int32(10), }, }, } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) workload.Labels[v1beta1.RolloutIDLabel] = "1" CreateObject(workload) WaitCloneSetAllPodsReady(workload) // v1 -> v2, start rollout action By("Update cloneset env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) time.Sleep(time.Second * 2) // wait step 2 complete By("wait step(2) pause") WaitRolloutCanaryStepPaused(rollout.Name, 2) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 1) // scale up replicas, 5 -> 10 By("scaling up CloneSet from 5 -> 10") Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) workload.Spec.Replicas = utilpointer.Int32(10) UpdateCloneSet(workload) Eventually(func() bool { object := &v1beta1.Rollout{} Expect(GetObject(rollout.Name, object)).NotTo(HaveOccurred()) return object.Status.CanaryStatus.CanaryReadyReplicas == 4 }, 5*time.Minute, time.Second).Should(BeTrue()) // check pod batch label after scale By("check pod batch label after scale") CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 3) // resume rollout canary By("check rollout canary status success, resume rollout, and wait rollout canary complete") ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitCloneSetAllPodsReady(workload) By("rollout completed, and check pod batch label") CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 3) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "3", 2) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "4", 4) }) It("patch batch id to pods: rollback case", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.TrafficRoutings = nil addAnnotation(rollout, v1beta1.RollbackInBatchAnnotation, "true") rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "40%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "80%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{Duration: utilpointer.Int32(0)}, }, } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) workload.Labels[v1beta1.RolloutIDLabel] = "1" CreateObject(workload) WaitCloneSetAllPodsReady(workload) // v1 -> v2, start rollout action By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) By("wait step(2) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 1) By("wait step(3) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 3) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "3", 1) By("Update cloneSet env NODE_NAME from(version2) -> to(version1)") Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version1"}) workload.Labels[v1beta1.RolloutIDLabel] = "2" workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) time.Sleep(10 * time.Second) // make sure disable quickly rollback policy By("Wait step (1) paused") WaitRolloutCanaryStepPaused(rollout.Name, 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "1", 1) By("wait step(2) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "2", 1) By("wait step(3) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 3) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "3", 1) By("wait step(4) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 4) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "4", 1) By("Wait rollout complete") ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "2", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "3", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "4", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "5", 1) }) It("patch batch id to pods: only rollout-id changes", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.TrafficRoutings = nil addAnnotation(rollout, v1beta1.RollbackInBatchAnnotation, "true") rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "40%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "80%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{Duration: utilpointer.Int32(0)}, }, } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitCloneSetAllPodsReady(workload) By("Only update rollout id = '1', and start rollout") workload.Labels[v1beta1.RolloutIDLabel] = "1" workload.Annotations[util.InRolloutProgressingAnnotation] = "true" UpdateCloneSet(workload) By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) By("wait step(2) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 1) By("wait step(3) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 3) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "3", 1) By("Only update rollout id = '2', and check batch label again") workload.Labels[v1beta1.RolloutIDLabel] = "2" UpdateCloneSet(workload) By("wait step(3) pause again") WaitRolloutCanaryStepPaused(rollout.Name, 3) time.Sleep(30 * time.Second) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "2", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "3", 1) By("wait step(4) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 4) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "4", 1) By("Wait rollout complete") ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "2", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "3", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "4", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "5", 1) }) It("patch batch id to pods: only change rollout-id after rolling the first step", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.TrafficRoutings = nil addAnnotation(rollout, v1beta1.RollbackInBatchAnnotation, "true") rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "40%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "80%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{Duration: utilpointer.Int32(0)}, }, } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitCloneSetAllPodsReady(workload) By("Only update rollout id = '1', and start rollout") workload.Labels[v1beta1.RolloutIDLabel] = "1" workload.Annotations[util.InRolloutProgressingAnnotation] = "true" UpdateCloneSet(workload) By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) By("Only update rollout id = '2', and check batch label again") workload.Labels[v1beta1.RolloutIDLabel] = "2" UpdateCloneSet(workload) By("wait 30s") time.Sleep(30 * time.Second) By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "1", 1) By("wait step(2) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "2", 1) By("wait step(3) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 3) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "3", 1) By("wait step(4) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 4) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "4", 1) By("Wait rollout complete") ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "1", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "2", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "3", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "4", 1) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "2", "5", 1) }) }) KruiseDescribe("StatefulSet canary rollout with Ingress", func() { KruiseDescribe("Native StatefulSet rollout canary with Ingress", func() { It("V1->V2: Percentage, 20%,60% Succeeded", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, } rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "StatefulSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // headless service headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &apps.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/native_statefulset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitNativeStatefulSetPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision)) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action By("Update statefulset env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateNativeStatefulSet(workload) // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(*workload.Spec.UpdateStrategy.RollingUpdate.Partition).Should(BeNumerically("==", *workload.Spec.Replicas-1)) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress CheckIngressConfigured(&trafficContext{ stableRevision: stableRevision, canaryRevision: canaryRevision, service: service, selectorKey: apps.ControllerRevisionHashLabelKey, }, &rollout.Spec.Strategy.Canary.Steps[0]) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("resume rollout, and wait next step(2)") WaitRolloutCanaryStepPaused(rollout.Name, 2) // check stable, canary service & ingress CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(*workload.Spec.UpdateStrategy.RollingUpdate.Partition).Should(BeNumerically("==", *workload.Spec.Replicas-3)) // resume rollout ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitNativeStatefulSetPodsReady(workload) By("rollout completed, and check") // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevision)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version2")) } } time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) //Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(canaryRevision)) // scale up replicas 5 -> 6 workload.Spec.Replicas = utilpointer.Int32(6) UpdateNativeStatefulSet(workload) By("Update cloneSet replicas from(5) -> to(6)") time.Sleep(time.Second * 2) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) It("V1->V2: Percentage, 20%,40% and continuous release v3", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "StatefulSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &apps.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/native_statefulset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitNativeStatefulSetPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision)) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateNativeStatefulSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) canaryRevisionV1 := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout canary ResumeRolloutCanary(rollout.Name) time.Sleep(time.Second * 15) // v1 -> v2 -> v3, continuous release newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version3"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateNativeStatefulSet(workload) By("Update cloneSet env NODE_NAME from(version2) -> to(version3)") time.Sleep(time.Second * 10) // wait step 0 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.CanaryReplicas).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).ShouldNot(Equal(canaryRevisionV1)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) canaryRevisionV2 := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) // check stable, canary service & ingress CheckIngressConfigured(&trafficContext{ stableRevision: stableRevision, canaryRevision: canaryRevisionV2, service: service, selectorKey: apps.ControllerRevisionHashLabelKey, }, &rollout.Spec.Strategy.Canary.Steps[0]) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("check rollout canary status success, resume rollout, and wait rollout canary complete") WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitNativeStatefulSetPodsReady(workload) By("rollout completed, and check") // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevisionV2)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevisionV2)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version3")) } } time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) It("V1->V2: Percentage, 20%, and rollback(v1)", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "StatefulSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &apps.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/native_statefulset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitNativeStatefulSetPodsReady(workload) // check rollout status // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision)) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Image = "echoserver:failed" workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateNativeStatefulSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete time.Sleep(time.Minute * 2) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.CurrentStepState).Should(Equal(v1beta1.CanaryStepStateUpgrade)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout canary ResumeRolloutCanary(rollout.Name) time.Sleep(time.Second * 15) // rollback -> v1 newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version1"}) workload.Spec.Template.Spec.Containers[0].Image = "cilium/echoserver:latest" workload.Spec.Template.Spec.Containers[0].Env = newEnvs workload.Spec.Template.Spec.Containers[0].ImagePullPolicy = v1.PullIfNotPresent UpdateNativeStatefulSet(workload) By("Rollback deployment env NODE_NAME from(version2) -> to(version1)") time.Sleep(time.Second * 2) // StatefulSet will not remove the broken pod with failed image, we should delete it manually brokenPod := &v1.Pod{} Expect(GetObject(fmt.Sprintf("%v-%v", workload.Name, *workload.Spec.Replicas-1), brokenPod)).NotTo(HaveOccurred()) Expect(k8sClient.Delete(context.TODO(), brokenPod)).NotTo(HaveOccurred()) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitNativeStatefulSetPodsReady(workload) By("rollout completed, and check") // check progressing canceled Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(*workload.Spec.UpdateStrategy.RollingUpdate.Partition).Should(BeNumerically("==", 0)) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(stableRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(stableRevision)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version1")) } } }) It("V1->V2: Percentage, 20%,40%,60%,80%,100%, no traffic, Succeeded", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps/v1", Kind: "StatefulSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.TrafficRoutings = nil CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // headless-service headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &apps.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/native_statefulset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitNativeStatefulSetPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision)) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action //newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Image = "cilium/echoserver:1.10.2" UpdateNativeStatefulSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitNativeStatefulSetPodsReady(workload) By("rollout completed, and check") // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevision)) time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) //Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(canaryRevision)) // scale up replicas 5 -> 6 workload.Spec.Replicas = utilpointer.Int32(6) UpdateNativeStatefulSet(workload) By("Update cloneSet replicas from(5) -> to(6)") time.Sleep(time.Second * 2) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) }) KruiseDescribe("Advanced StatefulSet rollout canary with Ingress", func() { It("V1->V2: Percentage, 20%,60% Succeeded", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.Strategy.Canary.Steps = []v1beta1.CanaryStep{ { TrafficRoutingStrategy: v1beta1.TrafficRoutingStrategy{ Traffic: utilpointer.String("20%"), }, Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, } rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1beta1", Kind: "StatefulSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // headless service headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1beta1.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/advanced_statefulset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitAdvancedStatefulSetPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision)) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action By("Update statefulset env NODE_NAME from(version1) -> to(version2)") newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateAdvancedStatefulSet(workload) // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(*workload.Spec.UpdateStrategy.RollingUpdate.Partition).Should(BeNumerically("==", *workload.Spec.Replicas-1)) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // check stable, canary service & ingress CheckIngressConfigured(&trafficContext{ stableRevision: stableRevision, canaryRevision: canaryRevision, service: service, selectorKey: apps.ControllerRevisionHashLabelKey, }, &rollout.Spec.Strategy.Canary.Steps[0]) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("resume rollout, and wait next step(2)") WaitRolloutCanaryStepPaused(rollout.Name, 2) // check stable, canary service & ingress CheckIngressRestored(service.Name) // workload time.Sleep(time.Second * 10) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 3)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(*workload.Spec.UpdateStrategy.RollingUpdate.Partition).Should(BeNumerically("==", *workload.Spec.Replicas-3)) // resume rollout ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitAdvancedStatefulSetPodsReady(workload) By("rollout completed, and check") // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevision)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version2")) } } time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) //Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(canaryRevision)) // scale up replicas 5 -> 6 workload.Spec.Replicas = utilpointer.Int32(6) UpdateAdvancedStatefulSet(workload) By("Update cloneSet replicas from(5) -> to(6)") time.Sleep(time.Second * 2) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) It("V1->V2: Percentage, 20%,40% and continuous release v3", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1beta1", Kind: "StatefulSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1beta1.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/advanced_statefulset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitAdvancedStatefulSetPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision)) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateAdvancedStatefulSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) canaryRevisionV1 := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout canary ResumeRolloutCanary(rollout.Name) time.Sleep(time.Second * 15) // v1 -> v2 -> v3, continuous release newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version3"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateAdvancedStatefulSet(workload) By("Update cloneSet env NODE_NAME from(version2) -> to(version3)") time.Sleep(time.Second * 10) // wait step 0 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.CanaryReplicas).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).ShouldNot(Equal(canaryRevisionV1)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) canaryRevisionV2 := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) // check stable, canary service & ingress CheckIngressConfigured(&trafficContext{ stableRevision: stableRevision, canaryRevision: canaryRevisionV2, service: service, selectorKey: apps.ControllerRevisionHashLabelKey, }, &rollout.Spec.Strategy.Canary.Steps[0]) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("check rollout canary status success, resume rollout, and wait rollout canary complete") WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitAdvancedStatefulSetPodsReady(workload) By("rollout completed, and check") // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevisionV2)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevisionV2)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version3")) } } time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) It("V1->V2: Percentage, 20%, and rollback(v1)", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1beta1", Kind: "StatefulSet", Name: "echoserver", } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1beta1.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/advanced_statefulset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitAdvancedStatefulSetPodsReady(workload) // check rollout status // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision)) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Image = "echoserver:failed" workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateAdvancedStatefulSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete time.Sleep(time.Minute * 2) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.CurrentStepState).Should(Equal(v1beta1.CanaryStepStateUpgrade)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout canary ResumeRolloutCanary(rollout.Name) time.Sleep(time.Second * 15) // rollback -> v1 newEnvs = mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version1"}) workload.Spec.Template.Spec.Containers[0].Image = "cilium/echoserver:latest" workload.Spec.Template.Spec.Containers[0].Env = newEnvs workload.Spec.Template.Spec.Containers[0].ImagePullPolicy = v1.PullIfNotPresent UpdateAdvancedStatefulSet(workload) By("Rollback deployment env NODE_NAME from(version2) -> to(version1)") time.Sleep(time.Second * 2) // StatefulSet will not remove the broken pod with failed image, we should delete it manually brokenPod := &v1.Pod{} Expect(GetObject(fmt.Sprintf("%v-%v", workload.Name, *workload.Spec.Replicas-1), brokenPod)).NotTo(HaveOccurred()) Expect(k8sClient.Delete(context.TODO(), brokenPod)).NotTo(HaveOccurred()) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitAdvancedStatefulSetPodsReady(workload) By("rollout completed, and check") // check progressing canceled Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) // check service & ingress & deployment CheckIngressRestored(service.Name) // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(*workload.Spec.UpdateStrategy.RollingUpdate.Partition).Should(BeNumerically("==", 0)) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(stableRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(stableRevision)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version1")) } } }) It("V1->V2: Percentage, 20%,40%,60%,80%,100%, no traffic, Succeeded", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1beta1", Kind: "StatefulSet", Name: "echoserver", } rollout.Spec.Strategy.Canary.TrafficRoutings = nil CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // headless-service headlessService := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/headless_service.yaml", headlessService)).ToNot(HaveOccurred()) CreateObject(headlessService) // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &appsv1beta1.StatefulSet{} Expect(ReadYamlToObject("./test_data/rollout/advanced_statefulset.yaml", workload)).ToNot(HaveOccurred()) CreateObject(workload) WaitAdvancedStatefulSetPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(workload.Status.CurrentRevision)) stableRevision := rollout.Status.CanaryStatus.StableRevision By("check rollout status & paused success") // v1 -> v2, start rollout action //newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Image = "cilium/echoserver:1.10.2" UpdateAdvancedStatefulSet(workload) By("Update cloneSet env NODE_NAME from(version1) -> to(version2)") // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check workload status & paused Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 1)) By("check cloneSet status & paused success") // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseProgressing)) Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(stableRevision)) Expect(rollout.Status.CanaryStatus.CanaryRevision).Should(Equal(workload.Status.UpdateRevision)) Expect(rollout.Status.CanaryStatus.PodTemplateHash).Should(Equal(workload.Status.UpdateRevision)) canaryRevision := rollout.Status.CanaryStatus.PodTemplateHash Expect(rollout.Status.CanaryStatus.CurrentStepIndex).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.RolloutHash).Should(Equal(rollout.Annotations[util.RolloutHashAnnotation])) // resume rollout ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) WaitAdvancedStatefulSetPodsReady(workload) By("rollout completed, and check") // cloneset Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", 5)) Expect(workload.Status.CurrentRevision).Should(ContainSubstring(canaryRevision)) Expect(workload.Status.UpdateRevision).Should(ContainSubstring(canaryRevision)) time.Sleep(time.Second * 3) // check progressing succeed Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutCondition(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutCondition(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) //Expect(rollout.Status.CanaryStatus.StableRevision).Should(Equal(canaryRevision)) // scale up replicas 5 -> 6 workload.Spec.Replicas = utilpointer.Int32(6) UpdateAdvancedStatefulSet(workload) By("Update cloneSet replicas from(5) -> to(6)") time.Sleep(time.Second * 2) Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) }) }) // seems the next never be tested KruiseDescribe("Test", func() { It("failure threshold", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/rollout/rollout_v1beta1_partition_base.yaml", rollout)).ToNot(HaveOccurred()) rollout.Spec.WorkloadRef = v1beta1.ObjectRef{ APIVersion: "apps.kruise.io/v1alpha1", Kind: "CloneSet", Name: "echoserver", } rollout.Spec.Strategy.Canary = &v1beta1.CanaryStrategy{ FailureThreshold: &intstr.IntOrString{Type: intstr.String, StrVal: "20%"}, Steps: []v1beta1.CanaryStep{ { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "10%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "30%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "60%"}, Pause: v1beta1.RolloutPause{}, }, { Replicas: &intstr.IntOrString{Type: intstr.String, StrVal: "100%"}, Pause: v1beta1.RolloutPause{}, }, }, } CreateObject(rollout) By("Creating workload and waiting for all pods ready...") workload := &appsv1alpha1.CloneSet{} Expect(ReadYamlToObject("./test_data/rollout/cloneset.yaml", workload)).ToNot(HaveOccurred()) workload.Spec.Replicas = utilpointer.Int32(10) workload.Spec.UpdateStrategy.MaxUnavailable = &intstr.IntOrString{Type: intstr.String, StrVal: "100%"} CreateObject(workload) WaitCloneSetAllPodsReady(workload) checkUpdateReadyPods := func(lower, upper int32) bool { Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) return lower <= workload.Status.UpdatedReadyReplicas && workload.Status.UpdatedReadyReplicas <= upper } By("start rollout") workload.Labels[v1beta1.RolloutIDLabel] = "1" newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateCloneSet(workload) By("wait step(1) pause") WaitRolloutCanaryStepPaused(rollout.Name, 1) Expect(checkUpdateReadyPods(1, 1)).Should(BeTrue()) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "1", 1) By("wait step(2) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 2) Expect(checkUpdateReadyPods(2, 3)).Should(BeTrue()) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "2", 2) By("wait step(3) pause") ResumeRolloutCanary(rollout.Name) WaitRolloutCanaryStepPaused(rollout.Name, 3) Expect(checkUpdateReadyPods(4, 6)).Should(BeTrue()) CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "3", 3) // By("wait step(4) pause") // ResumeRolloutCanary(rollout.Name) // WaitRolloutCanaryStepPaused(rollout.Name, 4) // Expect(checkUpdateReadyPods(8, 10)).Should(BeTrue()) // CheckPodBatchLabel(workload.Namespace, workload.Spec.Selector, "1", "4", 4) }) }) KruiseDescribe("Canary rollout with multiple network providers", func() { It("V1->V2: Route traffic with header/queryParams/path matches and weight using rollout for Istio and Ingress", func() { By("Creating Rollout...") rollout := &v1beta1.Rollout{} Expect(ReadYamlToObject("./test_data/customNetworkProvider/rollout_with_multi_trafficrouting.yaml", rollout)).ToNot(HaveOccurred()) CreateObject(rollout) By("Creating workload and waiting for all pods ready...") // service service := &v1.Service{} Expect(ReadYamlToObject("./test_data/rollout/service.yaml", service)).ToNot(HaveOccurred()) CreateObject(service) // istio api vs := &unstructured.Unstructured{} Expect(ReadYamlToObject("./test_data/customNetworkProvider/virtualservice_without_destinationrule.yaml", vs)).ToNot(HaveOccurred()) vs.SetAPIVersion("networking.istio.io/v1alpha3") vs.SetKind("VirtualService") CreateObject(vs) // ingress ingress := &netv1.Ingress{} Expect(ReadYamlToObject("./test_data/rollout/nginx_ingress.yaml", ingress)).ToNot(HaveOccurred()) CreateObject(ingress) // workload workload := &apps.Deployment{} Expect(ReadYamlToObject("./test_data/rollout/deployment.yaml", workload)).ToNot(HaveOccurred()) workload.Spec.Replicas = utilpointer.Int32(4) CreateObject(workload) WaitDeploymentAllPodsReady(workload) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.Phase).Should(Equal(v1beta1.RolloutPhaseHealthy)) By("check rollout status & paused success") // v1 -> v2, start rollout action newEnvs := mergeEnvVar(workload.Spec.Template.Spec.Containers[0].Env, v1.EnvVar{Name: "NODE_NAME", Value: "version2"}) workload.Spec.Template.Spec.Containers[0].Env = newEnvs UpdateDeployment(workload) By("Update deployment env NODE_NAME from(version1) -> to(version2), routing traffic with header agent:pc to new version pods") time.Sleep(time.Second * 2) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Spec.Paused).Should(BeTrue()) // wait step 1 complete WaitRolloutCanaryStepPaused(rollout.Name, 1) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.CanaryReplicas).Should(BeNumerically("==", 1)) Expect(rollout.Status.CanaryStatus.CanaryReadyReplicas).Should(BeNumerically("==", 1)) // check virtualservice spec Expect(GetObject(vs.GetName(), vs)).NotTo(HaveOccurred()) expectedSpec := `{"gateways":["nginx-gateway"],"hosts":["*"],"http":[{"match":[{"uri":{"prefix":"/pc"}}],"route":[{"destination":{"host":"echoserver-canary"}}]},{"match":[{"queryParams":{"user-agent":{"exact":"pc"}}}],"route":[{"destination":{"host":"echoserver-canary"}}]},{"match":[{"headers":{"user-agent":{"exact":"pc"}}}],"route":[{"destination":{"host":"echoserver-canary"}}]},{"route":[{"destination":{"host":"echoserver"}}]}]}` Expect(util.DumpJSON(vs.Object["spec"])).Should(Equal(expectedSpec)) // check original spec annotation expectedAnno := `{"spec":{"gateways":["nginx-gateway"],"hosts":["*"],"http":[{"route":[{"destination":{"host":"echoserver"}}]}]}}` Expect(vs.GetAnnotations()[OriginalSpecAnnotation]).Should(Equal(expectedAnno)) // check canary-ingress spec cIngress := &netv1.Ingress{} Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(BeEmpty()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-by-header", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("user-agent")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-by-header-value", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("pc")) // resume rollout canary ResumeRolloutCanary(rollout.Name) By("Resume rollout, and wait next step(2), routing 50% traffic to new version pods") WaitRolloutCanaryStepPaused(rollout.Name, 2) // check rollout status Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) Expect(rollout.Status.CanaryStatus.CanaryReplicas).Should(BeNumerically("==", 2)) Expect(rollout.Status.CanaryStatus.CanaryReadyReplicas).Should(BeNumerically("==", 2)) // check virtualservice spec Expect(GetObject(vs.GetName(), vs)).NotTo(HaveOccurred()) expectedSpec = `{"gateways":["nginx-gateway"],"hosts":["*"],"http":[{"route":[{"destination":{"host":"echoserver"},"weight":50},{"destination":{"host":"echoserver-canary"},"weight":50}]}]}` Expect(util.DumpJSON(vs.Object["spec"])).Should(Equal(expectedSpec)) // check original spec annotation Expect(vs.GetAnnotations()[OriginalSpecAnnotation]).Should(Equal(expectedAnno)) // check canary-ingress spec Expect(GetObject(service.Name+"-canary", cIngress)).NotTo(HaveOccurred()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary", nginxIngressAnnotationDefaultPrefix)]).Should(Equal("true")) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-weight", nginxIngressAnnotationDefaultPrefix)]).Should(Equal(removePercentageSign(*rollout.Spec.Strategy.Canary.Steps[1].Traffic))) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-by-header", nginxIngressAnnotationDefaultPrefix)]).Should(BeEmpty()) Expect(cIngress.Annotations[fmt.Sprintf("%s/canary-by-header-value", nginxIngressAnnotationDefaultPrefix)]).Should(BeEmpty()) // resume rollout ResumeRolloutCanary(rollout.Name) WaitRolloutStatusPhase(rollout.Name, v1beta1.RolloutPhaseHealthy) By("rollout completed, and check") // check ingress & service & virtualservice & deployment // ingress Expect(GetObject(ingress.Name, ingress)).NotTo(HaveOccurred()) cIngress = &netv1.Ingress{} Expect(GetObject(fmt.Sprintf("%s-canary", ingress.Name), cIngress)).To(HaveOccurred()) // virtualservice Expect(GetObject(vs.GetName(), vs)).NotTo(HaveOccurred()) expectedSpec = `{"gateways":["nginx-gateway"],"hosts":["*"],"http":[{"route":[{"destination":{"host":"echoserver"}}]}]}` Expect(util.DumpJSON(vs.Object["spec"])).Should(Equal(expectedSpec)) Expect(vs.GetAnnotations()[OriginalSpecAnnotation]).Should(Equal("")) // service Expect(GetObject(service.Name, service)).NotTo(HaveOccurred()) Expect(service.Spec.Selector[apps.DefaultDeploymentUniqueLabelKey]).Should(Equal("")) cService := &v1.Service{} Expect(GetObject(fmt.Sprintf("%s-canary", service.Name), cService)).To(HaveOccurred()) // deployment Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) Expect(workload.Spec.Paused).Should(BeFalse()) Expect(workload.Status.UpdatedReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.Replicas).Should(BeNumerically("==", *workload.Spec.Replicas)) Expect(workload.Status.ReadyReplicas).Should(BeNumerically("==", *workload.Spec.Replicas)) for _, env := range workload.Spec.Template.Spec.Containers[0].Env { if env.Name == "NODE_NAME" { Expect(env.Value).Should(Equal("version2")) } } // check progressing succeed Expect(GetObject(rollout.Name, rollout)).NotTo(HaveOccurred()) cond := getRolloutConditionV1beta1(rollout.Status, v1beta1.RolloutConditionProgressing) Expect(cond.Reason).Should(Equal(v1beta1.ProgressingReasonCompleted)) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionFalse))) cond = getRolloutConditionV1beta1(rollout.Status, v1beta1.RolloutConditionSucceeded) Expect(string(cond.Status)).Should(Equal(string(metav1.ConditionTrue))) Expect(GetObject(workload.Name, workload)).NotTo(HaveOccurred()) WaitRolloutWorkloadGeneration(rollout.Name, workload.Generation) }) }) }) func removePercentageSign(input string) string { if input == "0" { return "0" } if strings.HasSuffix(input, "%") { return strings.TrimSuffix(input, "%") } fmt.Printf("input(%s) has no percentage sign!", input) return "" } func addAnnotation(obj metav1.Object, key, value string) { if obj.GetAnnotations() == nil { obj.SetAnnotations(make(map[string]string)) } obj.GetAnnotations()[key] = value }