forked from pingcap/dead-mans-switch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdead_mans_switch_test.go
67 lines (50 loc) · 1.34 KB
/
dead_mans_switch_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package main
import (
"testing"
"time"
"github.com/prometheus/client_golang/prometheus/testutil"
)
func TestDeadMansSwitchDoesntTrigger(t *testing.T) {
evaluateMessage := make(chan string)
defer close(evaluateMessage)
notify := ""
d := NewDeadMansSwitch(evaluateMessage, 100*time.Millisecond, func(summary, detail string) error {
notify = summary
return nil
})
go d.Run()
defer d.Stop()
evaluateMessage <- ""
time.Sleep(150 * time.Millisecond)
if notify != "" {
t.Fatal("deadman should not trigger!")
}
}
func TestDeadMansSwitchTrigger(t *testing.T) {
evaluateMessage := make(chan string)
defer close(evaluateMessage)
notify := ""
d := NewDeadMansSwitch(evaluateMessage, 100*time.Millisecond, func(summary, detail string) error {
notify = summary
return nil
})
go d.Run()
defer d.Stop()
time.Sleep(150 * time.Millisecond)
if notify != "WatchdogDown" {
t.Fatal("deadman should trigger!")
}
}
func TestEvaluateMessageNotNull(t *testing.T) {
evaluateMessage := make(chan string)
defer close(evaluateMessage)
d := NewDeadMansSwitch(evaluateMessage, 100*time.Millisecond, func(summary, detail string) error {
return nil
})
go d.Run()
defer d.Stop()
evaluateMessage <- "alert not as expected"
if testutil.ToFloat64(failedEvaluatePayload) == 0 {
t.Fatal("failedEvaluatePayload should be > 0 when evaluate failed")
}
}