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
|
package main_test
import (
"bytes"
"fmt"
"testing"
. "github.com/pseudomuto/protoc-gen-doc/cmd/protoc-gen-doc"
"github.com/stretchr/testify/require"
)
func TestCode(t *testing.T) {
f := ParseFlags(nil, []string{"app", "-help"})
require.Zero(t, f.Code())
f = ParseFlags(nil, []string{"app", "-whoawhoawhoa"})
require.Equal(t, 1, f.Code())
}
func TestHasMatch(t *testing.T) {
f := ParseFlags(nil, []string{"app", "-help"})
require.True(t, f.HasMatch())
f = ParseFlags(nil, []string{"app", "-version"})
require.True(t, f.HasMatch())
f = ParseFlags(nil, []string{"app", "-watthewhat"})
require.True(t, f.HasMatch())
f = ParseFlags(nil, []string{"app"})
require.False(t, f.HasMatch())
}
func TestShowHelp(t *testing.T) {
f := ParseFlags(nil, []string{"app", "-help"})
require.True(t, f.ShowHelp())
f = ParseFlags(nil, []string{"app", "-version"})
require.False(t, f.ShowHelp())
}
func TestShowVersion(t *testing.T) {
f := ParseFlags(nil, []string{"app", "-version"})
require.True(t, f.ShowVersion())
f = ParseFlags(nil, []string{"app", "-help"})
require.False(t, f.ShowVersion())
}
func TestPrintHelp(t *testing.T) {
buf := new(bytes.Buffer)
f := ParseFlags(buf, []string{"app"})
f.PrintHelp()
result := buf.String()
require.Contains(t, result, "Usage of app:\n\n")
require.Contains(t, result, "FLAGS\n")
require.Contains(t, result, "-help")
require.Contains(t, result, "-version")
}
func TestPrintVersion(t *testing.T) {
buf := new(bytes.Buffer)
f := ParseFlags(buf, []string{"app"})
f.PrintVersion()
// Normally, I'm not a fan of using constants like this in tests. However, having this break everytime the version
// changes is kinda poop, so I've used VERSION here.
require.Equal(t, fmt.Sprintf("app version %s\n", Version()), buf.String())
}
func TestInvalidFlags(t *testing.T) {
buf := new(bytes.Buffer)
f := ParseFlags(buf, []string{"app", "-wat"})
require.Contains(t, buf.String(), "flag provided but not defined: -wat\n")
require.True(t, f.HasMatch())
require.True(t, f.ShowHelp())
}
|
