-
Notifications
You must be signed in to change notification settings - Fork 0
/
io_test.go
82 lines (78 loc) · 1.7 KB
/
io_test.go
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
package main
import (
"bytes"
"errors"
"flag"
"strings"
"testing"
)
func TestParseArgs(t *testing.T) {
tests := []struct {
args []string
conf config
output string
err error
}{
{
args: []string{"-v"},
conf: config{version: true, write: false, paths: []string{}},
},
{
args: []string{"-h"},
output: usageText,
err: flag.ErrHelp,
},
{
args: []string{"-help"},
output: usageText,
err: flag.ErrHelp,
},
{
args: []string{"--help"},
output: usageText,
err: flag.ErrHelp,
},
{
// That’s the test where all is nil.
},
{
args: []string{"-w"},
conf: config{version: false, write: true, paths: []string{}},
},
{
args: []string{"path1", "path2"},
conf: config{version: false, write: false, paths: []string{"path1", "path2"}},
},
}
for _, tt := range tests {
test := tt
t.Run(strings.Join(test.args, " "), func(t *testing.T) {
t.Parallel()
conf, output, err := parseArgs(test.args)
if !errors.Is(err, test.err) {
t.Errorf("got: %v, want: %v", err, test.err)
if test.output == "" {
return
}
}
if test.output != "" {
if output != test.output {
t.Errorf("got: %s, want: %s", output, test.output)
}
return
}
wantConf := test.conf
if (*conf).version != wantConf.version {
t.Errorf("got: %t, want: %t", conf.version, wantConf.version)
}
if conf.write != wantConf.write {
t.Errorf("got: %t, want: %t", conf.write, wantConf.write)
}
bpaths := []byte(strings.Join(conf.paths, ""))
wantConfBPaths := []byte(strings.Join(wantConf.paths, ""))
if !bytes.Equal(bpaths, wantConfBPaths) {
t.Errorf("got: %v, want: %v", bpaths, wantConfBPaths)
}
})
}
}