energylink/pkg/services/alerting/reader_test.go

package alerting

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

func TestAlertingScheduler(t *testing.T) {
	Convey("Testing alert job selection", t, func() {
		/*
			mockFn := func() []m.AlertRule {
				return []m.AlertRule{
					{Id: 1, Title: "test 1"},
					{Id: 2, Title: "test 2"},
					{Id: 3, Title: "test 3"},
					{Id: 4, Title: "test 4"},
					{Id: 5, Title: "test 5"},
					{Id: 6, Title: "test 6"},
				}
			}

			Convey("single server", func() {
				scheduler := &Scheduler{
					jobs:           make(map[int64]*AlertJob, 0),
					runQueue:       make(chan *AlertJob, 1000),
					serverId:       "",
					serverPosition: 1,
					clusterSize:    1,
				}

				scheduler.updateJobs(mockFn)
				So(len(scheduler.jobs), ShouldEqual, 6)
			})

			Convey("two servers", func() {
				scheduler := &Scheduler{
					jobs:           make(map[int64]*AlertJob, 0),
					runQueue:       make(chan *AlertJob, 1000),
					serverId:       "",
					serverPosition: 1,
					clusterSize:    2,
				}

				scheduler.updateJobs(mockFn)
				So(len(scheduler.jobs), ShouldEqual, 3)
				So(scheduler.jobs[1].rule.Id, ShouldEqual, 1)
			})

			Convey("six servers", func() {
				scheduler := &Scheduler{
					jobs:           make(map[int64]*AlertJob, 0),
					runQueue:       make(chan *AlertJob, 1000),
					serverId:       "",
					serverPosition: 6,
					clusterSize:    6,
				}

				scheduler.updateJobs(mockFn)
				So(len(scheduler.jobs), ShouldEqual, 1)
				So(scheduler.jobs[6].rule.Id, ShouldEqual, 6)
			})

			Convey("more servers then alerts", func() {
				mockFn := func() []m.AlertRule {
					return []m.AlertRule{
						{Id: 1, Title: "test 1"},
					}
				}

				scheduler := &Scheduler{
					jobs:           make(map[int64]*AlertJob, 0),
					runQueue:       make(chan *AlertJob, 1000),
					serverId:       "",
					serverPosition: 3,
					clusterSize:    3,
				}

				scheduler.updateJobs(mockFn)
				So(len(scheduler.jobs), ShouldEqual, 0)
			})
		*/
	})
}