package cluster

import (
	"github.com/google/go-cmp/cmp"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/runtime"
	"kubesphere.io/kubesphere/pkg/api"
	clusterv1alpha1 "kubesphere.io/kubesphere/pkg/apis/cluster/v1alpha1"
	"kubesphere.io/kubesphere/pkg/apiserver/query"
	"kubesphere.io/kubesphere/pkg/client/clientset/versioned/fake"
	"kubesphere.io/kubesphere/pkg/client/informers/externalversions"
	"testing"
)

var clusters = []*clusterv1alpha1.Cluster{
	{
		ObjectMeta: metav1.ObjectMeta{
			Name: "foo",
			Labels: map[string]string{
				"cluster.kubesphere.io/region": "beijing",
				"cluster.kubesphere.io/group":  "development",
			},
		},
	},
	{
		ObjectMeta: metav1.ObjectMeta{
			Name: "bar",
			Labels: map[string]string{
				"cluster.kubesphere.io/region": "beijing",
				"cluster.kubesphere.io/group":  "production",
			},
		},
	},
	{
		ObjectMeta: metav1.ObjectMeta{
			Name: "whatever",
			Labels: map[string]string{
				"cluster.kubesphere.io/region": "shanghai",
				"cluster.kubesphere.io/group":  "testing",
			},
		},
	},
}

func clustersToInterface(clusters ...*clusterv1alpha1.Cluster) []interface{} {
	items := make([]interface{}, 0)

	for _, cluster := range clusters {
		items = append(items, cluster)
	}

	return items
}

func clustersToRuntimeObject(clusters ...*clusterv1alpha1.Cluster) []runtime.Object {
	items := make([]runtime.Object, 0)

	for _, cluster := range clusters {
		items = append(items, cluster)
	}

	return items
}

func TestClustersGetter(t *testing.T) {
	var testCases = []struct {
		description string
		query       *query.Query
		expected    *api.ListResult
	}{
		{
			description: "Test normal case",
			query: &query.Query{
				LabelSelector: "cluster.kubesphere.io/region=beijing",
				Ascending:     false,
			},
			expected: &api.ListResult{
				TotalItems: 2,
				Items:      clustersToInterface(clusters[0], clusters[1]),
			},
		},
	}

	client := fake.NewSimpleClientset(clustersToRuntimeObject(clusters...)...)
	informer := externalversions.NewSharedInformerFactory(client, 0)

	for _, cluster := range clusters {
		informer.Cluster().V1alpha1().Clusters().Informer().GetIndexer().Add(cluster)
	}

	for _, testCase := range testCases {

		clusterGetter := New(informer)
		t.Run(testCase.description, func(t *testing.T) {
			result, err := clusterGetter.List("", testCase.query)
			if err != nil {
				t.Error(err)
			}

			if diff := cmp.Diff(result, testCase.expected); len(diff) != 0 {
				t.Errorf("%T, got+ expected-, %s", testCase.expected, diff)
			}
		})
	}
}