forked from Griesbacher/check_x
-
Notifications
You must be signed in to change notification settings - Fork 0
/
States_test.go
110 lines (99 loc) · 2.55 KB
/
States_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package check_x
import (
"reflect"
"testing"
)
var stateToString = []struct {
input State
result string
}{
{OK, "OK"},
{Warning, "WARNING"},
{Critical, "CRITICAL"},
{Unknown, "UNKNOWN"},
}
func TestState_String(t *testing.T) {
for i, data := range stateToString {
if data.input.String() != data.result {
t.Errorf("%d - Got: %s - expected: %s", i, data.input.String(), data.result)
}
}
}
var intToState = []struct {
input int
result State
}{
{0, OK},
{1, Warning},
{2, Critical},
{3, Unknown},
{4, State{code: 4}},
}
func TestStateFromInt(t *testing.T) {
for i, data := range intToState {
if StateFromInt(data.input) != data.result {
t.Errorf("%d - Got: %d - expected: %s", i, data.input, data.result)
}
}
}
var stringToState = []struct {
input string
result State
}{
{"OK", OK},
{"Warning", Warning},
{"Critical", Critical},
{"Unknown", Unknown},
{"FOO", State{name: "FOO"}},
}
func TestStateFromString(t *testing.T) {
for i, data := range stringToState {
if StateFromString(data.input) != data.result {
t.Errorf("%d - Got: %s - expected: %s", i, data.input, data.result)
}
}
}
var worstState = []struct {
input States
result *State
err error
}{
{States{}, nil, ErrEmptyStates},
{States{OK}, &OK, nil},
{States{OK, Warning}, &Warning, nil},
{States{Critical, OK, Warning}, &Critical, nil},
{States{OK, Unknown, Warning, Critical}, &Unknown, nil},
}
func TestStates_GetWorst(t *testing.T) {
for i, data := range worstState {
if worst, err := data.input.GetWorst(); err != data.err {
t.Errorf("%d - Got: %s - expected: %s", i, err, data.err)
} else if (worst == nil) != (data.result == nil) {
t.Errorf("%d - Got: %s - expected: %s", i, worst, data.result)
} else if worst != nil && !reflect.DeepEqual(*worst, *data.result) {
t.Errorf("%d - Got: %s - expected: %s", i, worst, data.result)
}
}
}
var bestState = []struct {
input States
result *State
err error
}{
{States{}, nil, ErrEmptyStates},
{States{OK}, &OK, nil},
{States{OK, Warning}, &OK, nil},
{States{OK, Critical, Warning}, &OK, nil},
{States{OK, Warning, Unknown, Critical}, &OK, nil},
}
func TestStates_GetBest(t *testing.T) {
for i, data := range bestState {
if worst, err := data.input.GetBest(); err != data.err {
t.Errorf("%d - Got: %s - expected: %s", i, err, data.err)
} else if (worst == nil) != (data.result == nil) {
t.Errorf("%d - Got: %s - expected: %s", i, worst, data.result)
} else if worst != nil && !reflect.DeepEqual(*worst, *data.result) {
t.Errorf("%d - Got: %s - expected: %s", i, worst, data.result)
}
}
}