From f2a1206c20499e3dfe5ffb2ce0be03cf42c8c141 Mon Sep 17 00:00:00 2001 From: Marten Seemann Date: Tue, 26 Dec 2017 10:57:20 +0700 Subject: [PATCH] fix flaky send stream test --- send_stream_test.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/send_stream_test.go b/send_stream_test.go index f9629fcad..8e4df2250 100644 --- a/send_stream_test.go +++ b/send_stream_test.go @@ -486,13 +486,15 @@ var _ = Describe("Send Stream", func() { It("says when it has data for sending", func() { mockFC.EXPECT().UpdateSendWindow(gomock.Any()) - mockSender.EXPECT().onHasStreamData(streamID) + mockSender.EXPECT().onHasStreamData(streamID).Times(2) // once for Write, once for the MAX_STREAM_DATA frame done := make(chan struct{}) go func() { defer GinkgoRecover() - _, _ = str.Write([]byte("foobar")) + _, err := str.Write([]byte("foobar")) + Expect(err).ToNot(HaveOccurred()) close(done) }() + waitForWrite() str.handleMaxStreamDataFrame(&wire.MaxStreamDataFrame{ StreamID: streamID, ByteOffset: 42,