startup: make checklist body matching whitespace-tolerant

This commit is contained in:
Brad Stein 2026-04-07 13:57:54 -03:00
parent 2e44c29adf
commit 78faf9a123
2 changed files with 51 additions and 2 deletions

View File

@ -21,6 +21,7 @@ import (
"strings"
"sync"
"time"
"unicode"
"scm.bstein.dev/bstein/ananke/internal/config"
"scm.bstein.dev/bstein/ananke/internal/execx"
@ -2225,12 +2226,12 @@ func (o *Orchestrator) serviceCheckReady(ctx context.Context, check config.Servi
}
bodyContains := strings.TrimSpace(check.BodyContains)
if bodyContains != "" && !strings.Contains(strings.ToLower(body), strings.ToLower(bodyContains)) {
if bodyContains != "" && !checklistContains(body, bodyContains) {
return false, fmt.Sprintf("response missing expected marker %q", bodyContains)
}
bodyNotContains := strings.TrimSpace(check.BodyNotContains)
if bodyNotContains != "" && strings.Contains(strings.ToLower(body), strings.ToLower(bodyNotContains)) {
if bodyNotContains != "" && checklistContains(body, bodyNotContains) {
return false, fmt.Sprintf("response contained forbidden marker %q", bodyNotContains)
}
@ -2272,6 +2273,32 @@ func (o *Orchestrator) httpChecklistProbe(ctx context.Context, check config.Serv
return resp.StatusCode, string(body), nil
}
func checklistContains(body, marker string) bool {
bodyLower := strings.ToLower(body)
markerLower := strings.ToLower(marker)
if strings.Contains(bodyLower, markerLower) {
return true
}
bodyCompact := compactLowerNoSpace(bodyLower)
markerCompact := compactLowerNoSpace(markerLower)
if markerCompact == "" {
return true
}
return strings.Contains(bodyCompact, markerCompact)
}
func compactLowerNoSpace(s string) string {
var b strings.Builder
b.Grow(len(s))
for _, r := range s {
if unicode.IsSpace(r) {
continue
}
b.WriteRune(r)
}
return b.String()
}
func (o *Orchestrator) waitForStabilityWindow(ctx context.Context) error {
window := time.Duration(o.cfg.Startup.ServiceChecklistStabilitySec) * time.Second
if window <= 0 {

View File

@ -192,3 +192,25 @@ func TestServiceCheckReadyRequiresBodyContains(t *testing.T) {
t.Fatalf("expected service check to pass, detail=%s", detail)
}
}
func TestServiceCheckReadyBodyContainsIgnoresWhitespace(t *testing.T) {
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
w.WriteHeader(http.StatusOK)
_, _ = w.Write([]byte("{\n \"database\": \"ok\"\n}\n"))
}))
defer srv.Close()
orch := &Orchestrator{
log: log.New(os.Stdout, "", 0),
}
ok, detail := orch.serviceCheckReady(context.Background(), config.ServiceChecklistCheck{
Name: "grafana-api",
URL: srv.URL,
AcceptedStatuses: []int{200},
BodyContains: `"database":"ok"`,
TimeoutSeconds: 5,
})
if !ok {
t.Fatalf("expected whitespace-tolerant service check to pass, detail=%s", detail)
}
}