From da198b710b4487e42b62361cdcdefda5ebc54a70 Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Mon, 8 May 2023 10:56:59 +0200
Subject: [PATCH] transport: fix flaky stateless reset test (#3810)

---
 transport_test.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/transport_test.go b/transport_test.go
index 9441e4dec..f60db3b2a 100644
--- a/transport_test.go
+++ b/transport_test.go
@@ -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())