minitor-go/alert_test.go

152 lines
3.7 KiB
Go
Raw Permalink Normal View History

2019-10-03 23:30:49 +00:00
package main
import (
"log"
"testing"
)
func TestAlertIsValid(t *testing.T) {
cases := []struct {
alert Alert
expected bool
name string
}{
{Alert{Command: CommandOrShell{Command: []string{"echo", "test"}}}, true, "Command only"},
{Alert{Command: CommandOrShell{ShellCommand: "echo test"}}, true, "CommandShell only"},
2019-10-03 23:30:49 +00:00
{Alert{}, false, "No commands"},
}
for _, c := range cases {
log.Printf("Testing case %s", c.name)
actual := c.alert.IsValid()
if actual != c.expected {
t.Errorf("IsValid(%v), expected=%t actual=%t", c.name, c.expected, actual)
log.Printf("Case failed: %s", c.name)
}
log.Println("-----")
}
}
func TestAlertSend(t *testing.T) {
cases := []struct {
alert Alert
notice AlertNotice
expectedOutput string
expectErr bool
name string
2020-02-20 01:31:04 +00:00
pyCompat bool
2019-10-03 23:30:49 +00:00
}{
{
Alert{Command: CommandOrShell{Command: []string{"echo", "{{.MonitorName}}"}}},
2019-10-03 23:30:49 +00:00
AlertNotice{MonitorName: "test"},
"test\n",
false,
"Command with template",
2020-02-20 01:31:04 +00:00
false,
2019-10-03 23:30:49 +00:00
},
{
Alert{Command: CommandOrShell{ShellCommand: "echo {{.MonitorName}}"}},
2019-10-03 23:30:49 +00:00
AlertNotice{MonitorName: "test"},
"test\n",
false,
"Command shell with template",
2020-02-20 01:31:04 +00:00
false,
2019-10-03 23:30:49 +00:00
},
{
Alert{Command: CommandOrShell{Command: []string{"echo", "{{.Bad}}"}}},
2019-10-03 23:30:49 +00:00
AlertNotice{MonitorName: "test"},
"",
true,
"Command with bad template",
2020-02-20 01:31:04 +00:00
false,
2019-10-03 23:30:49 +00:00
},
{
Alert{Command: CommandOrShell{ShellCommand: "echo {{.Bad}}"}},
2019-10-03 23:30:49 +00:00
AlertNotice{MonitorName: "test"},
"",
true,
"Command shell with bad template",
2020-02-20 01:31:04 +00:00
false,
2019-10-03 23:30:49 +00:00
},
{
Alert{Command: CommandOrShell{ShellCommand: "echo {alert_message}"}},
AlertNotice{MonitorName: "test", FailureCount: 1},
"test check has failed 1 times\n",
false,
"Command shell with legacy template",
2020-02-20 01:31:04 +00:00
true,
},
// Test default log alert down
{
*NewLogAlert(),
AlertNotice{MonitorName: "Test", FailureCount: 1, IsUp: false},
"Test check has failed 1 times\n",
false,
"Default log alert down",
false,
},
// Test default log alert up
{
*NewLogAlert(),
AlertNotice{MonitorName: "Test", IsUp: true},
"Test has recovered\n",
false,
"Default log alert up",
false,
},
2019-10-03 23:30:49 +00:00
}
for _, c := range cases {
log.Printf("Testing case %s", c.name)
2020-02-20 01:31:04 +00:00
// Set PyCompat to value of compat flag
PyCompat = c.pyCompat
2019-10-03 23:30:49 +00:00
c.alert.BuildTemplates()
output, err := c.alert.Send(c.notice)
hasErr := (err != nil)
if output != c.expectedOutput {
t.Errorf("Send(%v output), expected=%v actual=%v", c.name, c.expectedOutput, output)
log.Printf("Case failed: %s", c.name)
}
if hasErr != c.expectErr {
t.Errorf("Send(%v err), expected=%v actual=%v", c.name, "Err", err)
log.Printf("Case failed: %s", c.name)
}
2020-02-20 01:31:04 +00:00
// Set PyCompat back to default value
PyCompat = false
2019-10-03 23:30:49 +00:00
log.Println("-----")
}
}
func TestAlertSendNoTemplates(t *testing.T) {
alert := Alert{}
notice := AlertNotice{}
output, err := alert.Send(notice)
if err == nil {
t.Errorf("Send(no template), expected=%v actual=%v", "Err", output)
}
log.Println("-----")
}
func TestAlertBuildTemplate(t *testing.T) {
cases := []struct {
alert Alert
expectErr bool
name string
}{
{Alert{Command: CommandOrShell{Command: []string{"echo", "test"}}}, false, "Command only"},
{Alert{Command: CommandOrShell{ShellCommand: "echo test"}}, false, "CommandShell only"},
2019-10-03 23:30:49 +00:00
{Alert{}, true, "No commands"},
}
for _, c := range cases {
log.Printf("Testing case %s", c.name)
err := c.alert.BuildTemplates()
hasErr := (err != nil)
if hasErr != c.expectErr {
t.Errorf("IsValid(%v), expected=%t actual=%t", c.name, c.expectErr, err)
log.Printf("Case failed: %s", c.name)
}
log.Println("-----")
}
}