main_test.go

package main

import (
	"log"
	"path/filepath"
	"testing"

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

func getKubeConfig() *KubectlConfig {
	const KUBECONFIG_FILENAME = "kubeconfig.yaml"
	return getKubeConfigFromPath(KUBECONFIG_FILENAME)
}

func getKubeConfigFromPath(filename string) *KubectlConfig {
	const TEST_DATA_PATH = "test/data"
	kubeConfig, err := ParseKubeConfig(filepath.Join(TEST_DATA_PATH, filename))
	if err != nil {
		log.Panic(err)
	}

	return kubeConfig
}

// cluster name conflict
func Test_Merge_ClusterNameConflict(t *testing.T) {
	var kubeConfig = getKubeConfig()
	var toBeAddedKubeConfig = getKubeConfigFromPath("valid-default-cluster.yaml")

	toBeAddedKubeConfig.Clusters[0].Name = kubeConfig.Clusters[0].Name

	_, err := Merge(*kubeConfig, *toBeAddedKubeConfig, "", toBeAddedKubeConfig.Clusters[0].Name)
	if err == nil {
		t.Fatal()
	}
}

// context name conflict
func Test_Merge_ContextNameConflict(t *testing.T) {
	var kubeConfig = getKubeConfig()
	var toBeAddedKubeConfig = getKubeConfigFromPath("valid-default-cluster.yaml")

	toBeAddedKubeConfig.Contexts[0].Name = kubeConfig.Contexts[0].Name

	_, err := Merge(*kubeConfig, *toBeAddedKubeConfig, "", toBeAddedKubeConfig.Contexts[0].Name)
	if err == nil {
		t.Fatal()
	}
}

// user name conflict
func Test_Merge_UserNameConflict(t *testing.T) {
	var kubeConfig = getKubeConfig()
	var toBeAddedKubeConfig = getKubeConfigFromPath("valid-default-cluster.yaml")

	toBeAddedKubeConfig.Users[0].Name = kubeConfig.Users[0].Name

	_, err := Merge(*kubeConfig, *toBeAddedKubeConfig, "", toBeAddedKubeConfig.Users[0].Name)
	if err == nil {
		t.Fatal()
	}
}

// multiple cluster
func Test_Merge_MultipleCluster(t *testing.T) {
	var kubeConfig = getKubeConfig()
	var toBeAddedKubeConfig = getKubeConfigFromPath("valid-default-cluster.yaml")

	var newObj KubectlClusterWithName
	newObj.Name = "new-cluster"
	toBeAddedKubeConfig.Clusters = append(toBeAddedKubeConfig.Clusters, &newObj)

	_, err := Merge(*kubeConfig, *toBeAddedKubeConfig, "", toBeAddedKubeConfig.Clusters[0].Name)
	if err == nil {
		t.Fatal()
	}
}

// multiple context
func Test_Merge_MultipleContext(t *testing.T) {
	var kubeConfig = getKubeConfig()
	var toBeAddedKubeConfig = getKubeConfigFromPath("valid-default-cluster.yaml")

	var newObj KubectlContextWithName
	newObj.Name = "new-context"
	toBeAddedKubeConfig.Contexts = append(toBeAddedKubeConfig.Contexts, &newObj)

	_, err := Merge(*kubeConfig, *toBeAddedKubeConfig, "", toBeAddedKubeConfig.Clusters[0].Name)
	if err == nil {
		t.Fatal()
	}
}

// multiple user
func Test_Merge_MultipleUser(t *testing.T) {
	var kubeConfig = getKubeConfig()
	var toBeAddedKubeConfig = getKubeConfigFromPath("valid-default-cluster.yaml")

	var newObj KubectlUserWithName
	newObj.Name = "new-user"
	toBeAddedKubeConfig.Users = append(toBeAddedKubeConfig.Users, &newObj)

	_, err := Merge(*kubeConfig, *toBeAddedKubeConfig, "", toBeAddedKubeConfig.Clusters[0].Name)
	if err == nil {
		t.Fatal()
	}
}

func Test_Merge_ExplicitlySetName(t *testing.T) {
	const fileName = "valid-default-cluster"
	var kubeConfig = getKubeConfig()
	var toBeAddedKubeConfig = getKubeConfigFromPath(fileName + ".yaml")
	var countBeforeMerge = len(kubeConfig.Clusters)

	var explicitName = uuid.New().String()
	toBeAddedKubeConfig.Clusters[0].Name = explicitName
	toBeAddedKubeConfig.Contexts[0].Name = explicitName
	toBeAddedKubeConfig.Users[0].Name = explicitName

	result, err := Merge(*kubeConfig, *toBeAddedKubeConfig, explicitName, fileName)
	assert.NoError(t, err)

	assert.Equal(t, len(result.Contexts), countBeforeMerge+1)
	assert.Equal(t, len(result.Clusters), countBeforeMerge+1)
	assert.Equal(t, len(result.Users), countBeforeMerge+1)

	assert.EqualValues(t, result.Contexts[countBeforeMerge].Name, explicitName)
	assert.EqualValues(t, result.Clusters[countBeforeMerge].Name, explicitName)
	assert.EqualValues(t, result.Users[countBeforeMerge].Name, explicitName)
}

func Test_Merge_ImplicitlySetName(t *testing.T) {
	const fileName = "valid-default-cluster"
	var kubeConfig = getKubeConfig()
	var toBeAddedKubeConfig = getKubeConfigFromPath(fileName + ".yaml")
	var countBeforeMerge = len(kubeConfig.Clusters)

	var explicitName = uuid.New().String()
	toBeAddedKubeConfig.Clusters[0].Name = explicitName
	toBeAddedKubeConfig.Contexts[0].Name = explicitName
	toBeAddedKubeConfig.Users[0].Name = explicitName

	result, err := Merge(*kubeConfig, *toBeAddedKubeConfig, "", fileName)
	assert.NoError(t, err)

	assert.Equal(t, len(result.Contexts), countBeforeMerge+1)
	assert.Equal(t, len(result.Clusters), countBeforeMerge+1)
	assert.Equal(t, len(result.Users), countBeforeMerge+1)

	assert.EqualValues(t, result.Contexts[countBeforeMerge].Name, fileName)
	assert.EqualValues(t, result.Clusters[countBeforeMerge].Name, fileName)
	assert.EqualValues(t, result.Users[countBeforeMerge].Name, fileName)
}