mirror of https://github.com/databricks/cli.git
135 lines
3.1 KiB
Go
135 lines
3.1 KiB
Go
package mutator_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/config/mutator"
|
|
"github.com/databricks/cli/bundle/config/resources"
|
|
"github.com/databricks/databricks-sdk-go/service/compute"
|
|
"github.com/databricks/databricks-sdk-go/service/jobs"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestOverrideDevelopment(t *testing.T) {
|
|
os.Setenv("DATABRICKS_CLUSTER_ID", "")
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Bundle: config.Bundle{
|
|
Mode: config.Development,
|
|
ComputeID: "newClusterID",
|
|
},
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"job1": {JobSettings: &jobs.JobSettings{
|
|
Name: "job1",
|
|
Tasks: []jobs.Task{
|
|
{
|
|
NewCluster: &compute.ClusterSpec{},
|
|
},
|
|
{
|
|
ExistingClusterId: "cluster2",
|
|
},
|
|
},
|
|
}},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
m := mutator.OverrideCompute()
|
|
err := m.Apply(context.Background(), bundle)
|
|
require.NoError(t, err)
|
|
assert.Nil(t, bundle.Config.Resources.Jobs["job1"].Tasks[0].NewCluster)
|
|
assert.Equal(t, "newClusterID", bundle.Config.Resources.Jobs["job1"].Tasks[0].ExistingClusterId)
|
|
assert.Equal(t, "newClusterID", bundle.Config.Resources.Jobs["job1"].Tasks[1].ExistingClusterId)
|
|
}
|
|
|
|
func TestOverrideDevelopmentEnv(t *testing.T) {
|
|
os.Setenv("DATABRICKS_CLUSTER_ID", "newClusterId")
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"job1": {JobSettings: &jobs.JobSettings{
|
|
Name: "job1",
|
|
Tasks: []jobs.Task{
|
|
{
|
|
NewCluster: &compute.ClusterSpec{},
|
|
},
|
|
{
|
|
ExistingClusterId: "cluster2",
|
|
},
|
|
},
|
|
}},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
m := mutator.OverrideCompute()
|
|
err := m.Apply(context.Background(), bundle)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "cluster2", bundle.Config.Resources.Jobs["job1"].Tasks[1].ExistingClusterId)
|
|
}
|
|
|
|
func TestOverrideProduction(t *testing.T) {
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Bundle: config.Bundle{
|
|
ComputeID: "newClusterID",
|
|
},
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"job1": {JobSettings: &jobs.JobSettings{
|
|
Name: "job1",
|
|
Tasks: []jobs.Task{
|
|
{
|
|
NewCluster: &compute.ClusterSpec{},
|
|
},
|
|
{
|
|
ExistingClusterId: "cluster2",
|
|
},
|
|
},
|
|
}},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
m := mutator.OverrideCompute()
|
|
err := m.Apply(context.Background(), bundle)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestOverrideProductionEnv(t *testing.T) {
|
|
os.Setenv("DATABRICKS_CLUSTER_ID", "newClusterId")
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"job1": {JobSettings: &jobs.JobSettings{
|
|
Name: "job1",
|
|
Tasks: []jobs.Task{
|
|
{
|
|
NewCluster: &compute.ClusterSpec{},
|
|
},
|
|
{
|
|
ExistingClusterId: "cluster2",
|
|
},
|
|
},
|
|
}},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
m := mutator.OverrideCompute()
|
|
err := m.Apply(context.Background(), bundle)
|
|
require.NoError(t, err)
|
|
}
|