add cluster lifecycle management test

Signed-off-by: Garrybest <[email protected]>

pkg/util/helper/taint_test.go

@@ -0,0 +1,134 @@
+package helper
+
+import (
+	"context"
+	"testing"
+
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	fakeclient "sigs.k8s.io/controller-runtime/pkg/client/fake"
+
+	clusterv1alpha1 "github.com/karmada-io/karmada/pkg/apis/cluster/v1alpha1"
+	"github.com/karmada-io/karmada/pkg/util/gclient"
+)
+
+var (
+	unreachableTaintTemplate = &corev1.Taint{
+		Key:    clusterv1alpha1.TaintClusterUnreachable,
+		Effect: corev1.TaintEffectNoExecute,
+	}
+
+	notReadyTaintTemplate = &corev1.Taint{
+		Key:    clusterv1alpha1.TaintClusterNotReady,
+		Effect: corev1.TaintEffectNoExecute,
+	}
+)
+
+func TestUpdateClusterControllerTaint(t *testing.T) {
+	type args struct {
+		taints         []corev1.Taint
+		taintsToAdd    []*corev1.Taint
+		taintsToRemove []*corev1.Taint
+	}
+	tests := []struct {
+		name       string
+		args       args
+		wantTaints []corev1.Taint
+		wantErr    bool
+	}{
+		{
+			name: "ready condition from true to false",
+			args: args{
+				taints:         nil,
+				taintsToAdd:    []*corev1.Taint{notReadyTaintTemplate.DeepCopy()},
+				taintsToRemove: []*corev1.Taint{unreachableTaintTemplate.DeepCopy()},
+			},
+			wantTaints: []corev1.Taint{*notReadyTaintTemplate},
+			wantErr:    false,
+		},
+		{
+			name: "ready condition from true to unknown",
+			args: args{
+				taints:         nil,
+				taintsToAdd:    []*corev1.Taint{unreachableTaintTemplate.DeepCopy()},
+				taintsToRemove: []*corev1.Taint{notReadyTaintTemplate.DeepCopy()},
+			},
+			wantTaints: []corev1.Taint{*unreachableTaintTemplate},
+			wantErr:    false,
+		},
+		{
+			name: "ready condition from false to unknown",
+			args: args{
+				taints:         []corev1.Taint{*notReadyTaintTemplate},
+				taintsToAdd:    []*corev1.Taint{unreachableTaintTemplate.DeepCopy()},
+				taintsToRemove: []*corev1.Taint{notReadyTaintTemplate.DeepCopy()},
+			},
+			wantTaints: []corev1.Taint{*unreachableTaintTemplate},
+			wantErr:    false,
+		},
+		{
+			name: "ready condition from false to true",
+			args: args{
+				taints:         []corev1.Taint{*notReadyTaintTemplate},
+				taintsToAdd:    []*corev1.Taint{},
+				taintsToRemove: []*corev1.Taint{notReadyTaintTemplate.DeepCopy(), unreachableTaintTemplate.DeepCopy()},
+			},
+			wantTaints: nil,
+			wantErr:    false,
+		},
+		{
+			name: "ready condition from unknown to true",
+			args: args{
+				taints:         []corev1.Taint{*unreachableTaintTemplate},
+				taintsToAdd:    []*corev1.Taint{},
+				taintsToRemove: []*corev1.Taint{notReadyTaintTemplate.DeepCopy(), unreachableTaintTemplate.DeepCopy()},
+			},
+			wantTaints: nil,
+			wantErr:    false,
+		},
+		{
+			name: "ready condition from unknown to false",
+			args: args{
+				taints:         []corev1.Taint{*unreachableTaintTemplate},
+				taintsToAdd:    []*corev1.Taint{notReadyTaintTemplate.DeepCopy()},
+				taintsToRemove: []*corev1.Taint{unreachableTaintTemplate.DeepCopy()},
+			},
+			wantTaints: []corev1.Taint{*notReadyTaintTemplate},
+			wantErr:    false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			ctx, cancel := context.WithCancel(context.Background())
+			defer cancel()
+
+			cluster := &clusterv1alpha1.Cluster{
+				ObjectMeta: metav1.ObjectMeta{Name: "member"},
+				Spec: clusterv1alpha1.ClusterSpec{
+					Taints: tt.args.taints,
+				},
+			}
+			c := fakeclient.NewClientBuilder().WithScheme(gclient.NewSchema()).WithObjects(cluster).Build()
+
+			if err := UpdateClusterControllerTaint(ctx, c, tt.args.taintsToAdd, tt.args.taintsToRemove, cluster); (err != nil) != tt.wantErr {
+				t.Errorf("UpdateClusterControllerTaint() error = %v, wantErr %v", err, tt.wantErr)
+			}
+
+			if err := c.Get(ctx, client.ObjectKey{Name: cluster.Name}, cluster); err != nil {
+				t.Fatalf("Failed to get cluster %s: %v", cluster.Name, err)
+			}
+
+			if len(cluster.Spec.Taints) != len(tt.wantTaints) {
+				t.Errorf("Cluster gotTaints = %v, want %v", cluster.Spec.Taints, tt.wantTaints)
+			}
+			for i := range cluster.Spec.Taints {
+				if cluster.Spec.Taints[i].Key != tt.wantTaints[i].Key ||
+					cluster.Spec.Taints[i].Value != tt.wantTaints[i].Value ||
+					cluster.Spec.Taints[i].Effect != tt.wantTaints[i].Effect {
+					t.Errorf("Cluster gotTaints = %v, want %v", cluster.Spec.Taints, tt.wantTaints)
+				}
+			}
+		})
+	}
+}

vendor/modules.txt

@@ -1450,6 +1450,7 @@ sigs.k8s.io/controller-runtime/pkg/certwatcher
 sigs.k8s.io/controller-runtime/pkg/client
 sigs.k8s.io/controller-runtime/pkg/client/apiutil
 sigs.k8s.io/controller-runtime/pkg/client/config
+sigs.k8s.io/controller-runtime/pkg/client/fake
 sigs.k8s.io/controller-runtime/pkg/cluster
 sigs.k8s.io/controller-runtime/pkg/config
 sigs.k8s.io/controller-runtime/pkg/config/v1alpha1