feat(platform-api): enforce request timeout

This commit is contained in:
lingniu
2026-07-04 00:55:43 +08:00
parent 081f8b5099
commit 4bfaf8c4c8
5 changed files with 57 additions and 1 deletions

View File

@@ -0,0 +1,29 @@
package app
import (
"net/http"
"net/http/httptest"
"testing"
"time"
)
func TestWithRequestTimeoutAddsContextDeadline(t *testing.T) {
handler := withRequestTimeout(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
deadline, ok := r.Context().Deadline()
if !ok {
t.Fatal("request context should have deadline")
}
if time.Until(deadline) > 250*time.Millisecond {
t.Fatalf("deadline too far away: %s", time.Until(deadline))
}
w.WriteHeader(http.StatusNoContent)
}), 200*time.Millisecond)
rec := httptest.NewRecorder()
req := httptest.NewRequest(http.MethodGet, "/api/ops/health", nil)
handler.ServeHTTP(rec, req)
if rec.Code != http.StatusNoContent {
t.Fatalf("status = %d", rec.Code)
}
}