mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
3cac10e598
* Always use cache: stop passing skipCache among ngalert functions * Add updated column * Scheduler initial draft * Add retry on failure * Allow settting/updating alert definition interval Set default interval if no interval is provided during alert definition creation. Keep existing alert definition interval if no interval is provided during alert definition update. * Parameterise alerting.Ticker to run on custom interval * Allow updating alert definition interval without having to provide the queries and expressions * Add schedule tests * Use xorm tags for having initialisms with consistent case in Go * Add ability to pause/unpause the scheduler * Add alert definition versioning * Optimise scheduler to fetch alert definition only when it's necessary * Change MySQL data column to mediumtext * Delete alert definition versions * Increase default scheduler interval to 10 seconds * Fix setting OrgID on updates * Add validation for alert definition name length * Recreate tables
86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
package ngalert
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/api/routing"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/registry"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/eval"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func setupTestEnv(t *testing.T) *AlertNG {
|
|
cfg := setting.NewCfg()
|
|
cfg.FeatureToggles = map[string]bool{"ngalert": true}
|
|
|
|
ng := overrideAlertNGInRegistry(cfg)
|
|
|
|
sqlStore := sqlstore.InitTestDB(t)
|
|
ng.SQLStore = sqlStore
|
|
|
|
err := ng.Init()
|
|
require.NoError(t, err)
|
|
return &ng
|
|
}
|
|
|
|
func overrideAlertNGInRegistry(cfg *setting.Cfg) AlertNG {
|
|
ng := AlertNG{
|
|
SQLStore: nil,
|
|
Cfg: cfg,
|
|
RouteRegister: routing.NewRouteRegister(),
|
|
log: log.New("ngalert-test"),
|
|
}
|
|
|
|
overrideServiceFunc := func(descriptor registry.Descriptor) (*registry.Descriptor, bool) {
|
|
if _, ok := descriptor.Instance.(*AlertNG); ok {
|
|
return ®istry.Descriptor{
|
|
Name: "AlertNG",
|
|
Instance: &ng,
|
|
InitPriority: descriptor.InitPriority,
|
|
}, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
registry.RegisterOverride(overrideServiceFunc)
|
|
|
|
return ng
|
|
}
|
|
|
|
func createTestAlertDefinition(t *testing.T, ng *AlertNG, intervalSeconds int64) *AlertDefinition {
|
|
cmd := saveAlertDefinitionCommand{
|
|
OrgID: 1,
|
|
Title: fmt.Sprintf("an alert definition %d", rand.Intn(1000)),
|
|
Condition: eval.Condition{
|
|
RefID: "A",
|
|
QueriesAndExpressions: []eval.AlertQuery{
|
|
{
|
|
Model: json.RawMessage(`{
|
|
"datasource": "__expr__",
|
|
"type":"math",
|
|
"expression":"2 + 2 > 1"
|
|
}`),
|
|
RelativeTimeRange: eval.RelativeTimeRange{
|
|
From: eval.Duration(5 * time.Hour),
|
|
To: eval.Duration(3 * time.Hour),
|
|
},
|
|
RefID: "A",
|
|
},
|
|
},
|
|
},
|
|
IntervalSeconds: &intervalSeconds,
|
|
}
|
|
err := ng.saveAlertDefinition(&cmd)
|
|
require.NoError(t, err)
|
|
t.Logf("alert definition: %d with interval: %d created", cmd.Result.ID, intervalSeconds)
|
|
return cmd.Result
|
|
}
|