http3: fix flaky TestClientStreamHijacking (#5169)

diff --git a/http3/client_test.go b/http3/client_test.go
index 48e8ad42f7fd3831b8103f5ee1b7630736dc52a3..d65c4f533294e9b5dbd0ff0eacd192e6c1e0a3eb 100644 (file)
--- a/http3/client_test.go
+++ b/http3/client_test.go
@@ -542,24 +542,26 @@ func testClientStreamHijacking(t *testing.T, bidirectional, doHijack bool, strea
 
        clientConn, serverConn := newConnPair(t)
 
-       buf := bytes.NewBuffer(quicvarint.Append(nil, 0x41))
+       b := quicvarint.Append(nil, 0x41)
        if bidirectional {
                str, err := serverConn.OpenStream()
                require.NoError(t, err)
-               _, err = str.Write(buf.Bytes())
+               _, err = str.Write(b)
                require.NoError(t, err)
 
                if streamReadErr != nil {
                        str.CancelWrite(1337)
+                       time.Sleep(scaleDuration(10 * time.Millisecond)) // wait for the reset to be received
                }
        } else {
                str, err := serverConn.OpenUniStream()
                require.NoError(t, err)
-               _, err = str.Write(buf.Bytes())
+               _, err = str.Write(b)
                require.NoError(t, err)
 
                if streamReadErr != nil {
                        str.CancelWrite(1337)
+                       time.Sleep(scaleDuration(10 * time.Millisecond)) // wait for the reset to be received
                }
        }