mirror of
https://github.com/grafana/grafana.git
synced 2025-07-29 13:52:25 +08:00
f64be062c4
* introduce tests
* improve allocation by %61
* ~%27 improvement
goos: darwin
goarch: arm64
pkg: github.com/grafana/grafana/pkg/promlib/converter
cpu: Apple M1 Pro
│ pmem.0.txt │ pmem.2.txt │
│ sec/op │ sec/op vs base │
ReadPrometheusStyleResult_FromFile/prom-query-range.json-10 117.72µ ± 6% 82.90µ ± 4% -29.57% (p=0.002 n=6)
│ pmem.0.txt │ pmem.2.txt │
│ B/op │ B/op vs base │
ReadPrometheusStyleResult_FromFile/prom-query-range.json-10 104.34Ki ± 0% 76.09Ki ± 0% -27.08% (p=0.002 n=6)
│ pmem.0.txt │ pmem.2.txt │
│ allocs/op │ allocs/op vs base │
ReadPrometheusStyleResult_FromFile/prom-query-range.json-10 2463.0 ± 0% 899.0 ± 0% -63.50% (p=0.002 n=6)
* add more tests
* remove comment lines
* read string as slice to prevent type conversion
* golang lint
* revert appendValueFromString improvement
* fix merging issues
128 lines
2.8 KiB
Go
128 lines
2.8 KiB
Go
package converter
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
sdkjsoniter "github.com/grafana/grafana-plugin-sdk-go/data/utils/jsoniter"
|
|
"github.com/grafana/grafana-plugin-sdk-go/experimental"
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const update = false
|
|
|
|
var files = []string{
|
|
"prom-labels",
|
|
"prom-matrix",
|
|
"prom-matrix-with-nans",
|
|
"prom-matrix-histogram-no-labels",
|
|
"prom-matrix-histogram-partitioned",
|
|
"prom-vector-histogram-no-labels",
|
|
"prom-vector",
|
|
"prom-string",
|
|
"prom-scalar",
|
|
"prom-series",
|
|
"prom-warnings",
|
|
"prom-warnings-no-data",
|
|
"prom-infos",
|
|
"prom-infos-no-data",
|
|
"prom-error",
|
|
"prom-exemplars-a",
|
|
"prom-exemplars-b",
|
|
"prom-exemplars-diff-labels",
|
|
"prom-query-range",
|
|
"prom-query-range-big",
|
|
"loki-streams-a",
|
|
"loki-streams-b",
|
|
"loki-streams-c",
|
|
}
|
|
|
|
func TestReadPromFrames(t *testing.T) {
|
|
for _, name := range files {
|
|
t.Run(name, runScenario(name, Options{}))
|
|
}
|
|
}
|
|
|
|
func runScenario(name string, opts Options) func(t *testing.T) {
|
|
return func(t *testing.T) {
|
|
// Safe to disable, this is a test.
|
|
// nolint:gosec
|
|
f, err := os.Open(path.Join("testdata", name+".json"))
|
|
require.NoError(t, err)
|
|
|
|
iter := jsoniter.Parse(sdkjsoniter.ConfigDefault, f, 1024)
|
|
rsp := ReadPrometheusStyleResult(iter, opts)
|
|
|
|
if strings.Contains(name, "error") {
|
|
require.Error(t, rsp.Error)
|
|
return
|
|
}
|
|
|
|
if strings.Contains(name, "warnings") {
|
|
hasWarning := false
|
|
for _, frame := range rsp.Frames {
|
|
for _, notice := range frame.Meta.Notices {
|
|
if notice.Severity == data.NoticeSeverityWarning {
|
|
hasWarning = true
|
|
break
|
|
}
|
|
}
|
|
if hasWarning {
|
|
break
|
|
}
|
|
}
|
|
|
|
require.True(t, hasWarning)
|
|
}
|
|
|
|
if strings.Contains(name, "infos") {
|
|
hasInfo := false
|
|
for _, frame := range rsp.Frames {
|
|
for _, notice := range frame.Meta.Notices {
|
|
if notice.Severity == data.NoticeSeverityInfo {
|
|
hasInfo = true
|
|
break
|
|
}
|
|
}
|
|
if hasInfo {
|
|
break
|
|
}
|
|
}
|
|
|
|
require.True(t, hasInfo)
|
|
}
|
|
|
|
require.NoError(t, rsp.Error)
|
|
|
|
fname := name + "-frame"
|
|
experimental.CheckGoldenJSONResponse(t, "testdata", fname, &rsp, update)
|
|
}
|
|
}
|
|
|
|
func TestTimeConversions(t *testing.T) {
|
|
// include millisecond precision
|
|
assert.Equal(t,
|
|
time.Date(2020, time.September, 14, 15, 22, 25, 479000000, time.UTC),
|
|
timeFromFloat(1600096945.479))
|
|
|
|
ti, err := timeFromLokiString("1645030246277587968")
|
|
require.NoError(t, err)
|
|
// Loki date parsing
|
|
assert.Equal(t,
|
|
time.Date(2022, time.February, 16, 16, 50, 46, 277587968, time.UTC),
|
|
ti)
|
|
|
|
ti, err = timeFromLokiString("2000000000000000000")
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t,
|
|
time.Date(2033, time.May, 18, 3, 33, 20, 0, time.UTC),
|
|
ti)
|
|
}
|