add expectError helper function to args_test

Author: umarcor

args_test.go

@@ -26,74 +26,31 @@ func expectSuccess(output string, err error, t *testing.T) {
 	}
 }
 
-func validWithInvalidArgs(err error, t *testing.T) {
+func expectError(err error, t *testing.T, ex string) {
 	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)
+	switch ex {
+	case "no":
+		expected = `unknown command "one" for "c"`
+	case "min":
+		expected = "requires at least 2 arg(s), only received 1"
+	case "max":
+		expected = "accepts at most 2 arg(s), received 3"
+	case "exact":
+		expected = "accepts 2 arg(s), received 3"
+	case "range":
+		expected = "accepts between 2 and 4 arg(s), received 1"
 	}
-}
 
-func noArgsWithArgs(err error, t *testing.T) {
-	if err == nil {
-		t.Fatal("Expected an error")
-	}
-	got := err.Error()
-	expected := `unknown command "one" 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)
-	}
-}
-
 // NoArgs
 
 func TestNoArgs(t *testing.T) {
@@ -105,13 +62,13 @@ func TestNoArgs(t *testing.T) {
 func TestNoArgsWithArgs(t *testing.T) {
 	c := getCommand(NoArgs, false)
 	_, err := executeCommand(c, "one")
-	noArgsWithArgs(err, t)
+	expectError(err, t, "no")
 }
 
 func TestNoArgsWithArgsWithValid(t *testing.T) {
 	c := getCommand(NoArgs, true)
 	_, err := executeCommand(c, "one")
-	noArgsWithArgs(err, t)
+	expectError(err, t, "no")
 }
 
 // ArbitraryArgs
@@ -131,7 +88,7 @@ func TestArbitraryArgsWithValid(t *testing.T) {
 func TestArbitraryArgsWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(ArbitraryArgs, true)
 	_, err := executeCommand(c, "a")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 // MinimumNArgs
@@ -151,25 +108,25 @@ func TestMinimumNArgsWithValid(t *testing.T) {
 func TestMinimumNArgsWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(MinimumNArgs(2), true)
 	_, err := executeCommand(c, "a", "b")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 func TestMinimumNArgsWithLessArgs(t *testing.T) {
 	c := getCommand(MinimumNArgs(2), false)
 	_, err := executeCommand(c, "a")
-	minimumNArgsWithLessArgs(err, t)
+	expectError(err, t, "min")
 }
 
 func TestMinimumNArgsWithLessArgsWithValid(t *testing.T) {
 	c := getCommand(MinimumNArgs(2), true)
 	_, err := executeCommand(c, "one")
-	minimumNArgsWithLessArgs(err, t)
+	expectError(err, t, "min")
 }
 
 func TestMinimumNArgsWithLessArgsWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(MinimumNArgs(2), true)
 	_, err := executeCommand(c, "a")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 // MaximumNArgs
@@ -189,25 +146,25 @@ func TestMaximumNArgsWithValid(t *testing.T) {
 func TestMaximumNArgsWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(MaximumNArgs(2), true)
 	_, err := executeCommand(c, "a", "b")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 func TestMaximumNArgsWithMoreArgs(t *testing.T) {
 	c := getCommand(MaximumNArgs(2), false)
 	_, err := executeCommand(c, "a", "b", "c")
-	maximumNArgsWithMoreArgs(err, t)
+	expectError(err, t, "max")
 }
 
 func TestMaximumNArgsWithMoreArgsWithValid(t *testing.T) {
 	c := getCommand(MaximumNArgs(2), true)
 	_, err := executeCommand(c, "one", "three", "two")
-	maximumNArgsWithMoreArgs(err, t)
+	expectError(err, t, "max")
 }
 
 func TestMaximumNArgsWithMoreArgsWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(MaximumNArgs(2), true)
 	_, err := executeCommand(c, "a", "b", "c")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 // ExactArgs
@@ -227,25 +184,25 @@ func TestExactArgsWithValid(t *testing.T) {
 func TestExactArgsWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(ExactArgs(3), true)
 	_, err := executeCommand(c, "three", "a", "two")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 func TestExactArgsWithInvalidCount(t *testing.T) {
 	c := getCommand(ExactArgs(2), false)
 	_, err := executeCommand(c, "a", "b", "c")
-	exactArgsWithInvalidCount(err, t)
+	expectError(err, t, "exact")
 }
 
 func TestExactArgsWithInvalidCountWithValid(t *testing.T) {
 	c := getCommand(ExactArgs(2), true)
 	_, err := executeCommand(c, "three", "one", "two")
-	exactArgsWithInvalidCount(err, t)
+	expectError(err, t, "exact")
 }
 
 func TestExactArgsWithInvalidCountWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(ExactArgs(2), true)
 	_, err := executeCommand(c, "three", "a", "two")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 // RangeArgs
@@ -265,25 +222,25 @@ func TestRangeArgsWithValid(t *testing.T) {
 func TestRangeArgsWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(RangeArgs(2, 4), true)
 	_, err := executeCommand(c, "three", "a", "two")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 func TestRangeArgsWithInvalidCount(t *testing.T) {
 	c := getCommand(RangeArgs(2, 4), false)
 	_, err := executeCommand(c, "a")
-	rangeArgsWithInvalidCount(err, t)
+	expectError(err, t, "range")
 }
 
 func TestRangeArgsWithInvalidCountWithValid(t *testing.T) {
 	c := getCommand(RangeArgs(2, 4), true)
 	_, err := executeCommand(c, "two")
-	rangeArgsWithInvalidCount(err, t)
+	expectError(err, t, "range")
 }
 
 func TestRangeArgsWithInvalidCountWithValidWithInvalidArgs(t *testing.T) {
 	c := getCommand(RangeArgs(2, 4), true)
 	_, err := executeCommand(c, "a")
-	validWithInvalidArgs(err, t)
+	expectError(err, t, "valid")
 }
 
 // Takes(No)Args