scheibling
240
filepathx/filepath_test.go
Normal file
240
filepathx/filepath_test.go
Normal file
@@ -0,0 +1,240 @@
|
||||
package filepathx
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
|
||||
"git.cloudyne.io/go/hiscaler-gox/slicex"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestDirs(t *testing.T) {
|
||||
root, _ := os.Getwd()
|
||||
testCases := []struct {
|
||||
Number int
|
||||
Path string
|
||||
Option WalkOption
|
||||
Files []string
|
||||
}{
|
||||
{
|
||||
1,
|
||||
"/a/b",
|
||||
WalkOption{},
|
||||
[]string{},
|
||||
},
|
||||
{
|
||||
2,
|
||||
root,
|
||||
WalkOption{
|
||||
CaseSensitive: false,
|
||||
FilterFunc: func(path string) bool {
|
||||
return filepath.Base(path) == "2"
|
||||
},
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"2"},
|
||||
},
|
||||
{
|
||||
3,
|
||||
root,
|
||||
WalkOption{
|
||||
CaseSensitive: false,
|
||||
Only: []string{"2"},
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"2"},
|
||||
},
|
||||
{
|
||||
4,
|
||||
root,
|
||||
WalkOption{
|
||||
CaseSensitive: false,
|
||||
Except: []string{"2"},
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"testdata", "1", "1.1", "1.1", "1.1.1"},
|
||||
},
|
||||
{
|
||||
5,
|
||||
root,
|
||||
WalkOption{
|
||||
CaseSensitive: false,
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"testdata", "1", "1.1", "1.1", "2", "1.1.1"},
|
||||
},
|
||||
{
|
||||
6,
|
||||
root + "/testdata",
|
||||
WalkOption{
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"1", "1.1", "1.1", "2", "1.1.1"},
|
||||
},
|
||||
{
|
||||
7,
|
||||
root + "/testdata",
|
||||
WalkOption{
|
||||
Recursive: false,
|
||||
},
|
||||
[]string{"1", "2"},
|
||||
},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
dirs := Dirs(testCase.Path, testCase.Option)
|
||||
for i, dir := range dirs {
|
||||
dirs[i] = filepath.Base(dir)
|
||||
}
|
||||
if !slicex.StringSliceEqual(dirs, testCase.Files, true, true, true) {
|
||||
t.Errorf("%d: except %v actual %v", testCase.Number, testCase.Files, dirs)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestFiles(t *testing.T) {
|
||||
root, _ := os.Getwd()
|
||||
testCases := []struct {
|
||||
Number int
|
||||
Path string
|
||||
Option WalkOption
|
||||
Files []string
|
||||
}{
|
||||
{
|
||||
1,
|
||||
"/a/b",
|
||||
WalkOption{},
|
||||
[]string{},
|
||||
},
|
||||
{
|
||||
2,
|
||||
root,
|
||||
WalkOption{
|
||||
CaseSensitive: false,
|
||||
FilterFunc: func(path string) bool {
|
||||
return filepath.Base(path) == "2.txt"
|
||||
},
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"2.txt"},
|
||||
},
|
||||
{
|
||||
3,
|
||||
root,
|
||||
WalkOption{
|
||||
CaseSensitive: false,
|
||||
Only: []string{"2.txt"},
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"2.txt"},
|
||||
},
|
||||
{
|
||||
4,
|
||||
root,
|
||||
WalkOption{
|
||||
CaseSensitive: false,
|
||||
Except: []string{"2.txt"},
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"filepath.go", "filepath_test.go", "1.1.txt", "中文_ZH (1).txt", "中文_ZH (1).txt", "中文_ZH (9).txt", "0.txt"},
|
||||
},
|
||||
{
|
||||
5,
|
||||
root,
|
||||
WalkOption{
|
||||
CaseSensitive: false,
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"filepath.go", "filepath_test.go", "1.1.txt", "2.txt", "中文_ZH (1).txt", "中文_ZH (1).txt", "中文_ZH (9).txt", "0.txt"},
|
||||
},
|
||||
{
|
||||
6,
|
||||
root + "/testdata",
|
||||
WalkOption{
|
||||
Recursive: true,
|
||||
},
|
||||
[]string{"1.1.txt", "2.txt", "中文_ZH (1).txt", "中文_ZH (1).txt", "中文_ZH (9).txt", "0.txt"},
|
||||
},
|
||||
{
|
||||
7,
|
||||
root + "/testdata",
|
||||
WalkOption{
|
||||
Recursive: false,
|
||||
},
|
||||
[]string{"0.txt"},
|
||||
},
|
||||
{
|
||||
8,
|
||||
root + "/testdata/1/1.1/1.1",
|
||||
WalkOption{
|
||||
Recursive: false,
|
||||
},
|
||||
[]string{"中文_ZH (1).txt"},
|
||||
},
|
||||
{
|
||||
9,
|
||||
"./testdata/1/1.1/1.1",
|
||||
WalkOption{
|
||||
Recursive: false,
|
||||
},
|
||||
[]string{"中文_ZH (1).txt"},
|
||||
},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
files := Files(testCase.Path, testCase.Option)
|
||||
for i, file := range files {
|
||||
files[i] = filepath.Base(file)
|
||||
}
|
||||
if !slicex.StringSliceEqual(files, testCase.Files, true, true, true) {
|
||||
t.Errorf("%d: except %v actual %v", testCase.Number, testCase.Files, files)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGenerateDirNames(t *testing.T) {
|
||||
tests := []struct {
|
||||
tag string
|
||||
string string
|
||||
n int
|
||||
level int
|
||||
caseSensitive bool
|
||||
dirs []string
|
||||
}{
|
||||
{"t1", "abc", 0, 1, true, []string{"abc"}},
|
||||
{"t2", "abc", 1, 1, true, []string{"a"}},
|
||||
{"t3", "abc", 1, 2, true, []string{"a", "b"}},
|
||||
{"t4", "abc", 1, 3, true, []string{"a", "b", "c"}},
|
||||
{"t5", "abc", 2, 1, true, []string{"ab"}},
|
||||
{"t6", "abc", 2, 2, true, []string{"ab", "c"}},
|
||||
{"t7", " a b c ", 2, 2, true, []string{"ab", "c"}},
|
||||
{"t7", " a b cdefghijklmn ", 2, 3, true, []string{"ab", "cd", "ef"}},
|
||||
{"t8", " a", 12, 3, true, []string{"a"}},
|
||||
{"t9", " a中文$b", 12, 3, true, []string{"ab"}},
|
||||
}
|
||||
for _, test := range tests {
|
||||
names := GenerateDirNames(test.string, test.n, test.level, test.caseSensitive)
|
||||
assert.Equal(t, test.dirs, names, test.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestExt(t *testing.T) {
|
||||
root, _ := os.Getwd()
|
||||
tests := []struct {
|
||||
tag string
|
||||
path string
|
||||
b []byte
|
||||
ext string
|
||||
}{
|
||||
{"t1", "/a/b", nil, ""},
|
||||
{"t2", "https://golang.org/doc/gopher/fiveyears.jpg", nil, ".jpg"},
|
||||
{"t3", filepath.Join(root, "/testdata/2/2.txt"), nil, ".txt"},
|
||||
{"t4", filepath.Join(root, "/testdata/2/1.jpg"), nil, ".jpg"},
|
||||
{"t5", filepath.Join(root, "/testdata/2/1.pdf"), nil, ".pdf"},
|
||||
{"t6", filepath.Join(root, "/testdata/2/1111.pdf"), nil, ".pdf"},
|
||||
{"t7", filepath.Join(root, "/testdata/1.xlsx"), nil, ".xlsx"},
|
||||
}
|
||||
for _, test := range tests {
|
||||
ext := Ext(test.path, test.b)
|
||||
assert.Equal(t, test.ext, ext, test.tag)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user