134 lines
2.9 KiB
Go
134 lines
2.9 KiB
Go
package complete
|
|
|
|
import (
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestPredicate(t *testing.T) {
|
|
t.Parallel()
|
|
initTests()
|
|
|
|
tests := []struct {
|
|
name string
|
|
p Predictor
|
|
argList []string
|
|
want []string
|
|
}{
|
|
{
|
|
name: "set",
|
|
p: PredictSet("a", "b", "c"),
|
|
want: []string{"a", "b", "c"},
|
|
},
|
|
{
|
|
name: "set with does",
|
|
p: PredictSet("./..", "./x"),
|
|
argList: []string{"./.", "./.."},
|
|
want: []string{"./.."},
|
|
},
|
|
{
|
|
name: "set/empty",
|
|
p: PredictSet(),
|
|
want: []string{},
|
|
},
|
|
{
|
|
name: "anything",
|
|
p: PredictAnything,
|
|
want: []string{},
|
|
},
|
|
{
|
|
name: "or: word with nil",
|
|
p: PredictOr(PredictSet("a"), nil),
|
|
want: []string{"a"},
|
|
},
|
|
{
|
|
name: "or: nil with word",
|
|
p: PredictOr(nil, PredictSet("a")),
|
|
want: []string{"a"},
|
|
},
|
|
{
|
|
name: "or: nil with nil",
|
|
p: PredictOr(PredictNothing, PredictNothing),
|
|
want: []string{},
|
|
},
|
|
{
|
|
name: "or: word with word with word",
|
|
p: PredictOr(PredictSet("a"), PredictSet("b"), PredictSet("c")),
|
|
want: []string{"a", "b", "c"},
|
|
},
|
|
{
|
|
name: "files/txt",
|
|
p: PredictFiles("*.txt"),
|
|
want: []string{"./", "./dir/", "./a.txt", "./b.txt", "./c.txt", "./.dot.txt"},
|
|
},
|
|
{
|
|
name: "files/txt",
|
|
p: PredictFiles("*.txt"),
|
|
argList: []string{"./dir/"},
|
|
want: []string{"./dir/"},
|
|
},
|
|
{
|
|
name: "complete files inside dir if it is the only match",
|
|
p: PredictFiles("foo"),
|
|
argList: []string{"./dir/", "./d"},
|
|
want: []string{"./dir/", "./dir/foo"},
|
|
},
|
|
{
|
|
name: "complete files inside dir when argList includes file name",
|
|
p: PredictFiles("*"),
|
|
argList: []string{"./dir/f", "./dir/foo"},
|
|
want: []string{"./dir/foo"},
|
|
},
|
|
{
|
|
name: "files/md",
|
|
p: PredictFiles("*.md"),
|
|
argList: []string{"", ".", "./"},
|
|
want: []string{"./", "./dir/", "./readme.md"},
|
|
},
|
|
{
|
|
name: "dirs",
|
|
p: PredictDirs("*"),
|
|
argList: []string{"./dir/", "./di", "di", "dir", "dir/"},
|
|
want: []string{"./dir/"},
|
|
},
|
|
{
|
|
name: "predict anything in dir",
|
|
p: PredictFiles("*"),
|
|
argList: []string{"./dir", "dir", "./dir/", "./di"},
|
|
want: []string{"./dir/", "./dir/foo", "./dir/bar"},
|
|
},
|
|
{
|
|
name: "root directories",
|
|
p: PredictDirs("*"),
|
|
argList: []string{"", ".", "./"},
|
|
want: []string{"./", "./dir/"},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
|
|
// no args in argList, means an empty argument
|
|
if len(tt.argList) == 0 {
|
|
tt.argList = append(tt.argList, "")
|
|
}
|
|
|
|
for _, arg := range tt.argList {
|
|
t.Run(tt.name+"/arg="+arg, func(t *testing.T) {
|
|
|
|
matches := tt.p.Predict(newArgs(strings.Split(arg, " ")))
|
|
|
|
sort.Strings(matches)
|
|
sort.Strings(tt.want)
|
|
|
|
got := strings.Join(matches, ",")
|
|
want := strings.Join(tt.want, ",")
|
|
|
|
if got != want {
|
|
t.Errorf("failed %s\ngot = %s\nwant: %s", t.Name(), got, want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
}
|