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

Merge pull request #2537 from lucas-clemente/fix-flaky-session-test

Browse Source 
fix flaky session unit test
This commit is contained in:
Marten Seemann
2020-05-13 20:53:45 +07:00
committed by GitHub
parent e569a3d98e 33694fd77b
commit 83be7daef3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
session_test.go
View File

@@ -1036,7 +1036,7 @@ var _ = Describe("Session", func() {
runSession() runSession()
p := getPacket(1) p := getPacket(1)
packer.EXPECT().PackPacket().Return(p, nil) packer.EXPECT().PackPacket().Return(p, nil)
packer.EXPECT().PackPacket().Return(nil, nil) packer.EXPECT().PackPacket().Return(nil, nil).AnyTimes()
sent := make(chan struct{}) sent := make(chan struct{})
mconn.EXPECT().Write(gomock.Any()).Do(func([]byte) { close(sent) }) mconn.EXPECT().Write(gomock.Any()).Do(func([]byte) { close(sent) })
qlogger.EXPECT().SentPacket(p.header, p.buffer.Len(), nil, []wire.Frame{}) qlogger.EXPECT().SentPacket(p.header, p.buffer.Len(), nil, []wire.Frame{})
@@ -1047,7 +1047,7 @@ var _ = Describe("Session", func() {
It("doesn't send packets if there's nothing to send", func() { It("doesn't send packets if there's nothing to send", func() {
sess.handshakeConfirmed = true sess.handshakeConfirmed = true
runSession() runSession()
packer.EXPECT().PackPacket().Return(nil, nil) packer.EXPECT().PackPacket().Return(nil, nil).AnyTimes()
sess.receivedPacketHandler.ReceivedPacket(0x035e, protocol.Encryption1RTT, time.Now(), true) sess.receivedPacketHandler.ReceivedPacket(0x035e, protocol.Encryption1RTT, time.Now(), true)
sess.scheduleSending() sess.scheduleSending()
time.Sleep(50 * time.Millisecond) // make sure there are no calls to mconn.Write() time.Sleep(50 * time.Millisecond) // make sure there are no calls to mconn.Write()
@@ -1074,7 +1074,7 @@ var _ = Describe("Session", func() {
fc.EXPECT().IsNewlyBlocked() fc.EXPECT().IsNewlyBlocked()
p := getPacket(1) p := getPacket(1)
packer.EXPECT().PackPacket().Return(p, nil) packer.EXPECT().PackPacket().Return(p, nil)
packer.EXPECT().PackPacket().Return(nil, nil) packer.EXPECT().PackPacket().Return(nil, nil).AnyTimes()
sess.connFlowController = fc sess.connFlowController = fc
runSession() runSession()
sent := make(chan struct{}) sent := make(chan struct{})