mirror of https://github.com/databricks/cli.git
89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
package resources
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/config/resources"
|
|
"github.com/databricks/databricks-sdk-go/service/jobs"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestLookup_EmptyBundle(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{},
|
|
},
|
|
}
|
|
|
|
_, err := Lookup(b, "foo")
|
|
require.Error(t, err)
|
|
assert.ErrorContains(t, err, "resource with key \"foo\" not found")
|
|
}
|
|
|
|
func TestLookup_NotFound(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"foo": {},
|
|
"bar": {},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
_, err := Lookup(b, "qux")
|
|
require.Error(t, err)
|
|
assert.ErrorContains(t, err, `resource with key "qux" not found`)
|
|
}
|
|
|
|
func TestLookup_MultipleFound(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"foo": {},
|
|
},
|
|
Pipelines: map[string]*resources.Pipeline{
|
|
"foo": {},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
_, err := Lookup(b, "foo")
|
|
require.Error(t, err)
|
|
assert.ErrorContains(t, err, `multiple resources with key "foo" found`)
|
|
}
|
|
|
|
func TestLookup_Nominal(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Resources: config.Resources{
|
|
Jobs: map[string]*resources.Job{
|
|
"foo": {
|
|
JobSettings: &jobs.JobSettings{
|
|
Name: "Foo job",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
// Lookup by key only.
|
|
out, err := Lookup(b, "foo")
|
|
if assert.NoError(t, err) {
|
|
assert.Equal(t, "Foo job", out.Resource.GetName())
|
|
}
|
|
|
|
// Lookup by type and key.
|
|
out, err = Lookup(b, "jobs.foo")
|
|
if assert.NoError(t, err) {
|
|
assert.Equal(t, "Foo job", out.Resource.GetName())
|
|
}
|
|
}
|