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

fix flaky handshake limiting test (#4281)

Browse source
This commit is contained in:
Marten Seemann 2024-01-29 11:26:03 +07:00 • committed by GitHub
parent c82c37a31c
commit 9b83ac230b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
server_test.go
View file

@ -664,6 +664,8 @@ var _ = Describe("Server", func() {
handshakeChan := make(chan struct{}) handshakeChan := make(chan struct{})
connChan := make(chan *MockQUICConn, 1) connChan := make(chan *MockQUICConn, 1)
var wg sync.WaitGroup
wg.Add(2 * limit)
serv.newConn = func( serv.newConn = func(
_ sendConn, _ sendConn,
runner connRunner, runner connRunner,
@ -687,7 +689,7 @@ var _ = Describe("Server", func() {
conn.EXPECT().handlePacket(gomock.Any()) conn.EXPECT().handlePacket(gomock.Any())
conn.EXPECT().run() conn.EXPECT().run()
conn.EXPECT().Context().Return(context.Background()) conn.EXPECT().Context().Return(context.Background())
conn.EXPECT().HandshakeComplete().Return(handshakeChan) conn.EXPECT().HandshakeComplete().Return(handshakeChan).Do(func() <-chan struct{} { wg.Done(); return nil })
return conn return conn
} }
@ -727,6 +729,7 @@ var _ = Describe("Server", func() {
connChan <- conn connChan <- conn
serv.handlePacket(getInitialWithRandomDestConnID()) serv.handlePacket(getInitialWithRandomDestConnID())
} }
wg.Wait()
}) })
It("limits the number of total handshakes", func() { It("limits the number of total handshakes", func() {