mirror of
https://github.com/zalando-incubator/kube-metrics-adapter.git
synced 2025-05-12 08:21:54 +00:00
113 lines
3.1 KiB
Go
113 lines
3.1 KiB
Go
package collector
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/oliveagle/jsonpath"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func compareMetricsGetter(t *testing.T, first, second *JSONPathMetricsGetter) {
|
|
require.Equal(t, first.jsonPath, second.jsonPath)
|
|
require.Equal(t, first.scheme, second.scheme)
|
|
require.Equal(t, first.path, second.path)
|
|
require.Equal(t, first.port, second.port)
|
|
}
|
|
|
|
func TestNewJSONPathMetricsGetter(t *testing.T) {
|
|
configNoAggregator := map[string]string{
|
|
"json-key": "$.value",
|
|
"scheme": "http",
|
|
"path": "/metrics",
|
|
"port": "9090",
|
|
}
|
|
jpath1, _ := jsonpath.Compile(configNoAggregator["json-key"])
|
|
getterNoAggregator, err1 := NewJSONPathMetricsGetter(configNoAggregator)
|
|
|
|
require.NoError(t, err1)
|
|
compareMetricsGetter(t, &JSONPathMetricsGetter{
|
|
jsonPath: jpath1,
|
|
scheme: "http",
|
|
path: "/metrics",
|
|
port: 9090,
|
|
}, getterNoAggregator)
|
|
|
|
configAggregator := map[string]string{
|
|
"json-key": "$.values",
|
|
"scheme": "http",
|
|
"path": "/metrics",
|
|
"port": "9090",
|
|
"aggregator": "avg",
|
|
}
|
|
jpath2, _ := jsonpath.Compile(configAggregator["json-key"])
|
|
getterAggregator, err2 := NewJSONPathMetricsGetter(configAggregator)
|
|
|
|
require.NoError(t, err2)
|
|
compareMetricsGetter(t, &JSONPathMetricsGetter{
|
|
jsonPath: jpath2,
|
|
scheme: "http",
|
|
path: "/metrics",
|
|
port: 9090,
|
|
aggregator: "avg",
|
|
}, getterAggregator)
|
|
|
|
configErrorJSONPath := map[string]string{
|
|
"json-key": "{}",
|
|
"scheme": "http",
|
|
"path": "/metrics",
|
|
"port": "9090",
|
|
}
|
|
|
|
_, err3 := NewJSONPathMetricsGetter(configErrorJSONPath)
|
|
require.Error(t, err3)
|
|
|
|
configErrorPort := map[string]string{
|
|
"json-key": "$.values",
|
|
"scheme": "http",
|
|
"path": "/metrics",
|
|
"port": "a9090",
|
|
}
|
|
|
|
_, err4 := NewJSONPathMetricsGetter(configErrorPort)
|
|
require.Error(t, err4)
|
|
}
|
|
|
|
func TestCastSlice(t *testing.T) {
|
|
res1, err1 := castSlice([]interface{}{1, 2, 3})
|
|
require.NoError(t, err1)
|
|
require.Equal(t, []float64{1.0, 2.0, 3.0}, res1)
|
|
|
|
res2, err2 := castSlice([]interface{}{float32(1.0), float32(2.0), float32(3.0)})
|
|
require.NoError(t, err2)
|
|
require.Equal(t, []float64{1.0, 2.0, 3.0}, res2)
|
|
|
|
res3, err3 := castSlice([]interface{}{float64(1.0), float64(2.0), float64(3.0)})
|
|
require.NoError(t, err3)
|
|
require.Equal(t, []float64{1.0, 2.0, 3.0}, res3)
|
|
|
|
res4, err4 := castSlice([]interface{}{1, 2, "some string"})
|
|
require.Errorf(t, err4, "slice was returned by JSONPath, but value inside is unsupported: %T", "string")
|
|
require.Equal(t, []float64(nil), res4)
|
|
}
|
|
|
|
func TestReduce(t *testing.T) {
|
|
average, err1 := reduce([]float64{1, 2, 3}, "avg")
|
|
require.NoError(t, err1)
|
|
require.Equal(t, 2.0, average)
|
|
|
|
min, err2 := reduce([]float64{1, 2, 3}, "min")
|
|
require.NoError(t, err2)
|
|
require.Equal(t, 1.0, min)
|
|
|
|
max, err3 := reduce([]float64{1, 2, 3}, "max")
|
|
require.NoError(t, err3)
|
|
require.Equal(t, 3.0, max)
|
|
|
|
sum, err4 := reduce([]float64{1, 2, 3}, "sum")
|
|
require.NoError(t, err4)
|
|
require.Equal(t, 6.0, sum)
|
|
|
|
_, err5 := reduce([]float64{1, 2, 3}, "inexistent_function")
|
|
require.Errorf(t, err5, "slice of numbers was returned by JSONPath, but no valid aggregator function was specified: %v", "inexistent_function")
|
|
}
|