mirror of https://github.com/databricks/cli.git
161 lines
3.4 KiB
Go
161 lines
3.4 KiB
Go
package sync
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/internal/testutil"
|
|
"github.com/databricks/cli/libs/fileset"
|
|
"github.com/databricks/cli/libs/git"
|
|
"github.com/databricks/cli/libs/vfs"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func setupFiles(t *testing.T) string {
|
|
dir := t.TempDir()
|
|
|
|
for _, f := range []([]string){
|
|
[]string{dir, "a.go"},
|
|
[]string{dir, "b.go"},
|
|
[]string{dir, "ab.go"},
|
|
[]string{dir, "abc.go"},
|
|
[]string{dir, "c.go"},
|
|
[]string{dir, "d.go"},
|
|
[]string{dir, ".databricks", "e.go"},
|
|
[]string{dir, "test", "sub1", "f.go"},
|
|
[]string{dir, "test", "sub1", "sub2", "g.go"},
|
|
[]string{dir, "test", "sub1", "sub2", "h.txt"},
|
|
} {
|
|
testutil.Touch(t, f...)
|
|
}
|
|
|
|
return dir
|
|
}
|
|
|
|
func TestGetFileSet(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
dir := setupFiles(t)
|
|
root := vfs.MustNew(dir)
|
|
fileSet, err := git.NewFileSetAtRoot(root)
|
|
require.NoError(t, err)
|
|
|
|
err = fileSet.EnsureValidGitIgnoreExists()
|
|
require.NoError(t, err)
|
|
|
|
inc, err := fileset.NewGlobSet(root, []string{})
|
|
require.NoError(t, err)
|
|
|
|
excl, err := fileset.NewGlobSet(root, []string{})
|
|
require.NoError(t, err)
|
|
|
|
s := &Sync{
|
|
SyncOptions: &SyncOptions{},
|
|
|
|
fileSet: fileSet,
|
|
includeFileSet: inc,
|
|
excludeFileSet: excl,
|
|
}
|
|
|
|
fileList, err := s.GetFileList(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, len(fileList), 10)
|
|
|
|
inc, err = fileset.NewGlobSet(root, []string{})
|
|
require.NoError(t, err)
|
|
|
|
excl, err = fileset.NewGlobSet(root, []string{"*.go"})
|
|
require.NoError(t, err)
|
|
|
|
s = &Sync{
|
|
SyncOptions: &SyncOptions{},
|
|
|
|
fileSet: fileSet,
|
|
includeFileSet: inc,
|
|
excludeFileSet: excl,
|
|
}
|
|
|
|
fileList, err = s.GetFileList(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, len(fileList), 2)
|
|
|
|
inc, err = fileset.NewGlobSet(root, []string{"./.databricks/*.go"})
|
|
require.NoError(t, err)
|
|
|
|
excl, err = fileset.NewGlobSet(root, []string{})
|
|
require.NoError(t, err)
|
|
|
|
s = &Sync{
|
|
SyncOptions: &SyncOptions{},
|
|
|
|
fileSet: fileSet,
|
|
includeFileSet: inc,
|
|
excludeFileSet: excl,
|
|
}
|
|
|
|
fileList, err = s.GetFileList(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, len(fileList), 11)
|
|
}
|
|
|
|
func TestRecursiveExclude(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
dir := setupFiles(t)
|
|
root := vfs.MustNew(dir)
|
|
fileSet, err := git.NewFileSetAtRoot(root)
|
|
require.NoError(t, err)
|
|
|
|
err = fileSet.EnsureValidGitIgnoreExists()
|
|
require.NoError(t, err)
|
|
|
|
inc, err := fileset.NewGlobSet(root, []string{})
|
|
require.NoError(t, err)
|
|
|
|
excl, err := fileset.NewGlobSet(root, []string{"test/**"})
|
|
require.NoError(t, err)
|
|
|
|
s := &Sync{
|
|
SyncOptions: &SyncOptions{},
|
|
|
|
fileSet: fileSet,
|
|
includeFileSet: inc,
|
|
excludeFileSet: excl,
|
|
}
|
|
|
|
fileList, err := s.GetFileList(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, len(fileList), 7)
|
|
}
|
|
|
|
func TestNegateExclude(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
dir := setupFiles(t)
|
|
root := vfs.MustNew(dir)
|
|
fileSet, err := git.NewFileSetAtRoot(root)
|
|
require.NoError(t, err)
|
|
|
|
err = fileSet.EnsureValidGitIgnoreExists()
|
|
require.NoError(t, err)
|
|
|
|
inc, err := fileset.NewGlobSet(root, []string{})
|
|
require.NoError(t, err)
|
|
|
|
excl, err := fileset.NewGlobSet(root, []string{"./*", "!*.txt"})
|
|
require.NoError(t, err)
|
|
|
|
s := &Sync{
|
|
SyncOptions: &SyncOptions{},
|
|
|
|
fileSet: fileSet,
|
|
includeFileSet: inc,
|
|
excludeFileSet: excl,
|
|
}
|
|
|
|
fileList, err := s.GetFileList(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, len(fileList), 1)
|
|
require.Equal(t, fileList[0].Relative, "test/sub1/sub2/h.txt")
|
|
}
|