diff --git a/conn_test.go b/conn_test.go index 591ef13..141c102 100644 --- a/conn_test.go +++ b/conn_test.go @@ -88,7 +88,7 @@ func TestHTTPRWConn(t *testing.T) { flusher.Flush() fw := &flushWrite{w: w, f: flusher} doneCh := make(chan struct{}) - c2 = NewConn(r.Body, fw, SetCloseHook(func() { + c2 = NewConn(r.Body, fw, SetWriteDelay(500*time.Millisecond), SetCloseHook(func() { // exit the handler close(doneCh) }))