github.com/jenkins-x/jx/v2@v2.1.155/pkg/kube/cluster/cluster_test.go (about)

     1  // +build unit
     2  
     3  package cluster_test
     4  
     5  import (
     6  	"testing"
     7  
     8  	. "github.com/petergtz/pegomock"
     9  	"github.com/stretchr/testify/assert"
    10  	"k8s.io/client-go/tools/clientcmd/api"
    11  
    12  	"github.com/jenkins-x/jx/v2/pkg/kube/cluster"
    13  	kube_test "github.com/jenkins-x/jx/v2/pkg/kube/mocks"
    14  	"github.com/jenkins-x/jx/v2/pkg/kube/vault"
    15  )
    16  
    17  func TestGetSimplifiedClusterName(t *testing.T) {
    18  	t.Parallel()
    19  	simpleName := cluster.SimplifiedClusterName("gke_jenkinsx-dev_europe-west1-b_my-cluster-name")
    20  
    21  	assert.Equal(t, "my-cluster-name", simpleName)
    22  }
    23  
    24  func TestShortClusterName(t *testing.T) {
    25  	kuber := kube_test.NewMockKuber()
    26  
    27  	config := api.Config{
    28  		CurrentContext: "myContext",
    29  		Contexts: map[string]*api.Context{
    30  			"myContext": {Cluster: "short-cluster-name"},
    31  		},
    32  	}
    33  	When(kuber.LoadConfig()).ThenReturn(&config, nil, nil)
    34  
    35  	clusterName, err := cluster.ShortName(kuber)
    36  
    37  	assert.NoError(t, err)
    38  	assert.Equal(t, "short-cluster-na", clusterName)
    39  
    40  	config = api.Config{
    41  		CurrentContext: "myContext",
    42  		Contexts: map[string]*api.Context{
    43  			"myContext": {Cluster: "short-cluster-na"},
    44  		},
    45  	}
    46  	When(kuber.LoadConfig()).ThenReturn(&config, nil, nil)
    47  
    48  	clusterName, err = cluster.ShortName(kuber)
    49  
    50  	assert.NoError(t, err)
    51  	assert.Equal(t, "short-cluster-na", clusterName)
    52  
    53  	config = api.Config{
    54  		CurrentContext: "myContext",
    55  		Contexts: map[string]*api.Context{
    56  			"myContext": {Cluster: "short-cluster-na-test"},
    57  		},
    58  	}
    59  	When(kuber.LoadConfig()).ThenReturn(&config, nil, nil)
    60  
    61  	clusterName, err = cluster.ShortName(kuber)
    62  
    63  	assert.NoError(t, err)
    64  	assert.Equal(t, "short-cluster-na", clusterName)
    65  }
    66  
    67  func TestClusterName(t *testing.T) {
    68  	kuber := kube_test.NewMockKuber()
    69  
    70  	config := api.Config{
    71  		CurrentContext: "myContext",
    72  		Contexts: map[string]*api.Context{
    73  			"myContext": {Cluster: "my-cluster-name"},
    74  		},
    75  	}
    76  	When(kuber.LoadConfig()).ThenReturn(&config, nil, nil)
    77  
    78  	clusterName, err := cluster.Name(kuber)
    79  
    80  	assert.NoError(t, err)
    81  	assert.Equal(t, "my-cluster-name", clusterName)
    82  }
    83  
    84  func TestSystemVaultNameForCluster(t *testing.T) {
    85  	actual := vault.SystemVaultNameForCluster("jstrachan-kp38")
    86  	assert.Equal(t, "jx-vault-jstrachan-kp3", actual, "system vault name")
    87  }