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

fix flaky handshake limiting test (#4270)

Browse source
This commit is contained in:
Marten Seemann 2024-01-25 00:53:20 -08:00 • committed by GitHub
parent baeec0f41c
commit d3974e1674
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
server_test.go
View file

@ -807,12 +807,17 @@ var _ = Describe("Server", func() {
_, err := serv.Accept(context.Background()) _, err := serv.Accept(context.Background())
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
} }
// make sure we can enqueue and accept more connections after that
for i := 0; i < limit; i++ { for i := 0; i < limit; i++ {
conn := NewMockQUICConn(mockCtrl) conn := NewMockQUICConn(mockCtrl)
conn.EXPECT().closeWithTransportError(gomock.Any()).MaxTimes(1) // called when the server is closed conn.EXPECT().closeWithTransportError(gomock.Any()).MaxTimes(1) // called when the server is closed
connChan <- conn connChan <- conn
serv.handlePacket(getInitialWithRandomDestConnID()) serv.handlePacket(getInitialWithRandomDestConnID())
} }
for i := 0; i < limit; i++ {
_, err := serv.Accept(context.Background())
Expect(err).ToNot(HaveOccurred())
}
}) })
}) })