go-icingadsl/Notifications_test.go at main · NETWAYS/go-icingadsl · GitHub

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
111
package icingadsl

import (
	"testing"
)

func TestParseNotificationType(t *testing.T) {
	testcases := []struct {
		expected NotificationType
		nt       string
	}{
		{
			expected: DowntimeStart,
			nt:       "downtimestart",
		},
		{
			expected: DowntimeEnd,
			nt:       "downtimeend",
		},
		{
			expected: DowntimeRemoved,
			nt:       "downtimeremoved",
		},
		{
			expected: Custom,
			nt:       "custom",
		},
		{
			expected: Acknowledgement,
			nt:       "acknowledgement",
		},
		{
			expected: Problem,
			nt:       "problem",
		},
		{
			expected: Recovery,
			nt:       "recovery",
		},
		{
			expected: FlappingStart,
			nt:       "flappingstart",
		},
		{
			expected: FlappingEnd,
			nt:       "flappingend",
		},
	}

	for _, tc := range testcases {
		nt, err := ParseNotificationType(tc.nt)

		if err != nil {
			t.Error("Did not expect error, got:", err)
		}

		if nt != tc.expected {
			t.Error("\nActual: ", nt, "\nExpected: ", tc.expected)
		}
	}
}

func TestNotificationTypeStringer(t *testing.T) {
	testcases := []struct {
		nt       NotificationType
		expected string
	}{
		{
			nt:       DowntimeStart,
			expected: "downtimestart",
		},
		{
			nt:       DowntimeEnd,
			expected: "downtimeend",
		},
		{
			nt:       DowntimeRemoved,
			expected: "downtimeremoved",
		},
		{
			nt:       Custom,
			expected: "custom",
		},
		{
			nt:       Acknowledgement,
			expected: "acknowledgement",
		},
		{
			nt:       Problem,
			expected: "problem",
		},
		{
			nt:       Recovery,
			expected: "recovery",
		},
		{
			nt:       FlappingStart,
			expected: "flappingstart",
		},
		{
			nt:       FlappingEnd,
			expected: "flappingend",
		},
	}

	for _, tc := range testcases {
		if tc.nt.String() != tc.expected {
			t.Error("\nActual: ", tc.nt.String(), "\nExpected: ", tc.expected)
		}
	}
}