init

internal/monitor/monitor_test.go

@@ -0,0 +1,84 @@
+package monitor
+
+import (
+	"context"
+	"net"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"runtime"
+	"testing"
+)
+
+func TestHTTPProbe(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(http.StatusNoContent) }))
+	defer srv.Close()
+	c := Probe(context.Background(), Input{Type: "http", Target: srv.URL, TimeoutMS: 1000, ExpectedMin: 200, ExpectedMax: 299})
+	if !c.OK || c.StatusCode != http.StatusNoContent {
+		t.Fatalf("unexpected check: %+v", c)
+	}
+}
+func TestTCPProbe(t *testing.T) {
+	ln, err := net.Listen("tcp", "127.0.0.1:0")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer ln.Close()
+	c := Probe(context.Background(), Input{Type: "tcp", Target: ln.Addr().String(), TimeoutMS: 1000})
+	if !c.OK {
+		t.Fatalf("unexpected check: %+v", c)
+	}
+}
+
+func TestAggregateServiceStatus(t *testing.T) {
+	cases := []struct {
+		name     string
+		monitors []Monitor
+		want     string
+	}{
+		{"all up", []Monitor{{Status: "up"}, {Status: "up"}}, "up"},
+		{"one fault", []Monitor{{Status: "up"}, {Status: "down"}}, "down"},
+		{"maintenance", []Monitor{{Status: "up"}, {Status: "maintenance"}}, "maintenance"},
+		{"empty", nil, "unknown"},
+	}
+	for _, tc := range cases {
+		t.Run(tc.name, func(t *testing.T) {
+			if got := aggregateStatus(tc.monitors); got != tc.want {
+				t.Fatalf("got %s want %s", got, tc.want)
+			}
+		})
+	}
+}
+
+func TestDockerProbeRunningAndHealthy(t *testing.T) {
+	if runtime.GOOS == "windows" {
+		t.Skip("shell fixture")
+	}
+	d := t.TempDir()
+	path := filepath.Join(d, "docker")
+	if err := os.WriteFile(path, []byte("#!/bin/sh\nprintf '%s\\n' '{\"Running\":true,\"Status\":\"running\",\"Health\":{\"Status\":\"healthy\"}}'\n"), 0755); err != nil {
+		t.Fatal(err)
+	}
+	t.Setenv("PATH", d+string(os.PathListSeparator)+os.Getenv("PATH"))
+	c := Probe(context.Background(), Input{Type: "docker", Target: "app", TimeoutMS: 1000, RequireHealthy: true})
+	if !c.OK {
+		t.Fatalf("expected healthy docker probe, got %+v", c)
+	}
+}
+
+func TestDockerProbeHealthRequiredWithoutHealthcheck(t *testing.T) {
+	if runtime.GOOS == "windows" {
+		t.Skip("shell fixture")
+	}
+	d := t.TempDir()
+	path := filepath.Join(d, "docker")
+	if err := os.WriteFile(path, []byte("#!/bin/sh\nprintf '%s\\n' '{\"Running\":true,\"Status\":\"running\",\"Health\":null}'\n"), 0755); err != nil {
+		t.Fatal(err)
+	}
+	t.Setenv("PATH", d+string(os.PathListSeparator)+os.Getenv("PATH"))
+	c := Probe(context.Background(), Input{Type: "docker", Target: "app", TimeoutMS: 1000, RequireHealthy: true})
+	if c.OK || c.Message != "container has no healthcheck" {
+		t.Fatalf("unexpected docker check: %+v", c)
+	}
+}