fix: timeout handler not implementing http.Flusher (#3225)
This commit is contained in:
@@ -164,6 +164,24 @@ func TestTimeoutHijack(t *testing.T) {
|
||||
})
|
||||
}
|
||||
|
||||
func TestTimeoutFlush(t *testing.T) {
|
||||
timeoutHandler := TimeoutHandler(time.Minute)
|
||||
handler := timeoutHandler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
flusher, ok := w.(http.Flusher)
|
||||
if !ok {
|
||||
http.Error(w, "Streaming unsupported!", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
flusher.Flush()
|
||||
}))
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "http://localhost", http.NoBody)
|
||||
resp := httptest.NewRecorder()
|
||||
handler.ServeHTTP(resp, req)
|
||||
assert.Equal(t, http.StatusOK, resp.Code)
|
||||
}
|
||||
|
||||
func TestTimeoutPusher(t *testing.T) {
|
||||
handler := &timeoutWriter{
|
||||
w: mockedPusher{},
|
||||
|
||||
Reference in New Issue
Block a user