grafana/pkg/tsdb/stackdriver/stackdriver_test.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)
			})
		})
	})
}