grafana/pkg/services/live/pipeline/rule_cache_segmented_test.go

package pipeline

import (
	"context"
	"testing"

	"github.com/stretchr/testify/require"
)

type testBuilder struct{}

func (t *testBuilder) BuildRules(_ context.Context, _ int64) ([]*LiveChannelRule, error) {
	return []*LiveChannelRule{
		{
			OrgId:   1,
			Pattern: "stream/telegraf/cpu",
		},
		{
			OrgId:   1,
			Pattern: "stream/telegraf/:metric",
		},
		{
			OrgId:   1,
			Pattern: "stream/telegraf/:metric/:extra",
		},
		{
			OrgId:   1,
			Pattern: "stream/boom:er",
		},
	}, nil
}

func TestStorage_Get(t *testing.T) {
	s := NewCacheSegmentedTree(&testBuilder{})
	rule, ok, err := s.Get(1, "stream/telegraf/cpu")
	require.NoError(t, err)
	require.True(t, ok)
	require.Equal(t, "stream/telegraf/cpu", rule.Pattern)

	rule, ok, err = s.Get(1, "stream/telegraf/mem")
	require.NoError(t, err)
	require.True(t, ok)
	require.Equal(t, "stream/telegraf/:metric", rule.Pattern)

	rule, ok, err = s.Get(1, "stream/telegraf/mem/rss")
	require.NoError(t, err)
	require.True(t, ok)
	require.Equal(t, "stream/telegraf/:metric/:extra", rule.Pattern)

	rule, ok, err = s.Get(1, "stream/booms")
	require.NoError(t, err)
	require.True(t, ok)
	require.Equal(t, "stream/boom:er", rule.Pattern)
}

func BenchmarkRuleGet(b *testing.B) {
	s := NewCacheSegmentedTree(&testBuilder{})
	for i := 0; i < b.N; i++ {
		_, ok, err := s.Get(1, "stream/telegraf/cpu")
		if err != nil || !ok {
			b.Fatal("unexpected return values")
		}
	}
}