1 files changed, 95 insertions, 0 deletions
diff --git a/integrationtests/dtailhealth_test.go b/integrationtests/dtailhealth_test.go
new file mode 100644
index 0000000..8c0918f
--- /dev/null
+++ b/integrationtests/dtailhealth_test.go
@@ -0,0 +1,95 @@
+package integrationtests
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/mimecast/dtail/internal/config"
+)
+
+func TestDTailHealthCheck(t *testing.T) {
+	if !config.Env("DTAIL_RUN_INTEGRATION_TESTS") {
+		t.Log("Skipping")
+		return
+	}
+	stdoutFile := "dtailhealth.stdout.tmp"
+	expectedStdoutFile := "dtailhealth.expected"
+
+	t.Log("Serverless check, is supposed to exit with warning state.")
+	exitCode, err := runCommand(context.TODO(), t, stdoutFile, "../dtailhealth")
+	if exitCode != 1 {
+		t.Error(fmt.Sprintf("Expected exit code '1' but got '%d': %v", exitCode, err))
+		return
+	}
+
+	if err := compareFiles(t, stdoutFile, expectedStdoutFile); err != nil {
+		t.Error(err)
+		return
+	}
+	os.Remove(stdoutFile)
+}
+
+func TestDTailHealthCheck2(t *testing.T) {
+	if !config.Env("DTAIL_RUN_INTEGRATION_TESTS") {
+		t.Log("Skipping")
+		return
+	}
+	stdoutFile := "dtailhealth2.stdout.tmp"
+	expectedStdoutFile := "dtailhealth2.expected"
+
+	t.Log("Negative test, is supposed to exit with a critical state.")
+	exitCode, err := runCommand(context.TODO(), t, stdoutFile,
+		"../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, stdoutFile, expectedStdoutFile); err != nil {
+		t.Error(err)
+		return
+	}
+
+	os.Remove(stdoutFile)
+}
+
+func TestDTailHealthCheck3(t *testing.T) {
+	if !config.Env("DTAIL_RUN_INTEGRATION_TESTS") {
+		t.Log("Skipping")
+		return
+	}
+	stdoutFile := "dtailhealth3.stdout.tmp"
+	expectedStdoutFile := "dtailhealth3.expected"
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	_, _, _, err := startCommand(ctx, t,
+		"../dserver",
+		"--logger", "stdout",
+		"--logLevel", "trace",
+		"--bindAddress", "localhost",
+		"--port", "4242",
+	)
+	if err != nil {
+		t.Error(err)
+		return
+	}
+
+	_, err = runCommandRetry(ctx, t, 10, stdoutFile,
+		"../dtailhealth", "--server", "localhost:4242")
+	if err != nil {
+		t.Error(err)
+		return
+	}
+
+	if err := compareFiles(t, stdoutFile, expectedStdoutFile); err != nil {
+		t.Error(err)
+		return
+	}
+
+	os.Remove(stdoutFile)
+}