You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
1.8 KiB
79 lines
1.8 KiB
package cli
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func Test_SettingFlags(t *testing.T) {
|
|
msg := ""
|
|
app := NewApp()
|
|
app.Flags = []Flag{
|
|
StringFlag{"foo", "default", "a string flag"},
|
|
IntFlag{"bar", 42, "an int flag"},
|
|
BoolFlag{"bat", "a bool flag"},
|
|
}
|
|
app.Action = func(c *Context) {
|
|
expect(t, c.String("foo"), "hello world")
|
|
expect(t, c.Int("bar"), 245)
|
|
expect(t, c.Bool("bat"), true)
|
|
msg = "foobar"
|
|
}
|
|
app.Run([]string{"command", "--foo", "hello world", "--bar", "245", "--bat"})
|
|
expect(t, msg, "foobar")
|
|
}
|
|
|
|
func Test_FlagDefaults(t *testing.T) {
|
|
msg := ""
|
|
app := NewApp()
|
|
app.Flags = []Flag{
|
|
StringFlag{"foo", "default", "a string flag"},
|
|
IntFlag{"bar", 42, "an int flag"},
|
|
BoolFlag{"bat", "a bool flag"},
|
|
}
|
|
app.Action = func(c *Context) {
|
|
expect(t, c.String("foo"), "default")
|
|
expect(t, c.Int("bar"), 42)
|
|
expect(t, c.Bool("bat"), false)
|
|
msg = "foobar"
|
|
}
|
|
app.Run([]string{"command"})
|
|
expect(t, msg, "foobar")
|
|
}
|
|
|
|
func TestCommands(t *testing.T) {
|
|
app := NewApp()
|
|
app.Flags = []Flag{
|
|
StringFlag{"name", "jeremy", "a name to print"},
|
|
}
|
|
app.Commands = []Command{
|
|
{
|
|
Name: "print",
|
|
Flags: []Flag{
|
|
IntFlag{"age", 50, "the age of the person"},
|
|
},
|
|
Action: func(c *Context) {
|
|
expect(t, c.GlobalString("name"), "jordie")
|
|
expect(t, c.Int("age"), 21)
|
|
},
|
|
},
|
|
}
|
|
app.Action = func(c *Context) {
|
|
t.Error("default action should not be called")
|
|
}
|
|
app.Run([]string{"command", "--name", "jordie", "print", "--age", "21"})
|
|
}
|
|
|
|
/* Test Helpers */
|
|
func expect(t *testing.T, a interface{}, b interface{}) {
|
|
if a != b {
|
|
t.Errorf("Expected %v (type %v) - Got %v (type %v)", b, reflect.TypeOf(b), a, reflect.TypeOf(a))
|
|
}
|
|
}
|
|
|
|
func refute(t *testing.T, a interface{}, b interface{}) {
|
|
if a == b {
|
|
t.Errorf("Did not expect %v (type %v) - Got %v (type %v)", b, reflect.TypeOf(b), a, reflect.TypeOf(a))
|
|
}
|
|
}
|