mirror of https://github.com/databricks/cli.git
101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
|
package dynloc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/databricks/cli/libs/dyn"
|
||
|
|
"github.com/databricks/cli/libs/dyn/merge"
|
||
|
|
"github.com/databricks/cli/libs/dyn/yamlloader"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func load(t *testing.T, path string) dyn.Value {
|
||
|
|
matches, err := filepath.Glob(path + "/*.yml")
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotEmpty(t, matches)
|
||
|
|
|
||
|
|
// Load all files.
|
||
|
|
vout := dyn.NilValue
|
||
|
|
for _, match := range matches {
|
||
|
|
buf, err := os.ReadFile(match)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
v, err := yamlloader.LoadYAML(match, bytes.NewBuffer(buf))
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
vout, err = merge.Merge(vout, v)
|
||
|
|
require.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return vout
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLocations_Default(t *testing.T) {
|
||
|
|
v := load(t, "testdata/default")
|
||
|
|
locs, err := Build(v)
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, 1, locs.Version)
|
||
|
|
assert.Equal(t, []string{"testdata/default/a.yml", "testdata/default/b.yml"}, locs.Files)
|
||
|
|
assert.Equal(t, map[string][][]int{
|
||
|
|
"a": {{0, 2, 3}},
|
||
|
|
"a.b": {{0, 2, 6}},
|
||
|
|
"b": {{1, 2, 3}},
|
||
|
|
"b.c": {{1, 2, 6}},
|
||
|
|
}, locs.Locations)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLocations_DefaultWithBasePath(t *testing.T) {
|
||
|
|
v := load(t, "testdata/default")
|
||
|
|
locs, err := Build(v, WithBasePath("testdata/default"))
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, 1, locs.Version)
|
||
|
|
assert.Equal(t, []string{"a.yml", "b.yml"}, locs.Files)
|
||
|
|
assert.Equal(t, map[string][][]int{
|
||
|
|
"a": {{0, 2, 3}},
|
||
|
|
"a.b": {{0, 2, 6}},
|
||
|
|
"b": {{1, 2, 3}},
|
||
|
|
"b.c": {{1, 2, 6}},
|
||
|
|
}, locs.Locations)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLocations_Override(t *testing.T) {
|
||
|
|
v := load(t, "testdata/override")
|
||
|
|
locs, err := Build(v)
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, 1, locs.Version)
|
||
|
|
assert.Equal(t, []string{"testdata/override/a.yml", "testdata/override/b.yml"}, locs.Files)
|
||
|
|
|
||
|
|
// Note: specific ordering of locations is described in [merge.Merge].
|
||
|
|
assert.Equal(t, map[string][][]int{
|
||
|
|
"a": {
|
||
|
|
{0, 2, 3},
|
||
|
|
{1, 2, 3},
|
||
|
|
},
|
||
|
|
"a.b": {
|
||
|
|
{1, 2, 6},
|
||
|
|
{0, 2, 6},
|
||
|
|
},
|
||
|
|
}, locs.Locations)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLocations_MaxDepth(t *testing.T) {
|
||
|
|
v := load(t, "testdata/depth")
|
||
|
|
|
||
|
|
var locs Locations
|
||
|
|
var err error
|
||
|
|
|
||
|
|
// Test with no max depth.
|
||
|
|
locs, err = Build(v)
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Len(t, locs.Locations, 5)
|
||
|
|
|
||
|
|
// Test with max depth and see that the number of locations is reduced.
|
||
|
|
locs, err = Build(v, WithMaxDepth(3))
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Len(t, locs.Locations, 3)
|
||
|
|
}
|