From 36942452e5e802abfc3ae9d458b17cd105e0c6ba Mon Sep 17 00:00:00 2001
From: ChrisLiu <chrisliu1995@163.com>
Date: Tue, 12 Sep 2023 22:39:04 +0800
Subject: [PATCH] add gameserverset controller unit tests

Signed-off-by: ChrisLiu <chrisliu1995@163.com>
---
 .../gameserverset_controller_test.go          | 130 ++++++++++++++++++
 1 file changed, 130 insertions(+)
 create mode 100644 pkg/controllers/gameserverset/gameserverset_controller_test.go

diff --git a/pkg/controllers/gameserverset/gameserverset_controller_test.go b/pkg/controllers/gameserverset/gameserverset_controller_test.go
new file mode 100644
index 00000000..5a0e2002
--- /dev/null
+++ b/pkg/controllers/gameserverset/gameserverset_controller_test.go
@@ -0,0 +1,130 @@
+package gameserverset
+
+import (
+	"context"
+	appspub "github.com/openkruise/kruise-api/apps/pub"
+	kruiseV1beta1 "github.com/openkruise/kruise-api/apps/v1beta1"
+	gameKruiseV1alpha1 "github.com/openkruise/kruise-game/apis/v1alpha1"
+	"github.com/openkruise/kruise-game/pkg/util"
+	apps "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+	"k8s.io/utils/pointer"
+	"reflect"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/client/fake"
+	"testing"
+)
+
+func TestInitAsts(t *testing.T) {
+	tests := []struct {
+		gss  *gameKruiseV1alpha1.GameServerSet
+		asts *kruiseV1beta1.StatefulSet
+	}{
+		{
+			gss: &gameKruiseV1alpha1.GameServerSet{
+				TypeMeta: metav1.TypeMeta{
+					Kind:       "GameServerSet",
+					APIVersion: "game.kruise.io/v1alpha1",
+				},
+				ObjectMeta: metav1.ObjectMeta{
+					Namespace: "xxx",
+					Name:      "case0",
+					UID:       "xxx0",
+				},
+				Spec: gameKruiseV1alpha1.GameServerSetSpec{
+					Replicas: pointer.Int32(5),
+					UpdateStrategy: gameKruiseV1alpha1.UpdateStrategy{
+						Type:          apps.RollingUpdateStatefulSetStrategyType,
+						RollingUpdate: &gameKruiseV1alpha1.RollingUpdateStatefulSetStrategy{},
+					},
+				},
+			},
+			asts: &kruiseV1beta1.StatefulSet{
+				TypeMeta: metav1.TypeMeta{
+					Kind:       "StatefulSet",
+					APIVersion: "apps.kruise.io/v1beta1",
+				},
+				ObjectMeta: metav1.ObjectMeta{
+					Namespace: "xxx",
+					Name:      "case0",
+					OwnerReferences: []metav1.OwnerReference{
+						{
+							APIVersion:         "game.kruise.io/v1alpha1",
+							Kind:               "GameServerSet",
+							Name:               "case0",
+							UID:                "xxx0",
+							Controller:         pointer.BoolPtr(true),
+							BlockOwnerDeletion: pointer.BoolPtr(true),
+						},
+					},
+					ResourceVersion: "1",
+				},
+				Spec: kruiseV1beta1.StatefulSetSpec{
+					Replicas:            pointer.Int32(5),
+					PodManagementPolicy: apps.ParallelPodManagement,
+					ServiceName:         "case0",
+					Selector: &metav1.LabelSelector{
+						MatchLabels: map[string]string{gameKruiseV1alpha1.GameServerOwnerGssKey: "case0"},
+					},
+					UpdateStrategy: kruiseV1beta1.StatefulSetUpdateStrategy{
+						Type: apps.RollingUpdateStatefulSetStrategyType,
+						RollingUpdate: &kruiseV1beta1.RollingUpdateStatefulSetStrategy{
+							UnorderedUpdate: &kruiseV1beta1.UnorderedUpdateStrategy{
+								PriorityStrategy: &appspub.UpdatePriorityStrategy{
+									OrderPriority: []appspub.UpdatePriorityOrderTerm{
+										{
+											OrderedKey: gameKruiseV1alpha1.GameServerUpdatePriorityKey,
+										},
+									},
+								},
+							},
+						},
+					},
+					Template: corev1.PodTemplateSpec{
+						ObjectMeta: metav1.ObjectMeta{
+							Labels: map[string]string{
+								gameKruiseV1alpha1.GameServerOwnerGssKey: "case0",
+							},
+						},
+						Spec: corev1.PodSpec{
+							ReadinessGates: []corev1.PodReadinessGate{
+								{
+									ConditionType: appspub.InPlaceUpdateReady,
+								},
+							},
+						},
+					},
+					ScaleStrategy: &kruiseV1beta1.StatefulSetScaleStrategy{},
+				},
+			},
+		},
+	}
+
+	for i, test := range tests {
+		var objs []client.Object
+		c := fake.NewClientBuilder().WithScheme(scheme).WithObjects(objs...).Build()
+		reconcile := &GameServerSetReconciler{
+			Client: c,
+			Scheme: scheme,
+		}
+		if err := reconcile.initAsts(test.gss); err != nil {
+			t.Errorf("case %d: %s", i, err.Error())
+		}
+		initAsts := &kruiseV1beta1.StatefulSet{}
+		if err := reconcile.Get(context.TODO(), types.NamespacedName{
+			Namespace: test.gss.Namespace,
+			Name:      test.gss.Name,
+		}, initAsts); err != nil {
+			t.Errorf("case %d: %s", i, err.Error())
+		}
+		if test.asts.Annotations == nil {
+			test.asts.Annotations = make(map[string]string)
+		}
+		test.asts.Annotations[gameKruiseV1alpha1.AstsHashKey] = util.GetAstsHash(test.gss)
+		if !reflect.DeepEqual(initAsts, test.asts) {
+			t.Errorf("expect asts %v but got %v", test.asts, initAsts)
+		}
+	}
+}