fix flaky TestSendStreamStopSending (#4832)

This commit is contained in:
Marten Seemann
2025-01-03 23:17:37 +08:00
committed by GitHub
parent ea6d198c3d
commit 9b55fbe0be

View File

@@ -763,7 +763,7 @@ func TestSendStreamStopSending(t *testing.T) {
mockSender := NewMockStreamSender(mockCtrl)
str := newSendStream(context.Background(), streamID, mockSender, mockFC)
mockSender.EXPECT().onHasStreamData(streamID, str)
mockSender.EXPECT().onHasStreamData(streamID, str).MaxTimes(2)
_, err := (&writerWithTimeout{Writer: str, Timeout: time.Second}).Write([]byte("foobar"))
require.NoError(t, err)
mockFC.EXPECT().SendWindowSize().Return(protocol.MaxByteCount)