summaryrefslogtreecommitdiffstats
path: root/clover/cloverctl/src/cloverctl/cmd/start_testplan.go
blob: b516ad6339e2e41a5cf201179bbaa1883b598731 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
// 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"
    "strings"
    "gopkg.in/resty.v1"
    "github.com/spf13/cobra"
    "cloverkube"
)


var testplanstartCmd = &cobra.Command{
    Use:   "testplan",
    Short: "Start a test for a given test plan",
    Long: `Specify number of slaves to use with '-s' flag. Default is 0 slaves,
which runs tests only from jmeter-master.`,
    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")
}

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], ",")

    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 {
        panic(err.Error())
    }
    fmt.Printf("\n%v\n", resp)

}