-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
status_test.go
68 lines (60 loc) · 1.98 KB
/
status_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
// Copyright (c) 2024, Roel Schut. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package healthcheck
import (
"net/http"
"testing"
"github.com/stretchr/testify/assert"
)
func TestStatusCode(t *testing.T) {
statuses := []Status{StatusUnknown, StatusHealthy, StatusUnhealthy}
for _, stat := range statuses {
t.Run(stat.String(), func(t *testing.T) {
assert.Equal(t, stat, StatusCode(stat.StatusCode()))
})
}
}
func TestStatus_StatusCode(t *testing.T) {
tests := map[Status]int{
StatusHealthy: http.StatusOK,
StatusUnhealthy: http.StatusServiceUnavailable,
StatusUnknown: http.StatusTooEarly,
-3: http.StatusInternalServerError,
}
for stat, want := range tests {
assert.Equal(t, want, stat.StatusCode())
}
}
func TestStatus_ExitCode(t *testing.T) {
tests := map[Status]int{
StatusHealthy: 0,
StatusUnhealthy: int(StatusUnhealthy),
StatusUnknown: 100,
-3: -3,
}
for stat, want := range tests {
assert.Equal(t, want, stat.ExitCode())
}
}
func TestCombine(t *testing.T) {
tests := map[string]struct {
a, b, want Status
}{
"both unknown": {StatusUnknown, StatusUnknown, StatusUnknown},
"both healthy": {StatusHealthy, StatusHealthy, StatusHealthy},
"both unhealthy": {StatusUnhealthy, StatusUnhealthy, StatusUnhealthy},
"unknown + healthy": {StatusUnknown, StatusHealthy, StatusHealthy},
"unknown + unhealthy": {StatusUnknown, StatusUnhealthy, StatusUnhealthy},
"healthy + unknown": {StatusHealthy, StatusUnknown, StatusUnhealthy},
"healthy + unhealthy": {StatusHealthy, StatusUnhealthy, StatusUnhealthy},
"unhealthy + unknown": {StatusUnhealthy, StatusUnknown, StatusUnhealthy},
"unhealthy + healthy": {StatusUnhealthy, StatusHealthy, StatusUnhealthy},
"invalid": {3, -3, StatusUnknown},
}
for name, tc := range tests {
t.Run(name, func(t *testing.T) {
assert.Equal(t, tc.want, Combine(tc.a, tc.b))
})
}
}