mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
790e1feb93
* streamline initialization of test databases, support on-disk sqlite test db * clean up test databases * introduce testsuite helper * use testsuite everywhere we use a test db * update documentation * improve error handling * disable entity integration test until we can figure out locking error
68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package tests
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/services/live/model"
|
|
"github.com/grafana/grafana/pkg/tests/testsuite"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
testsuite.Run(m)
|
|
}
|
|
|
|
func TestIntegrationLiveMessage(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping integration test")
|
|
}
|
|
storage := SetupTestStorage(t)
|
|
|
|
getQuery := &model.GetLiveMessageQuery{
|
|
OrgID: 1,
|
|
Channel: "test_channel",
|
|
}
|
|
_, ok, err := storage.GetLiveMessage(getQuery)
|
|
require.NoError(t, err)
|
|
require.False(t, ok)
|
|
|
|
saveQuery := &model.SaveLiveMessageQuery{
|
|
OrgID: 1,
|
|
Channel: "test_channel",
|
|
Data: []byte(`{}`),
|
|
}
|
|
err = storage.SaveLiveMessage(saveQuery)
|
|
require.NoError(t, err)
|
|
|
|
msg, ok, err := storage.GetLiveMessage(getQuery)
|
|
require.NoError(t, err)
|
|
require.True(t, ok)
|
|
require.Equal(t, int64(1), msg.OrgID)
|
|
require.Equal(t, "test_channel", msg.Channel)
|
|
require.Equal(t, json.RawMessage(`{}`), msg.Data)
|
|
require.NotZero(t, msg.Published)
|
|
|
|
// try saving again, should be replaced.
|
|
saveQuery2 := &model.SaveLiveMessageQuery{
|
|
OrgID: 1,
|
|
Channel: "test_channel",
|
|
Data: []byte(`{"input": "hello"}`),
|
|
}
|
|
err = storage.SaveLiveMessage(saveQuery2)
|
|
require.NoError(t, err)
|
|
|
|
getQuery2 := &model.GetLiveMessageQuery{
|
|
OrgID: 1,
|
|
Channel: "test_channel",
|
|
}
|
|
msg2, ok, err := storage.GetLiveMessage(getQuery2)
|
|
require.NoError(t, err)
|
|
require.True(t, ok)
|
|
require.Equal(t, int64(1), msg2.OrgID)
|
|
require.Equal(t, "test_channel", msg2.Channel)
|
|
require.Equal(t, json.RawMessage(`{"input": "hello"}`), msg2.Data)
|
|
require.NotZero(t, msg2.Published)
|
|
}
|