mirror of https://github.com/databricks/cli.git
98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package validate
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/config/resources"
|
|
"github.com/databricks/cli/libs/diag"
|
|
"github.com/databricks/databricks-sdk-go/service/jobs"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestJobClusterKeyDefined(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"job1": {
|
|
JobSettings: &jobs.JobSettings{
|
|
Name: "job1",
|
|
JobClusters: []jobs.JobCluster{
|
|
{JobClusterKey: "do-not-exist"},
|
|
},
|
|
Tasks: []jobs.Task{
|
|
{JobClusterKey: "do-not-exist"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
diags := bundle.ApplyReadOnly(context.Background(), bundle.ReadOnly(b), JobClusterKeyDefined())
|
|
require.Len(t, diags, 0)
|
|
require.NoError(t, diags.Error())
|
|
}
|
|
|
|
func TestJobClusterKeyNotDefined(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"job1": {
|
|
JobSettings: &jobs.JobSettings{
|
|
Name: "job1",
|
|
Tasks: []jobs.Task{
|
|
{JobClusterKey: "do-not-exist"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
diags := bundle.ApplyReadOnly(context.Background(), bundle.ReadOnly(b), JobClusterKeyDefined())
|
|
require.Len(t, diags, 1)
|
|
require.NoError(t, diags.Error())
|
|
require.Equal(t, diags[0].Severity, diag.Warning)
|
|
require.Equal(t, diags[0].Summary, "job_cluster_key do-not-exist is not defined")
|
|
}
|
|
|
|
func TestJobClusterKeyDefinedInDifferentJob(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"job1": {
|
|
JobSettings: &jobs.JobSettings{
|
|
Name: "job1",
|
|
Tasks: []jobs.Task{
|
|
{JobClusterKey: "do-not-exist"},
|
|
},
|
|
},
|
|
},
|
|
"job2": {
|
|
JobSettings: &jobs.JobSettings{
|
|
Name: "job2",
|
|
JobClusters: []jobs.JobCluster{
|
|
{JobClusterKey: "do-not-exist"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
diags := bundle.ApplyReadOnly(context.Background(), bundle.ReadOnly(b), JobClusterKeyDefined())
|
|
require.Len(t, diags, 1)
|
|
require.NoError(t, diags.Error())
|
|
require.Equal(t, diags[0].Severity, diag.Warning)
|
|
require.Equal(t, diags[0].Summary, "job_cluster_key do-not-exist is not defined")
|
|
}
|