mirror of
https://github.com/spf13/cobra
synced 2025-05-05 12:57:22 +00:00
args_test: add additional helper functions
* noArgsWithArgs * validWithInvalidArgs * minimumNArgsWithLessArgs * maximumNArgsWithMoreArgs * exactArgsWithInvalidCount * rangeArgsWithInvalidCount
This commit is contained in:
parent
989590274f
commit
4120f6e4dc
1 changed files with 78 additions and 85 deletions
163
args_test.go
163
args_test.go
|
@ -26,6 +26,72 @@ func expectSuccess(output string, err error, t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func validWithInvalidArgs(err error, t *testing.T) {
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func noArgsWithArgs(err error, t *testing.T) {
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("Expected an error")
|
||||||
|
}
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func maximumNArgsWithMoreArgs(err error, t *testing.T) {
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("Expected an error")
|
||||||
|
}
|
||||||
|
got := err.Error()
|
||||||
|
expected := "accepts at most 2 arg(s), received 3"
|
||||||
|
if got != expected {
|
||||||
|
t.Fatalf("Expected %q, got %q", expected, got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func rangeArgsWithInvalidCount(err error, t *testing.T) {
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("Expected an error")
|
||||||
|
}
|
||||||
|
got := err.Error()
|
||||||
|
expected := "accepts between 2 and 4 arg(s), received 1"
|
||||||
|
if got != expected {
|
||||||
|
t.Fatalf("Expected %q, got %q", expected, got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestNoArgs(t *testing.T) {
|
func TestNoArgs(t *testing.T) {
|
||||||
c := getCommand(NoArgs, false)
|
c := getCommand(NoArgs, false)
|
||||||
output, err := executeCommand(c)
|
output, err := executeCommand(c)
|
||||||
|
@ -34,39 +100,20 @@ func TestNoArgs(t *testing.T) {
|
||||||
|
|
||||||
func TestNoArgsWithArgs(t *testing.T) {
|
func TestNoArgsWithArgs(t *testing.T) {
|
||||||
c := getCommand(NoArgs, false)
|
c := getCommand(NoArgs, false)
|
||||||
|
|
||||||
_, err := executeCommand(c, "illegal")
|
_, err := executeCommand(c, "illegal")
|
||||||
if err == nil {
|
noArgsWithArgs(err, t)
|
||||||
t.Fatal("Expected an error")
|
|
||||||
}
|
|
||||||
|
|
||||||
got := err.Error()
|
|
||||||
expected := `unknown command "illegal" for "c"`
|
|
||||||
if got != expected {
|
|
||||||
t.Errorf("Expected: %q, got: %q", expected, got)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestOnlyValidArgs(t *testing.T) {
|
func TestOnlyValidArgs(t *testing.T) {
|
||||||
c := getCommand(OnlyValidArgs, true)
|
c := getCommand(OnlyValidArgs, true)
|
||||||
|
|
||||||
output, err := executeCommand(c, "one", "two")
|
output, err := executeCommand(c, "one", "two")
|
||||||
expectSuccess(output, err, t)
|
expectSuccess(output, err, t)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestOnlyValidArgsWithInvalidArgs(t *testing.T) {
|
func TestOnlyValidArgsWithInvalidArgs(t *testing.T) {
|
||||||
c := getCommand(OnlyValidArgs, true)
|
c := getCommand(OnlyValidArgs, true)
|
||||||
|
|
||||||
_, err := executeCommand(c, "a")
|
_, err := executeCommand(c, "a")
|
||||||
if err == nil {
|
validWithInvalidArgs(err, t)
|
||||||
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)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestArbitraryArgs(t *testing.T) {
|
func TestArbitraryArgs(t *testing.T) {
|
||||||
|
@ -84,16 +131,7 @@ func TestMinimumNArgs(t *testing.T) {
|
||||||
func TestMinimumNArgsWithLessArgs(t *testing.T) {
|
func TestMinimumNArgsWithLessArgs(t *testing.T) {
|
||||||
c := getCommand(MinimumNArgs(2), false)
|
c := getCommand(MinimumNArgs(2), false)
|
||||||
_, err := executeCommand(c, "a")
|
_, err := executeCommand(c, "a")
|
||||||
|
minimumNArgsWithLessArgs(err, 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)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestMaximumNArgs(t *testing.T) {
|
func TestMaximumNArgs(t *testing.T) {
|
||||||
|
@ -105,16 +143,7 @@ func TestMaximumNArgs(t *testing.T) {
|
||||||
func TestMaximumNArgsWithMoreArgs(t *testing.T) {
|
func TestMaximumNArgsWithMoreArgs(t *testing.T) {
|
||||||
c := getCommand(MaximumNArgs(2), false)
|
c := getCommand(MaximumNArgs(2), false)
|
||||||
_, err := executeCommand(c, "a", "b", "c")
|
_, err := executeCommand(c, "a", "b", "c")
|
||||||
|
maximumNArgsWithMoreArgs(err, t)
|
||||||
if err == nil {
|
|
||||||
t.Fatal("Expected an error")
|
|
||||||
}
|
|
||||||
|
|
||||||
got := err.Error()
|
|
||||||
expected := "accepts at most 2 arg(s), received 3"
|
|
||||||
if got != expected {
|
|
||||||
t.Fatalf("Expected %q, got %q", expected, got)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestExactArgs(t *testing.T) {
|
func TestExactArgs(t *testing.T) {
|
||||||
|
@ -126,52 +155,25 @@ func TestExactArgs(t *testing.T) {
|
||||||
func TestExactArgsWithInvalidCount(t *testing.T) {
|
func TestExactArgsWithInvalidCount(t *testing.T) {
|
||||||
c := getCommand(ExactArgs(2), false)
|
c := getCommand(ExactArgs(2), false)
|
||||||
_, err := executeCommand(c, "a", "b", "c")
|
_, err := executeCommand(c, "a", "b", "c")
|
||||||
|
exactArgsWithInvalidCount(err, 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)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestExactValidArgs(t *testing.T) {
|
func TestExactValidArgs(t *testing.T) {
|
||||||
c := getCommand(ExactValidArgs(3), true)
|
c := getCommand(ExactValidArgs(3), true)
|
||||||
output, err := executeCommand(c, "two", "three", "one")
|
output, err := executeCommand(c, "three", "one", "two")
|
||||||
expectSuccess(output, err, t)
|
expectSuccess(output, err, t)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestExactValidArgsWithInvalidCount(t *testing.T) {
|
func TestExactValidArgsWithInvalidCount(t *testing.T) {
|
||||||
c := getCommand(ExactValidArgs(2), false)
|
c := getCommand(ExactValidArgs(2), false)
|
||||||
_, err := executeCommand(c, "two", "three", "one")
|
_, err := executeCommand(c, "three", "one", "two")
|
||||||
|
exactArgsWithInvalidCount(err, 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)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestExactValidArgsWithInvalidArgs(t *testing.T) {
|
func TestExactValidArgsWithInvalidArgs(t *testing.T) {
|
||||||
c := getCommand(ExactValidArgs(1), true)
|
c := getCommand(ExactValidArgs(3), true)
|
||||||
|
_, err := executeCommand(c, "three", "a", "two")
|
||||||
_, err := executeCommand(c, "a")
|
validWithInvalidArgs(err, t)
|
||||||
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)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestRangeArgs(t *testing.T) {
|
func TestRangeArgs(t *testing.T) {
|
||||||
|
@ -183,16 +185,7 @@ func TestRangeArgs(t *testing.T) {
|
||||||
func TestRangeArgsWithInvalidCount(t *testing.T) {
|
func TestRangeArgsWithInvalidCount(t *testing.T) {
|
||||||
c := getCommand(RangeArgs(2, 4), false)
|
c := getCommand(RangeArgs(2, 4), false)
|
||||||
_, err := executeCommand(c, "a")
|
_, err := executeCommand(c, "a")
|
||||||
|
rangeArgsWithInvalidCount(err, t)
|
||||||
if err == nil {
|
|
||||||
t.Fatal("Expected an error")
|
|
||||||
}
|
|
||||||
|
|
||||||
got := err.Error()
|
|
||||||
expected := "accepts between 2 and 4 arg(s), received 1"
|
|
||||||
if got != expected {
|
|
||||||
t.Fatalf("Expected %q, got %q", expected, got)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestRootTakesNoArgs(t *testing.T) {
|
func TestRootTakesNoArgs(t *testing.T) {
|
||||||
|
|
Loading…
Add table
Reference in a new issue