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

transport: fix flaky stateless reset test (#3810)

Browse source
This commit is contained in:
Marten Seemann 2023-05-08 10:56:59 +02:00 • committed by GitHub
parent 22d6b0e626
commit da198b710b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
transport_test.go
View file

@ -235,13 +235,16 @@ var _ = Describe("Transport", func() {
Expect(err).ToNot(HaveOccurred())
b = append(b, token[:]...)
conn := NewMockPacketHandler(mockCtrl)
destroyed := make(chan struct{})
gomock.InOrder(
phm.EXPECT().GetByResetToken(token).Return(conn, true),
conn.EXPECT().destroy(gomock.Any()).Do(func(err error) {
Expect(err).To(MatchError(&StatelessResetError{Token: token}))
close(destroyed)
}),
)
packetChan <- packetToRead{data: b}
Eventually(destroyed).Should(BeClosed())
// shutdown
phm.EXPECT().Close(gomock.Any())