energylink/pkg/services/alerting/executor_test.go

package alerting

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

func TestAlertingExecutor(t *testing.T) {
	Convey("Test alert execution", t, func() {
		executor := &ExecutorImpl{}

		Convey("Show return ok since avg is above 2", func() {
			rule := m.AlertRule{CritLevel: 10, CritOperator: "<", Aggregator: "sum"}

			timeSeries := []*m.TimeSeries{
				m.NewTimeSeries("test1", [][2]float64{{2, 0}}),
			}

			result := executor.ValidateRule(rule, timeSeries)
			So(result.State, ShouldEqual, m.AlertStateOk)
		})

		Convey("Show return critical since below 2", func() {
			rule := m.AlertRule{CritLevel: 10, CritOperator: ">", Aggregator: "sum"}

			timeSeries := []*m.TimeSeries{
				m.NewTimeSeries("test1", [][2]float64{{2, 0}}),
			}

			result := executor.ValidateRule(rule, timeSeries)
			So(result.State, ShouldEqual, m.AlertStateCritical)
		})

		Convey("Show return critical since sum is above 10", func() {
			rule := m.AlertRule{CritLevel: 10, CritOperator: "<", Aggregator: "sum"}

			timeSeries := []*m.TimeSeries{
				m.NewTimeSeries("test1", [][2]float64{{9, 0}, {9, 0}}),
			}

			result := executor.ValidateRule(rule, timeSeries)
			So(result.State, ShouldEqual, m.AlertStateCritical)
		})

		Convey("Show return ok since avg is below 10", func() {
			rule := m.AlertRule{CritLevel: 10, CritOperator: "<", Aggregator: "avg"}

			timeSeries := []*m.TimeSeries{
				m.NewTimeSeries("test1", [][2]float64{{9, 0}, {9, 0}}),
			}

			result := executor.ValidateRule(rule, timeSeries)
			So(result.State, ShouldEqual, m.AlertStateOk)
		})

		Convey("Show return ok since min is below 10", func() {
			rule := m.AlertRule{CritLevel: 10, CritOperator: "<", Aggregator: "min"}

			timeSeries := []*m.TimeSeries{
				m.NewTimeSeries("test1", [][2]float64{{11, 0}, {9, 0}}),
			}

			result := executor.ValidateRule(rule, timeSeries)
			So(result.State, ShouldEqual, m.AlertStateOk)
		})

		Convey("Show return ok since max is above 10", func() {
			rule := m.AlertRule{CritLevel: 10, CritOperator: "<", Aggregator: "max"}

			timeSeries := []*m.TimeSeries{
				m.NewTimeSeries("test1", [][2]float64{{1, 0}, {11, 0}}),
			}

			result := executor.ValidateRule(rule, timeSeries)
			So(result.State, ShouldEqual, m.AlertStateCritical)
		})
	})
}