alerting_alert_rule_test.go

package gapi

import (
	"encoding/json"
	"testing"
	"time"

	"github.com/gobs/pretty"
)

func TestAlertRules(t *testing.T) {
	t.Run("get alert rule succeeds", func(t *testing.T) {
		client := gapiTestTools(t, 200, getAlertRuleJSON)

		alertRule, err := client.AlertRule("123abcd")

		if err != nil {
			t.Error(err)
		}
		if alertRule.UID != "123abcd" {
			t.Errorf("incorrect UID - expected %s got %s", "123abcd", alertRule.UID)
		}
	})

	t.Run("get alert rule group succeeds", func(t *testing.T) {
		client := gapiTestTools(t, 200, getAlertRuleGroupJSON)

		group, err := client.AlertRuleGroup("project_test", "eval_group_1")

		if err != nil {
			t.Error(err)
		}
		t.Log(pretty.PrettyFormat(group))
		if group.Title != "eval_group_1" {
			t.Errorf("incorrect title - expected %s got %s", "eval_group_1", group.Title)
		}
		if group.FolderUID != "project_test" {
			t.Errorf("incorrect folderUID - expected %s got %s", "project_test", group.FolderUID)
		}
		if len(group.Rules) != 1 {
			t.Errorf("wrong number of rules, got %d", len(group.Rules))
		}
	})

	t.Run("get non-existent alert rule fails", func(t *testing.T) {
		client := gapiTestTools(t, 404, "")

		alertRule, err := client.AlertRule("does not exist")

		if err == nil {
			t.Errorf("expected error but got nil")
			t.Log(pretty.PrettyFormat(alertRule))
		}
	})

	t.Run("get non-existent rule group fails", func(t *testing.T) {
		client := gapiTestTools(t, 404, "")

		group, err := client.AlertRuleGroup("d8-gk06nz", "does not exist")

		if err == nil {
			t.Errorf("expected error but got nil")
			t.Log(pretty.PrettyFormat(group))
		}
	})

	t.Run("create alert rule succeeds", func(t *testing.T) {
		client := gapiTestTools(t, 201, writeAlertRuleJSON)
		alertRule := createAlertRule()

		uid, err := client.NewAlertRule(&alertRule)

		if err != nil {
			t.Error(err)
		}
		if uid != "123abcd" {
			t.Errorf("unexpected UID returned, got %s", uid)
		}
	})

	t.Run("set alert rule group succeeds", func(t *testing.T) {
		client := gapiTestTools(t, 200, getAlertRuleGroupJSON)
		group := createAlertRuleGroup()

		err := client.SetAlertRuleGroup(group)

		if err != nil {
			t.Error(err)
		}
	})

	t.Run("update alert rule succeeds", func(t *testing.T) {
		client := gapiTestTools(t, 200, writeAlertRuleJSON)
		alertRule := createAlertRule()
		alertRule.UID = "foobar"

		err := client.UpdateAlertRule(&alertRule)

		if err != nil {
			t.Error(err)
		}
	})

	t.Run("delete alert rule succeeds", func(t *testing.T) {
		client := gapiTestTools(t, 204, "")

		err := client.DeleteAlertRule("123abcd")

		if err != nil {
			t.Error(err)
		}
	})
}

func createAlertRuleGroup() RuleGroup {
	return RuleGroup{
		Title:     "eval_group_1",
		FolderUID: "project_test",
		Interval:  120,
		Rules:     []AlertRule{createAlertRule()},
	}
}

func createAlertRule() AlertRule {
	return AlertRule{
		Condition:    "A",
		Data:         createAlertQueries(),
		ExecErrState: ErrOK,
		FolderUID:    "project_test",
		NoDataState:  NoDataOk,
		OrgID:        1,
		RuleGroup:    "eval_group_1",
		Title:        "Always in alarm",
		ForDuration:  60 * time.Second,
	}
}

func createAlertQueries() []*AlertQuery {
	alertQueries := make([]*AlertQuery, 1)
	alertQueries[0] = &AlertQuery{
		DatasourceUID:     "-100",
		Model:             json.RawMessage(`{"datasourceUid":"-100","model":{"conditions":[{"evaluator":{"params":[0,0],"type":"gt"},"operator":{"type":"and"},"query":{"params":[]},"reducer":{"params":[],"type":"avg"},"type":"query"}],"datasource":{"type":"__expr__","uid":"__expr__"},"expression":"1 == 1","hide":false,"intervalMs":1000,"maxDataPoints":43200,"refId":"A","type":"math"},"queryType":"","refId":"A","relativeTimeRange":{"from":0,"to":0}}`),
		QueryType:         "",
		RefID:             "A",
		RelativeTimeRange: RelativeTimeRange{From: 0, To: 0},
	}
	return alertQueries
}

const writeAlertRuleJSON = `
	{
	"conditions": "A",
	"data": [{"datasourceUid":"-100","model":{"conditions":[{"evaluator":{"params":[0,0],"type":"gt"},"operator":{"type":"and"},"query":{"params":[]},"reducer":{"params":[],"type":"avg"},"type":"query"}],"datasource":{"type":"__expr__","uid":"__expr__"},"expression":"1 == 1","hide":false,"intervalMs":1000,"maxDataPoints":43200,"refId":"A","type":"math"},"queryType":"","refId":"A","relativeTimeRange":{"from":0,"to":0}}],
	"uid": "123abcd",
	"execErrState": "OK",
	"folderUID": "project_test",
	"noDataState": "OK",
	"orgId": 1,
	"ruleGroup": "eval_group_1",
	"title": "Always in alarm",
	"for": "1m"
}
`

const getAlertRuleJSON = `
	{
	"conditions": "A",
	"data": [{"datasourceUid":"-100","model":{"conditions":[{"evaluator":{"params":[0,0],"type":"gt"},"operator":{"type":"and"},"query":{"params":[]},"reducer":{"params":[],"type":"avg"},"type":"query"}],"datasource":{"type":"__expr__","uid":"__expr__"},"expression":"1 == 1","hide":false,"intervalMs":1000,"maxDataPoints":43200,"refId":"A","type":"math"},"queryType":"","refId":"A","relativeTimeRange":{"from":0,"to":0}}],
	"execErrState": "OK",
	"folderUID": "project_test",
	"noDataState": "OK",
	"orgId": 1,
	"uid": "123abcd",
	"ruleGroup": "eval_group_1",
	"title": "Always in alarm",
	"for": "1m"
}
`

const getAlertRuleGroupJSON = `
{
	"title": "eval_group_1",
	"folderUid": "project_test",
	"interval": 60,
	"rules": [
	  {
		"id": 212,
		"uid": "HW7RYci4z",
		"orgID": 1,
		"folderUID": "project_test",
		"ruleGroup": "eval_group_1",
		"title": "Always in alarm",
		"condition": "A",
		"data": [
		  {
			"refId": "A",
			"queryType": "",
			"relativeTimeRange": {
			  "from": 0,
			  "to": 0
			},
			"datasourceUid": "-100",
			"model": {
			  "datasourceUid": "-100",
			  "intervalMs": 1000,
			  "maxDataPoints": 43200,
			  "model": {
				"conditions": [
				  {
					"evaluator": {
					  "params": [
						0,
						0
					  ],
					  "type": "gt"
					},
					"operator": {
					  "type": "and"
					},
					"query": {
					  "params": []
					},
					"reducer": {
					  "params": [],
					  "type": "avg"
					},
					"type": "query"
				  }
				],
				"datasource": {
				  "type": "__expr__",
				  "uid": "__expr__"
				},
				"expression": "1 == 1",
				"hide": false,
				"intervalMs": 1000,
				"maxDataPoints": 43200,
				"refId": "A",
				"type": "math"
			  },
			  "queryType": "",
			  "refId": "A",
			  "relativeTimeRange": {
				"from": 0,
				"to": 0
			  }
			}
		  }
		],
		"updated": "2022-08-12T15:44:43-05:00",
		"noDataState": "OK",
		"execErrState": "OK",
		"for": "2m"
	  }
	]
  }`