JMS #4: Getting some cli tests going. Testing out flags

This commit is contained in:
Jeremy Saenz
2013-07-18 17:29:06 -07:00
parent 04496e4364
commit 3f76896c4f
7 changed files with 64 additions and 38 deletions

View File

@@ -17,7 +17,7 @@ func NewContext(flagSet *flag.FlagSet) *Context {
return &Context{flagSet}
}
func (c *Context) IntFlag(name string) int {
func (c *Context) Int(name string) int {
flag := c.flagSet.Lookup(name)
if flag != nil {
val, err := strconv.Atoi(flag.Value.String())
@@ -30,7 +30,7 @@ func (c *Context) IntFlag(name string) int {
}
}
func (c *Context) BoolFlag(name string) bool {
func (c *Context) Bool(name string) bool {
flag := c.flagSet.Lookup(name)
if flag != nil {
val, err := strconv.ParseBool(flag.Value.String())
@@ -43,7 +43,7 @@ func (c *Context) BoolFlag(name string) bool {
}
}
func (c *Context) StringFlag(name string) string {
func (c *Context) String(name string) string {
flag := c.flagSet.Lookup(name)
if flag != nil {
return flag.Value.String()