mirror of https://github.com/databricks/cli.git
136 lines
3.7 KiB
Go
136 lines
3.7 KiB
Go
package mutator
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/config/variable"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/databricks/databricks-sdk-go/experimental/mocks"
|
|
"github.com/databricks/databricks-sdk-go/service/compute"
|
|
"github.com/databricks/databricks-sdk-go/service/iam"
|
|
)
|
|
|
|
func TestResolveClusterReference(t *testing.T) {
|
|
clusterRef1 := "Some Custom Cluster"
|
|
clusterRef2 := "Some Other Name"
|
|
justString := "random string"
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Variables: map[string]*variable.Variable{
|
|
"my-cluster-id-1": {
|
|
Lookup: &variable.Lookup{
|
|
Cluster: clusterRef1,
|
|
},
|
|
},
|
|
"my-cluster-id-2": {
|
|
Lookup: &variable.Lookup{
|
|
Cluster: clusterRef2,
|
|
},
|
|
},
|
|
"some-variable": {
|
|
Value: &justString,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
m := mocks.NewMockWorkspaceClient(t)
|
|
b.SetWorkpaceClient(m.WorkspaceClient)
|
|
clusterApi := m.GetMockClustersAPI()
|
|
clusterApi.EXPECT().GetByClusterName(mock.Anything, clusterRef1).Return(&compute.ClusterDetails{
|
|
ClusterId: "1234-5678-abcd",
|
|
}, nil)
|
|
clusterApi.EXPECT().GetByClusterName(mock.Anything, clusterRef2).Return(&compute.ClusterDetails{
|
|
ClusterId: "9876-5432-xywz",
|
|
}, nil)
|
|
|
|
err := bundle.Apply(context.Background(), b, ResolveResourceReferences())
|
|
require.NoError(t, err)
|
|
require.Equal(t, "1234-5678-abcd", *b.Config.Variables["my-cluster-id-1"].Value)
|
|
require.Equal(t, "9876-5432-xywz", *b.Config.Variables["my-cluster-id-2"].Value)
|
|
}
|
|
|
|
func TestResolveNonExistentClusterReference(t *testing.T) {
|
|
clusterRef := "Random"
|
|
justString := "random string"
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Variables: map[string]*variable.Variable{
|
|
"my-cluster-id": {
|
|
Lookup: &variable.Lookup{
|
|
Cluster: clusterRef,
|
|
},
|
|
},
|
|
"some-variable": {
|
|
Value: &justString,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
m := mocks.NewMockWorkspaceClient(t)
|
|
b.SetWorkpaceClient(m.WorkspaceClient)
|
|
clusterApi := m.GetMockClustersAPI()
|
|
clusterApi.EXPECT().GetByClusterName(mock.Anything, clusterRef).Return(nil, fmt.Errorf("ClusterDetails named '%s' does not exist", clusterRef))
|
|
|
|
err := bundle.Apply(context.Background(), b, ResolveResourceReferences())
|
|
require.ErrorContains(t, err, "failed to resolve cluster: Random, err: ClusterDetails named 'Random' does not exist")
|
|
}
|
|
|
|
func TestNoLookupIfVariableIsSet(t *testing.T) {
|
|
clusterRef := "donotexist"
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Variables: map[string]*variable.Variable{
|
|
"my-cluster-id": {
|
|
Lookup: &variable.Lookup{
|
|
Cluster: clusterRef,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
m := mocks.NewMockWorkspaceClient(t)
|
|
b.SetWorkpaceClient(m.WorkspaceClient)
|
|
|
|
b.Config.Variables["my-cluster-id"].Set("random value")
|
|
|
|
err := bundle.Apply(context.Background(), b, ResolveResourceReferences())
|
|
require.NoError(t, err)
|
|
require.Equal(t, "random value", *b.Config.Variables["my-cluster-id"].Value)
|
|
}
|
|
|
|
func TestResolveServicePrincipal(t *testing.T) {
|
|
spName := "Some SP name"
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Variables: map[string]*variable.Variable{
|
|
"my-sp": {
|
|
Lookup: &variable.Lookup{
|
|
ServicePrincipal: spName,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
m := mocks.NewMockWorkspaceClient(t)
|
|
b.SetWorkpaceClient(m.WorkspaceClient)
|
|
spApi := m.GetMockServicePrincipalsAPI()
|
|
spApi.EXPECT().GetByDisplayName(mock.Anything, spName).Return(&iam.ServicePrincipal{
|
|
Id: "1234",
|
|
ApplicationId: "app-1234",
|
|
}, nil)
|
|
|
|
err := bundle.Apply(context.Background(), b, ResolveResourceReferences())
|
|
require.NoError(t, err)
|
|
require.Equal(t, "app-1234", *b.Config.Variables["my-sp"].Value)
|
|
}
|