Fix tests

Signed-off-by: Pavol Loffay <[email protected]>

cmd/docs/command.go

@@ -53,7 +53,7 @@ func Command(v *viper.Viper) *cobra.Command {
 			case "rst":
 				doc.GenReSTTree(cmd, "./")
 			default:
-				return errors.New("undefined format")
+				return errors.New(fmt.Sprintf("undefined value of %v, possible values are: %v", formatFlag, formats))
 			}
 
 			return nil

cmd/docs/command_test.go

@@ -28,19 +28,25 @@ func TestCommand(t *testing.T) {
 	tests := []struct{
 		file string
 		flag string
+		err string
 	}{
 		{file: "docs.md"},
 		{file: "docs.1", flag: "--format=man"},
 		{file: "docs.rst", flag: "--format=rst"},
+		{flag: "--format=foo", err: "undefined value of format, possible values are: [md man rst]"},
 	}
 	for _, test := range tests {
 		v := viper.New()
 		cmd := Command(v)
 		cmd.ParseFlags([]string{test.flag})
-		cmd.RunE(cmd, []string{})
-		f, err := ioutil.ReadFile(test.file)
-		require.NoError(t, err)
-		assert.True(t, strings.Contains(string(f), "documentation"))
+		err := cmd.Execute()
+		if err == nil {
+			f, err := ioutil.ReadFile(test.file)
+			require.NoError(t, err)
+			assert.True(t, strings.Contains(string(f), "documentation"))
+		} else {
+			assert.Equal(t, test.err, err.Error())
+		}
 	}
 }