fix: ignore timeout on websocket (#1802)

This commit is contained in:
Kevin Wan
2022-04-18 20:14:46 +08:00
committed by GitHub
parent d58cf7a12a
commit 92b450eb11
2 changed files with 20 additions and 0 deletions

View File

@@ -79,6 +79,19 @@ func TestTimeoutPanic(t *testing.T) {
})
}
func TestTimeoutWebsocket(t *testing.T) {
timeoutHandler := TimeoutHandler(time.Millisecond)
handler := timeoutHandler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
time.Sleep(time.Millisecond * 10)
}))
req := httptest.NewRequest(http.MethodGet, "http://localhost", nil)
req.Header.Set(headerUpgrade, valueWebsocket)
resp := httptest.NewRecorder()
handler.ServeHTTP(resp, req)
assert.Equal(t, http.StatusOK, resp.Code)
}
func TestTimeoutWroteHeaderTwice(t *testing.T) {
timeoutHandler := TimeoutHandler(time.Minute)
handler := timeoutHandler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {