1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
package complete
import (
"sort"
"strings"
"testing"
)
func TestPredicate(t *testing.T) {
t.Parallel()
initTests()
tests := []struct {
name string
p Predictor
arg string
want []string
}{
{
name: "set",
p: PredictSet("a", "b", "c"),
want: []string{"a", "b", "c"},
},
{
name: "set with does",
p: PredictSet("./..", "./x"),
arg: "./.",
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{"./a.txt", "./b.txt", "./c.txt", "./.dot.txt"},
},
{
name: "files/txt",
p: PredictFiles("*.txt"),
arg: "./dir/",
want: []string{},
},
{
name: "files/x",
p: PredictFiles("x"),
arg: "./dir/",
want: []string{"./dir/x"},
},
{
name: "files/*",
p: PredictFiles("x*"),
arg: "./dir/",
want: []string{"./dir/x"},
},
{
name: "files/md",
p: PredictFiles("*.md"),
want: []string{"./readme.md"},
},
{
name: "dirs",
p: PredictDirs("*"),
arg: "./dir/",
want: []string{"./dir/"},
},
{
name: "dirs and files",
p: PredictFilesOrDirs("*"),
arg: "./dir",
want: []string{"./dir/", "./dir/x"},
},
{
name: "dirs",
p: PredictDirs("*"),
want: []string{"./", "./dir/"},
},
{
name: "subdir",
p: PredictFiles("*"),
arg: "./dir/",
want: []string{"./dir/x"},
},
}
for _, tt := range tests {
t.Run(tt.name+"?arg='"+tt.arg+"'", func(t *testing.T) {
matches := tt.p.Predict(newArgs(strings.Split(tt.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)
}
})
}
}
|