Skip to content

Commit

Permalink
more test coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
elri committed Nov 22, 2022
1 parent 3daffbf commit 7e1ee74
Show file tree
Hide file tree
Showing 2 changed files with 33 additions and 10 deletions.
7 changes: 7 additions & 0 deletions config_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,13 @@ func Test_SetEnvsToParse(t *testing.T) {
func Test_ConfigDefault(t *testing.T) {
var err error

// don't care if default file doesn't exist
err = SetDefaultFile("doesntexist.yml")
assert.NotNil(t, err)

err = SetUpConfiguration(new(TestConfig))
assert.Nil(t, err)

// set default config file
err = SetDefaultFile(DEFAULT_TEST_CONFIG)
assert.Nil(t, err)
Expand Down
36 changes: 26 additions & 10 deletions flags_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -183,6 +183,7 @@ func Test_ParseBoolFLag(t *testing.T) {
}

func Test_ParseFlags(t *testing.T) {
var err error
flagSet := testInit()

_ = flagSet.Bool(b, false, "usage")
Expand All @@ -193,15 +194,17 @@ func Test_ParseFlags(t *testing.T) {
_ = flagSet.Uint(ui, 20, "usage")
_ = flagSet.Duration(d, 5*time.Second, "usage")

flags := map[string]*flag.Flag{
b: LookupFlag(b),
f64: LookupFlag(f64),
i: LookupFlag(i),
i64: LookupFlag(i64),
str: LookupFlag(str),
ui: LookupFlag(ui),
d: LookupFlag(d)}
// only given flag is default flag '-write def conf'
SetFlagSetArgs([]string{"-write-def-conf"})
err = ParseFlags()
assert.Nil(t, err)

//only given flag is default flag '-print-conf'
SetFlagSetArgs([]string{"-print-conf"})
err = ParseFlags()
assert.Nil(t, err)

//Parsed
args := []string{
"-f64", "3.1415",
"-i", "342",
Expand All @@ -210,10 +213,18 @@ func Test_ParseFlags(t *testing.T) {
}
SetFlagSetArgs(args)

//parse
err := ParseFlags()
err = ParseFlags()
assert.Nil(t, err)

flags := map[string]*flag.Flag{
b: LookupFlag(b),
f64: LookupFlag(f64),
i: LookupFlag(i),
i64: LookupFlag(i64),
str: LookupFlag(str),
ui: LookupFlag(ui),
d: LookupFlag(d)}

for _, arg := range args {
argSplit := strings.Split(arg, "-")
if len(argSplit) > 1 {
Expand Down Expand Up @@ -359,6 +370,11 @@ func Test_afterParse(t *testing.T) {
f.Value = &FlagValue{Value: f.Value}
})

// no flags added if parse hasn't happened
// continueonerror is default so nothing happens
flagSet.VisitAll(afterParse())
assert.Equal(t, flags, make(map[string]interface{}))

err := flagSet.Parse(args)
assert.Nil(t, err)

Expand Down

0 comments on commit 7e1ee74

Please sign in to comment.