mirror of https://github.com/databricks/cli.git
128 lines
3.0 KiB
Go
128 lines
3.0 KiB
Go
package mutator_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/config/mutator"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func touch(t *testing.T, path, file string) {
|
|
f, err := os.Create(filepath.Join(path, file))
|
|
require.NoError(t, err)
|
|
f.Close()
|
|
}
|
|
|
|
func TestProcessRootIncludesEmpty(t *testing.T) {
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Path: ".",
|
|
},
|
|
}
|
|
_, err := mutator.ProcessRootIncludes().Apply(context.Background(), bundle)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestProcessRootIncludesAbs(t *testing.T) {
|
|
// remove this once equivalent tests for windows have been set up
|
|
// or this test has been fixed for windows
|
|
// date: 28 Nov 2022
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip("skipping temperorilty to make windows unit tests green")
|
|
}
|
|
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Path: ".",
|
|
Include: []string{
|
|
"/tmp/*.yml",
|
|
},
|
|
},
|
|
}
|
|
_, err := mutator.ProcessRootIncludes().Apply(context.Background(), bundle)
|
|
require.Error(t, err)
|
|
assert.Contains(t, err.Error(), "must be relative paths")
|
|
}
|
|
|
|
func TestProcessRootIncludesSingleGlob(t *testing.T) {
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Path: t.TempDir(),
|
|
Include: []string{
|
|
"*.yml",
|
|
},
|
|
},
|
|
}
|
|
|
|
touch(t, bundle.Config.Path, "bundle.yml")
|
|
touch(t, bundle.Config.Path, "a.yml")
|
|
touch(t, bundle.Config.Path, "b.yml")
|
|
|
|
ms, err := mutator.ProcessRootIncludes().Apply(context.Background(), bundle)
|
|
require.NoError(t, err)
|
|
|
|
var names []string
|
|
for _, m := range ms {
|
|
names = append(names, m.Name())
|
|
}
|
|
|
|
assert.NotContains(t, names, "ProcessInclude(bundle.yml)")
|
|
assert.Contains(t, names, "ProcessInclude(a.yml)")
|
|
assert.Contains(t, names, "ProcessInclude(b.yml)")
|
|
assert.Equal(t, []string{"a.yml", "b.yml"}, bundle.Config.Include)
|
|
}
|
|
|
|
func TestProcessRootIncludesMultiGlob(t *testing.T) {
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Path: t.TempDir(),
|
|
Include: []string{
|
|
"a*.yml",
|
|
"b*.yml",
|
|
},
|
|
},
|
|
}
|
|
|
|
touch(t, bundle.Config.Path, "a1.yml")
|
|
touch(t, bundle.Config.Path, "b1.yml")
|
|
|
|
ms, err := mutator.ProcessRootIncludes().Apply(context.Background(), bundle)
|
|
require.NoError(t, err)
|
|
|
|
var names []string
|
|
for _, m := range ms {
|
|
names = append(names, m.Name())
|
|
}
|
|
|
|
assert.Contains(t, names, "ProcessInclude(a1.yml)")
|
|
assert.Contains(t, names, "ProcessInclude(b1.yml)")
|
|
assert.Equal(t, []string{"a1.yml", "b1.yml"}, bundle.Config.Include)
|
|
}
|
|
|
|
func TestProcessRootIncludesRemoveDups(t *testing.T) {
|
|
bundle := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Path: t.TempDir(),
|
|
Include: []string{
|
|
"*.yml",
|
|
"*.yml",
|
|
},
|
|
},
|
|
}
|
|
|
|
touch(t, bundle.Config.Path, "a.yml")
|
|
|
|
ms, err := mutator.ProcessRootIncludes().Apply(context.Background(), bundle)
|
|
require.NoError(t, err)
|
|
assert.Len(t, ms, 1)
|
|
assert.Equal(t, "ProcessInclude(a.yml)", ms[0].Name())
|
|
assert.Equal(t, []string{"a.yml"}, bundle.Config.Include)
|
|
}
|