mirror of https://github.com/grafana/grafana.git
97 lines
3.3 KiB
Go
97 lines
3.3 KiB
Go
package stackdriver
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/tsdb"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
func TestStackdriver(t *testing.T) {
|
|
Convey("Stackdriver", t, func() {
|
|
executor := &StackdriverExecutor{}
|
|
|
|
Convey("Parse query from frontend", func() {
|
|
fromStart := time.Date(2018, 3, 15, 13, 0, 0, 0, time.UTC).In(time.Local)
|
|
tsdbQuery := &tsdb.TsdbQuery{
|
|
TimeRange: &tsdb.TimeRange{
|
|
From: fmt.Sprintf("%v", fromStart.Unix()*1000),
|
|
To: fmt.Sprintf("%v", fromStart.Add(34*time.Minute).Unix()*1000),
|
|
},
|
|
Queries: []*tsdb.Query{
|
|
{
|
|
Model: simplejson.NewFromAny(map[string]interface{}{
|
|
"target": "target",
|
|
"metricType": "time_series",
|
|
}),
|
|
RefId: "A",
|
|
},
|
|
},
|
|
}
|
|
queries, err := executor.parseQueries(tsdbQuery)
|
|
So(err, ShouldBeNil)
|
|
|
|
So(len(queries), ShouldEqual, 1)
|
|
So(queries[0].RefID, ShouldEqual, "A")
|
|
So(queries[0].Target, ShouldEqual, "target")
|
|
So(len(queries[0].Params), ShouldEqual, 4)
|
|
So(queries[0].Params["interval.startTime"][0], ShouldEqual, "2018-03-15T13:00:00Z")
|
|
So(queries[0].Params["interval.endTime"][0], ShouldEqual, "2018-03-15T13:34:00Z")
|
|
So(queries[0].Params["aggregation.perSeriesAligner"][0], ShouldEqual, "ALIGN_NONE")
|
|
So(queries[0].Params["filter"][0], ShouldEqual, "time_series")
|
|
})
|
|
|
|
Convey("Parse stackdriver response for data aggregated to one time series", func() {
|
|
var data StackDriverResponse
|
|
|
|
jsonBody, err := ioutil.ReadFile("./test-data/1-series-response-agg-one-metric.json")
|
|
So(err, ShouldBeNil)
|
|
err = json.Unmarshal(jsonBody, &data)
|
|
So(err, ShouldBeNil)
|
|
So(len(data.TimeSeries), ShouldEqual, 1)
|
|
|
|
res, err := executor.parseResponse(data, "A")
|
|
So(err, ShouldBeNil)
|
|
|
|
So(len(res.Series), ShouldEqual, 1)
|
|
So(res.Series[0].Name, ShouldEqual, "serviceruntime.googleapis.com/api/request_count")
|
|
So(len(res.Series[0].Points), ShouldEqual, 3)
|
|
|
|
So(res.Series[0].Points[0][0].Float64, ShouldEqual, 1.0666666666667)
|
|
So(res.Series[0].Points[1][0].Float64, ShouldEqual, 1.05)
|
|
So(res.Series[0].Points[2][0].Float64, ShouldEqual, 0.05)
|
|
})
|
|
|
|
Convey("Parse stackdriver response for data with no aggregation", func() {
|
|
var data StackDriverResponse
|
|
|
|
jsonBody, err := ioutil.ReadFile("./test-data/2-series-response-no-agg.json")
|
|
So(err, ShouldBeNil)
|
|
err = json.Unmarshal(jsonBody, &data)
|
|
So(err, ShouldBeNil)
|
|
So(len(data.TimeSeries), ShouldEqual, 3)
|
|
|
|
res, err := executor.parseResponse(data, "A")
|
|
So(err, ShouldBeNil)
|
|
|
|
Convey("Should add labels to metric name", func() {
|
|
So(len(res.Series), ShouldEqual, 3)
|
|
So(res.Series[0].Name, ShouldEqual, "compute.googleapis.com/instance/cpu/usage_time collector-asia-east-1")
|
|
So(res.Series[1].Name, ShouldEqual, "compute.googleapis.com/instance/cpu/usage_time collector-europe-west-1")
|
|
So(res.Series[2].Name, ShouldEqual, "compute.googleapis.com/instance/cpu/usage_time collector-us-east-1")
|
|
|
|
So(len(res.Series[0].Points), ShouldEqual, 3)
|
|
So(res.Series[0].Points[0][0].Float64, ShouldEqual, 9.7730520330369)
|
|
So(res.Series[0].Points[1][0].Float64, ShouldEqual, 9.7323568146676)
|
|
So(res.Series[0].Points[2][0].Float64, ShouldEqual, 9.8566497180145)
|
|
})
|
|
})
|
|
})
|
|
}
|