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
|
package integrationtests
import (
"context"
"fmt"
"os"
"testing"
"github.com/mimecast/dtail/internal/config"
)
func TestDTailHealth1(t *testing.T) {
if !config.Env("DTAIL_INTEGRATION_TEST_RUN_MODE") {
t.Log("Skipping")
return
}
outFile := "dtailhealth1.stdout.tmp"
expectedOutFile := "dtailhealth1.expected"
t.Log("Serverless check, is supposed to exit with warning state.")
exitCode, err := runCommand(context.TODO(), t, outFile, "../dtailhealth")
if exitCode != 1 {
t.Error(fmt.Sprintf("Expected exit code '1' but got '%d': %v", exitCode, err))
return
}
if err := compareFiles(t, outFile, expectedOutFile); err != nil {
t.Error(err)
return
}
os.Remove(outFile)
}
func TestDTailHealth2(t *testing.T) {
if !config.Env("DTAIL_INTEGRATION_TEST_RUN_MODE") {
t.Log("Skipping")
return
}
outFile := "dtailhealth2.stdout.tmp"
expectedOutFile := "dtailhealth2.expected"
t.Log("Negative test, is supposed to exit with a critical state.")
exitCode, err := runCommand(context.TODO(), t, outFile,
"../dtailhealth", "--server", "example:1")
if exitCode != 2 {
t.Error(fmt.Sprintf("Expected exit code '2' but got '%d': %v", exitCode, err))
return
}
if err := compareFiles(t, outFile, expectedOutFile); err != nil {
t.Error(err)
return
}
os.Remove(outFile)
}
func TestDTailHealthCheck3(t *testing.T) {
if !config.Env("DTAIL_INTEGRATION_TEST_RUN_MODE") {
t.Log("Skipping")
return
}
outFile := "dtailhealth3.stdout.tmp"
port := getUniquePortNumber()
bindAddress := "localhost"
expectedOut := fmt.Sprintf("OK: All fine at %s:%d :-)", bindAddress, port)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
_, _, _, err := startCommand(ctx, t,
"", "../dserver",
"--cfg", "none",
"--logger", "stdout",
"--logLevel", "trace",
"--bindAddress", bindAddress,
"--port", fmt.Sprintf("%d", port),
)
if err != nil {
t.Error(err)
return
}
_, err = runCommandRetry(ctx, t, 10, outFile,
"../dtailhealth", "--server", fmt.Sprintf("%s:%d", bindAddress, port))
if err != nil {
t.Error(err)
return
}
if err := fileContainsStr(t, outFile, expectedOut); err != nil {
t.Error(err)
return
}
os.Remove(outFile)
}
|
