|
|
|
@ -6,19 +6,18 @@ import ( |
|
|
|
|
"testing" |
|
|
|
|
|
|
|
|
|
"github.com/grafana/grafana/pkg/setting" |
|
|
|
|
. "github.com/smartystreets/goconvey/convey" |
|
|
|
|
"github.com/stretchr/testify/require" |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
func TestInfluxdbResponseParser(t *testing.T) { |
|
|
|
|
Convey("Influxdb response parser", t, func() { |
|
|
|
|
Convey("Response parser", func() { |
|
|
|
|
t.Run("Influxdb response parser should parse everything normally", func(t *testing.T) { |
|
|
|
|
parser := &ResponseParser{} |
|
|
|
|
|
|
|
|
|
cfg := setting.NewCfg() |
|
|
|
|
err := cfg.Load(&setting.CommandLineArgs{ |
|
|
|
|
HomePath: "../../../", |
|
|
|
|
}) |
|
|
|
|
So(err, ShouldBeNil) |
|
|
|
|
require.NoError(t, err) |
|
|
|
|
|
|
|
|
|
response := &Response{ |
|
|
|
|
Results: []Result{ |
|
|
|
@ -43,31 +42,21 @@ func TestInfluxdbResponseParser(t *testing.T) { |
|
|
|
|
|
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
Convey("can parse all series", func() { |
|
|
|
|
So(len(result.Series), ShouldEqual, 2) |
|
|
|
|
}) |
|
|
|
|
require.Len(t, result.Series, 2) |
|
|
|
|
|
|
|
|
|
Convey("can parse all points", func() { |
|
|
|
|
So(len(result.Series[0].Points), ShouldEqual, 3) |
|
|
|
|
So(len(result.Series[1].Points), ShouldEqual, 3) |
|
|
|
|
}) |
|
|
|
|
require.Len(t, result.Series[0].Points, 3) |
|
|
|
|
require.Len(t, result.Series[1].Points, 3) |
|
|
|
|
|
|
|
|
|
Convey("can parse multi row result", func() { |
|
|
|
|
So(result.Series[0].Points[1][0].Float64, ShouldEqual, float64(222)) |
|
|
|
|
So(result.Series[1].Points[1][0].Float64, ShouldEqual, float64(333)) |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Points[1][0].Float64, float64(222)) |
|
|
|
|
require.Equal(t, result.Series[1].Points[1][0].Float64, float64(333)) |
|
|
|
|
|
|
|
|
|
Convey("can parse null points", func() { |
|
|
|
|
So(result.Series[0].Points[2][0].Valid, ShouldBeFalse) |
|
|
|
|
}) |
|
|
|
|
require.False(t, result.Series[0].Points[2][0].Valid) |
|
|
|
|
|
|
|
|
|
Convey("can format series names", func() { |
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "cpu.mean { datacenter: America }") |
|
|
|
|
So(result.Series[1].Name, ShouldEqual, "cpu.sum { datacenter: America }") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "cpu.mean { datacenter: America }") |
|
|
|
|
require.Equal(t, result.Series[1].Name, "cpu.sum { datacenter: America }") |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
Convey("Response parser with alias", func() { |
|
|
|
|
t.Run("Influxdb response parser with alias", func(t *testing.T) { |
|
|
|
|
parser := &ResponseParser{} |
|
|
|
|
|
|
|
|
|
response := &Response{ |
|
|
|
@ -91,105 +80,77 @@ func TestInfluxdbResponseParser(t *testing.T) { |
|
|
|
|
}, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
Convey("$ alias", func() { |
|
|
|
|
Convey("simple alias", func() { |
|
|
|
|
query := &Query{Alias: "series alias"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "series alias") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "series alias") |
|
|
|
|
|
|
|
|
|
Convey("measurement alias", func() { |
|
|
|
|
query := &Query{Alias: "alias $m $measurement", Measurement: "10m"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias $m $measurement", Measurement: "10m"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias 10m 10m") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias 10m 10m") |
|
|
|
|
|
|
|
|
|
Convey("column alias", func() { |
|
|
|
|
query := &Query{Alias: "alias $col", Measurement: "10m"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias $col", Measurement: "10m"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias mean") |
|
|
|
|
So(result.Series[1].Name, ShouldEqual, "alias sum") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias mean") |
|
|
|
|
require.Equal(t, result.Series[1].Name, "alias sum") |
|
|
|
|
|
|
|
|
|
Convey("tag alias", func() { |
|
|
|
|
query := &Query{Alias: "alias $tag_datacenter"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias $tag_datacenter"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias America") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias America") |
|
|
|
|
|
|
|
|
|
Convey("segment alias", func() { |
|
|
|
|
query := &Query{Alias: "alias $1"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias $1"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias upc") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias upc") |
|
|
|
|
|
|
|
|
|
Convey("segment position out of bound", func() { |
|
|
|
|
query := &Query{Alias: "alias $5"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias $5"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias $5") |
|
|
|
|
}) |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias $5") |
|
|
|
|
|
|
|
|
|
Convey("[[]] alias", func() { |
|
|
|
|
Convey("simple alias", func() { |
|
|
|
|
query := &Query{Alias: "series alias"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "series alias"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "series alias") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "series alias") |
|
|
|
|
|
|
|
|
|
Convey("measurement alias", func() { |
|
|
|
|
query := &Query{Alias: "alias [[m]] [[measurement]]", Measurement: "10m"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias [[m]] [[measurement]]", Measurement: "10m"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias 10m 10m") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias 10m 10m") |
|
|
|
|
|
|
|
|
|
Convey("column alias", func() { |
|
|
|
|
query := &Query{Alias: "alias [[col]]", Measurement: "10m"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias [[col]]", Measurement: "10m"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias mean") |
|
|
|
|
So(result.Series[1].Name, ShouldEqual, "alias sum") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias mean") |
|
|
|
|
require.Equal(t, result.Series[1].Name, "alias sum") |
|
|
|
|
|
|
|
|
|
Convey("tag alias", func() { |
|
|
|
|
query := &Query{Alias: "alias [[tag_datacenter]]"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias [[tag_datacenter]]"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias America") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias America") |
|
|
|
|
|
|
|
|
|
Convey("tag alias with periods", func() { |
|
|
|
|
query := &Query{Alias: "alias [[tag_dc.region.name]]"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias [[tag_dc.region.name]]"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias Northeast") |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias Northeast") |
|
|
|
|
|
|
|
|
|
Convey("tag alias with hyphens", func() { |
|
|
|
|
query := &Query{Alias: "alias [[tag_cluster-name]]"} |
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
query = &Query{Alias: "alias [[tag_cluster-name]]"} |
|
|
|
|
result = parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
So(result.Series[0].Name, ShouldEqual, "alias Cluster") |
|
|
|
|
}) |
|
|
|
|
}) |
|
|
|
|
require.Equal(t, result.Series[0].Name, "alias Cluster") |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
Convey("Response parser with errors", func() { |
|
|
|
|
t.Run("Influxdb response parser with errors", func(t *testing.T) { |
|
|
|
|
parser := &ResponseParser{} |
|
|
|
|
|
|
|
|
|
cfg := setting.NewCfg() |
|
|
|
|
err := cfg.Load(&setting.CommandLineArgs{ |
|
|
|
|
HomePath: "../../../", |
|
|
|
|
}) |
|
|
|
|
So(err, ShouldBeNil) |
|
|
|
|
require.Nil(t, err) |
|
|
|
|
|
|
|
|
|
response := &Response{ |
|
|
|
|
Results: []Result{ |
|
|
|
@ -217,19 +178,12 @@ func TestInfluxdbResponseParser(t *testing.T) { |
|
|
|
|
|
|
|
|
|
result := parser.Parse(response, query) |
|
|
|
|
|
|
|
|
|
Convey("can parse all series", func() { |
|
|
|
|
So(len(result.Series), ShouldEqual, 2) |
|
|
|
|
}) |
|
|
|
|
require.Len(t, result.Series, 2) |
|
|
|
|
|
|
|
|
|
Convey("can parse all points", func() { |
|
|
|
|
So(len(result.Series[0].Points), ShouldEqual, 3) |
|
|
|
|
So(len(result.Series[1].Points), ShouldEqual, 3) |
|
|
|
|
}) |
|
|
|
|
require.Len(t, result.Series[0].Points, 3) |
|
|
|
|
require.Len(t, result.Series[1].Points, 3) |
|
|
|
|
|
|
|
|
|
Convey("can parse errors ", func() { |
|
|
|
|
So(result.Error, ShouldNotBeNil) |
|
|
|
|
So(result.Error.Error(), ShouldEqual, "query-timeout limit exceeded") |
|
|
|
|
}) |
|
|
|
|
}) |
|
|
|
|
require.Error(t, result.Error) |
|
|
|
|
require.Equal(t, result.Error.Error(), "query-timeout limit exceeded") |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|