databricks-cli/libs/sync/sync_test.go

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)
}