完善本地质量门禁与资金链路测试
This commit is contained in:
@@ -0,0 +1,75 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func TestHealthLiveDoesNotDependOnReadiness(t *testing.T) {
|
||||
gin.SetMode(gin.TestMode)
|
||||
h := NewHealthHandler(map[string]ReadinessCheck{
|
||||
"mysql": func(context.Context) error { return errors.New("down") },
|
||||
})
|
||||
recorder := httptest.NewRecorder()
|
||||
ctx, _ := gin.CreateTestContext(recorder)
|
||||
ctx.Request = httptest.NewRequest(http.MethodGet, "/health", nil)
|
||||
|
||||
h.Live(ctx)
|
||||
|
||||
if recorder.Code != http.StatusOK {
|
||||
t.Fatalf("status = %d, want %d", recorder.Code, http.StatusOK)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthReadyReportsDependencyState(t *testing.T) {
|
||||
gin.SetMode(gin.TestMode)
|
||||
tests := []struct {
|
||||
name string
|
||||
checks map[string]ReadinessCheck
|
||||
wantStatus int
|
||||
wantBody string
|
||||
}{
|
||||
{
|
||||
name: "all dependencies ready",
|
||||
checks: map[string]ReadinessCheck{
|
||||
"mysql": func(context.Context) error { return nil },
|
||||
"redis": func(context.Context) error { return nil },
|
||||
},
|
||||
wantStatus: http.StatusOK,
|
||||
wantBody: `"status":"ok"`,
|
||||
},
|
||||
{
|
||||
name: "dependency unavailable",
|
||||
checks: map[string]ReadinessCheck{
|
||||
"mysql": func(context.Context) error { return errors.New("down") },
|
||||
"redis": func(context.Context) error { return nil },
|
||||
},
|
||||
wantStatus: http.StatusServiceUnavailable,
|
||||
wantBody: `"mysql":"unavailable"`,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
h := NewHealthHandler(tt.checks)
|
||||
recorder := httptest.NewRecorder()
|
||||
ctx, _ := gin.CreateTestContext(recorder)
|
||||
ctx.Request = httptest.NewRequest(http.MethodGet, "/health/ready", nil)
|
||||
|
||||
h.Ready(ctx)
|
||||
|
||||
if recorder.Code != tt.wantStatus {
|
||||
t.Fatalf("status = %d, want %d; body=%s", recorder.Code, tt.wantStatus, recorder.Body.String())
|
||||
}
|
||||
if body := recorder.Body.String(); !strings.Contains(body, tt.wantBody) {
|
||||
t.Fatalf("body = %s, want substring %s", body, tt.wantBody)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user