Method for setting policies and tests

Author: alexweav

alerting_notification_policy.go

@@ -1,9 +1,9 @@
 package gapi
 
 import (
+	"bytes"
 	"encoding/json"
 	"fmt"
-	"regexp"
 )
 
 // Represents a notification routing tree in Grafana Alerting.
@@ -36,7 +36,6 @@ type Matcher struct {
 	Type  MatchType
 	Name  string
 	Value string
-	re    *regexp.Regexp
 }
 
 type MatchType int
@@ -87,13 +86,6 @@ func (m *Matchers) UnmarshalJSON(data []byte) error {
 			Name:  rawMatcher[0],
 			Value: rawMatcher[2],
 		}
-		if matchType == MatchRegexp || matchType == MatchNotRegexp {
-			re, err := regexp.Compile("^(?:" + rawMatcher[2] + ")$")
-			if err != nil {
-				return err
-			}
-			matcher.re = re
-		}
 		*m = append(*m, matcher)
 	}
 	return nil
@@ -111,8 +103,18 @@ func (m Matchers) MarshalJSON() ([]byte, error) {
 	return json.Marshal(result)
 }
 
+// NotificationPolicy fetches the notification policy tree.
 func (c *Client) NotificationPolicy() (NotificationPolicy, error) {
 	np := NotificationPolicy{}
 	err := c.request("GET", "/api/v1/provisioning/policies", nil, nil, &np)
 	return np, err
 }
+
+// SetNotificationPolicy sets the notification policy tree.
+func (c *Client) SetNotificationPolicy(np *NotificationPolicy) error {
+	req, err := json.Marshal(np)
+	if err != nil {
+		return err
+	}
+	return c.request("PUT", "/api/v1/provisioning/policies", nil, bytes.NewBuffer(req), nil)
+}

alerting_notification_policy_test.go

@@ -10,10 +10,6 @@ func TestNotificationPolicies(t *testing.T) {
 	t.Run("get policy tree succeeds", func(t *testing.T) {
 		server, client := gapiTestTools(t, 200, notificationPolicyJSON)
 		defer server.Close()
-		/*cfg := Config{
-			BasicAuth: url.UserPassword("admin", "admin"),
-		}
-		client, _ := New("http://localhost:3000", cfg)*/
 
 		np, err := client.NotificationPolicy()
 
@@ -28,6 +24,52 @@ func TestNotificationPolicies(t *testing.T) {
 			t.Errorf("wrong number of routes returned, got %#v", np)
 		}
 	})
+
+	t.Run("set policy tree succeeds", func(t *testing.T) {
+		server, client := gapiTestTools(t, 202, `{"message":"created"}`)
+		defer server.Close()
+		np := createNotificationPolicy()
+
+		err := client.SetNotificationPolicy(&np)
+
+		if err != nil {
+			t.Error(err)
+		}
+	})
+}
+
+func createNotificationPolicy() NotificationPolicy {
+	return NotificationPolicy{
+		Receiver: "grafana-default-email",
+		GroupBy:  []string{"asdfasdf", "alertname"},
+		Routes: []SpecificPolicy{
+			{
+				Receiver: "grafana-default-email",
+				ObjectMatchers: Matchers{
+					{
+						Type:  MatchNotEqual,
+						Name:  "abc",
+						Value: "def",
+					},
+				},
+				Continue: true,
+			},
+			{
+				Receiver: "grafana-default-email",
+				ObjectMatchers: Matchers{
+					{
+						Type:  MatchRegexp,
+						Name:  "jkl",
+						Value: "something.*",
+					},
+				},
+				Continue: false,
+			},
+		},
+		GroupWait:      "10s",
+		GroupInterval:  "5m",
+		RepeatInterval: "4h",
+	}
 }
 
 const notificationPolicyJSON = `