Duration sent to cloud when starting a test with stages

stats/cloud/client.go

@@ -90,14 +90,16 @@ type TestRun struct {
 	Name        string              `json:"name"`
 	ProjectID   int                 `json:"project_id,omitempty"`
 	Threasholds map[string][]string `json:"thresholds"`
+	// Duration of test in seconds. -1 for unknown length, 0 for continuous running.
+	Duration int64 `json:"duration"`
 }
 
 type CreateTestRunResponse struct {
 	ReferenceID string `json:"reference_id"`
 }
 
-func (c *Client) CreateTestRun(name string, thresholds map[string][]string) *CreateTestRunResponse {
-	testRun := TestRun{Name: name, Threasholds: thresholds}
+func (c *Client) CreateTestRun(name string, thresholds map[string][]string, duration int64) *CreateTestRunResponse {
+	testRun := TestRun{Name: name, Threasholds: thresholds, Duration: duration}
 
 	url := fmt.Sprintf("%s/tests", c.baseURL)
 	req, err := c.NewRequest("POST", url, testRun)

stats/cloud/collector.go

@@ -15,6 +15,7 @@ import (
 type Collector struct {
 	referenceID string
 
+	duration   int64
 	thresholds map[string][]string
 	client     *Client
 }
@@ -31,10 +32,17 @@ func New(fname string, opts lib.Options) (*Collector, error) {
 		}
 	}
 
+	// Sum test duration from options. -1 for unknown duration.
+	var duration int64 = -1
+	if len(opts.Stages) > 0 {
+		duration = sumStages(opts.Stages)
+	}
+
 	return &Collector{
 		referenceID: referenceID,
 		thresholds:  thresholds,
 		client:      NewClient(token),
+		duration:    duration,
 	}, nil
 }
 
@@ -46,7 +54,7 @@ func (c *Collector) Init() {
 
 	// TODO fix this and add proper error handling
 	if c.referenceID == "" {
-		response := c.client.CreateTestRun(name, c.thresholds)
+		response := c.client.CreateTestRun(name, c.thresholds, c.duration)
 		if response != nil {
 			c.referenceID = response.ReferenceID
 		}
@@ -88,3 +96,12 @@ func (c *Collector) Collect(samples []stats.Sample) {
 		c.client.PushMetric(c.referenceID, cloudSamples)
 	}
 }
+
+func sumStages(stages []lib.Stage) int64 {
+	var total time.Duration
+	for _, stage := range stages {
+		total += stage.Duration
+	}
+
+	return int64(total.Seconds())
+}