cobra/args_test.go

324 lines
8.0 KiB
Go
Raw Normal View History

2017-10-31 18:57:17 +00:00
package cobra
import (
"fmt"
2017-10-31 18:57:17 +00:00
"strings"
"testing"
)
func getCommand(args PositionalArgs, withValid bool) *Command {
c := &Command{
Use: "c",
Args: args,
Run: emptyRun,
}
if withValid {
c.ValidArgs = []string{"one", "two", "three"}
}
return c
}
2017-10-31 18:57:17 +00:00
func expectSuccess(output string, err error, t *testing.T) {
2017-10-31 18:57:17 +00:00
if output != "" {
t.Errorf("Unexpected output: %v", output)
2017-10-31 18:57:17 +00:00
}
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
}
func validWithInvalidArgs(err error, t *testing.T) {
2017-10-31 18:57:17 +00:00
if err == nil {
t.Fatal("Expected an error")
}
got := err.Error()
expected := `invalid argument "a" for "c"`
if got != expected {
t.Errorf("Expected: %q, got: %q", expected, got)
}
}
2017-10-31 18:57:17 +00:00
func noArgsWithArgs(err error, t *testing.T) {
if err == nil {
t.Fatal("Expected an error")
}
2017-10-31 18:57:17 +00:00
got := err.Error()
expected := `unknown command "illegal" for "c"`
if got != expected {
t.Errorf("Expected: %q, got: %q", expected, got)
}
}
func minimumNArgsWithLessArgs(err error, t *testing.T) {
if err == nil {
t.Fatal("Expected an error")
}
got := err.Error()
expected := "requires at least 2 arg(s), only received 1"
if got != expected {
t.Fatalf("Expected %q, got %q", expected, got)
2017-10-31 18:57:17 +00:00
}
}
2017-10-31 18:57:17 +00:00
func maximumNArgsWithMoreArgs(err error, t *testing.T) {
if err == nil {
t.Fatal("Expected an error")
2017-10-31 18:57:17 +00:00
}
got := err.Error()
expected := "accepts at most 2 arg(s), received 3"
if got != expected {
t.Fatalf("Expected %q, got %q", expected, got)
2017-10-31 18:57:17 +00:00
}
}
func exactArgsWithInvalidCount(err error, t *testing.T) {
if err == nil {
t.Fatal("Expected an error")
}
got := err.Error()
expected := "accepts 2 arg(s), received 3"
if got != expected {
t.Fatalf("Expected %q, got %q", expected, got)
2017-10-31 18:57:17 +00:00
}
}
2017-10-31 18:57:17 +00:00
func rangeArgsWithInvalidCount(err error, t *testing.T) {
2017-10-31 18:57:17 +00:00
if err == nil {
t.Fatal("Expected an error")
}
got := err.Error()
expected := "accepts between 2 and 4 arg(s), received 1"
2017-10-31 18:57:17 +00:00
if got != expected {
t.Fatalf("Expected %q, got %q", expected, got)
2017-10-31 18:57:17 +00:00
}
}
func TestNoArgs(t *testing.T) {
c := getCommand(NoArgs, false)
output, err := executeCommand(c)
expectSuccess(output, err, t)
}
func TestNoArgsWithArgs(t *testing.T) {
c := getCommand(NoArgs, false)
_, err := executeCommand(c, "illegal")
noArgsWithArgs(err, t)
}
func TestOnlyValidArgs(t *testing.T) {
c := getCommand(OnlyValidArgs, true)
output, err := executeCommand(c, "one", "two")
expectSuccess(output, err, t)
}
func TestOnlyValidArgsWithInvalidArgs(t *testing.T) {
c := getCommand(OnlyValidArgs, true)
_, err := executeCommand(c, "a")
validWithInvalidArgs(err, t)
}
2017-10-31 18:57:17 +00:00
func TestArbitraryArgs(t *testing.T) {
c := getCommand(ArbitraryArgs, false)
2017-10-31 18:57:17 +00:00
output, err := executeCommand(c, "a", "b")
expectSuccess(output, err, t)
2017-10-31 18:57:17 +00:00
}
func TestMinimumNArgs(t *testing.T) {
c := getCommand(MinimumNArgs(2), false)
2017-10-31 18:57:17 +00:00
output, err := executeCommand(c, "a", "b", "c")
expectSuccess(output, err, t)
2017-10-31 18:57:17 +00:00
}
func TestMinimumNArgsWithLessArgs(t *testing.T) {
c := getCommand(MinimumNArgs(2), false)
2017-10-31 18:57:17 +00:00
_, err := executeCommand(c, "a")
minimumNArgsWithLessArgs(err, t)
2017-10-31 18:57:17 +00:00
}
func TestMaximumNArgs(t *testing.T) {
c := getCommand(MaximumNArgs(3), false)
2017-10-31 18:57:17 +00:00
output, err := executeCommand(c, "a", "b")
expectSuccess(output, err, t)
2017-10-31 18:57:17 +00:00
}
func TestMaximumNArgsWithMoreArgs(t *testing.T) {
c := getCommand(MaximumNArgs(2), false)
2017-10-31 18:57:17 +00:00
_, err := executeCommand(c, "a", "b", "c")
maximumNArgsWithMoreArgs(err, t)
2017-10-31 18:57:17 +00:00
}
func TestExactArgs(t *testing.T) {
c := getCommand(ExactArgs(3), false)
2017-10-31 18:57:17 +00:00
output, err := executeCommand(c, "a", "b", "c")
expectSuccess(output, err, t)
2017-10-31 18:57:17 +00:00
}
func TestExactArgsWithInvalidCount(t *testing.T) {
c := getCommand(ExactArgs(2), false)
2017-10-31 18:57:17 +00:00
_, err := executeCommand(c, "a", "b", "c")
exactArgsWithInvalidCount(err, t)
2017-10-31 18:57:17 +00:00
}
2018-10-21 14:01:21 +00:00
func TestExactValidArgs(t *testing.T) {
c := getCommand(ExactValidArgs(3), true)
output, err := executeCommand(c, "three", "one", "two")
expectSuccess(output, err, t)
2018-10-21 14:01:21 +00:00
}
func TestExactValidArgsWithInvalidCount(t *testing.T) {
c := getCommand(ExactValidArgs(2), false)
_, err := executeCommand(c, "three", "one", "two")
exactArgsWithInvalidCount(err, t)
2018-10-21 14:01:21 +00:00
}
func TestExactValidArgsWithInvalidArgs(t *testing.T) {
c := getCommand(ExactValidArgs(3), true)
_, err := executeCommand(c, "three", "a", "two")
validWithInvalidArgs(err, t)
2018-10-21 14:01:21 +00:00
}
2017-10-31 18:57:17 +00:00
func TestRangeArgs(t *testing.T) {
c := getCommand(RangeArgs(2, 4), false)
2017-10-31 18:57:17 +00:00
output, err := executeCommand(c, "a", "b", "c")
expectSuccess(output, err, t)
2017-10-31 18:57:17 +00:00
}
func TestRangeArgsWithInvalidCount(t *testing.T) {
c := getCommand(RangeArgs(2, 4), false)
2017-10-31 18:57:17 +00:00
_, err := executeCommand(c, "a")
rangeArgsWithInvalidCount(err, t)
2017-10-31 18:57:17 +00:00
}
func TestRootTakesNoArgs(t *testing.T) {
rootCmd := &Command{Use: "root", Run: emptyRun}
childCmd := &Command{Use: "child", Run: emptyRun}
rootCmd.AddCommand(childCmd)
_, err := executeCommand(rootCmd, "illegal", "args")
if err == nil {
t.Fatal("Expected an error")
}
got := err.Error()
expected := `unknown command "illegal" for "root"`
if !strings.Contains(got, expected) {
t.Errorf("expected %q, got %q", expected, got)
}
}
func TestRootTakesArgs(t *testing.T) {
rootCmd := &Command{Use: "root", Args: ArbitraryArgs, Run: emptyRun}
childCmd := &Command{Use: "child", Run: emptyRun}
rootCmd.AddCommand(childCmd)
_, err := executeCommand(rootCmd, "legal", "args")
if err != nil {
t.Errorf("Unexpected error: %v", err)
}
}
func TestChildTakesNoArgs(t *testing.T) {
rootCmd := &Command{Use: "root", Run: emptyRun}
childCmd := &Command{Use: "child", Args: NoArgs, Run: emptyRun}
rootCmd.AddCommand(childCmd)
_, err := executeCommand(rootCmd, "child", "illegal", "args")
if err == nil {
t.Fatal("Expected an error")
}
got := err.Error()
expected := `unknown command "illegal" for "root child"`
if !strings.Contains(got, expected) {
t.Errorf("expected %q, got %q", expected, got)
}
}
func TestChildTakesArgs(t *testing.T) {
rootCmd := &Command{Use: "root", Run: emptyRun}
childCmd := &Command{Use: "child", Args: ArbitraryArgs, Run: emptyRun}
rootCmd.AddCommand(childCmd)
_, err := executeCommand(rootCmd, "child", "legal", "args")
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
}
func TestMatchAll(t *testing.T) {
// Somewhat contrived example check that ensures there are exactly 3
// arguments, and each argument is exactly 2 bytes long.
pargs := MatchAll(
ExactArgs(3),
func(cmd *Command, args []string) error {
for _, arg := range args {
if len([]byte(arg)) != 2 {
return fmt.Errorf("expected to be exactly 2 bytes long")
}
}
return nil
},
)
testCases := map[string]struct {
args []string
fail bool
}{
"happy path": {
[]string{"aa", "bb", "cc"},
false,
},
"incorrect number of args": {
[]string{"aa", "bb", "cc", "dd"},
true,
},
"incorrect number of bytes in one arg": {
[]string{"aa", "bb", "abc"},
true,
},
}
rootCmd := &Command{Use: "root", Args: pargs, Run: emptyRun}
for name, tc := range testCases {
t.Run(name, func(t *testing.T) {
_, err := executeCommand(rootCmd, tc.args...)
if err != nil && !tc.fail {
t.Errorf("unexpected: %v\n", err)
}
if err == nil && tc.fail {
t.Errorf("expected error")
}
})
}
}
// This test make sure we keep backwards-compatibility with respect
// to the legacyArgs() function.
// It makes sure the root command accepts arguments if it does not have
// sub-commands.
func TestLegacyArgsRootAcceptsArgs(t *testing.T) {
rootCmd := &Command{Use: "root", Args: nil, Run: emptyRun}
_, err := executeCommand(rootCmd, "somearg")
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
}
// This test make sure we keep backwards-compatibility with respect
// to the legacyArgs() function.
// It makes sure a sub-command accepts arguments and further sub-commands
func TestLegacyArgsSubcmdAcceptsArgs(t *testing.T) {
rootCmd := &Command{Use: "root", Args: nil, Run: emptyRun}
childCmd := &Command{Use: "child", Args: nil, Run: emptyRun}
grandchildCmd := &Command{Use: "grandchild", Args: nil, Run: emptyRun}
rootCmd.AddCommand(childCmd)
childCmd.AddCommand(grandchildCmd)
_, err := executeCommand(rootCmd, "child", "somearg")
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
}