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
|
package internal
import (
"encoding/json"
"os"
"path/filepath"
"testing"
"time"
)
func TestPersistJSONReport(t *testing.T) {
tmpDir := t.TempDir()
htmlFile := filepath.Join(tmpDir, "status.html")
conf := config{HTMLStatusFile: htmlFile}
now := time.Now().Unix()
s := state{
checks: map[string]checkState{
"CriticalCheck": {
Status: nagiosCritical,
PrevStatus: nagiosOk,
Epoch: now,
Output: "boom",
},
"StaleCheck": {
Status: nagiosWarning,
PrevStatus: nagiosWarning,
Epoch: now - 1000,
Output: "stale warning",
},
"OkCheck": {
Status: nagiosOk,
PrevStatus: nagiosOk,
Epoch: now,
Output: "all good",
},
},
staleEpoch: now - 100,
}
subject := "GOGIOS Report [C:1 W:1 U:0 S:1 SU:0 OK:1]"
if err := persistJSONReport(s, subject, conf); err != nil {
t.Fatalf("persistJSONReport() error = %v", err)
}
jsonFile := filepath.Join(tmpDir, "status.json")
data, err := os.ReadFile(jsonFile)
if err != nil {
t.Fatalf("failed to read JSON file: %v", err)
}
var report jsonReport
if err := json.Unmarshal(data, &report); err != nil {
t.Fatalf("failed to unmarshal JSON report: %v", err)
}
if report.LastUpdated == "" {
t.Fatal("lastUpdated is empty")
}
if _, err := time.Parse(time.RFC3339, report.LastUpdated); err != nil {
t.Fatalf("lastUpdated is not RFC3339: %v", err)
}
if report.Subject != subject {
t.Fatalf("subject = %q, want %q", report.Subject, subject)
}
if report.Summary.Critical != 1 || report.Summary.Warning != 1 || report.Summary.Stale != 1 || report.Summary.Ok != 1 {
t.Fatalf("unexpected summary: %+v", report.Summary)
}
if len(report.Sections.StatusChanged) != 1 || report.Sections.StatusChanged[0].Name != "CriticalCheck" {
t.Fatalf("unexpected statusChanged: %+v", report.Sections.StatusChanged)
}
if len(report.Sections.Unhandled) != 1 || report.Sections.Unhandled[0].Name != "CriticalCheck" {
t.Fatalf("unexpected unhandled: %+v", report.Sections.Unhandled)
}
if len(report.Sections.Stale) != 1 || report.Sections.Stale[0].Name != "StaleCheck" {
t.Fatalf("unexpected stale: %+v", report.Sections.Stale)
}
if len(report.Sections.Ok) != 1 || report.Sections.Ok[0].Name != "OkCheck" {
t.Fatalf("unexpected ok: %+v", report.Sections.Ok)
}
if len(report.Sections.Suppressed) != 0 {
t.Fatalf("unexpected suppressed: %+v", report.Sections.Suppressed)
}
}
|
