|
@@ -1,84 +0,0 @@
|
|
|
-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)
|
|
|
|
|
- })
|
|
|
|
|
- */
|
|
|
|
|
- })
|
|
|
|
|
-}
|
|
|