mirror of
https://github.com/grafana/grafana.git
synced 2025-02-20 11:48:34 -06:00
85 lines
2.4 KiB
Go
85 lines
2.4 KiB
Go
package managedstream
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type testPublisher struct {
|
|
t *testing.T
|
|
}
|
|
|
|
func (p *testPublisher) publish(_ int64, _ string, _ []byte) error {
|
|
return nil
|
|
}
|
|
|
|
func TestNewManagedStream(t *testing.T) {
|
|
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.
|
|
}
|