Live: display stream rate, fix duplicate channels in list response (#37365)

pkg/services/live/managedstream/runner_test.go

@@ -2,22 +2,83 @@ package managedstream
 
 import (
 	"testing"
+	"time"
 
+	"github.com/grafana/grafana-plugin-sdk-go/data"
 	"github.com/stretchr/testify/require"
 )
 
 type testPublisher struct {
-	orgID int64
-	t     *testing.T
+	t *testing.T
 }
 
-func (p *testPublisher) publish(orgID int64, _ string, _ []byte) error {
-	require.Equal(p.t, p.orgID, orgID)
+func (p *testPublisher) publish(_ int64, _ string, _ []byte) error {
 	return nil
 }
 
 func TestNewManagedStream(t *testing.T) {
-	publisher := &testPublisher{orgID: 1, t: t}
-	c := NewManagedStream("a", publisher.publish, NewMemoryFrameCache())
+	publisher := &testPublisher{t: t}
+	c := NewManagedStream("a", 1, publisher.publish, NewMemoryFrameCache())
 	require.NotNil(t, c)
 }
+
+func TestManagedStreamMinuteRate(t *testing.T) {
+	publisher := &testPublisher{t: t}
+	c := NewManagedStream("a", 1, publisher.publish, NewMemoryFrameCache())
+	require.NotNil(t, c)
+
+	c.incRate("test1", time.Now().Unix())
+	require.Equal(t, int64(1), c.minuteRate("test1"))
+	require.Equal(t, int64(0), c.minuteRate("test2"))
+	c.incRate("test1", time.Now().Unix())
+	require.Equal(t, int64(2), c.minuteRate("test1"))
+
+	nowUnix := time.Now().Unix()
+	for i := 0; i < 1000; i++ {
+		unixTime := nowUnix + int64(i)
+		c.incRate("test3", unixTime)
+	}
+	require.Equal(t, int64(60), c.minuteRate("test3"))
+
+	c.incRate("test3", nowUnix+999)
+	require.Equal(t, int64(61), c.minuteRate("test3"))
+}
+
+func TestGetManagedStreams(t *testing.T) {
+	publisher := &testPublisher{t: t}
+	frameCache := NewMemoryFrameCache()
+	runner := NewRunner(publisher.publish, frameCache)
+	s1, err := runner.GetOrCreateStream(1, "test1")
+	require.NoError(t, err)
+	s2, err := runner.GetOrCreateStream(1, "test2")
+	require.NoError(t, err)
+
+	managedChannels, err := runner.GetManagedChannels(1)
+	require.NoError(t, err)
+	require.Len(t, managedChannels, 3) // 3 hardcoded testdata streams.
+
+	err = s1.Push("cpu1", data.NewFrame("cpu1"))
+	require.NoError(t, err)
+
+	err = s1.Push("cpu2", data.NewFrame("cpu2"))
+	require.NoError(t, err)
+
+	err = s2.Push("cpu1", data.NewFrame("cpu1"))
+	require.NoError(t, err)
+
+	managedChannels, err = runner.GetManagedChannels(1)
+	require.NoError(t, err)
+	require.Len(t, managedChannels, 6) // 3 hardcoded testdata streams + 3 test channels.
+	require.Equal(t, "stream/test1/cpu1", managedChannels[3].Channel)
+	require.Equal(t, "stream/test1/cpu2", managedChannels[4].Channel)
+	require.Equal(t, "stream/test2/cpu1", managedChannels[5].Channel)
+
+	// Different org.
+	s3, err := runner.GetOrCreateStream(2, "test1")
+	require.NoError(t, err)
+	err = s3.Push("cpu1", data.NewFrame("cpu1"))
+	require.NoError(t, err)
+	managedChannels, err = runner.GetManagedChannels(1)
+	require.NoError(t, err)
+	require.Len(t, managedChannels, 6) // Not affected by other org.
+}