energylink/pkg/services/sqlstore/alert_rule_changes_test.go

package sqlstore

import (
	"testing"

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

var (
	FakeOrgId int64 = 2
)

func TestAlertRuleChangesDataAccess(t *testing.T) {

	Convey("Testing Alert rule changes data access", t, func() {
		InitTestDB(t)

		testDash := insertTestDashboard("dashboard with alerts", 2, "alert")
		var err error

		Convey("When dashboard is removed", func() {
			items := []*m.AlertRule{
				{
					PanelId:      1,
					DashboardId:  testDash.Id,
					Query:        "Query",
					QueryRefId:   "A",
					WarnLevel:    30,
					CritLevel:    50,
					WarnOperator: ">",
					CritOperator: ">",
					Frequency:    10,
					Name:         "Alerting title",
					Description:  "Alerting description",
					QueryRange:   3600,
					Aggregator:   "avg",
					OrgId:        FakeOrgId,
				},
			}

			cmd := m.SaveAlertsCommand{
				Alerts:      items,
				DashboardId: testDash.Id,
				OrgId:       FakeOrgId,
				UserId:      2,
			}

			SaveAlerts(&cmd)

			er := GetAlertRuleChanges(query)
			So(er, ShouldBeNil)
			So(len(query.Result), ShouldEqual, 1)

			err = DeleteDashboard(&m.DeleteDashboardCommand{
				OrgId: FakeOrgId,
				Slug:  testDash.Slug,
			})

			So(err, ShouldBeNil)

			Convey("Alerts should be removed", func() {
				query := m.GetAlertsQuery{DashboardId: testDash.Id, OrgId: 1}
				err2 := HandleAlertsQuery(&query)

				So(testDash.Id, ShouldEqual, 1)
				So(err2, ShouldBeNil)
				So(len(query.Result), ShouldEqual, 0)
			})

			Convey("should add one more alert_rule_change", func() {
				query := &m.GetAlertChangesQuery{OrgId: FakeOrgId}
				er := GetAlertRuleChanges(query)
				So(er, ShouldBeNil)
				So(len(query.Result), ShouldEqual, 2)
			})

			Convey("add 4 updates", func() {
				sess := x.NewSession()
				SaveAlertChange("UPDATED", items[0], sess)
				SaveAlertChange("UPDATED", items[0], sess)
				SaveAlertChange("UPDATED", items[0], sess)
				SaveAlertChange("UPDATED", items[0], sess)
				sess.Commit()

				Convey("query for max one change", func() {
					query := &m.GetAlertChangesQuery{OrgId: FakeOrgId, Limit: 1}
					er := GetAlertRuleChanges(query)
					So(er, ShouldBeNil)
					So(len(query.Result), ShouldEqual, 1)
				})

				Convey("query for all since id 5", func() {
					query := &m.GetAlertChangesQuery{OrgId: FakeOrgId, SinceId: 5}
					er := GetAlertRuleChanges(query)
					So(er, ShouldBeNil)
					So(len(query.Result), ShouldEqual, 2)
				})
			})
		})
	})
}