1 files changed, 14 insertions, 9 deletions

diff --git a/clover/cloverctl/src/cloverctl/cmd/start_testplan.go b/clover/cloverctl/src/cloverctl/cmd/start_testplan.go
index b516ad6..9e664c0 100644
--- a/clover/cloverctl/src/cloverctl/cmd/start_testplan.go
+++ b/clover/cloverctl/src/cloverctl/cmd/start_testplan.go
@@ -9,6 +9,7 @@ package cmd
 
 import (
     "fmt"
+    "os"
     "strings"
     "gopkg.in/resty.v1"
     "github.com/spf13/cobra"
@@ -18,20 +19,21 @@ import (
 
 var testplanstartCmd = &cobra.Command{
     Use:   "testplan",
-    Short: "Start a test for a given test plan",
+    Short: "Start test for configured test plan",
     Long: `Specify number of slaves to use with '-s' flag. Default is 0 slaves,
-which runs tests only from jmeter-master.`,
+which runs tests from jmeter-master only.`,
     Run: func(cmd *cobra.Command, args []string) {
         startTest()
-        //fmt.Printf("%v\n", cmd.Parent().CommandPath())
     },
 }
 var num_slaves int
 
 func init() {
     startCmd.AddCommand(testplanstartCmd)
-    testplanstartCmd.PersistentFlags().StringVarP(&cloverFile, "file", "f", "", "Currently unused")
-    testplanstartCmd.PersistentFlags().IntVarP(&num_slaves, "slaves", "s", 0, "Number of slaves to use")
+    testplanstartCmd.PersistentFlags().StringVarP(&cloverFile, "file", "f", "",
+                                                  "Currently unused")
+    testplanstartCmd.PersistentFlags().IntVarP(&num_slaves, "slaves", "s", 0,
+                                               "Number of slaves to use")
 }
 
 func startTest() {
@@ -39,21 +41,24 @@ func startTest() {
     ips := cloverkube.GetPodsIP("clover-jmeter-slave", "default")
     fmt.Printf("\njmeter-slaves found: %s\n", ips)
     if num_slaves > len(ips) {
-        fmt.Printf("Number of slaves specified must be less than found: %d\n", len(ips))
+        fmt.Printf("Number of slaves specified must be less than found: %d\n",
+                   len(ips))
         return
     }
     ip_list := strings.Join(ips[0:num_slaves], ",")
 
+    checkControllerIP()
     url := controllerIP + "/jmeter/start"
     resp, err := resty.R().
     SetHeader("Content-Type", "application/json").
-    SetBody(fmt.Sprintf(`{"num_slaves":"%d", "slave_list":"%s"}`, num_slaves, ip_list)).
+    SetBody(fmt.Sprintf(`{"num_slaves":"%d", "slave_list":"%s"}`, num_slaves,
+                        ip_list)).
     Post(url)
     if err != nil {
-        panic(err.Error())
+        fmt.Printf("Cannot connect to controller: %v\n", err)
+        os.Exit(1)
     }
     fmt.Printf("\n%v\n", resp)
-
 }