karmada/pkg/util/cluster_test.go

package util

import (
	"context"
	"reflect"
	"testing"

	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	clusterv1alpha1 "github.com/karmada-io/karmada/pkg/apis/cluster/v1alpha1"
	karmadaclientset "github.com/karmada-io/karmada/pkg/generated/clientset/versioned"
	karmadaclientsetfake "github.com/karmada-io/karmada/pkg/generated/clientset/versioned/fake"
)

func newCluster(name string) *clusterv1alpha1.Cluster {
	return &clusterv1alpha1.Cluster{
		ObjectMeta: metav1.ObjectMeta{
			Name: name,
		},
		Spec:   clusterv1alpha1.ClusterSpec{},
		Status: clusterv1alpha1.ClusterStatus{},
	}
}

func withAPIEndPoint(cluster *clusterv1alpha1.Cluster, apiEndPoint string) *clusterv1alpha1.Cluster {
	cluster.Spec.APIEndpoint = apiEndPoint
	return cluster
}

func withSyncMode(cluster *clusterv1alpha1.Cluster, syncMode clusterv1alpha1.ClusterSyncMode) *clusterv1alpha1.Cluster {
	cluster.Spec.SyncMode = syncMode
	return cluster
}

func withID(cluster *clusterv1alpha1.Cluster, id string) *clusterv1alpha1.Cluster {
	cluster.Spec.ID = id
	return cluster
}

func TestCreateOrUpdateClusterObject(t *testing.T) {
	fakeClient := karmadaclientsetfake.NewSimpleClientset()
	type args struct {
		controlPlaneClient karmadaclientset.Interface
		clusterObj         *clusterv1alpha1.Cluster
		mutate             func(*clusterv1alpha1.Cluster)
		aop                func() func()
	}
	tests := []struct {
		name    string
		args    args
		want    *clusterv1alpha1.Cluster
		wantErr bool
	}{
		{
			name: "cluster exist, and update cluster",
			args: args{
				controlPlaneClient: fakeClient,
				clusterObj:         newCluster(ClusterMember1),
				mutate: func(cluster *clusterv1alpha1.Cluster) {
					cluster.Spec.SyncMode = clusterv1alpha1.Pull
				},
				aop: func() func() {
					cluster := withAPIEndPoint(newCluster(ClusterMember1), "https://127.0.0.1:6443")
					_, _ = fakeClient.ClusterV1alpha1().Clusters().Create(context.TODO(), cluster, metav1.CreateOptions{})
					return func() {
						_ = fakeClient.ClusterV1alpha1().Clusters().Delete(context.TODO(), cluster.Name, metav1.DeleteOptions{})
					}
				},
			},
			want: withSyncMode(withAPIEndPoint(newCluster(ClusterMember1), "https://127.0.0.1:6443"), clusterv1alpha1.Pull),
		},
		{
			name: "cluster not exist, and create cluster",
			args: args{
				controlPlaneClient: fakeClient,
				clusterObj:         newCluster(ClusterMember1),
				mutate: func(cluster *clusterv1alpha1.Cluster) {
					cluster.Spec.SyncMode = clusterv1alpha1.Pull
				},
				aop: func() func() {
					cluster := withSyncMode(newCluster(ClusterMember1), clusterv1alpha1.Pull)
					return func() {
						_ = fakeClient.ClusterV1alpha1().Clusters().Delete(context.TODO(), cluster.Name, metav1.DeleteOptions{})
					}
				},
			},
			want: withSyncMode(newCluster(ClusterMember1), clusterv1alpha1.Pull),
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if tt.args.aop != nil {
				cancel := tt.args.aop()
				defer cancel()
			}
			got, err := CreateOrUpdateClusterObject(tt.args.controlPlaneClient, tt.args.clusterObj, tt.args.mutate)
			if (err != nil) != tt.wantErr {
				t.Errorf("CreateOrUpdateClusterObject() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("CreateOrUpdateClusterObject() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestIsClusterIDUnique(t *testing.T) {
	tests := []struct {
		name           string
		existedCluster []*clusterv1alpha1.Cluster
		id             string
		want           bool
		clustername    string
	}{
		{
			name: "no cluster", id: "1", want: true,
			existedCluster: []*clusterv1alpha1.Cluster{},
		},
		{
			name: "existed id", id: "1", want: false, clustername: "cluster-1",
			existedCluster: []*clusterv1alpha1.Cluster{withID(newCluster("cluster-1"), "1")},
		},
		{
			name: "unique id", id: "2", want: true,
			existedCluster: []*clusterv1alpha1.Cluster{withID(newCluster("cluster-1"), "1")},
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			fakeClient := karmadaclientsetfake.NewSimpleClientset()

			for _, cluster := range tc.existedCluster {
				_, err := fakeClient.ClusterV1alpha1().Clusters().Create(context.TODO(), cluster, metav1.CreateOptions{})
				if err != nil {
					t.Fatal(err)
				}
			}

			ok, name, err := IsClusterIdentifyUnique(fakeClient, tc.id)
			if err != nil {
				t.Fatal(err)
			}

			if ok != tc.want {
				t.Errorf("expected value: %v, but got: %v", tc.want, ok)
			}

			if !ok && name != tc.clustername {
				t.Errorf("expected clustername: %v, but got: %v", tc.clustername, name)
			}
		})
	}
}