Added tests and readme

2026-05-12 06:20:51 +00:00
parent 7e85bb0fcc
commit 139c224a31
23 changed files with 1449 additions and 97 deletions
@@ -0,0 +1,118 @@
+package checks
+
+import (
+	"context"
+	"net"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/jasper/quptime/internal/config"
+)
+
+func TestHTTPProberHappyPath(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
+		w.WriteHeader(200)
+		w.Write([]byte("hello world"))
+	}))
+	defer srv.Close()
+
+	res := Run(context.Background(), &config.Check{
+		ID: "c", Type: config.CheckHTTP, Target: srv.URL,
+		Timeout: 5 * time.Second, ExpectStatus: 200,
+	})
+	if !res.OK {
+		t.Errorf("expected OK, got %+v", res)
+	}
+}
+
+func TestHTTPProberBodyMatch(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
+		w.WriteHeader(200)
+		w.Write([]byte("the magic word is xyzzy and other stuff"))
+	}))
+	defer srv.Close()
+
+	hit := Run(context.Background(), &config.Check{
+		ID: "c", Type: config.CheckHTTP, Target: srv.URL,
+		Timeout: 5 * time.Second, BodyMatch: "xyzzy",
+	})
+	if !hit.OK {
+		t.Errorf("expected match, got %+v", hit)
+	}
+
+	miss := Run(context.Background(), &config.Check{
+		ID: "c", Type: config.CheckHTTP, Target: srv.URL,
+		Timeout: 5 * time.Second, BodyMatch: "absent",
+	})
+	if miss.OK {
+		t.Errorf("expected miss, got %+v", miss)
+	}
+	if !strings.Contains(miss.Detail, "body match") {
+		t.Errorf("detail unexpected: %q", miss.Detail)
+	}
+}
+
+func TestHTTPProberStatusMismatch(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
+		w.WriteHeader(500)
+	}))
+	defer srv.Close()
+
+	res := Run(context.Background(), &config.Check{
+		ID: "c", Type: config.CheckHTTP, Target: srv.URL, Timeout: 5 * time.Second,
+	})
+	if res.OK {
+		t.Errorf("500 should fail check, got %+v", res)
+	}
+}
+
+func TestTCPProberHappyPath(t *testing.T) {
+	ln, err := net.Listen("tcp", "127.0.0.1:0")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer ln.Close()
+	go func() {
+		for {
+			c, err := ln.Accept()
+			if err != nil {
+				return
+			}
+			c.Close()
+		}
+	}()
+
+	res := Run(context.Background(), &config.Check{
+		ID: "c", Type: config.CheckTCP, Target: ln.Addr().String(),
+		Timeout: 2 * time.Second,
+	})
+	if !res.OK {
+		t.Errorf("expected OK, got %+v", res)
+	}
+}
+
+func TestTCPProberRefusedConnection(t *testing.T) {
+	// Listen and immediately close so the address is known-bad.
+	ln, _ := net.Listen("tcp", "127.0.0.1:0")
+	addr := ln.Addr().String()
+	ln.Close()
+
+	res := Run(context.Background(), &config.Check{
+		ID: "c", Type: config.CheckTCP, Target: addr, Timeout: 1 * time.Second,
+	})
+	if res.OK {
+		t.Errorf("dead address should fail check, got %+v", res)
+	}
+}
+
+func TestRunUnknownCheckType(t *testing.T) {
+	res := Run(context.Background(), &config.Check{
+		ID: "c", Type: "bogus", Target: "x",
+	})
+	if res.OK {
+		t.Error("unknown check type should not succeed")
+	}
+}