databricks-cli/libs/git/fileset_test.go

package git

import (
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestFileSetRecursiveListFiles(t *testing.T) {
	fileSet, err := NewFileSet("./testdata")
	require.NoError(t, err)
	files, err := fileSet.RecursiveListFiles("./testdata")
	require.NoError(t, err)
	require.Len(t, files, 6)
	assert.Equal(t, filepath.Join(".gitignore"), files[0].Relative)
	assert.Equal(t, filepath.Join("a", ".gitignore"), files[1].Relative)
	assert.Equal(t, filepath.Join("a", "b", ".gitignore"), files[2].Relative)
	assert.Equal(t, filepath.Join("a", "b", "world.txt"), files[3].Relative)
	assert.Equal(t, filepath.Join("a", "hello.txt"), files[4].Relative)
	assert.Equal(t, filepath.Join("databricks.yml"), files[5].Relative)
}

func TestFileSetNonCleanRoot(t *testing.T) {
	// Test what happens if the root directory can be simplified.
	// Path simplification is done by most filepath functions.
	// This should yield the same result as above test.
	fileSet, err := NewFileSet("./testdata/../testdata")
	require.NoError(t, err)
	files, err := fileSet.All()
	require.NoError(t, err)
	assert.Len(t, files, 6)
}

func TestFilesetAddsCacheDirToGitIgnore(t *testing.T) {
	projectDir := t.TempDir()
	fileSet, err := NewFileSet(projectDir)
	require.NoError(t, err)
	fileSet.EnsureValidGitIgnoreExists()

	gitIgnorePath := filepath.Join(projectDir, ".gitignore")
	assert.FileExists(t, gitIgnorePath)
	fileBytes, err := os.ReadFile(gitIgnorePath)
	assert.NoError(t, err)
	assert.Contains(t, string(fileBytes), ".databricks")
}

func TestFilesetDoesNotCacheDirToGitIgnoreIfAlreadyPresent(t *testing.T) {
	projectDir := t.TempDir()
	gitIgnorePath := filepath.Join(projectDir, ".gitignore")

	fileSet, err := NewFileSet(projectDir)
	require.NoError(t, err)
	err = os.WriteFile(gitIgnorePath, []byte(".databricks"), 0o644)
	require.NoError(t, err)

	fileSet.EnsureValidGitIgnoreExists()

	b, err := os.ReadFile(gitIgnorePath)
	require.NoError(t, err)
	assert.Equal(t, 1, strings.Count(string(b), ".databricks"))
}