Live: experimental HA with Redis (#34851)

pkg/services/live/managedstream/cache_memory_test.go

@@ -0,0 +1,69 @@
+package managedstream
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/grafana/grafana-plugin-sdk-go/data"
+
+	"github.com/stretchr/testify/require"
+)
+
+func testFrameCache(t *testing.T, c FrameCache) {
+	// Create new frame and update cache.
+	frame := data.NewFrame("hello")
+	frameJsonCache, err := data.FrameToJSONCache(frame)
+	require.NoError(t, err)
+
+	updated, err := c.Update(1, "test", frameJsonCache)
+	require.NoError(t, err)
+	require.True(t, updated)
+
+	// Make sure channel is active.
+	channels, err := c.GetActiveChannels(1)
+	require.NoError(t, err)
+	schema, ok := channels["test"]
+	require.True(t, ok)
+	require.NotZero(t, schema)
+
+	// Make sure the same frame does not update schema.
+	updated, err = c.Update(1, "test", frameJsonCache)
+	require.NoError(t, err)
+	require.False(t, updated)
+
+	// Now construct new frame with updated schema.
+	newFrame := data.NewFrame("hello", data.NewField("new_field", nil, []int64{}))
+	frameJsonCache, err = data.FrameToJSONCache(newFrame)
+	require.NoError(t, err)
+
+	// Make sure schema updated.
+	updated, err = c.Update(1, "test", frameJsonCache)
+	require.NoError(t, err)
+	require.True(t, updated)
+
+	// Add the same with another orgID and make sure schema updated.
+	updated, err = c.Update(2, "test", frameJsonCache)
+	require.NoError(t, err)
+	require.True(t, updated)
+
+	// Make sure that the last frame successfully saved in cache.
+	frameJSON, ok, err := c.GetFrame(1, "test")
+	require.NoError(t, err)
+	require.True(t, ok)
+
+	var f data.Frame
+	err = json.Unmarshal(frameJSON, &f)
+	require.NoError(t, err)
+	require.Equal(t, "new_field", f.Fields[0].Name)
+
+	// Make sure channel has updated schema.
+	channels, err = c.GetActiveChannels(1)
+	require.NoError(t, err)
+	require.NotEqual(t, string(channels["test"]), string(schema))
+}
+
+func TestMemoryFrameCache(t *testing.T) {
+	c := NewMemoryFrameCache()
+	require.NotNil(t, c)
+	testFrameCache(t, c)
+}