mirror of
https://github.com/grafana/grafana.git
synced 2025-08-06 20:59:35 +08:00
Chore: Move away from convey - introduce testify (#32295)
* Generic OAuth: Prevent adding duplicated users * Chore: Move away from convey - introduce testify * Revert commits from another branch * Fix formatting
This commit is contained in:

committed by
GitHub

parent
66548878fe
commit
95de48261b
@ -6,230 +6,184 @@ import (
|
|||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/grafana/grafana/pkg/setting"
|
"github.com/grafana/grafana/pkg/setting"
|
||||||
. "github.com/smartystreets/goconvey/convey"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestInfluxdbResponseParser(t *testing.T) {
|
func TestInfluxdbResponseParser(t *testing.T) {
|
||||||
Convey("Influxdb response parser", t, func() {
|
t.Run("Influxdb response parser should parse everything normally", func(t *testing.T) {
|
||||||
Convey("Response parser", func() {
|
parser := &ResponseParser{}
|
||||||
parser := &ResponseParser{}
|
|
||||||
|
|
||||||
cfg := setting.NewCfg()
|
cfg := setting.NewCfg()
|
||||||
err := cfg.Load(&setting.CommandLineArgs{
|
err := cfg.Load(&setting.CommandLineArgs{
|
||||||
HomePath: "../../../",
|
HomePath: "../../../",
|
||||||
})
|
})
|
||||||
So(err, ShouldBeNil)
|
require.NoError(t, err)
|
||||||
|
|
||||||
response := &Response{
|
response := &Response{
|
||||||
Results: []Result{
|
Results: []Result{
|
||||||
{
|
{
|
||||||
Series: []Row{
|
Series: []Row{
|
||||||
{
|
{
|
||||||
Name: "cpu",
|
Name: "cpu",
|
||||||
Columns: []string{"time", "mean", "sum"},
|
Columns: []string{"time", "mean", "sum"},
|
||||||
Tags: map[string]string{"datacenter": "America"},
|
Tags: map[string]string{"datacenter": "America"},
|
||||||
Values: [][]interface{}{
|
Values: [][]interface{}{
|
||||||
{json.Number("111"), json.Number("222"), json.Number("333")},
|
{json.Number("111"), json.Number("222"), json.Number("333")},
|
||||||
{json.Number("111"), json.Number("222"), json.Number("333")},
|
{json.Number("111"), json.Number("222"), json.Number("333")},
|
||||||
{json.Number("111"), json.Number("null"), json.Number("333")},
|
{json.Number("111"), json.Number("null"), json.Number("333")},
|
||||||
},
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
},
|
||||||
|
}
|
||||||
|
|
||||||
query := &Query{}
|
query := &Query{}
|
||||||
|
|
||||||
result := parser.Parse(response, query)
|
result := parser.Parse(response, query)
|
||||||
|
|
||||||
Convey("can parse all series", func() {
|
require.Len(t, result.Series, 2)
|
||||||
So(len(result.Series), ShouldEqual, 2)
|
|
||||||
})
|
|
||||||
|
|
||||||
Convey("can parse all points", func() {
|
require.Len(t, result.Series[0].Points, 3)
|
||||||
So(len(result.Series[0].Points), ShouldEqual, 3)
|
require.Len(t, result.Series[1].Points, 3)
|
||||||
So(len(result.Series[1].Points), ShouldEqual, 3)
|
|
||||||
})
|
|
||||||
|
|
||||||
Convey("can parse multi row result", func() {
|
require.Equal(t, result.Series[0].Points[1][0].Float64, float64(222))
|
||||||
So(result.Series[0].Points[1][0].Float64, ShouldEqual, float64(222))
|
require.Equal(t, result.Series[1].Points[1][0].Float64, float64(333))
|
||||||
So(result.Series[1].Points[1][0].Float64, ShouldEqual, float64(333))
|
|
||||||
})
|
|
||||||
|
|
||||||
Convey("can parse null points", func() {
|
require.False(t, result.Series[0].Points[2][0].Valid)
|
||||||
So(result.Series[0].Points[2][0].Valid, ShouldBeFalse)
|
|
||||||
})
|
|
||||||
|
|
||||||
Convey("can format series names", func() {
|
require.Equal(t, result.Series[0].Name, "cpu.mean { datacenter: America }")
|
||||||
So(result.Series[0].Name, ShouldEqual, "cpu.mean { datacenter: America }")
|
require.Equal(t, result.Series[1].Name, "cpu.sum { datacenter: America }")
|
||||||
So(result.Series[1].Name, ShouldEqual, "cpu.sum { datacenter: America }")
|
})
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
Convey("Response parser with alias", func() {
|
t.Run("Influxdb response parser with alias", func(t *testing.T) {
|
||||||
parser := &ResponseParser{}
|
parser := &ResponseParser{}
|
||||||
|
|
||||||
response := &Response{
|
response := &Response{
|
||||||
Results: []Result{
|
Results: []Result{
|
||||||
{
|
{
|
||||||
Series: []Row{
|
Series: []Row{
|
||||||
{
|
{
|
||||||
Name: "cpu.upc",
|
Name: "cpu.upc",
|
||||||
Columns: []string{"time", "mean", "sum"},
|
Columns: []string{"time", "mean", "sum"},
|
||||||
Tags: map[string]string{
|
Tags: map[string]string{
|
||||||
"datacenter": "America",
|
"datacenter": "America",
|
||||||
"dc.region.name": "Northeast",
|
"dc.region.name": "Northeast",
|
||||||
"cluster-name": "Cluster",
|
"cluster-name": "Cluster",
|
||||||
},
|
},
|
||||||
Values: [][]interface{}{
|
Values: [][]interface{}{
|
||||||
{json.Number("111"), json.Number("222"), json.Number("333")},
|
{json.Number("111"), json.Number("222"), json.Number("333")},
|
||||||
},
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
},
|
||||||
|
}
|
||||||
|
|
||||||
Convey("$ alias", func() {
|
query := &Query{Alias: "series alias"}
|
||||||
Convey("simple alias", func() {
|
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"}
|
||||||
query := &Query{Alias: "alias $m $measurement", Measurement: "10m"}
|
result = parser.Parse(response, query)
|
||||||
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"}
|
||||||
query := &Query{Alias: "alias $col", Measurement: "10m"}
|
result = parser.Parse(response, query)
|
||||||
result := parser.Parse(response, query)
|
|
||||||
|
|
||||||
So(result.Series[0].Name, ShouldEqual, "alias mean")
|
require.Equal(t, result.Series[0].Name, "alias mean")
|
||||||
So(result.Series[1].Name, ShouldEqual, "alias sum")
|
require.Equal(t, result.Series[1].Name, "alias sum")
|
||||||
})
|
|
||||||
|
|
||||||
Convey("tag alias", func() {
|
query = &Query{Alias: "alias $tag_datacenter"}
|
||||||
query := &Query{Alias: "alias $tag_datacenter"}
|
result = parser.Parse(response, query)
|
||||||
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"}
|
||||||
query := &Query{Alias: "alias $1"}
|
result = parser.Parse(response, query)
|
||||||
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"}
|
||||||
query := &Query{Alias: "alias $5"}
|
result = parser.Parse(response, query)
|
||||||
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() {
|
query = &Query{Alias: "series alias"}
|
||||||
Convey("simple alias", func() {
|
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"}
|
||||||
query := &Query{Alias: "alias [[m]] [[measurement]]", Measurement: "10m"}
|
result = parser.Parse(response, query)
|
||||||
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"}
|
||||||
query := &Query{Alias: "alias [[col]]", Measurement: "10m"}
|
result = parser.Parse(response, query)
|
||||||
result := parser.Parse(response, query)
|
|
||||||
|
|
||||||
So(result.Series[0].Name, ShouldEqual, "alias mean")
|
require.Equal(t, result.Series[0].Name, "alias mean")
|
||||||
So(result.Series[1].Name, ShouldEqual, "alias sum")
|
require.Equal(t, result.Series[1].Name, "alias sum")
|
||||||
})
|
|
||||||
|
|
||||||
Convey("tag alias", func() {
|
query = &Query{Alias: "alias [[tag_datacenter]]"}
|
||||||
query := &Query{Alias: "alias [[tag_datacenter]]"}
|
result = parser.Parse(response, query)
|
||||||
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]]"}
|
||||||
query := &Query{Alias: "alias [[tag_dc.region.name]]"}
|
result = parser.Parse(response, query)
|
||||||
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]]"}
|
||||||
query := &Query{Alias: "alias [[tag_cluster-name]]"}
|
result = parser.Parse(response, query)
|
||||||
result := parser.Parse(response, query)
|
|
||||||
|
|
||||||
So(result.Series[0].Name, ShouldEqual, "alias Cluster")
|
require.Equal(t, result.Series[0].Name, "alias Cluster")
|
||||||
})
|
})
|
||||||
})
|
|
||||||
|
t.Run("Influxdb response parser with errors", func(t *testing.T) {
|
||||||
|
parser := &ResponseParser{}
|
||||||
|
|
||||||
|
cfg := setting.NewCfg()
|
||||||
|
err := cfg.Load(&setting.CommandLineArgs{
|
||||||
|
HomePath: "../../../",
|
||||||
})
|
})
|
||||||
|
require.Nil(t, err)
|
||||||
|
|
||||||
Convey("Response parser with errors", func() {
|
response := &Response{
|
||||||
parser := &ResponseParser{}
|
Results: []Result{
|
||||||
|
{
|
||||||
cfg := setting.NewCfg()
|
Series: []Row{
|
||||||
err := cfg.Load(&setting.CommandLineArgs{
|
{
|
||||||
HomePath: "../../../",
|
Name: "cpu",
|
||||||
})
|
Columns: []string{"time", "mean", "sum"},
|
||||||
So(err, ShouldBeNil)
|
Tags: map[string]string{"datacenter": "America"},
|
||||||
|
Values: [][]interface{}{
|
||||||
response := &Response{
|
{json.Number("111"), json.Number("222"), json.Number("333")},
|
||||||
Results: []Result{
|
{json.Number("111"), json.Number("222"), json.Number("333")},
|
||||||
{
|
{json.Number("111"), json.Number("null"), json.Number("333")},
|
||||||
Series: []Row{
|
|
||||||
{
|
|
||||||
Name: "cpu",
|
|
||||||
Columns: []string{"time", "mean", "sum"},
|
|
||||||
Tags: map[string]string{"datacenter": "America"},
|
|
||||||
Values: [][]interface{}{
|
|
||||||
{json.Number("111"), json.Number("222"), json.Number("333")},
|
|
||||||
{json.Number("111"), json.Number("222"), json.Number("333")},
|
|
||||||
{json.Number("111"), json.Number("null"), json.Number("333")},
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
|
||||||
Err: fmt.Errorf("query-timeout limit exceeded"),
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
}
|
{
|
||||||
|
Err: fmt.Errorf("query-timeout limit exceeded"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
query := &Query{}
|
query := &Query{}
|
||||||
|
|
||||||
result := parser.Parse(response, query)
|
result := parser.Parse(response, query)
|
||||||
|
|
||||||
Convey("can parse all series", func() {
|
require.Len(t, result.Series, 2)
|
||||||
So(len(result.Series), ShouldEqual, 2)
|
|
||||||
})
|
|
||||||
|
|
||||||
Convey("can parse all points", func() {
|
require.Len(t, result.Series[0].Points, 3)
|
||||||
So(len(result.Series[0].Points), ShouldEqual, 3)
|
require.Len(t, result.Series[1].Points, 3)
|
||||||
So(len(result.Series[1].Points), ShouldEqual, 3)
|
|
||||||
})
|
|
||||||
|
|
||||||
Convey("can parse errors ", func() {
|
require.Error(t, result.Error)
|
||||||
So(result.Error, ShouldNotBeNil)
|
require.Equal(t, result.Error.Error(), "query-timeout limit exceeded")
|
||||||
So(result.Error.Error(), ShouldEqual, "query-timeout limit exceeded")
|
|
||||||
})
|
|
||||||
})
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user