110 lines
3 KiB
Go
110 lines
3 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(map[string]bool)
|
||
|
|
ignore := NewIgnoreList(".ppignore", true, false)
|
||
|
|
|
||
|
|
done["foo.jpg"] = true
|
||
|
|
|
||
|
|
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"])
|
||
|
|
assert.True(t, done["testdata/directory"])
|
||
|
|
assert.Equal(t, 2, len(done))
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("symlink", func(t *testing.T) {
|
||
|
|
done := make(map[string]bool)
|
||
|
|
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"])
|
||
|
|
assert.True(t, done["testdata/directory/subdirectory/symlink"])
|
||
|
|
assert.True(t, done["testdata/directory/subdirectory/symlink/self"])
|
||
|
|
assert.True(t, done["testdata/directory/subdirectory/symlink/self/self"])
|
||
|
|
assert.Equal(t, 4, len(done))
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("godirwalk", func(t *testing.T) {
|
||
|
|
done := make(map[string]bool)
|
||
|
|
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.IsDir()
|
||
|
|
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] = true
|
||
|
|
|
||
|
|
if textName := TypeText.Find(fileName, false); textName != "" {
|
||
|
|
done[textName] = true
|
||
|
|
}
|
||
|
|
|
||
|
|
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/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)
|
||
|
|
})
|
||
|
|
}
|