mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
Alerting: Repurpose rule testing endpoint to return potential alerts (#69755)
* Alerting: Repurpose rule testing endpoint to return potential alerts This feature replaces the existing no-longer in-use grafana ruler testing API endpoint /api/v1/rule/test/grafana. The new endpoint returns a list of potential alerts created by the given alert rule, including built-in + interpolated labels and annotations. The key priority of this endpoint is that it is intended to be as true as possible to what would be generated by the ruler except that the resulting alerts are not filtered to only Resolved / Firing and ready to be sent. This means that the endpoint will, among other things: - Attach static annotations and labels from the rule configuration to the alert instances. - Attach dynamic annotations from the datasource to the alert instances. - Attach built-in labels and annotations created by the Grafana Ruler (such as alertname and grafana_folder) to the alert instances. - Interpolate templated annotations / labels and accept allowed template functions.
This commit is contained in:
Matthew Jacobson
GitHub
@@ -2119,237 +2119,6 @@ func TestIntegrationEval(t *testing.T) {
|
||||
expectedStatusCode func() int
|
||||
expectedResponse func() string
|
||||
expectedMessage func() string
|
||||
}{
|
||||
{
|
||||
desc: "alerting condition",
|
||||
payload: `
|
||||
{
|
||||
"grafana_condition": {
|
||||
"condition": "A",
|
||||
"data": [
|
||||
{
|
||||
"refId": "A",
|
||||
"relativeTimeRange": {
|
||||
"from": 18000,
|
||||
"to": 10800
|
||||
},
|
||||
"datasourceUid":"__expr__",
|
||||
"model": {
|
||||
"type":"math",
|
||||
"expression":"1 < 2"
|
||||
}
|
||||
}
|
||||
],
|
||||
"now": "2021-04-11T14:38:14Z"
|
||||
}
|
||||
}
|
||||
`,
|
||||
expectedMessage: func() string { return "" },
|
||||
expectedStatusCode: func() int { return http.StatusOK },
|
||||
expectedResponse: func() string {
|
||||
return `{
|
||||
"instances": [
|
||||
{
|
||||
"schema": {
|
||||
"name": "evaluation results",
|
||||
"fields": [
|
||||
{
|
||||
"name": "State",
|
||||
"type": "string",
|
||||
"typeInfo": {
|
||||
"frame": "string"
|
||||
}
|
||||
},
|
||||
{
|
||||
"name": "Info",
|
||||
"type": "string",
|
||||
"typeInfo": {
|
||||
"frame": "string"
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"data": {
|
||||
"values": [
|
||||
[
|
||||
"Alerting"
|
||||
],
|
||||
[
|
||||
"[ var='A' labels={} value=1 ]"
|
||||
]
|
||||
]
|
||||
}
|
||||
}
|
||||
]
|
||||
}`
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "normal condition",
|
||||
payload: `
|
||||
{
|
||||
"grafana_condition": {
|
||||
"condition": "A",
|
||||
"data": [
|
||||
{
|
||||
"refId": "A",
|
||||
"relativeTimeRange": {
|
||||
"from": 18000,
|
||||
"to": 10800
|
||||
},
|
||||
"datasourceUid": "__expr__",
|
||||
"model": {
|
||||
"type":"math",
|
||||
"expression":"1 > 2"
|
||||
}
|
||||
}
|
||||
],
|
||||
"now": "2021-04-11T14:38:14Z"
|
||||
}
|
||||
}
|
||||
`,
|
||||
expectedMessage: func() string { return "" },
|
||||
expectedStatusCode: func() int { return http.StatusOK },
|
||||
expectedResponse: func() string {
|
||||
return `{
|
||||
"instances": [
|
||||
{
|
||||
"schema": {
|
||||
"name": "evaluation results",
|
||||
"fields": [
|
||||
{
|
||||
"name": "State",
|
||||
"type": "string",
|
||||
"typeInfo": {
|
||||
"frame": "string"
|
||||
}
|
||||
},
|
||||
{
|
||||
"name": "Info",
|
||||
"type": "string",
|
||||
"typeInfo": {
|
||||
"frame": "string"
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"data": {
|
||||
"values": [
|
||||
[
|
||||
"Normal"
|
||||
],
|
||||
[
|
||||
"[ var='A' labels={} value=0 ]"
|
||||
]
|
||||
]
|
||||
}
|
||||
}
|
||||
]
|
||||
}`
|
||||
},
|
||||
},
|
||||
{
|
||||
desc: "condition not found in any query or expression",
|
||||
payload: `
|
||||
{
|
||||
"grafana_condition": {
|
||||
"condition": "B",
|
||||
"data": [
|
||||
{
|
||||
"refId": "A",
|
||||
"relativeTimeRange": {
|
||||
"from": 18000,
|
||||
"to": 10800
|
||||
},
|
||||
"datasourceUid": "__expr__",
|
||||
"model": {
|
||||
"type":"math",
|
||||
"expression":"1 > 2"
|
||||
}
|
||||
}
|
||||
],
|
||||
"now": "2021-04-11T14:38:14Z"
|
||||
}
|
||||
}
|
||||
`,
|
||||
expectedStatusCode: func() int { return http.StatusBadRequest },
|
||||
expectedMessage: func() string {
|
||||
return "invalid condition: condition B does not exist, must be one of [A]"
|
||||
},
|
||||
expectedResponse: func() string { return "" },
|
||||
},
|
||||
{
|
||||
desc: "unknown query datasource",
|
||||
payload: `
|
||||
{
|
||||
"grafana_condition": {
|
||||
"condition": "A",
|
||||
"data": [
|
||||
{
|
||||
"refId": "A",
|
||||
"relativeTimeRange": {
|
||||
"from": 18000,
|
||||
"to": 10800
|
||||
},
|
||||
"datasourceUid": "unknown",
|
||||
"model": {
|
||||
}
|
||||
}
|
||||
],
|
||||
"now": "2021-04-11T14:38:14Z"
|
||||
}
|
||||
}
|
||||
`,
|
||||
expectedStatusCode: func() int {
|
||||
if setting.IsEnterprise {
|
||||
return http.StatusUnauthorized
|
||||
}
|
||||
return http.StatusBadRequest
|
||||
},
|
||||
expectedMessage: func() string {
|
||||
if setting.IsEnterprise {
|
||||
return "user is not authorized to query one or many data sources used by the rule"
|
||||
}
|
||||
return "invalid condition: failed to build query 'A': data source not found"
|
||||
},
|
||||
expectedResponse: func() string { return "" },
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range testCases {
|
||||
t.Run(tc.desc, func(t *testing.T) {
|
||||
u := fmt.Sprintf("http://grafana:password@%s/api/v1/rule/test/grafana", grafanaListedAddr)
|
||||
r := strings.NewReader(tc.payload)
|
||||
// nolint:gosec
|
||||
resp, err := http.Post(u, "application/json", r)
|
||||
require.NoError(t, err)
|
||||
t.Cleanup(func() {
|
||||
err := resp.Body.Close()
|
||||
require.NoError(t, err)
|
||||
})
|
||||
b, err := io.ReadAll(resp.Body)
|
||||
require.NoError(t, err)
|
||||
res := Response{}
|
||||
err = json.Unmarshal(b, &res)
|
||||
require.NoError(t, err)
|
||||
|
||||
assert.Equal(t, tc.expectedStatusCode(), resp.StatusCode)
|
||||
if tc.expectedResponse() != "" {
|
||||
require.JSONEq(t, tc.expectedResponse(), string(b))
|
||||
}
|
||||
if tc.expectedMessage() != "" {
|
||||
assert.Equal(t, tc.expectedMessage(), res.Message)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// test eval queries and expressions
|
||||
testCases = []struct {
|
||||
desc string
|
||||
payload string
|
||||
expectedStatusCode func() int
|
||||
expectedResponse func() string
|
||||
expectedMessage func() string
|
||||
}{
|
||||
{
|
||||
desc: "alerting condition",
|
||||
|
||||
408
pkg/tests/api/alerting/api_testing_test.go
Normal file
408
pkg/tests/api/alerting/api_testing_test.go
Normal file
@@ -0,0 +1,408 @@
|
||||
package alerting
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
alertingModels "github.com/grafana/alerting/models"
|
||||
amv2 "github.com/prometheus/alertmanager/api/v2/models"
|
||||
"github.com/prometheus/common/model"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/grafana/grafana/pkg/expr"
|
||||
"github.com/grafana/grafana/pkg/services/accesscontrol"
|
||||
"github.com/grafana/grafana/pkg/services/accesscontrol/resourcepermissions"
|
||||
"github.com/grafana/grafana/pkg/services/datasources"
|
||||
apimodels "github.com/grafana/grafana/pkg/services/ngalert/api/tooling/definitions"
|
||||
ngmodels "github.com/grafana/grafana/pkg/services/ngalert/models"
|
||||
"github.com/grafana/grafana/pkg/services/org"
|
||||
"github.com/grafana/grafana/pkg/services/user"
|
||||
"github.com/grafana/grafana/pkg/setting"
|
||||
"github.com/grafana/grafana/pkg/tests/testinfra"
|
||||
"github.com/grafana/grafana/pkg/util"
|
||||
)
|
||||
|
||||
const (
|
||||
TESTDATA_UID = "testdata"
|
||||
)
|
||||
|
||||
func TestGrafanaRuleConfig(t *testing.T) {
|
||||
dir, path := testinfra.CreateGrafDir(t, testinfra.GrafanaOpts{
|
||||
DisableLegacyAlerting: true,
|
||||
EnableUnifiedAlerting: true,
|
||||
DisableAnonymous: true,
|
||||
AppModeProduction: true,
|
||||
EnableFeatureToggles: []string{},
|
||||
EnableLog: false,
|
||||
})
|
||||
|
||||
grafanaListedAddr, env := testinfra.StartGrafanaEnv(t, dir, path)
|
||||
|
||||
userId := createUser(t, env.SQLStore, user.CreateUserCommand{
|
||||
DefaultOrgRole: string(org.RoleAdmin),
|
||||
Password: "admin",
|
||||
Login: "admin",
|
||||
})
|
||||
|
||||
apiCli := newAlertingApiClient(grafanaListedAddr, "admin", "admin")
|
||||
|
||||
dsCmd := &datasources.AddDataSourceCommand{
|
||||
Name: "TestDatasource",
|
||||
Type: "testdata",
|
||||
Access: datasources.DS_ACCESS_PROXY,
|
||||
UID: TESTDATA_UID,
|
||||
UserID: userId,
|
||||
OrgID: 1,
|
||||
}
|
||||
_, err := env.Server.HTTPServer.DataSourcesService.AddDataSource(context.Background(), dsCmd)
|
||||
require.NoError(t, err)
|
||||
|
||||
dynamicLabels := []string{"GA", "FL", "AL", "AZ"}
|
||||
dynamicLabelsJson, _ := json.Marshal(&dynamicLabels)
|
||||
testdataQueryModel := json.RawMessage(fmt.Sprintf(`{
|
||||
"refId": "A",
|
||||
"hide": false,
|
||||
"scenarioId": "usa",
|
||||
"usa": {
|
||||
"mode": "timeseries",
|
||||
"period": "1m",
|
||||
"states": %s,
|
||||
"fields": [
|
||||
"baz"
|
||||
]
|
||||
}
|
||||
}`, string(dynamicLabelsJson)))
|
||||
|
||||
genRule := func(ruleGen func() apimodels.PostableExtendedRuleNode) apimodels.PostableExtendedRuleNodeExtended {
|
||||
return apimodels.PostableExtendedRuleNodeExtended{
|
||||
Rule: ruleGen(),
|
||||
NamespaceUID: "NamespaceUID",
|
||||
NamespaceTitle: "NamespaceTitle",
|
||||
}
|
||||
}
|
||||
|
||||
t.Run("valid rule should accept request", func(t *testing.T) {
|
||||
status, body := apiCli.SubmitRuleForTesting(t, genRule(alertRuleGen()))
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
})
|
||||
|
||||
t.Run("valid rule should return alerts in response", func(t *testing.T) {
|
||||
status, body := apiCli.SubmitRuleForTesting(t, genRule(alertRuleGen()))
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 1)
|
||||
})
|
||||
|
||||
t.Run("valid rule should return static annotations", func(t *testing.T) {
|
||||
rule := genRule(testdataRule(testdataQueryModel, nil, nil))
|
||||
rule.Rule.Annotations = map[string]string{
|
||||
"foo": "bar",
|
||||
"foo2": "bar2",
|
||||
}
|
||||
status, body := apiCli.SubmitRuleForTesting(t, rule)
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 4)
|
||||
for _, alert := range result {
|
||||
require.Equal(t, "bar", alert.Annotations["foo"])
|
||||
require.Equal(t, "bar2", alert.Annotations["foo2"])
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("valid rule should return static labels", func(t *testing.T) {
|
||||
rule := genRule(testdataRule(testdataQueryModel, nil, nil))
|
||||
rule.Rule.Labels = map[string]string{
|
||||
"foo": "bar",
|
||||
"foo2": "bar2",
|
||||
}
|
||||
status, body := apiCli.SubmitRuleForTesting(t, rule)
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 4)
|
||||
for _, alert := range result {
|
||||
require.Equal(t, "bar", alert.Labels["foo"])
|
||||
require.Equal(t, "bar2", alert.Labels["foo2"])
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("valid rule should return interpolated annotations", func(t *testing.T) {
|
||||
rule := genRule(testdataRule(testdataQueryModel, nil, nil))
|
||||
rule.Rule.Annotations = map[string]string{
|
||||
"value": "{{ $value }}",
|
||||
"values.B": "{{ $values.B }}",
|
||||
"values.C": "{{ $values.C }}",
|
||||
}
|
||||
status, body := apiCli.SubmitRuleForTesting(t, rule)
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 4)
|
||||
for i, alert := range result {
|
||||
require.NotEmpty(t, alert.Annotations["values.B"])
|
||||
require.NotEmpty(t, alert.Annotations["values.C"])
|
||||
valueB := fmt.Sprintf("[ var='B' labels={state=%s} value=%s ]", dynamicLabels[i], alert.Annotations["values.B"])
|
||||
valueC := fmt.Sprintf("[ var='C' labels={state=%s} value=%s ]", dynamicLabels[i], alert.Annotations["values.C"])
|
||||
require.Contains(t, alert.Annotations["value"], valueB)
|
||||
require.Contains(t, alert.Annotations["value"], valueC)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("valid rule should return interpolated labels", func(t *testing.T) {
|
||||
rule := genRule(testdataRule(testdataQueryModel, nil, nil))
|
||||
rule.Rule.Labels = map[string]string{
|
||||
"value": "{{ $value }}",
|
||||
"values.B": "{{ $values.B }}",
|
||||
"values.C": "{{ $values.C }}",
|
||||
}
|
||||
status, body := apiCli.SubmitRuleForTesting(t, rule)
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 4)
|
||||
for i, alert := range result {
|
||||
require.NotEmpty(t, alert.Labels["values.B"])
|
||||
require.NotEmpty(t, alert.Labels["values.C"])
|
||||
valueB := fmt.Sprintf("[ var='B' labels={state=%s} value=%s ]", dynamicLabels[i], alert.Labels["values.B"])
|
||||
valueC := fmt.Sprintf("[ var='C' labels={state=%s} value=%s ]", dynamicLabels[i], alert.Labels["values.C"])
|
||||
require.Contains(t, alert.Labels["value"], valueB)
|
||||
require.Contains(t, alert.Labels["value"], valueC)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("valid rule should use functions with annotations", func(t *testing.T) {
|
||||
rule := genRule(testdataRule(testdataQueryModel, nil, nil))
|
||||
rule.Rule.Annotations = map[string]string{
|
||||
"externalURL": "{{ externalURL }}",
|
||||
"humanize": "{{ humanize 1000.0 }}",
|
||||
}
|
||||
status, body := apiCli.SubmitRuleForTesting(t, rule)
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 4)
|
||||
for _, alert := range result {
|
||||
require.Equal(t, "http://localhost:3000/", alert.Annotations["externalURL"])
|
||||
require.Equal(t, "1k", alert.Annotations["humanize"])
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("valid rule should use functions with labels", func(t *testing.T) {
|
||||
rule := genRule(testdataRule(testdataQueryModel, nil, nil))
|
||||
rule.Rule.Labels = map[string]string{
|
||||
"externalURL": "{{ externalURL }}",
|
||||
"humanize": "{{ humanize 1000.0 }}",
|
||||
}
|
||||
status, body := apiCli.SubmitRuleForTesting(t, rule)
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 4)
|
||||
for _, alert := range result {
|
||||
require.Equal(t, "http://localhost:3000/", alert.Labels["externalURL"])
|
||||
require.Equal(t, "1k", alert.Labels["humanize"])
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("valid rule should return dynamic labels", func(t *testing.T) {
|
||||
rule := genRule(testdataRule(testdataQueryModel, nil, nil))
|
||||
status, body := apiCli.SubmitRuleForTesting(t, rule)
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 4)
|
||||
for i, alert := range result {
|
||||
require.Equal(t, dynamicLabels[i], alert.Labels["state"])
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("valid rule should return built-in labels", func(t *testing.T) {
|
||||
rule := genRule(testdataRule(testdataQueryModel, nil, nil))
|
||||
status, body := apiCli.SubmitRuleForTesting(t, rule)
|
||||
require.Equal(t, http.StatusOK, status)
|
||||
var result []amv2.PostableAlert
|
||||
require.NoErrorf(t, json.Unmarshal([]byte(body), &result), "cannot parse response to data frame")
|
||||
require.Len(t, result, 4)
|
||||
for _, alert := range result {
|
||||
require.Equal(t, rule.Rule.GrafanaManagedAlert.Title, alert.Labels[model.AlertNameLabel])
|
||||
require.Equal(t, rule.NamespaceUID, alert.Labels[alertingModels.NamespaceUIDLabel])
|
||||
require.Equal(t, rule.NamespaceTitle, alert.Labels[ngmodels.FolderTitleLabel])
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("invalid rule should reject request", func(t *testing.T) {
|
||||
req := genRule(alertRuleGen())
|
||||
req.Rule = apimodels.PostableExtendedRuleNode{}
|
||||
status, _ := apiCli.SubmitRuleForTesting(t, req)
|
||||
require.Equal(t, http.StatusBadRequest, status)
|
||||
})
|
||||
|
||||
t.Run("authentication permissions", func(t *testing.T) {
|
||||
if !setting.IsEnterprise {
|
||||
t.Skip("Enterprise-only test")
|
||||
}
|
||||
|
||||
testUserId := createUser(t, env.SQLStore, user.CreateUserCommand{
|
||||
DefaultOrgRole: "DOESNOTEXIST", // Needed so that the SignedInUser has OrgId=1. Otherwise, datasource will not be found.
|
||||
Password: "test",
|
||||
Login: "test",
|
||||
})
|
||||
|
||||
testUserApiCli := newAlertingApiClient(grafanaListedAddr, "test", "test")
|
||||
|
||||
t.Run("fail if can't read rules", func(t *testing.T) {
|
||||
status, body := testUserApiCli.SubmitRuleForTesting(t, genRule(testdataRule(testdataQueryModel, nil, nil)))
|
||||
require.Contains(t, body, accesscontrol.ActionAlertingRuleRead)
|
||||
require.Equalf(t, http.StatusForbidden, status, "Response: %s", body)
|
||||
})
|
||||
|
||||
// access control permissions store
|
||||
permissionsStore := resourcepermissions.NewStore(env.SQLStore)
|
||||
_, err := permissionsStore.SetUserResourcePermission(context.Background(),
|
||||
accesscontrol.GlobalOrgID,
|
||||
accesscontrol.User{ID: testUserId},
|
||||
resourcepermissions.SetResourcePermissionCommand{
|
||||
Actions: []string{
|
||||
accesscontrol.ActionAlertingRuleRead,
|
||||
},
|
||||
Resource: "folders",
|
||||
ResourceID: "*",
|
||||
ResourceAttribute: "uid",
|
||||
}, nil)
|
||||
require.NoError(t, err)
|
||||
testUserApiCli.ReloadCachedPermissions(t)
|
||||
|
||||
t.Run("fail if can't query data sources", func(t *testing.T) {
|
||||
status, body := testUserApiCli.SubmitRuleForTesting(t, genRule(testdataRule(testdataQueryModel, nil, nil)))
|
||||
require.Contains(t, body, "user is not authorized to query one or many data sources used by the rule")
|
||||
require.Equalf(t, http.StatusUnauthorized, status, "Response: %s", body)
|
||||
})
|
||||
|
||||
_, err = permissionsStore.SetUserResourcePermission(context.Background(),
|
||||
accesscontrol.GlobalOrgID,
|
||||
accesscontrol.User{ID: testUserId},
|
||||
resourcepermissions.SetResourcePermissionCommand{
|
||||
Actions: []string{
|
||||
datasources.ActionQuery,
|
||||
},
|
||||
Resource: "datasources",
|
||||
ResourceID: TESTDATA_UID,
|
||||
ResourceAttribute: "uid",
|
||||
}, nil)
|
||||
require.NoError(t, err)
|
||||
testUserApiCli.ReloadCachedPermissions(t)
|
||||
|
||||
t.Run("succeed if can query data sources", func(t *testing.T) {
|
||||
status, body := testUserApiCli.SubmitRuleForTesting(t, genRule(testdataRule(testdataQueryModel, nil, nil)))
|
||||
require.Equalf(t, http.StatusOK, status, "Response: %s", body)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func testdataRule(queryModel json.RawMessage, labels map[string]string, annotations map[string]string) func() apimodels.PostableExtendedRuleNode {
|
||||
return func() apimodels.PostableExtendedRuleNode {
|
||||
forDuration := model.Duration(10 * time.Second)
|
||||
return apimodels.PostableExtendedRuleNode{
|
||||
ApiRuleNode: &apimodels.ApiRuleNode{
|
||||
For: &forDuration,
|
||||
Labels: labels,
|
||||
Annotations: annotations,
|
||||
},
|
||||
GrafanaManagedAlert: &apimodels.PostableGrafanaRule{
|
||||
Title: fmt.Sprintf("rule-%s", util.GenerateShortUID()),
|
||||
Condition: "C",
|
||||
Data: []apimodels.AlertQuery{
|
||||
{
|
||||
RefID: "A",
|
||||
RelativeTimeRange: apimodels.RelativeTimeRange{From: 600, To: 0},
|
||||
DatasourceUID: TESTDATA_UID,
|
||||
Model: queryModel,
|
||||
},
|
||||
{ // Simple reduce last A.
|
||||
RefID: "B",
|
||||
RelativeTimeRange: apimodels.RelativeTimeRange{From: 0, To: 0},
|
||||
DatasourceUID: expr.DatasourceUID,
|
||||
Model: json.RawMessage(`{
|
||||
"refId": "B",
|
||||
"hide": false,
|
||||
"type": "reduce",
|
||||
"datasource": {
|
||||
"uid": "__expr__",
|
||||
"type": "__expr__"
|
||||
},
|
||||
"conditions": [
|
||||
{
|
||||
"type": "query",
|
||||
"evaluator": {
|
||||
"params": [],
|
||||
"type": "gt"
|
||||
},
|
||||
"operator": {
|
||||
"type": "and"
|
||||
},
|
||||
"query": {
|
||||
"params": [
|
||||
"B"
|
||||
]
|
||||
},
|
||||
"reducer": {
|
||||
"params": [],
|
||||
"type": "last"
|
||||
}
|
||||
}
|
||||
],
|
||||
"reducer": "last",
|
||||
"expression": "A"
|
||||
}`),
|
||||
},
|
||||
{ // Threshold B > 0.
|
||||
RefID: "C",
|
||||
RelativeTimeRange: apimodels.RelativeTimeRange{From: 0, To: 0},
|
||||
DatasourceUID: expr.DatasourceUID,
|
||||
Model: json.RawMessage(`{
|
||||
"refId": "C",
|
||||
"hide": false,
|
||||
"type": "threshold",
|
||||
"datasource": {
|
||||
"uid": "__expr__",
|
||||
"type": "__expr__"
|
||||
},
|
||||
"conditions": [
|
||||
{
|
||||
"type": "query",
|
||||
"evaluator": {
|
||||
"params": [
|
||||
0
|
||||
],
|
||||
"type": "gt"
|
||||
},
|
||||
"operator": {
|
||||
"type": "and"
|
||||
},
|
||||
"query": {
|
||||
"params": [
|
||||
"C"
|
||||
]
|
||||
},
|
||||
"reducer": {
|
||||
"params": [],
|
||||
"type": "last"
|
||||
}
|
||||
}
|
||||
],
|
||||
"expression": "B"
|
||||
}`),
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -334,3 +334,22 @@ func (a apiClient) SubmitRuleForBacktesting(t *testing.T, config apimodels.Backt
|
||||
require.NoError(t, err)
|
||||
return resp.StatusCode, string(b)
|
||||
}
|
||||
|
||||
func (a apiClient) SubmitRuleForTesting(t *testing.T, config apimodels.PostableExtendedRuleNodeExtended) (int, string) {
|
||||
t.Helper()
|
||||
buf := bytes.Buffer{}
|
||||
enc := json.NewEncoder(&buf)
|
||||
err := enc.Encode(config)
|
||||
require.NoError(t, err)
|
||||
|
||||
u := fmt.Sprintf("%s/api/v1/rule/test/grafana", a.url)
|
||||
// nolint:gosec
|
||||
resp, err := http.Post(u, "application/json", &buf)
|
||||
require.NoError(t, err)
|
||||
defer func() {
|
||||
_ = resp.Body.Close()
|
||||
}()
|
||||
b, err := io.ReadAll(resp.Body)
|
||||
require.NoError(t, err)
|
||||
return resp.StatusCode, string(b)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user