feat(platform-api): return traceable timeout errors

This commit is contained in:
lingniu
2026-07-04 00:57:53 +08:00
parent 4bfaf8c4c8
commit 377cfcdfca
2 changed files with 99 additions and 1 deletions

View File

@@ -1,6 +1,7 @@
package app
import (
"encoding/json"
"net/http"
"net/http/httptest"
"testing"
@@ -27,3 +28,31 @@ func TestWithRequestTimeoutAddsContextDeadline(t *testing.T) {
t.Fatalf("status = %d", rec.Code)
}
}
func TestWithRequestTimeoutReturnsEnvelopeWithTraceID(t *testing.T) {
handler := withRequestTimeout(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
<-r.Context().Done()
}), time.Millisecond)
rec := httptest.NewRecorder()
req := httptest.NewRequest(http.MethodGet, "/api/ops/health", nil)
req.Header.Set("X-Trace-Id", "trace-from-test")
handler.ServeHTTP(rec, req)
if rec.Code != http.StatusServiceUnavailable {
t.Fatalf("status = %d body=%s", rec.Code, rec.Body.String())
}
var body struct {
Error struct {
Code string `json:"code"`
Message string `json:"message"`
} `json:"error"`
TraceID string `json:"traceId"`
}
if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
t.Fatalf("timeout response should be JSON: %v body=%s", err, rec.Body.String())
}
if body.Error.Code != "REQUEST_TIMEOUT" || body.Error.Message != "请求处理超时" || body.TraceID != "trace-from-test" {
t.Fatalf("unexpected timeout envelope: %+v body=%s", body, rec.Body.String())
}
}