energylink/pkg/services/alerting/handler_test.go

package alerting

import (
	"testing"

	. "github.com/smartystreets/goconvey/convey"
)

type conditionStub struct {
	firing bool
}

func (c *conditionStub) Eval(context *AlertResultContext) {
	context.Firing = c.firing
}

func TestAlertingExecutor(t *testing.T) {
	Convey("Test alert execution", t, func() {
		handler := NewHandler()

		Convey("Show return triggered with single passing condition", func() {
			context := NewAlertResultContext(&AlertRule{
				Conditions: []AlertCondition{&conditionStub{
					firing: true,
				}},
			})

			handler.eval(context)
			So(context.Firing, ShouldEqual, true)
		})

		Convey("Show return false with not passing condition", func() {
			context := NewAlertResultContext(&AlertRule{
				Conditions: []AlertCondition{
					&conditionStub{firing: true},
					&conditionStub{firing: false},
				},
			})

			handler.eval(context)
			So(context.Firing, ShouldEqual, false)
		})

		// 	Convey("Show return critical since below 2", func() {
		// 		rule := &AlertRule{
		// 			Critical:    Level{Value: 10, Operator: "<"},
		// 			Transformer: transformers.NewAggregationTransformer("avg"),
		// 		}
		//
		// 		timeSeries := []*tsdb.TimeSeries{
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{2, 0}}),
		// 		}
		//
		// 		result := executor.evaluateRule(rule, timeSeries)
		// 		So(result.State, ShouldEqual, alertstates.Critical)
		// 	})
		//
		// 	Convey("Show return critical since sum is above 10", func() {
		// 		rule := &AlertRule{
		// 			Critical:    Level{Value: 10, Operator: ">"},
		// 			Transformer: transformers.NewAggregationTransformer("sum"),
		// 		}
		//
		// 		timeSeries := []*tsdb.TimeSeries{
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{9, 0}, {9, 0}}),
		// 		}
		//
		// 		result := executor.evaluateRule(rule, timeSeries)
		// 		So(result.State, ShouldEqual, alertstates.Critical)
		// 	})
		//
		// 	Convey("Show return ok since avg is below 10", func() {
		// 		rule := &AlertRule{
		// 			Critical:    Level{Value: 10, Operator: ">"},
		// 			Transformer: transformers.NewAggregationTransformer("avg"),
		// 		}
		//
		// 		timeSeries := []*tsdb.TimeSeries{
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{9, 0}, {9, 0}}),
		// 		}
		//
		// 		result := executor.evaluateRule(rule, timeSeries)
		// 		So(result.State, ShouldEqual, alertstates.Ok)
		// 	})
		//
		// 	Convey("Show return ok since min is below 10", func() {
		// 		rule := &AlertRule{
		// 			Critical:    Level{Value: 10, Operator: ">"},
		// 			Transformer: transformers.NewAggregationTransformer("avg"),
		// 		}
		//
		// 		timeSeries := []*tsdb.TimeSeries{
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{11, 0}, {9, 0}}),
		// 		}
		//
		// 		result := executor.evaluateRule(rule, timeSeries)
		// 		So(result.State, ShouldEqual, alertstates.Ok)
		// 	})
		//
		// 	Convey("Show return ok since max is above 10", func() {
		// 		rule := &AlertRule{
		// 			Critical:    Level{Value: 10, Operator: ">"},
		// 			Transformer: transformers.NewAggregationTransformer("max"),
		// 		}
		//
		// 		timeSeries := []*tsdb.TimeSeries{
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{6, 0}, {11, 0}}),
		// 		}
		//
		// 		result := executor.evaluateRule(rule, timeSeries)
		// 		So(result.State, ShouldEqual, alertstates.Critical)
		// 	})
		//
		// })
		//
		// Convey("muliple time series", func() {
		// 	Convey("both are ok", func() {
		// 		rule := &AlertRule{
		// 			Critical:    Level{Value: 10, Operator: ">"},
		// 			Transformer: transformers.NewAggregationTransformer("avg"),
		// 		}
		//
		// 		timeSeries := []*tsdb.TimeSeries{
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{2, 0}}),
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{2, 0}}),
		// 		}
		//
		// 		result := executor.evaluateRule(rule, timeSeries)
		// 		So(result.State, ShouldEqual, alertstates.Ok)
		// 	})
		//
		// 	Convey("first serie is good, second is critical", func() {
		// 		rule := &AlertRule{
		// 			Critical:    Level{Value: 10, Operator: ">"},
		// 			Transformer: transformers.NewAggregationTransformer("avg"),
		// 		}
		//
		// 		timeSeries := []*tsdb.TimeSeries{
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{2, 0}}),
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{11, 0}}),
		// 		}
		//
		// 		result := executor.evaluateRule(rule, timeSeries)
		// 		So(result.State, ShouldEqual, alertstates.Critical)
		// 	})
		//
		// 	Convey("first serie is warn, second is critical", func() {
		// 		rule := &AlertRule{
		// 			Critical:    Level{Value: 10, Operator: ">"},
		// 			Warning:     Level{Value: 5, Operator: ">"},
		// 			Transformer: transformers.NewAggregationTransformer("avg"),
		// 		}
		//
		// 		timeSeries := []*tsdb.TimeSeries{
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{6, 0}}),
		// 			tsdb.NewTimeSeries("test1", [][2]float64{{11, 0}}),
		// 		}
		//
		// 		result := executor.evaluateRule(rule, timeSeries)
		// 		So(result.State, ShouldEqual, alertstates.Critical)
		// 	})
		// })
	})
}