energylink/pkg/services/notifications/notifications_test.go

package notifications

import (
	"testing"

	"github.com/grafana/grafana/pkg/bus"
	m "github.com/grafana/grafana/pkg/models"
	"github.com/grafana/grafana/pkg/setting"
	. "github.com/smartystreets/goconvey/convey"
)

type testTriggeredAlert struct {
	ActualValue float64
	Name        string
	State       string
}

func TestNotifications(t *testing.T) {

	Convey("Given the notifications service", t, func() {
		bus.ClearBusHandlers()

		setting.StaticRootPath = "../../../public/"
		setting.Smtp.Enabled = true
		setting.Smtp.TemplatesPattern = "emails/*.html"
		setting.Smtp.FromAddress = "from@address.com"

		err := Init()
		So(err, ShouldBeNil)

		var sentMsg *Message
		addToMailQueue = func(msg *Message) {
			sentMsg = msg
		}

		Convey("When sending reset email password", func() {
			err := sendResetPasswordEmail(&m.SendResetPasswordEmailCommand{User: &m.User{Email: "asd@asd.com"}})
			So(err, ShouldBeNil)
			So(sentMsg.Body, ShouldContainSubstring, "body")
			So(sentMsg.Subject, ShouldEqual, "Reset your Grafana password - asd@asd.com")
			So(sentMsg.Body, ShouldNotContainSubstring, "Subject")
		})

		Convey("Alert notifications", func() {
			// Convey("When sending reset email password", func() {
			// 	cmd := &m.SendEmailCommand{
			// 		Data: map[string]interface{}{
			// 			"Name":           "Name",
			// 			"State":          "Critical",
			// 			"Description":    "Description",
			// 			"DashboardLink":  "http://localhost:3000/dashboard/db/alerting",
			// 			"AlertPageUrl":   "http://localhost:3000/alerting",
			// 			"DashboardImage": "http://localhost:3000/render/dashboard-solo/db/alerting?from=1466169458375&to=1466171258375&panelId=1&width=1000&height=500",
			// 			"TriggeredAlerts": []testTriggeredAlert{
			// 				{Name: "desktop", State: "Critical", ActualValue: 13},
			// 				{Name: "mobile", State: "Warn", ActualValue: 5},
			// 			},
			// 		},
			// 		To:       []string{"asd@asd.com "},
			// 		Template: "alert_notification.html",
			// 	}
			//
			// 	err := sendEmailCommandHandler(cmd)
			// 	So(err, ShouldBeNil)
			//
			// 	So(sentMsg.Body, ShouldContainSubstring, "Alertstate: Critical")
			// 	So(sentMsg.Body, ShouldContainSubstring, "http://localhost:3000/dashboard/db/alerting")
			// 	So(sentMsg.Body, ShouldContainSubstring, "Critical")
			// 	So(sentMsg.Body, ShouldContainSubstring, "Warn")
			// 	So(sentMsg.Body, ShouldContainSubstring, "mobile")
			// 	So(sentMsg.Body, ShouldContainSubstring, "desktop")
			// 	So(sentMsg.Subject, ShouldContainSubstring, "Grafana Alert: [ Critical ] ")
			// })
			//
			// Convey("given critical", func() {
			// 	cmd := &m.SendEmailCommand{
			// 		Data: map[string]interface{}{
			// 			"Name":           "Name",
			// 			"State":          "Warn",
			// 			"Description":    "Description",
			// 			"DashboardLink":  "http://localhost:3000/dashboard/db/alerting",
			// 			"DashboardImage": "http://localhost:3000/render/dashboard-solo/db/alerting?from=1466169458375&to=1466171258375&panelId=1&width=1000&height=500",
			// 			"AlertPageUrl":   "http://localhost:3000/alerting",
			// 			"TriggeredAlerts": []testTriggeredAlert{
			// 				{Name: "desktop", State: "Critical", ActualValue: 13},
			// 				{Name: "mobile", State: "Warn", ActualValue: 5},
			// 			},
			// 		},
			// 		To:       []string{"asd@asd.com "},
			// 		Template: "alert_notification.html",
			// 	}
			//
			// 	err := sendEmailCommandHandler(cmd)
			// 	So(err, ShouldBeNil)
			// 	So(sentMsg.Body, ShouldContainSubstring, "Alertstate: Warn")
			// 	So(sentMsg.Body, ShouldContainSubstring, "http://localhost:3000/dashboard/db/alerting")
			// 	So(sentMsg.Body, ShouldContainSubstring, "Critical")
			// 	So(sentMsg.Body, ShouldContainSubstring, "Warn")
			// 	So(sentMsg.Body, ShouldContainSubstring, "mobile")
			// 	So(sentMsg.Body, ShouldContainSubstring, "desktop")
			// 	So(sentMsg.Subject, ShouldContainSubstring, "Grafana Alert: [ Warn ]")
			// })
			//
			// Convey("given ok", func() {
			// 	cmd := &m.SendEmailCommand{
			// 		Data: map[string]interface{}{
			// 			"Name":          "Name",
			// 			"State":         "Ok",
			// 			"Description":   "Description",
			// 			"DashboardLink": "http://localhost:3000/dashboard/db/alerting",
			// 			"AlertPageUrl":  "http://localhost:3000/alerting",
			// 		},
			// 		To:       []string{"asd@asd.com "},
			// 		Template: "alert_notification.html",
			// 	}
			//
			// 	err := sendEmailCommandHandler(cmd)
			// 	So(err, ShouldBeNil)
			// 	So(sentMsg.Subject, ShouldContainSubstring, "Grafana Alert: [ Ok ]")
			// })
		})
	})
}