feat(alerting): converter for db model to notification

2016-06-14 16:56:14 +02:00
parent 6eca26e8ec
commit 9a8416416d
5 changed files with 259 additions and 30 deletions
@@ -0,0 +1,68 @@
+package alerting
+
+import (
+	"testing"
+
+	"reflect"
+
+	"github.com/grafana/grafana/pkg/components/simplejson"
+	m "github.com/grafana/grafana/pkg/models"
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func TestAlertNotificationExtraction(t *testing.T) {
+
+	Convey("Parsing alert notification from settings", t, func() {
+		Convey("Parsing email notification from settings", func() {
+			json := `
+            {
+                "from": "alerting@grafana.org",
+                "to": "ops@grafana.org"
+            }`
+
+			settingsJSON, _ := simplejson.NewJson([]byte(json))
+			model := &m.AlertNotification{
+				Name:     "ops",
+				Type:     "email",
+				Settings: settingsJSON,
+			}
+
+			not, err := NewNotificationFromDBModel(model)
+
+			So(err, ShouldBeNil)
+			So(not.Name, ShouldEqual, "ops")
+			So(not.Type, ShouldEqual, "email")
+			So(reflect.TypeOf(not.Notifierr).Elem().String(), ShouldEqual, "alerting.EmailNotifier")
+
+			email := not.Notifierr.(*EmailNotifier)
+			So(email.To, ShouldEqual, "ops@grafana.org")
+			So(email.From, ShouldEqual, "alerting@grafana.org")
+		})
+
+		Convey("Parsing webhook notification from settings", func() {
+			json := `
+            {
+                "url": "http://localhost:3000",
+                "username": "username",
+                "password": "password"
+            }`
+
+			settingsJSON, _ := simplejson.NewJson([]byte(json))
+			model := &m.AlertNotification{
+				Name:     "slack",
+				Type:     "webhook",
+				Settings: settingsJSON,
+			}
+
+			not, err := NewNotificationFromDBModel(model)
+
+			So(err, ShouldBeNil)
+			So(not.Name, ShouldEqual, "slack")
+			So(not.Type, ShouldEqual, "webhook")
+			So(reflect.TypeOf(not.Notifierr).Elem().String(), ShouldEqual, "alerting.WebhookNotifier")
+
+			webhook := not.Notifierr.(*WebhookNotifier)
+			So(webhook.Url, ShouldEqual, "http://localhost:3000")
+		})
+	})
+}