photoprism/pkg/fs/walk_test.go
Michael Mayer a73ee48213 Import/Convert: Improve symbolic link handling to support files #1049
Signed-off-by: Michael Mayer <michael@photoprism.app>
2022-07-07 19:54:22 +02:00

115 lines
3.4 KiB
Go

package fs
import (
"testing"
"github.com/karrick/godirwalk"
"github.com/stretchr/testify/assert"
)
func TestSkipWalk(t *testing.T) {
t.Run("done", func(t *testing.T) {
done := make(Done)
ignore := NewIgnoreList(".ppignore", true, false)
done["foo.jpg"] = Found
if skip, result := SkipWalk("testdata/directory", true, false, done, ignore); skip {
assert.Nil(t, result)
} else {
t.Fatal("should be skipped")
}
assert.True(t, done["foo.jpg"].Exists())
assert.True(t, done["testdata/directory"].Exists())
assert.Equal(t, 2, len(done))
})
t.Run("symlink", func(t *testing.T) {
done := make(Done)
ignore := NewIgnoreList(".ppignore", true, false)
if skip, result := SkipWalk("testdata/directory/subdirectory/symlink", false, true, done, ignore); skip {
assert.Nil(t, result)
} else {
t.Fatal("should be skipped")
}
if skip, result := SkipWalk("testdata/directory/subdirectory/symlink/self", false, true, done, ignore); skip {
assert.Error(t, result)
} else {
t.Fatal("should be skipped")
}
if skip, result := SkipWalk("testdata/directory/subdirectory/symlink/self/self", false, true, done, ignore); skip {
assert.Error(t, result)
} else {
t.Fatal("should be skipped")
}
assert.True(t, done["testdata/linked"].Exists())
assert.True(t, done["testdata/directory/subdirectory/symlink"].Exists())
assert.True(t, done["testdata/directory/subdirectory/symlink/self"].Exists())
assert.True(t, done["testdata/directory/subdirectory/symlink/self/self"].Exists())
assert.Equal(t, 4, len(done))
})
t.Run("godirwalk", func(t *testing.T) {
done := make(Done)
var skipped []string
var skippedDirs []string
testPath := "testdata"
ignore := NewIgnoreList(".ppignore", true, false)
err := godirwalk.Walk(testPath, &godirwalk.Options{
Callback: func(fileName string, info *godirwalk.Dirent) error {
isDir, _ := info.IsDirOrSymlinkToDir()
isSymlink := info.IsSymlink()
if skip, result := SkipWalk(fileName, isDir, isSymlink, done, ignore); skip {
if result != nil {
skippedDirs = append(skippedDirs, fileName)
} else {
skipped = append(skipped, fileName)
}
return result
}
done[fileName] = Found
if textName := TextFile.Find(fileName, false); textName != "" {
done[textName] = Found
}
return nil
},
Unsorted: false,
FollowSymbolicLinks: true,
})
assert.NoError(t, err)
assert.Contains(t, skippedDirs, "testdata/directory/subdirectory/.hiddendir")
expectSkipped := []string{
"testdata", "testdata/directory",
"testdata/directory/.ppignore",
"testdata/directory/bar.txt",
"testdata/directory/baz.xml",
"testdata/directory/subdirectory",
"testdata/directory/subdirectory/.hiddenfile",
"testdata/directory/subdirectory/.ppignore",
"testdata/directory/subdirectory/animals",
"testdata/directory/subdirectory/animals/.ppignore",
"testdata/directory/subdirectory/animals/dog.json",
"testdata/directory/subdirectory/animals/gopher.json",
"testdata/directory/subdirectory/animals/gopher.md",
"testdata/directory/subdirectory/example.txt",
"testdata/directory/subdirectory/foo.txt",
"testdata/directory/subdirectory/symlink",
"testdata/directory/subdirectory/symlink/somefile.txt",
"testdata/directory/subdirectory/symlink/test.md",
"testdata/directory/subdirectory/symlink/test.txt"}
assert.Equal(t, expectSkipped, skipped)
})
}