mirror of https://github.com/grafana/grafana.git
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
package ngalert
|
|
|
|
import (
|
|
"context"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/events"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/infra/tracing"
|
|
"github.com/grafana/grafana/pkg/services/folder"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/schedule"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/tests/fakes"
|
|
"github.com/grafana/grafana/pkg/util"
|
|
)
|
|
|
|
func Test_subscribeToFolderChanges(t *testing.T) {
|
|
orgID := rand.Int63()
|
|
folder := &folder.Folder{
|
|
ID: 0,
|
|
UID: util.GenerateShortUID(),
|
|
Title: "Folder" + util.GenerateShortUID(),
|
|
}
|
|
rules := models.GenerateAlertRules(5, models.AlertRuleGen(models.WithOrgID(orgID), models.WithNamespace(folder)))
|
|
|
|
bus := bus.ProvideBus(tracing.InitializeTracerForTest())
|
|
db := fakes.NewRuleStore(t)
|
|
db.Folders[orgID] = append(db.Folders[orgID], folder)
|
|
db.PutRule(context.Background(), rules...)
|
|
|
|
scheduler := &schedule.FakeScheduleService{}
|
|
scheduler.On("UpdateAlertRule", mock.Anything, mock.Anything).Return()
|
|
|
|
subscribeToFolderChanges(log.New("test"), bus, db, scheduler)
|
|
|
|
err := bus.Publish(context.Background(), &events.FolderTitleUpdated{
|
|
Timestamp: time.Now(),
|
|
Title: "Folder" + util.GenerateShortUID(),
|
|
ID: folder.ID,
|
|
UID: folder.UID,
|
|
OrgID: orgID,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
require.Eventuallyf(t, func() bool {
|
|
return len(db.GetRecordedCommands(func(cmd interface{}) (interface{}, bool) {
|
|
c, ok := cmd.(fakes.GenericRecordedQuery)
|
|
if !ok || c.Name != "IncreaseVersionForAllRulesInNamespace" {
|
|
return nil, false
|
|
}
|
|
return c, true
|
|
})) > 0
|
|
}, time.Second, 10*time.Millisecond, "expected to call db store method but nothing was called")
|
|
|
|
var calledTimes int
|
|
require.Eventuallyf(t, func() bool {
|
|
for _, call := range scheduler.Calls {
|
|
if call.Method == "UpdateAlertRule" {
|
|
calledTimes++
|
|
}
|
|
}
|
|
return calledTimes == len(rules)
|
|
}, time.Second, 10*time.Millisecond, "scheduler was expected to be called %d times but called %d", len(rules), calledTimes)
|
|
|
|
for _, rule := range rules {
|
|
scheduler.AssertCalled(t, "UpdateAlertRule", rule.GetKey(), rule.Version)
|
|
}
|
|
}
|