mirror of
https://github.com/grafana/grafana.git
synced 2024-11-27 03:11:01 -06:00
43b15b92ad
* propagate notificationservice down to the notifiers * replace dispatch in result handler * remove dispatch from the rule reader * remove dispatch from eval context * remove dispatch from alerting usage * remove dispatch from alerting usage * remove dispatch from notifier * attempt to fix tests in alerting * hello linter, my old friend; also disable some tests for now * use mocks to fix the tests * resolving wire providers * make linter happy * remove yet another bus.dispatch * fix tests using store mock
129 lines
3.1 KiB
Go
129 lines
3.1 KiB
Go
package alerting
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAlertingUsageStats(t *testing.T) {
|
|
store := &AlertStoreMock{}
|
|
ae := &AlertEngine{
|
|
sqlStore: store,
|
|
}
|
|
|
|
store.getAllAlerts = func(ctx context.Context, query *models.GetAllAlertsQuery) error {
|
|
var createFake = func(file string) *simplejson.Json {
|
|
// Ignore gosec warning G304 since it's a test
|
|
// nolint:gosec
|
|
content, err := ioutil.ReadFile(file)
|
|
require.NoError(t, err, "expected to be able to read file")
|
|
|
|
j, err := simplejson.NewJson(content)
|
|
require.NoError(t, err)
|
|
return j
|
|
}
|
|
|
|
query.Result = []*models.Alert{
|
|
{Id: 1, Settings: createFake("testdata/settings/one_condition.json")},
|
|
{Id: 2, Settings: createFake("testdata/settings/two_conditions.json")},
|
|
{Id: 2, Settings: createFake("testdata/settings/three_conditions.json")},
|
|
{Id: 3, Settings: createFake("testdata/settings/empty.json")},
|
|
}
|
|
return nil
|
|
}
|
|
|
|
store.getDataSource = func(ctx context.Context, query *models.GetDataSourceQuery) error {
|
|
ds := map[int64]*models.DataSource{
|
|
1: {Type: "influxdb"},
|
|
2: {Type: "graphite"},
|
|
3: {Type: "prometheus"},
|
|
4: {Type: "prometheus"},
|
|
}
|
|
|
|
r, exist := ds[query.Id]
|
|
if !exist {
|
|
return models.ErrDataSourceNotFound
|
|
}
|
|
|
|
query.Result = r
|
|
return nil
|
|
}
|
|
|
|
result, err := ae.QueryUsageStats(context.Background())
|
|
require.NoError(t, err, "getAlertingUsage should not return error")
|
|
|
|
expected := map[string]int{
|
|
"prometheus": 4,
|
|
"graphite": 2,
|
|
}
|
|
|
|
for k := range expected {
|
|
if expected[k] != result.DatasourceUsage[k] {
|
|
t.Errorf("result mismatch for %s. got %v expected %v", k, result.DatasourceUsage[k], expected[k])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParsingAlertRuleSettings(t *testing.T) {
|
|
tcs := []struct {
|
|
name string
|
|
file string
|
|
expected []int64
|
|
shouldErr require.ErrorAssertionFunc
|
|
}{
|
|
{
|
|
name: "can parse single condition",
|
|
file: "testdata/settings/one_condition.json",
|
|
expected: []int64{3},
|
|
shouldErr: require.NoError,
|
|
},
|
|
{
|
|
name: "can parse multiple conditions",
|
|
file: "testdata/settings/two_conditions.json",
|
|
expected: []int64{3, 2},
|
|
shouldErr: require.NoError,
|
|
},
|
|
{
|
|
name: "can parse empty json",
|
|
file: "testdata/settings/empty.json",
|
|
expected: []int64{},
|
|
shouldErr: require.NoError,
|
|
},
|
|
{
|
|
name: "can handle nil content",
|
|
expected: []int64{},
|
|
shouldErr: require.NoError,
|
|
},
|
|
}
|
|
|
|
ae := &AlertEngine{}
|
|
|
|
for _, tc := range tcs {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
var settings json.Marshaler
|
|
if tc.file != "" {
|
|
content, err := ioutil.ReadFile(tc.file)
|
|
require.NoError(t, err, "expected to be able to read file")
|
|
|
|
settings, err = simplejson.NewJson(content)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
result, err := ae.parseAlertRuleModel(settings)
|
|
|
|
tc.shouldErr(t, err)
|
|
diff := cmp.Diff(tc.expected, result)
|
|
if diff != "" {
|
|
t.Errorf("result mismatch (-want +got) %s\n", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|