package mapper

import (
	"testing"

	clusterType "edge-infra.dev/pkg/edge/constants/api/cluster"
	"edge-infra.dev/pkg/edge/constants/api/fleet"

	"github.com/stretchr/testify/assert"
)

func TestToClusterAndFleetType(t *testing.T) {
	gkeLabelKeys := []string{"cluster-infra", "gke"}
	invalidLabelKeys := []string{"invalid-cluster", "invalid-fleet"}

	gkeFleet, err := ToFleetType(gkeLabelKeys)
	assert.Nil(t, err)
	gkeCluster, err := ToClusterType(gkeLabelKeys)
	assert.Nil(t, err)

	assert.Equal(t, fleet.Cluster, gkeFleet)
	assert.Equal(t, clusterType.GKE, gkeCluster)

	invalidFleet, err := ToFleetType(invalidLabelKeys)
	assert.Error(t, err)
	assert.Empty(t, invalidFleet)

	invalidCluster, err := ToClusterType(invalidLabelKeys)
	assert.Error(t, err)
	assert.Empty(t, invalidCluster)
}