main_test.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)
		}
	}
}

func TestFirstRunAlerts(t *testing.T) {
	cases := []struct {
		config        Config
		expectErr     bool
		startupAlerts []string
		name          string
	}{
		{
			config:        Config{},
			expectErr:     false,
			startupAlerts: []string{},
			name:          "Empty",
		},
		{
			config:        Config{},
			expectErr:     true,
			startupAlerts: []string{"missing"},
			name:          "Unknown",
		},
		{
			config: Config{
				Alerts: map[string]*Alert{
					"good": {
						Command: CommandOrShell{Command: []string{"true"}},
					},
				},
			},
			expectErr:     false,
			startupAlerts: []string{"good"},
			name:          "Successful alert",
		},
		{
			config: Config{
				Alerts: map[string]*Alert{
					"bad": {
						Name:    "bad",
						Command: CommandOrShell{Command: []string{"false"}},
					},
				},
			},
			expectErr:     true,
			startupAlerts: []string{"bad"},
			name:          "Failed alert",
		},
	}

	for _, c := range cases {
		err := c.config.Init()
		if err != nil {
			t.Errorf("sendFirstRunAlerts(%s): unexpected error reading config: %v", c.name, err)
		}

		err = sendStartupAlerts(&c.config, c.startupAlerts)
		if err == nil && c.expectErr {
			t.Errorf("sendFirstRunAlerts(%s): Expected error, the code did not error", c.name)
		} else if err != nil && !c.expectErr {
			t.Errorf("sendFirstRunAlerts(%s): Did not expect an error, but we got one anyway: %v", c.name, err)
		}
	}
}