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

fix flaky server unit test

Browse Source
This commit is contained in:
Marten Seemann
2020-02-25 14:19:19 +07:00
parent 51f51d5a71
commit 6c02bd3a30
1 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
server_test.go
View File

@@ -806,19 +806,14 @@ var _ = Describe("Server", func() {
return sess return sess
} }
var wg sync.WaitGroup phm.EXPECT().GetStatelessResetToken(gomock.Any()).Times(protocol.MaxAcceptQueueSize)
wg.Add(protocol.MaxAcceptQueueSize) phm.EXPECT().Add(gomock.Any(), gomock.Any()).Return(true).Times(2 * protocol.MaxAcceptQueueSize)
for i := 0; i < protocol.MaxAcceptQueueSize; i++ { for i := 0; i < protocol.MaxAcceptQueueSize; i++ {
go func() {
defer GinkgoRecover()
defer wg.Done()
phm.EXPECT().GetStatelessResetToken(gomock.Any())
phm.EXPECT().Add(gomock.Any(), gomock.Any()).Return(true).Times(2)
serv.handlePacket(getInitialWithRandomDestConnID()) serv.handlePacket(getInitialWithRandomDestConnID())
Consistently(conn.dataWritten).ShouldNot(Receive())
}()
} }
wg.Wait()
Eventually(func() int32 { return atomic.LoadInt32(&serv.sessionQueueLen) }).Should(BeEquivalentTo(protocol.MaxAcceptQueueSize))
Consistently(conn.dataWritten).ShouldNot(Receive())
p := getInitialWithRandomDestConnID() p := getInitialWithRandomDestConnID()
hdr := parseHeader(p.data) hdr := parseHeader(p.data)