Merge pull request #98 from grafana/alexweav/notif-policy-reset

Alerting: Add method for resetting notification policies

Author: alexweav

alerting_notification_policy.go

@@ -7,7 +7,7 @@ import (
 )
 
 // Represents a notification routing tree in Grafana Alerting.
-type NotificationPolicy struct {
+type NotificationPolicyTree struct {
 	Receiver       string           `json:"receiver,omitempty"`
 	GroupBy        []string         `json:"group_by,omitempty"`
 	Routes         []SpecificPolicy `json:"routes,omitempty"`
@@ -104,17 +104,21 @@ func (m Matchers) MarshalJSON() ([]byte, error) {
 }
 
 // NotificationPolicy fetches the notification policy tree.
-func (c *Client) NotificationPolicy() (NotificationPolicy, error) {
-	np := NotificationPolicy{}
+func (c *Client) NotificationPolicyTree() (NotificationPolicyTree, error) {
+	np := NotificationPolicyTree{}
 	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 {
+func (c *Client) SetNotificationPolicyTree(np *NotificationPolicyTree) error {
 	req, err := json.Marshal(np)
 	if err != nil {
 		return err
 	}
 	return c.request("PUT", "/api/v1/provisioning/policies", nil, bytes.NewBuffer(req), nil)
 }
+
+func (c *Client) ResetNotificationPolicyTree() error {
+	return c.request("DELETE", "/api/v1/provisioning/policies", nil, nil, nil)
+}

alerting_notification_policy_test.go

@@ -11,7 +11,7 @@ func TestNotificationPolicies(t *testing.T) {
 		server, client := gapiTestTools(t, 200, notificationPolicyJSON)
 		defer server.Close()
 
-		np, err := client.NotificationPolicy()
+		np, err := client.NotificationPolicyTree()
 
 		if err != nil {
 			t.Error(err)
@@ -30,16 +30,27 @@ func TestNotificationPolicies(t *testing.T) {
 		defer server.Close()
 		np := createNotificationPolicy()
 
-		err := client.SetNotificationPolicy(&np)
+		err := client.SetNotificationPolicyTree(&np)
+
+		if err != nil {
+			t.Error(err)
+		}
+	})
+
+	t.Run("reset policy tree succeeds", func(t *testing.T) {
+		server, client := gapiTestTools(t, 200, notificationPolicyJSON)
+		defer server.Close()
+
+		err := client.ResetNotificationPolicyTree()
 
 		if err != nil {
 			t.Error(err)
 		}
 	})
 }
 
-func createNotificationPolicy() NotificationPolicy {
-	return NotificationPolicy{
+func createNotificationPolicy() NotificationPolicyTree {
+	return NotificationPolicyTree{
 		Receiver: "grafana-default-email",
 		GroupBy:  []string{"asdfasdf", "alertname"},
 		Routes: []SpecificPolicy{