1 package mapper
2
3 import (
4 "testing"
5
6 clusterType "edge-infra.dev/pkg/edge/constants/api/cluster"
7 "edge-infra.dev/pkg/edge/constants/api/fleet"
8
9 "github.com/stretchr/testify/assert"
10 )
11
12 func TestToClusterAndFleetType(t *testing.T) {
13 gkeLabelKeys := []string{"cluster-infra", "gke"}
14 invalidLabelKeys := []string{"invalid-cluster", "invalid-fleet"}
15
16 gkeFleet, err := ToFleetType(gkeLabelKeys)
17 assert.Nil(t, err)
18 gkeCluster, err := ToClusterType(gkeLabelKeys)
19 assert.Nil(t, err)
20
21 assert.Equal(t, fleet.Cluster, gkeFleet)
22 assert.Equal(t, clusterType.GKE, gkeCluster)
23
24 invalidFleet, err := ToFleetType(invalidLabelKeys)
25 assert.Error(t, err)
26 assert.Empty(t, invalidFleet)
27
28 invalidCluster, err := ToClusterType(invalidLabelKeys)
29 assert.Error(t, err)
30 assert.Empty(t, invalidCluster)
31 }
32
View as plain text