energylink/pkg/services/alerting/notifier_test.go

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")
		})
	})
}