Implement read policy tree and tests

Author: alexweav

alerting_notification_policy.go

@@ -4,17 +4,16 @@ import (
 	"encoding/json"
 	"fmt"
 	"regexp"
-	"time"
 )
 
 // Represents a notification routing tree in Grafana Alerting.
 type NotificationPolicy struct {
 	Receiver       string           `json:"receiver,omitempty"`
 	GroupBy        []string         `json:"group_by,omitempty"`
 	Routes         []SpecificPolicy `json:"routes,omitempty"`
-	GroupWait      time.Duration    `json:"group_wait,omitempty"`
-	GroupInterval  time.Duration    `json:"group_interval,omitempty"`
-	RepeatInterval time.Duration    `json:"repeat_interval,omitempty"`
+	GroupWait      string           `json:"group_wait,omitempty"`
+	GroupInterval  string           `json:"group_interval,omitempty"`
+	RepeatInterval string           `json:"repeat_interval,omitempty"`
 	Provenance     string           `json:"provenance,omitempty"`
 }
 
@@ -26,9 +25,9 @@ type SpecificPolicy struct {
 	MuteTimeIntervals []string         `json:"mute_time_intervals,omitempty"`
 	Continue          bool             `json:"continue"`
 	Routes            []SpecificPolicy `json:"routes,omitempty"`
-	GroupWait         time.Duration    `json:"group_wait,omitempty"`
-	GroupInterval     time.Duration    `json:"group_interval,omitempty"`
-	RepeatInterval    time.Duration    `json:"repeat_interval,omitempty"`
+	GroupWait         string           `json:"group_wait,omitempty"`
+	GroupInterval     string           `json:"group_interval,omitempty"`
+	RepeatInterval    string           `json:"repeat_interval,omitempty"`
 }
 
 type Matchers []Matcher
@@ -111,3 +110,9 @@ func (m Matchers) MarshalJSON() ([]byte, error) {
 	}
 	return json.Marshal(result)
 }
+
+func (c *Client) NotificationPolicy() (NotificationPolicy, error) {
+	np := NotificationPolicy{}
+	err := c.request("GET", "/api/v1/provisioning/policies", nil, nil, &np)
+	return np, err
+}

alerting_notification_policy_test.go

@@ -0,0 +1,59 @@
+package gapi
+
+import (
+	"testing"
+
+	"github.com/gobs/pretty"
+)
+
+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()
+
+		if err != nil {
+			t.Error(err)
+		}
+		t.Log(pretty.PrettyFormat(np))
+		if np.Receiver != "grafana-default-email" {
+			t.Errorf("wrong receiver, got %#v", np.Receiver)
+		}
+		if len(np.Routes) != 1 {
+			t.Errorf("wrong number of routes returned, got %#v", np)
+		}
+	})
+}
+
+const notificationPolicyJSON = `
+{
+	"receiver": "grafana-default-email",
+	"group_by": [
+		"..."
+	],
+	"routes": [
+		{
+			"receiver": "grafana-default-email",
+			"object_matchers": [
+				[
+					"a",
+					"=",
+					"b"
+				],
+				[
+					"asdf",
+					"!=",
+					"jk"
+				]
+			]
+		}
+	],
+	"group_wait": "5s",
+	"group_interval": "1m",
+	"repeat_interval": "1h"
+}`