mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-04 12:47:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #2384 from lucas-clemente/fix-flaky-server-test

Browse source 
fix flaky server unit test
This commit is contained in:
Marten Seemann 2020-02-28 15:06:22 +07:00 • committed by GitHub
commit 715654ace9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

17
server_test.go
View file

@ -875,19 +875,14 @@ var _ = Describe("Server", func() {
return sess
}
var wg sync.WaitGroup
wg.Add(protocol.MaxAcceptQueueSize)
phm.EXPECT().GetStatelessResetToken(gomock.Any()).Times(protocol.MaxAcceptQueueSize)
phm.EXPECT().Add(gomock.Any(), gomock.Any()).Return(true).Times(2 * protocol.MaxAcceptQueueSize)
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())
Consistently(conn.dataWritten).ShouldNot(Receive())
}()
serv.handlePacket(getInitialWithRandomDestConnID())
}
wg.Wait()
Eventually(func() int32 { return atomic.LoadInt32(&serv.sessionQueueLen) }).Should(BeEquivalentTo(protocol.MaxAcceptQueueSize))
Consistently(conn.dataWritten).ShouldNot(Receive())
p := getInitialWithRandomDestConnID()
hdr := parseHeader(p.data)