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

fix flaky TestServerClose (#4973)

Browse Source
This commit is contained in:
Marten Seemann
2025-03-06 08:30:52 +01:00
committed by GitHub
parent 08a89e9f8b
commit 565b7f75ab
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server_test.go
View File

@@ -737,10 +737,10 @@ func TestServerClose(t *testing.T) {
done := make(chan struct{}, numConns)
for range numConns {
conn := NewMockQUICConn(mockCtrl)
conn.EXPECT().run()
conn.EXPECT().handlePacket(gomock.Any())
conn.EXPECT().Context().Return(context.Background())
conn.EXPECT().HandshakeComplete().Return(make(chan struct{})) // doesn't complete handshake
conn.EXPECT().run().MaxTimes(1)
conn.EXPECT().handlePacket(gomock.Any()).MaxTimes(1)
conn.EXPECT().Context().Return(context.Background()).MaxTimes(1)
conn.EXPECT().HandshakeComplete().Return(make(chan struct{})).MaxTimes(1) // doesn't complete handshake
conn.EXPECT().closeWithTransportError(ConnectionRefused).Do(func(TransportErrorCode) { done <- struct{}{} })
conns = append(conns, conn)
}