http3: fix flaky TestConnControlStreamFailure (#5188)

This commit is contained in:
Marten Seemann
2025-05-31 16:21:17 +08:00
committed by GitHub
parent f8a1711ce7
commit eb427b599e

View File

@@ -207,6 +207,12 @@ func testConnControlStreamFailures(t *testing.T, data []byte, readErr error, exp
_, err = controlStr.Write(quicvarint.Append(nil, streamTypeControlStream)) _, err = controlStr.Write(quicvarint.Append(nil, streamTypeControlStream))
require.NoError(t, err) require.NoError(t, err)
done := make(chan struct{})
go func() {
defer close(done)
conn.handleUnidirectionalStreams(nil)
}()
switch readErr { switch readErr {
case nil: case nil:
_, err = controlStr.Write(data) _, err = controlStr.Write(data)
@@ -223,11 +229,6 @@ func testConnControlStreamFailures(t *testing.T, data []byte, readErr error, exp
conn.openRequestStream(context.Background(), nil, nil, true, 1000) conn.openRequestStream(context.Background(), nil, nil, true, 1000)
done := make(chan struct{})
go func() {
defer close(done)
conn.handleUnidirectionalStreams(nil)
}()
select { select {
case <-serverConn.Context().Done(): case <-serverConn.Context().Done():
require.ErrorIs(t, require.ErrorIs(t,