package sync import ( "context" "os" "path/filepath" "testing" "github.com/databricks/cli/libs/fileset" "github.com/databricks/cli/libs/git" "github.com/stretchr/testify/require" ) func createFile(dir string, name string) error { f, err := os.Create(filepath.Join(dir, name)) if err != nil { return err } return f.Close() } func setupFiles(t *testing.T) string { dir := t.TempDir() err := createFile(dir, "a.go") require.NoError(t, err) err = createFile(dir, "b.go") require.NoError(t, err) err = createFile(dir, "ab.go") require.NoError(t, err) err = createFile(dir, "abc.go") require.NoError(t, err) err = createFile(dir, "c.go") require.NoError(t, err) err = createFile(dir, "d.go") require.NoError(t, err) dbDir := filepath.Join(dir, ".databricks") err = os.Mkdir(dbDir, 0755) require.NoError(t, err) err = createFile(dbDir, "e.go") require.NoError(t, err) testDir := filepath.Join(dir, "test") err = os.Mkdir(testDir, 0755) require.NoError(t, err) sub1 := filepath.Join(testDir, "sub1") err = os.Mkdir(sub1, 0755) require.NoError(t, err) err = createFile(sub1, "f.go") require.NoError(t, err) sub2 := filepath.Join(sub1, "sub2") err = os.Mkdir(sub2, 0755) require.NoError(t, err) err = createFile(sub2, "g.go") require.NoError(t, err) return dir } func TestGetFileSet(t *testing.T) { ctx := context.Background() dir := setupFiles(t) fileSet, err := git.NewFileSet(dir) require.NoError(t, err) err = fileSet.EnsureValidGitIgnoreExists() require.NoError(t, err) inc, err := fileset.NewGlobSet(dir, []string{}) require.NoError(t, err) excl, err := fileset.NewGlobSet(dir, []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), 9) inc, err = fileset.NewGlobSet(dir, []string{}) require.NoError(t, err) excl, err = fileset.NewGlobSet(dir, []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), 1) inc, err = fileset.NewGlobSet(dir, []string{".databricks/*"}) require.NoError(t, err) excl, err = fileset.NewGlobSet(dir, []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) } func TestRecursiveExclude(t *testing.T) { ctx := context.Background() dir := setupFiles(t) fileSet, err := git.NewFileSet(dir) require.NoError(t, err) err = fileSet.EnsureValidGitIgnoreExists() require.NoError(t, err) inc, err := fileset.NewGlobSet(dir, []string{}) require.NoError(t, err) excl, err := fileset.NewGlobSet(dir, []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) }