John Sirois
705865fa70
Previously StringSlice flag defaults were not respected. These use special handling and that handling did not take account of default values, always over-writing them. Expand the CommandLine interface to take advantage of the underlying codegansta/cli.Context IsSet method and guard default string slice overwrites with an IsSet test. Signed-off-by: John Sirois <john.sirois@gmail.com>
101 lines
1.9 KiB
Go
101 lines
1.9 KiB
Go
package commandstest
|
|
|
|
import (
|
|
"github.com/codegangsta/cli"
|
|
)
|
|
|
|
type FakeFlagger struct {
|
|
Data map[string]interface{}
|
|
}
|
|
|
|
type FakeCommandLine struct {
|
|
LocalFlags, GlobalFlags *FakeFlagger
|
|
HelpShown, VersionShown bool
|
|
CliArgs []string
|
|
}
|
|
|
|
func (ff FakeFlagger) String(key string) string {
|
|
if value, ok := ff.Data[key]; ok {
|
|
return value.(string)
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (ff FakeFlagger) StringSlice(key string) []string {
|
|
if value, ok := ff.Data[key]; ok {
|
|
return value.([]string)
|
|
}
|
|
return []string{}
|
|
}
|
|
|
|
func (ff FakeFlagger) Int(key string) int {
|
|
if value, ok := ff.Data[key]; ok {
|
|
return value.(int)
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (ff FakeFlagger) Bool(key string) bool {
|
|
if value, ok := ff.Data[key]; ok {
|
|
return value.(bool)
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) IsSet(key string) bool {
|
|
_, ok := fcli.LocalFlags.Data[key]
|
|
return ok
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) String(key string) string {
|
|
return fcli.LocalFlags.String(key)
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) StringSlice(key string) []string {
|
|
return fcli.LocalFlags.StringSlice(key)
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) Int(key string) int {
|
|
return fcli.LocalFlags.Int(key)
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) Bool(key string) bool {
|
|
if fcli.LocalFlags == nil {
|
|
return false
|
|
}
|
|
return fcli.LocalFlags.Bool(key)
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) GlobalString(key string) string {
|
|
return fcli.GlobalFlags.String(key)
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) Generic(name string) interface{} {
|
|
return fcli.LocalFlags.Data[name]
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) FlagNames() []string {
|
|
flagNames := []string{}
|
|
for key := range fcli.LocalFlags.Data {
|
|
flagNames = append(flagNames, key)
|
|
}
|
|
|
|
return flagNames
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) ShowHelp() {
|
|
fcli.HelpShown = true
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) Application() *cli.App {
|
|
return cli.NewApp()
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) Args() cli.Args {
|
|
return fcli.CliArgs
|
|
}
|
|
|
|
func (fcli *FakeCommandLine) ShowVersion() {
|
|
fcli.VersionShown = true
|
|
}
|