main_test.go

package main

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestMainCheck_matches(t *testing.T) {
	testcases := map[string]struct {
		input    string
		regex    []string
		expected bool
	}{
		"simple-t": {
			input:    "unittest",
			regex:    []string{"unit", "foobar"},
			expected: true,
		},
		"simple-f": {
			input:    "unittest",
			regex:    []string{"barfoo", "foobar"},
			expected: false,
		},
	}

	for name, tc := range testcases {
		t.Run(name, func(t *testing.T) {
			actual := matches(tc.input, tc.regex)
			assert.Equal(t, tc.expected, actual)
		})
	}
}

func TestMainAlerts_GetPerfdata(t *testing.T) {
	testcases := map[string]struct {
		ao       AlertOverview
		expected string
	}{
		"simple-overview": {
			ao:       AlertOverview{},
			expected: "'alerts'=0 'alerts_high'=0 'alerts_medium'=0 'alerts_low'=0",
		},
	}

	for name, tc := range testcases {
		t.Run(name, func(t *testing.T) {
			actual := tc.ao.GetPerfdata()
			assert.Equal(t, tc.expected, actual)
		})
	}
}

func TestMainAlerts_GetOutput(t *testing.T) {
	testcases := map[string]struct {
		ao       AlertOverview
		expected string
	}{
		"simple-overview": {
			ao: AlertOverview{
				Total:  1,
				Output: []string{"test"},
			},
			expected: "\n## Alerts\ntest\n",
		},
	}

	for name, tc := range testcases {
		t.Run(name, func(t *testing.T) {
			actual := tc.ao.GetOutput()
			assert.Equal(t, tc.expected, actual)
		})
	}
}

func TestMainAlerts_GetStatus(t *testing.T) {
	testcases := map[string]struct {
		ao       AlertOverview
		expected int
	}{
		"simple-overview": {
			ao:       AlertOverview{},
			expected: 0,
		},
		"simple-warning": {
			ao: AlertOverview{
				Medium: 1,
			},
			expected: 1,
		},
		"simple-critical": {
			ao: AlertOverview{
				High: 1,
			},
			expected: 2,
		},
	}

	for name, tc := range testcases {
		t.Run(name, func(t *testing.T) {
			actual := tc.ao.GetStatus()
			assert.Equal(t, tc.expected, actual)
		})
	}
}