gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

fix racy session unit test

Browse Source
This commit is contained in:
Marten Seemann
2020-05-06 11:15:12 +07:00
parent 6bcb4e69ee
commit 5a169b3c9a
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
session_test.go
View File

@@ -1055,13 +1055,16 @@ var _ = Describe("Session", func() {
It("sends ACK only packets", func() { It("sends ACK only packets", func() {
sph := mockackhandler.NewMockSentPacketHandler(mockCtrl) sph := mockackhandler.NewMockSentPacketHandler(mockCtrl)
sph.EXPECT().TimeUntilSend().AnyTimes()
sph.EXPECT().GetLossDetectionTimeout().AnyTimes() sph.EXPECT().GetLossDetectionTimeout().AnyTimes()
sph.EXPECT().SendMode().Return(ackhandler.SendAck) sph.EXPECT().SendMode().Return(ackhandler.SendAck)
sph.EXPECT().ShouldSendNumPackets().Return(1000) sph.EXPECT().ShouldSendNumPackets().Return(1000)
packer.EXPECT().MaybePackAckPacket(false) done := make(chan struct{})
runSession() packer.EXPECT().MaybePackAckPacket(false).Do(func(bool) { close(done) })
sess.sentPacketHandler = sph sess.sentPacketHandler = sph
Expect(sess.sendPackets()).To(Succeed()) runSession()
sess.scheduleSending()
Eventually(done).Should(BeClosed())
}) })
It("adds a BLOCKED frame when it is connection-level flow control blocked", func() { It("adds a BLOCKED frame when it is connection-level flow control blocked", func() {