add clonesets scale ut

Signed-off-by: liheng.zms <[email protected]>

Author: zmberg

pkg/controller/cloneset/sync/cloneset_scale_test.go

@@ -18,9 +18,11 @@ package sync
 
 import (
 	"context"
+	"fmt"
 	"reflect"
 	"sort"
 	"testing"
+	"time"
 
 	appspub "github.com/openkruise/kruise/apis/apps/pub"
 	appsv1alpha1 "github.com/openkruise/kruise/apis/apps/v1alpha1"
@@ -32,12 +34,29 @@ import (
 	v1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/util/intstr"
+	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	"k8s.io/apimachinery/pkg/util/sets"
+	"k8s.io/client-go/kubernetes/scheme"
 	"k8s.io/client-go/tools/record"
+	corev1 "k8s.io/kubernetes/pkg/apis/core/v1"
+	kubecontroller "k8s.io/kubernetes/pkg/controller"
+	utilpointer "k8s.io/utils/pointer"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/client/fake"
 )
 
+var (
+	kscheme *runtime.Scheme
+)
+
+func init() {
+	kscheme = runtime.NewScheme()
+	utilruntime.Must(appsv1alpha1.AddToScheme(scheme.Scheme))
+	utilruntime.Must(corev1.AddToScheme(kscheme))
+}
+
 func newFakeControl() *realControl {
 	return &realControl{
 		Client:   fake.NewClientBuilder().Build(),
@@ -527,3 +546,119 @@ func TestGetOrGenAvailableIDs(t *testing.T) {
 		t.Fatalf("expected got random id, but actually %v", id)
 	}
 }
+
+func TestScale(t *testing.T) {
+	cases := []struct {
+		name             string
+		getCloneSets     func() [2]*appsv1alpha1.CloneSet
+		getRevisions     func() [2]string
+		getPods          func() []*v1.Pod
+		expectedPodsLen  int
+		expectedModified bool
+	}{
+		{
+			name: "cloneSet(replicas=3,maxUnavailable=20%,partition=nil,maxSurge=nil,minReadySeconds=9999), pods=5, and scale replicas 5 -> 3",
+			getCloneSets: func() [2]*appsv1alpha1.CloneSet {
+				obj := &appsv1alpha1.CloneSet{
+					ObjectMeta: metav1.ObjectMeta{
+						Name: "sample",
+					},
+					Spec: appsv1alpha1.CloneSetSpec{
+						Replicas:        utilpointer.Int32(3),
+						MinReadySeconds: 9999,
+						UpdateStrategy: appsv1alpha1.CloneSetUpdateStrategy{
+							MaxUnavailable: &intstr.IntOrString{
+								Type:   intstr.String,
+								StrVal: "20%",
+							},
+						},
+					},
+				}
+				return [2]*appsv1alpha1.CloneSet{obj.DeepCopy(), obj.DeepCopy()}
+			},
+			getRevisions: func() [2]string {
+				return [2]string{"sample-b976d4544", "sample-b976d4544"}
+			},
+			getPods: func() []*v1.Pod {
+				t := time.Now().Add(-time.Second * 10)
+				obj := &v1.Pod{
+					ObjectMeta: metav1.ObjectMeta{
+						Name: "sample",
+						Labels: map[string]string{
+							apps.ControllerRevisionHashLabelKey: "sample-b976d4544",
+						},
+					},
+					Spec: v1.PodSpec{
+						Containers: []v1.Container{
+							{
+								Name:  "main",
+								Image: "sample:v1",
+							},
+						},
+					},
+					Status: v1.PodStatus{
+						Phase: v1.PodRunning,
+						Conditions: []v1.PodCondition{
+							{
+								Type:               v1.PodReady,
+								Status:             v1.ConditionTrue,
+								LastTransitionTime: metav1.Time{Time: t},
+							},
+						},
+					},
+				}
+				return generatePods(obj, 5)
+			},
+			expectedPodsLen:  3,
+			expectedModified: true,
+		},
+	}
+
+	for _, cs := range cases {
+		t.Run(cs.name, func(t *testing.T) {
+			fClient := fake.NewClientBuilder().WithScheme(kscheme).Build()
+			pods := cs.getPods()
+			for _, pod := range pods {
+				err := fClient.Create(context.TODO(), pod)
+				if err != nil {
+					t.Fatalf(err.Error())
+				}
+			}
+			rControl := &realControl{
+				Client:   fClient,
+				recorder: record.NewFakeRecorder(10),
+			}
+			modified, err := rControl.Scale(cs.getCloneSets()[0], cs.getCloneSets()[1], cs.getRevisions()[0], cs.getRevisions()[1], pods, nil)
+			if err != nil {
+				t.Fatalf(err.Error())
+			}
+			if cs.expectedModified != modified {
+				t.Fatalf("expect(%v), but get(%v)", cs.expectedModified, modified)
+			}
+			podList := &v1.PodList{}
+			err = fClient.List(context.TODO(), podList, &client.ListOptions{})
+			if err != nil {
+				t.Fatalf(err.Error())
+			}
+			actives := 0
+			for _, pod := range podList.Items {
+				if kubecontroller.IsPodActive(&pod) {
+					actives++
+				}
+			}
+			if actives != cs.expectedPodsLen {
+				t.Fatalf("expect(%v), but get(%v)", cs.expectedPodsLen, actives)
+			}
+		})
+	}
+}
+
+func generatePods(base *v1.Pod, replicas int) []*v1.Pod {
+	objs := make([]*v1.Pod, 0, replicas)
+	for i := 0; i < replicas; i++ {
+		obj := base.DeepCopy()
+		obj.Name = fmt.Sprintf("%s-%d", base.Name, i)
+		objs = append(objs, obj)
+	}
+	return objs
+}