forked from quic-go/quic-go
http3: fix flaky TestClientStreamHijacking (#5169)
This commit is contained in:
@@ -542,24 +542,26 @@ func testClientStreamHijacking(t *testing.T, bidirectional, doHijack bool, strea
|
|||||||
|
|
||||||
clientConn, serverConn := newConnPair(t)
|
clientConn, serverConn := newConnPair(t)
|
||||||
|
|
||||||
buf := bytes.NewBuffer(quicvarint.Append(nil, 0x41))
|
b := quicvarint.Append(nil, 0x41)
|
||||||
if bidirectional {
|
if bidirectional {
|
||||||
str, err := serverConn.OpenStream()
|
str, err := serverConn.OpenStream()
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
_, err = str.Write(buf.Bytes())
|
_, err = str.Write(b)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
if streamReadErr != nil {
|
if streamReadErr != nil {
|
||||||
str.CancelWrite(1337)
|
str.CancelWrite(1337)
|
||||||
|
time.Sleep(scaleDuration(10 * time.Millisecond)) // wait for the reset to be received
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
str, err := serverConn.OpenUniStream()
|
str, err := serverConn.OpenUniStream()
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
_, err = str.Write(buf.Bytes())
|
_, err = str.Write(b)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
if streamReadErr != nil {
|
if streamReadErr != nil {
|
||||||
str.CancelWrite(1337)
|
str.CancelWrite(1337)
|
||||||
|
time.Sleep(scaleDuration(10 * time.Millisecond)) // wait for the reset to be received
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user