package kustomization

import (
	"testing"

	"github.com/stretchr/testify/assert"
	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	clusterApi "edge-infra.dev/pkg/edge/apis/cluster/v1alpha1"
	"edge-infra.dev/pkg/edge/constants/api/cluster"
	"edge-infra.dev/pkg/edge/constants/api/fleet"
)

func TestForemanKustomization(t *testing.T) {
	foremanCluster := &clusterApi.Cluster{
		ObjectMeta: v1.ObjectMeta{
			Name:      "foreman0",
			Namespace: "edge-flux",
		},
		Spec: clusterApi.ClusterSpec{
			Name:      "foreman0",
			Fleet:     fleet.TopLevel,
			Type:      cluster.GKE,
			Location:  "us-central-c",
			ProjectID: "ret-edge-dev0-foreman",
		},
	}
	foremanKustomization := New().
		SetCluster(foremanCluster).
		UseDefaultForemanRules().
		CreateClusterKustomizations()
	assert.Equal(t, len(foremanKustomization), 6)
}

func TestStoreKustomization(t *testing.T) {
	storeCluster := &clusterApi.Cluster{
		ObjectMeta: v1.ObjectMeta{
			Name:      "8c4e4f90-e26c-46cf-9ad2-411305b33cb0",
			Namespace: "edge-flux",
		},
		Spec: clusterApi.ClusterSpec{
			Name:      "test-store",
			Fleet:     fleet.Store,
			Type:      cluster.GKE,
			Location:  "us-central-c",
			ProjectID: "ret-edge-fdghufdbghhs",
		},
	}
	storeKustomization := New().
		SetCluster(storeCluster).
		UseDefaultStoreRules().
		CreateClusterKustomizations()
	assert.Equal(t, len(storeKustomization), 10)
}

func TestWarehouseKustomization(t *testing.T) {
	storeCluster := &clusterApi.Cluster{
		ObjectMeta: v1.ObjectMeta{
			Name:      "8c4e4f90-e26c-46cf-9ad2-411305b33cb0",
			Namespace: "edge-flux",
		},
		Spec: clusterApi.ClusterSpec{
			Name:      "test-store",
			Fleet:     fleet.Store,
			Type:      cluster.GKE,
			Location:  "us-central-c",
			ProjectID: "ret-edge-fdghufdbghhs",
		},
	}
	storeKustomization := New().
		SetCluster(storeCluster).
		UseWarehouseRules().
		CreateClusterKustomizations()
	assert.Equal(t, len(storeKustomization), 2)
}