1 files changed, 64 insertions, 0 deletions
diff --git a/clover/cloverctl/src/cloverctl/cmd/start_testplan.go b/clover/cloverctl/src/cloverctl/cmd/start_testplan.go
new file mode 100644
index 0000000..9e664c0
--- /dev/null
+++ b/clover/cloverctl/src/cloverctl/cmd/start_testplan.go
@@ -0,0 +1,64 @@
+// Copyright (c) Authors of Clover
+//
+// All rights reserved. This program and the accompanying materials
+// are made available under the terms of the Apache License, Version 2.0
+// which accompanies this distribution, and is available at
+// http://www.apache.org/licenses/LICENSE-2.0
+
+package cmd
+
+import (
+    "fmt"
+    "os"
+    "strings"
+    "gopkg.in/resty.v1"
+    "github.com/spf13/cobra"
+    "cloverkube"
+)
+
+
+var testplanstartCmd = &cobra.Command{
+    Use:   "testplan",
+    Short: "Start test for configured test plan",
+    Long: `Specify number of slaves to use with '-s' flag. Default is 0 slaves,
+which runs tests from jmeter-master only.`,
+    Run: func(cmd *cobra.Command, args []string) {
+        startTest()
+    },
+}
+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")
+}
+
+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))
+        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)).
+    Post(url)
+    if err != nil {
+        fmt.Printf("Cannot connect to controller: %v\n", err)
+        os.Exit(1)
+    }
+    fmt.Printf("\n%v\n", resp)
+}
+
+