minitor-go/main_test.go

115 lines
2.4 KiB
Go
Raw Permalink Normal View History

package main
import "testing"
func TestCheckMonitors(t *testing.T) {
cases := []struct {
2019-10-04 23:17:36 +00:00
config Config
expectErr bool
name string
}{
{
2019-10-04 23:17:36 +00:00
config: Config{},
expectErr: false,
name: "Empty",
},
{
config: Config{
Monitors: []*Monitor{
&Monitor{
Name: "Success",
Command: CommandOrShell{Command: []string{"true"}},
},
},
},
2019-10-04 23:17:36 +00:00
expectErr: false,
name: "Monitor success, no alerts",
},
{
config: Config{
Monitors: []*Monitor{
&Monitor{
Name: "Failure",
Command: CommandOrShell{Command: []string{"false"}},
AlertAfter: 1,
},
&Monitor{
Name: "Failure",
Command: CommandOrShell{Command: []string{"false"}},
AlertDown: []string{"unknown"},
AlertAfter: 1,
},
},
},
2019-10-04 23:17:36 +00:00
expectErr: false,
name: "Monitor failure, no and unknown alerts",
},
{
config: Config{
Monitors: []*Monitor{
&Monitor{
Name: "Success",
Command: CommandOrShell{Command: []string{"ls"}},
alertCount: 1,
},
&Monitor{
Name: "Success",
Command: CommandOrShell{Command: []string{"true"}},
AlertUp: []string{"unknown"},
alertCount: 1,
},
},
},
2019-10-04 23:17:36 +00:00
expectErr: false,
name: "Monitor recovery, no alerts",
},
{
config: Config{
Monitors: []*Monitor{
&Monitor{
Name: "Failure",
Command: CommandOrShell{Command: []string{"false"}},
AlertDown: []string{"good"},
AlertAfter: 1,
},
},
Alerts: map[string]*Alert{
"good": &Alert{
Command: CommandOrShell{Command: []string{"true"}},
},
},
},
2019-10-04 23:17:36 +00:00
expectErr: false,
name: "Monitor failure, successful alert",
},
{
config: Config{
Monitors: []*Monitor{
&Monitor{
Name: "Failure",
Command: CommandOrShell{Command: []string{"false"}},
AlertDown: []string{"bad"},
AlertAfter: 1,
},
},
Alerts: map[string]*Alert{
"bad": &Alert{
Name: "bad",
Command: CommandOrShell{Command: []string{"false"}},
},
},
},
2019-10-04 23:17:36 +00:00
expectErr: true,
name: "Monitor failure, bad alert",
},
}
for _, c := range cases {
2019-10-04 23:17:36 +00:00
c.config.Init()
err := checkMonitors(&c.config)
if err == nil && c.expectErr {
t.Errorf("checkMonitors(%s): Expected panic, the code did not panic", c.name)
}
}
}