The open and composable observability and data visualization platform. Visualize metrics, logs, and traces from multiple sources like Prometheus, Loki, Elasticsearch, InfluxDB, Postgres and many more.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
grafana/pkg/services/live/pipeline/frame_output_changelog_test.go

90 lines
2.6 KiB

package pipeline
import (
"context"
"testing"
"time"
"github.com/golang/mock/gomock"
"github.com/grafana/grafana-plugin-sdk-go/data"
"github.com/stretchr/testify/require"
)
func TestChangeLogOutput_NoPreviousFrame_SingleRow(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
mockStorage := NewMockFrameGetSetter(mockCtrl)
mockStorage.EXPECT().Get(gomock.Any(), gomock.Any()).DoAndReturn(func(orgID int64, channel string) (*data.Frame, bool, error) {
return nil, false, nil
})
mockStorage.EXPECT().Set(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
outputter := NewChangeLogFrameOutput(mockStorage, ChangeLogOutputConfig{
FieldName: "test",
Channel: "stream/test/no_previous_frame",
})
f1 := data.NewField("time", nil, make([]time.Time, 1))
f1.Set(0, time.Now())
f2 := data.NewField("test", nil, make([]*float64, 1))
f2.SetConcrete(0, 20.0)
frame := data.NewFrame("test", f1, f2)
channelFrames, err := outputter.OutputFrame(context.Background(), Vars{}, frame)
require.NoError(t, err)
require.Len(t, channelFrames, 1)
changeFrame := channelFrames[0].Frame
require.Len(t, changeFrame.Fields, 3)
var x *float64
var y = 20.0
require.Equal(t, x, changeFrame.Fields[1].At(0).(*float64))
require.Equal(t, &y, changeFrame.Fields[2].At(0))
}
func TestChangeLogOutput_NoPreviousFrame_MultipleRows(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
mockStorage := NewMockFrameGetSetter(mockCtrl)
mockStorage.EXPECT().Get(gomock.Any(), gomock.Any()).DoAndReturn(func(orgID int64, channel string) (*data.Frame, bool, error) {
return nil, false, nil
}).Times(1)
mockStorage.EXPECT().Set(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
outputter := NewChangeLogFrameOutput(mockStorage, ChangeLogOutputConfig{
FieldName: "test",
Channel: "stream/test/no_previous_frame",
})
f1 := data.NewField("time", nil, make([]time.Time, 2))
f1.Set(0, time.Now())
f1.Set(1, time.Now())
f2 := data.NewField("test", nil, make([]*float64, 2))
f2.SetConcrete(0, 5.0)
f2.SetConcrete(1, 20.0)
frame := data.NewFrame("test", f1, f2)
channelFrames, err := outputter.OutputFrame(context.Background(), Vars{}, frame)
require.NoError(t, err)
require.Len(t, channelFrames, 1)
changeFrame := channelFrames[0].Frame
require.Len(t, changeFrame.Fields, 3)
var x *float64
var y = 5.0
require.Equal(t, x, changeFrame.Fields[1].At(0).(*float64))
require.Equal(t, &y, changeFrame.Fields[2].At(0))
var z = 5.0
var v = 20.0
require.Equal(t, &z, changeFrame.Fields[1].At(1).(*float64))
require.Equal(t, &v, changeFrame.Fields[2].At(1))
}