Riot mark as required kind only if APIMode is unset

BBVA · Mar 15, 2019 · a4591c8 · a4591c8
1 parent 6531835
commit a4591c8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/riot.go b/tests/riot.go
@@ -151,6 +151,10 @@ func newRiotCommand() *cobra.Command {
 				}()
 			}
 
+			if !APIMode && riot.Config.Kind == "" {
+				log.Fatal("Argument `kind` is required")
+			}
+
 		},
 		Run: func(cmd *cobra.Command, args []string) {
 			riot.Start(APIMode)
@@ -174,8 +178,6 @@ func newRiotCommand() *cobra.Command {
 	f.UintVar(&riot.Config.MaxGoRoutines, "r", 10, "Set the concurrency value")
 	f.UintVar(&riot.Config.Offset, "offset", 0, "The starting version from which we start the load")
 
-	cmd.MarkFlagRequired("kind")
-
 	return cmd
 }