89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package util
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/warewulf/warewulf/internal/pkg/testenv"
|
|
"github.com/warewulf/warewulf/internal/pkg/wwlog"
|
|
)
|
|
|
|
func TryCreatePath(t *testing.T, elem ...string) {
|
|
err := os.MkdirAll(filepath.Join(elem...), os.ModePerm)
|
|
if err != nil {
|
|
t.Errorf("Failed creating dir: %v", err)
|
|
t.FailNow()
|
|
}
|
|
}
|
|
|
|
func Test_FindFiles(t *testing.T) {
|
|
var tests = map[string]struct {
|
|
createFiles []string
|
|
findFiles []string
|
|
}{
|
|
"no files": {
|
|
createFiles: []string{},
|
|
findFiles: nil,
|
|
},
|
|
"single file": {
|
|
createFiles: []string{"testfile"},
|
|
findFiles: []string{"testfile"},
|
|
},
|
|
"nested file": {
|
|
createFiles: []string{"testdir/testfile"},
|
|
findFiles: []string{"testdir/", "testdir/testfile"},
|
|
},
|
|
}
|
|
|
|
for name, tt := range tests {
|
|
t.Run(name, func(t *testing.T) {
|
|
env := testenv.New(t)
|
|
defer env.RemoveAll(t)
|
|
env.MkdirAll(t, "/test")
|
|
for _, file_ := range tt.createFiles {
|
|
env.CreateFile(t, filepath.Join("/test", file_))
|
|
}
|
|
|
|
files := FindFiles(env.GetPath("/test"))
|
|
assert.Equal(t, tt.findFiles, files)
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_FindFilterFiles(t *testing.T) {
|
|
wwlog.SetLogLevel(wwlog.DEBUG)
|
|
dir, err := os.MkdirTemp(os.TempDir(), "warewulf-test")
|
|
if err != nil {
|
|
t.Errorf("Failed creating tmpdir: %v", err)
|
|
t.FailNow()
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
TryCreatePath(t, dir, "boot")
|
|
TryCreatePath(t, dir, "usr", "local")
|
|
TryCreatePath(t, dir, "usr", "bin")
|
|
TryCreatePath(t, dir, "usr", "usr", "local")
|
|
TryCreatePath(t, dir, "bin")
|
|
TryCreatePath(t, dir, "lib")
|
|
|
|
assert.NoError(t, os.Symlink("/path/to/target", filepath.Join(dir, "symlink")))
|
|
|
|
files, err := FindFilterFiles(dir, []string{"boot", "usr", "bin", "symlink"}, []string{"/b*/", "/usr/local"}, true)
|
|
|
|
if err != nil {
|
|
t.Errorf("FindFilerFiles failed: %v", err)
|
|
t.FailNow()
|
|
}
|
|
|
|
expected := []string{"usr", "usr/bin", "usr/usr", "usr/usr/local", "symlink"}
|
|
sort.Strings(expected)
|
|
sort.Strings(files)
|
|
if !reflect.DeepEqual(files, expected) {
|
|
t.Errorf("expected %v, got %v", expected, files)
|
|
t.FailNow()
|
|
}
|
|
}
|