Add test for FindFilterFiles
Signed-off-by: Tobias Ribizel <mail@ribizel.de>
This commit is contained in:
50
internal/pkg/util/util_test.go
Normal file
50
internal/pkg/util/util_test.go
Normal file
@@ -0,0 +1,50 @@
|
||||
package util
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
"sort"
|
||||
"testing"
|
||||
|
||||
"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_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")
|
||||
|
||||
files, err := FindFilterFiles(dir, []string{"boot", "usr", "bin"}, []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"}
|
||||
sort.Strings(expected)
|
||||
sort.Strings(files)
|
||||
if !reflect.DeepEqual(files, expected) {
|
||||
t.Errorf("expected %v, got %v", expected, files)
|
||||
t.FailNow()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user