135 lines
3.0 KiB
Go
135 lines
3.0 KiB
Go
package agent
|
|
|
|
import (
|
|
"github.com/hashicorp/consul/consul/structs"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type MockNotify struct {
|
|
state map[string]string
|
|
updates map[string]int
|
|
output map[string]string
|
|
}
|
|
|
|
func (m *MockNotify) UpdateCheck(id, status, output string) {
|
|
m.state[id] = status
|
|
old := m.updates[id]
|
|
m.updates[id] = old + 1
|
|
m.output[id] = output
|
|
}
|
|
|
|
func expectStatus(t *testing.T, script, status string) {
|
|
mock := &MockNotify{
|
|
state: make(map[string]string),
|
|
updates: make(map[string]int),
|
|
output: make(map[string]string),
|
|
}
|
|
check := &CheckMonitor{
|
|
Notify: mock,
|
|
CheckID: "foo",
|
|
Script: script,
|
|
Interval: 25 * time.Millisecond,
|
|
Logger: log.New(os.Stderr, "", log.LstdFlags),
|
|
}
|
|
check.Start()
|
|
defer check.Stop()
|
|
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
// Should have at least 2 updates
|
|
if mock.updates["foo"] < 2 {
|
|
t.Fatalf("should have 2 updates %v", mock.updates)
|
|
}
|
|
|
|
if mock.state["foo"] != status {
|
|
t.Fatalf("should be %v %v", status, mock.state)
|
|
}
|
|
}
|
|
|
|
func TestCheckMonitor_Passing(t *testing.T) {
|
|
expectStatus(t, "exit 0", structs.HealthPassing)
|
|
}
|
|
|
|
func TestCheckMonitor_Warning(t *testing.T) {
|
|
expectStatus(t, "exit 1", structs.HealthWarning)
|
|
}
|
|
|
|
func TestCheckMonitor_Critical(t *testing.T) {
|
|
expectStatus(t, "exit 2", structs.HealthCritical)
|
|
}
|
|
|
|
func TestCheckMonitor_BadCmd(t *testing.T) {
|
|
expectStatus(t, "foobarbaz", structs.HealthCritical)
|
|
}
|
|
|
|
func TestCheckMonitor_LimitOutput(t *testing.T) {
|
|
mock := &MockNotify{
|
|
state: make(map[string]string),
|
|
updates: make(map[string]int),
|
|
output: make(map[string]string),
|
|
}
|
|
check := &CheckMonitor{
|
|
Notify: mock,
|
|
CheckID: "foo",
|
|
Script: "dd if=/dev/urandom bs=8192 count=10",
|
|
Interval: 25 * time.Millisecond,
|
|
Logger: log.New(os.Stderr, "", log.LstdFlags),
|
|
}
|
|
check.Start()
|
|
defer check.Stop()
|
|
|
|
time.Sleep(50 * time.Millisecond)
|
|
|
|
// Allow for extra bytes for the truncation message
|
|
if len(mock.output["foo"]) > CheckBufSize+100 {
|
|
t.Fatalf("output size is too long")
|
|
}
|
|
}
|
|
|
|
func TestCheckTTL(t *testing.T) {
|
|
mock := &MockNotify{
|
|
state: make(map[string]string),
|
|
updates: make(map[string]int),
|
|
output: make(map[string]string),
|
|
}
|
|
check := &CheckTTL{
|
|
Notify: mock,
|
|
CheckID: "foo",
|
|
TTL: 20 * time.Millisecond,
|
|
Logger: log.New(os.Stderr, "", log.LstdFlags),
|
|
}
|
|
check.Start()
|
|
defer check.Stop()
|
|
|
|
time.Sleep(10 * time.Millisecond)
|
|
check.SetStatus(structs.HealthPassing, "")
|
|
|
|
if mock.updates["foo"] != 1 {
|
|
t.Fatalf("should have 1 updates %v", mock.updates)
|
|
}
|
|
|
|
if mock.state["foo"] != structs.HealthPassing {
|
|
t.Fatalf("should be passing %v", mock.state)
|
|
}
|
|
|
|
// Ensure we don't fail early
|
|
time.Sleep(15 * time.Millisecond)
|
|
if mock.updates["foo"] != 1 {
|
|
t.Fatalf("should have 1 updates %v", mock.updates)
|
|
}
|
|
|
|
// Wait for the TTL to expire
|
|
time.Sleep(15 * time.Millisecond)
|
|
|
|
if mock.updates["foo"] != 2 {
|
|
t.Fatalf("should have 2 updates %v", mock.updates)
|
|
}
|
|
|
|
if mock.state["foo"] != structs.HealthCritical {
|
|
t.Fatalf("should be critical %v", mock.state)
|
|
}
|
|
}
|