Merge pull request #1084 from lucas-clemente/fix-1083

fix flaky send stream test
This commit is contained in:
Marten Seemann
2018-01-03 09:55:32 +07:00
committed by GitHub

View File

@@ -486,13 +486,15 @@ var _ = Describe("Send Stream", func() {
It("says when it has data for sending", func() { It("says when it has data for sending", func() {
mockFC.EXPECT().UpdateSendWindow(gomock.Any()) 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{}) done := make(chan struct{})
go func() { go func() {
defer GinkgoRecover() defer GinkgoRecover()
_, _ = str.Write([]byte("foobar")) _, err := str.Write([]byte("foobar"))
Expect(err).ToNot(HaveOccurred())
close(done) close(done)
}() }()
waitForWrite()
str.handleMaxStreamDataFrame(&wire.MaxStreamDataFrame{ str.handleMaxStreamDataFrame(&wire.MaxStreamDataFrame{
StreamID: streamID, StreamID: streamID,
ByteOffset: 42, ByteOffset: 42,