Expressions: Move GEL into core as expressions (#29072)

* comes from grafana/gel-app * remove transform plugin code * move __expr__ and -100 constants to expr pkg * set OrgID on request plugin context * use gtime for resample duration * in resample, rename "rule" to "window", use gtime for duration, parse duration before exec * remove gel entry from plugins-bundled/external.json which creates an empty array for plugins
2020-11-19 07:17:00 -05:00
parent f01c3f35e1
commit 0cb29d337a
44 changed files with 5770 additions and 401 deletions
@@ -0,0 +1,240 @@
+package mathexp
+
+import (
+	"math"
+	"testing"
+	"time"
+
+	"github.com/google/go-cmp/cmp"
+	"github.com/grafana/grafana-plugin-sdk-go/data"
+	"github.com/stretchr/testify/require"
+)
+
+var seriesWithNil = Vars{
+	"A": Results{
+		[]Value{
+			makeSeries("temp", nil, tp{
+				time.Unix(5, 0), float64Pointer(2),
+			}, tp{
+				time.Unix(10, 0), nil,
+			}),
+		},
+	},
+}
+
+var seriesEmpty = Vars{
+	"A": Results{
+		[]Value{
+			makeSeries("temp", nil),
+		},
+	},
+}
+
+func TestSeriesReduce(t *testing.T) {
+	var tests = []struct {
+		name        string
+		red         string
+		vars        Vars
+		varToReduce string
+		errIs       require.ErrorAssertionFunc
+		resultsIs   require.ComparisonAssertionFunc
+		results     Results
+	}{
+		{
+			name:        "foo reduction will error",
+			red:         "foo",
+			varToReduce: "A",
+			vars:        aSeriesNullableTime,
+			errIs:       require.Error,
+			resultsIs:   require.Equal,
+		},
+		{
+			name:        "sum series",
+			red:         "sum",
+			varToReduce: "A",
+			vars:        aSeriesNullableTime,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("sum_", nil, float64Pointer(3)),
+				},
+			},
+		},
+		{
+			name:        "sum series with a nil value",
+			red:         "sum",
+			varToReduce: "A",
+			vars:        seriesWithNil,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("sum_", nil, NaN),
+				},
+			},
+		},
+		{
+			name:        "sum empty series",
+			red:         "sum",
+			varToReduce: "A",
+			vars:        seriesEmpty,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("sum_", nil, float64Pointer(0)),
+				},
+			},
+		},
+		{
+			name:        "mean series with a nil value",
+			red:         "mean",
+			varToReduce: "A",
+			vars:        seriesWithNil,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("mean_", nil, NaN),
+				},
+			},
+		},
+		{
+			name:        "mean empty series",
+			red:         "mean",
+			varToReduce: "A",
+			vars:        seriesEmpty,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("mean_", nil, NaN),
+				},
+			},
+		},
+		{
+			name:        "min series with a nil value",
+			red:         "min",
+			varToReduce: "A",
+			vars:        seriesWithNil,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("min_", nil, NaN),
+				},
+			},
+		},
+		{
+			name:        "min empty series",
+			red:         "min",
+			varToReduce: "A",
+			vars:        seriesEmpty,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("min_", nil, NaN),
+				},
+			},
+		},
+		{
+			name:        "max series with a nil value",
+			red:         "max",
+			varToReduce: "A",
+			vars:        seriesWithNil,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("max_", nil, NaN),
+				},
+			},
+		},
+		{
+			name:        "max empty series",
+			red:         "max",
+			varToReduce: "A",
+			vars:        seriesEmpty,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("max_", nil, NaN),
+				},
+			},
+		},
+		{
+			name:        "mean series",
+			red:         "mean",
+			varToReduce: "A",
+			vars:        aSeriesNullableTime,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("mean_", nil, float64Pointer(1.5)),
+				},
+			},
+		},
+		{
+			name:        "count empty series",
+			red:         "count",
+			varToReduce: "A",
+			vars:        seriesEmpty,
+			errIs:       require.NoError,
+			resultsIs:   require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("count_", nil, float64Pointer(0)),
+				},
+			},
+		},
+		{
+			name:        "mean series with labels",
+			red:         "mean",
+			varToReduce: "A",
+			vars: Vars{
+				"A": Results{
+					[]Value{
+						makeSeriesNullableTime("temp", data.Labels{"host": "a"}, nullTimeTP{
+							unixTimePointer(5, 0), float64Pointer(2),
+						}, nullTimeTP{
+							unixTimePointer(10, 0), float64Pointer(1),
+						}),
+					},
+				},
+			},
+			errIs:     require.NoError,
+			resultsIs: require.Equal,
+			results: Results{
+				[]Value{
+					makeNumber("mean_", data.Labels{"host": "a"}, float64Pointer(1.5)),
+				},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			results := Results{}
+			seriesSet := tt.vars[tt.varToReduce]
+			for _, series := range seriesSet.Values {
+				ns, err := series.Value().(*Series).Reduce(tt.red)
+				tt.errIs(t, err)
+				if err != nil {
+					return
+				}
+				results.Values = append(results.Values, ns)
+			}
+			opt := cmp.Comparer(func(x, y float64) bool {
+				return (math.IsNaN(x) && math.IsNaN(y)) || x == y
+			})
+			options := append([]cmp.Option{opt}, data.FrameTestCompareOptions()...)
+			if diff := cmp.Diff(tt.results, results, options...); diff != "" {
+				t.Errorf("Result mismatch (-want +got):\n%s", diff)
+			}
+		})
+	}
+}