Merge pull request #103 from grafana/alexweav/alerting-write-rule

Alerting: Add method for PUTting a whole rule group

Author: alexweav

alerting_alert_rule.go

@@ -80,6 +80,19 @@ func (c *Client) AlertRuleGroup(folderUID string, name string) (RuleGroup, error
 	return result, err
 }
 
+// SetAlertRuleGroup overwrites an existing rule group on the server.
+func (c *Client) SetAlertRuleGroup(group RuleGroup) error {
+	folderUID := group.FolderUID
+	name := group.Title
+	req, err := json.Marshal(group)
+	if err != nil {
+		return err
+	}
+
+	uri := fmt.Sprintf("/api/v1/provisioning/folder/%s/rule-groups/%s", folderUID, name)
+	return c.request("PUT", uri, nil, bytes.NewBuffer(req), nil)
+}
+
 // NewAlertRule creates a new alert rule and returns its UID.
 func (c *Client) NewAlertRule(ar *AlertRule) (string, error) {
 	req, err := json.Marshal(ar)

alerting_alert_rule_test.go

@@ -55,10 +55,23 @@ func TestAlertRules(t *testing.T) {
 		}
 	})
 
+	t.Run("get non-existent rule group fails", func(t *testing.T) {
+		server, client := gapiTestTools(t, 404, "")
+		defer server.Close()
+
+		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) {
 		server, client := gapiTestTools(t, 201, writeAlertRuleJSON)
 		defer server.Close()
 		alertRule := createAlertRule()
+
 		uid, err := client.NewAlertRule(&alertRule)
 
 		if err != nil {
@@ -69,6 +82,18 @@ func TestAlertRules(t *testing.T) {
 		}
 	})
 
+	t.Run("set alert rule group succeeds", func(t *testing.T) {
+		server, client := gapiTestTools(t, 200, getAlertRuleGroupJSON)
+		defer server.Close()
+		group := createAlertRuleGroup()
+
+		err := client.SetAlertRuleGroup(group)
+
+		if err != nil {
+			t.Error(err)
+		}
+	})
+
 	t.Run("update alert rule succeeds", func(t *testing.T) {
 		server, client := gapiTestTools(t, 200, writeAlertRuleJSON)
 		defer server.Close()
@@ -94,6 +119,15 @@ func TestAlertRules(t *testing.T) {
 	})
 }
 
+func createAlertRuleGroup() RuleGroup {
+	return RuleGroup{
+		Title:     "eval_group_1",
+		FolderUID: "project_test",
+		Interval:  120,
+		Rules:     []AlertRule{createAlertRule()},
+	}
+}
+
 func createAlertRule() AlertRule {
 	return AlertRule{
 		Condition:    "A",