mirror of
https://github.com/grafana/grafana.git
synced 2024-11-30 12:44:10 -06:00
88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package features
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/services/live/model"
|
|
"github.com/grafana/grafana/pkg/services/user"
|
|
)
|
|
|
|
func TestNewBroadcastRunner(t *testing.T) {
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
d := NewMockLiveMessageStore(mockCtrl)
|
|
br := NewBroadcastRunner(d)
|
|
require.NotNil(t, br)
|
|
}
|
|
|
|
func TestBroadcastRunner_OnSubscribe(t *testing.T) {
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
mockDispatcher := NewMockLiveMessageStore(mockCtrl)
|
|
|
|
channel := "stream/channel/test"
|
|
data := json.RawMessage(`{}`)
|
|
|
|
mockDispatcher.EXPECT().GetLiveMessage(&model.GetLiveMessageQuery{
|
|
OrgID: 1,
|
|
Channel: channel,
|
|
}).DoAndReturn(func(query *model.GetLiveMessageQuery) (model.LiveMessage, bool, error) {
|
|
return model.LiveMessage{
|
|
Data: data,
|
|
}, true, nil
|
|
}).Times(1)
|
|
|
|
br := NewBroadcastRunner(mockDispatcher)
|
|
require.NotNil(t, br)
|
|
handler, err := br.GetHandlerForPath("test")
|
|
require.NoError(t, err)
|
|
reply, status, err := handler.OnSubscribe(
|
|
context.Background(),
|
|
&user.SignedInUser{OrgID: 1, UserID: 2},
|
|
model.SubscribeEvent{Channel: channel, Path: "test"},
|
|
)
|
|
require.NoError(t, err)
|
|
require.Equal(t, backend.SubscribeStreamStatusOK, status)
|
|
require.Equal(t, data, reply.Data)
|
|
require.True(t, reply.Presence)
|
|
require.True(t, reply.JoinLeave)
|
|
require.False(t, reply.Recover)
|
|
}
|
|
|
|
func TestBroadcastRunner_OnPublish(t *testing.T) {
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
mockDispatcher := NewMockLiveMessageStore(mockCtrl)
|
|
|
|
channel := "stream/channel/test"
|
|
data := json.RawMessage(`{}`)
|
|
var orgID int64 = 1
|
|
|
|
mockDispatcher.EXPECT().SaveLiveMessage(&model.SaveLiveMessageQuery{
|
|
OrgID: orgID,
|
|
Channel: channel,
|
|
Data: data,
|
|
}).DoAndReturn(func(query *model.SaveLiveMessageQuery) error {
|
|
return nil
|
|
}).Times(1)
|
|
|
|
br := NewBroadcastRunner(mockDispatcher)
|
|
require.NotNil(t, br)
|
|
handler, err := br.GetHandlerForPath("test")
|
|
require.NoError(t, err)
|
|
reply, status, err := handler.OnPublish(
|
|
context.Background(),
|
|
&user.SignedInUser{OrgID: 1, UserID: 2},
|
|
model.PublishEvent{Channel: channel, Path: "test", Data: data},
|
|
)
|
|
require.NoError(t, err)
|
|
require.Equal(t, backend.PublishStreamStatusOK, status)
|
|
require.Equal(t, data, reply.Data)
|
|
}
|