137 lines
2.8 KiB
Go
137 lines
2.8 KiB
Go
package main
|
|
|
|
import "testing"
|
|
|
|
func TestCheckMonitors(t *testing.T) {
|
|
cases := []struct {
|
|
config Config
|
|
expectErr bool
|
|
name string
|
|
}{
|
|
{
|
|
config: Config{},
|
|
expectErr: false,
|
|
name: "Empty",
|
|
},
|
|
{
|
|
config: Config{
|
|
Monitors: []*Monitor{
|
|
{
|
|
Name: "Success",
|
|
Command: CommandOrShell{Command: []string{"true"}},
|
|
},
|
|
},
|
|
},
|
|
expectErr: false,
|
|
name: "Monitor success, no alerts",
|
|
},
|
|
{
|
|
config: Config{
|
|
Monitors: []*Monitor{
|
|
{
|
|
Name: "Failure",
|
|
Command: CommandOrShell{Command: []string{"false"}},
|
|
AlertAfter: 1,
|
|
},
|
|
},
|
|
},
|
|
expectErr: false,
|
|
name: "Monitor failure, no alerts",
|
|
},
|
|
{
|
|
config: Config{
|
|
Monitors: []*Monitor{
|
|
{
|
|
Name: "Success",
|
|
Command: CommandOrShell{Command: []string{"ls"}},
|
|
alertCount: 1,
|
|
},
|
|
},
|
|
},
|
|
expectErr: false,
|
|
name: "Monitor recovery, no alerts",
|
|
},
|
|
{
|
|
config: Config{
|
|
Monitors: []*Monitor{
|
|
{
|
|
Name: "Failure",
|
|
Command: CommandOrShell{Command: []string{"false"}},
|
|
AlertDown: []string{"unknown"},
|
|
AlertAfter: 1,
|
|
},
|
|
},
|
|
},
|
|
expectErr: true,
|
|
name: "Monitor failure, unknown alerts",
|
|
},
|
|
{
|
|
config: Config{
|
|
Monitors: []*Monitor{
|
|
{
|
|
Name: "Success",
|
|
Command: CommandOrShell{Command: []string{"true"}},
|
|
AlertUp: []string{"unknown"},
|
|
alertCount: 1,
|
|
},
|
|
},
|
|
},
|
|
expectErr: true,
|
|
name: "Monitor recovery, unknown alerts",
|
|
},
|
|
{
|
|
config: Config{
|
|
Monitors: []*Monitor{
|
|
{
|
|
Name: "Failure",
|
|
Command: CommandOrShell{Command: []string{"false"}},
|
|
AlertDown: []string{"good"},
|
|
AlertAfter: 1,
|
|
},
|
|
},
|
|
Alerts: map[string]*Alert{
|
|
"good": {
|
|
Command: CommandOrShell{Command: []string{"true"}},
|
|
},
|
|
},
|
|
},
|
|
expectErr: false,
|
|
name: "Monitor failure, successful alert",
|
|
},
|
|
{
|
|
config: Config{
|
|
Monitors: []*Monitor{
|
|
{
|
|
Name: "Failure",
|
|
Command: CommandOrShell{Command: []string{"false"}},
|
|
AlertDown: []string{"bad"},
|
|
AlertAfter: 1,
|
|
},
|
|
},
|
|
Alerts: map[string]*Alert{
|
|
"bad": {
|
|
Name: "bad",
|
|
Command: CommandOrShell{Command: []string{"false"}},
|
|
},
|
|
},
|
|
},
|
|
expectErr: true,
|
|
name: "Monitor failure, bad alert",
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
err := c.config.Init()
|
|
if err != nil {
|
|
t.Errorf("checkMonitors(%s): unexpected error reading config: %v", c.name, err)
|
|
}
|
|
|
|
err = checkMonitors(&c.config)
|
|
if err == nil && c.expectErr {
|
|
t.Errorf("checkMonitors(%s): Expected panic, the code did not panic", c.name)
|
|
} else if err != nil && !c.expectErr {
|
|
t.Errorf("checkMonitors(%s): Did not expect an error, but we got one anyway: %v", c.name, err)
|
|
}
|
|
}
|
|
}
|