From 54b97a507905d78f3b46bf4fb52b84ed804fd5f4 Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Mon, 20 Jan 2025 22:05:50 -0800
Subject: [PATCH] http3: fix flaky transport tests (#4900)

---
 http3/transport_test.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/http3/transport_test.go b/http3/transport_test.go
index a2f90378..2ddc83cb 100644
--- a/http3/transport_test.go
+++ b/http3/transport_test.go
@@ -221,7 +221,7 @@ func TestTransportConnectionReuse(t *testing.T) {
 	conn := mockquic.NewMockEarlyConnection(mockCtrl)
 	handshakeChan := make(chan struct{})
 	close(handshakeChan)
-	conn.EXPECT().HandshakeComplete().Return(handshakeChan).MaxTimes(2)
+	conn.EXPECT().HandshakeComplete().Return(handshakeChan).AnyTimes()
 	var dialCount int
 	tr := &Transport{
 		Dial: func(context.Context, string, *tls.Config, *quic.Config) (quic.EarlyConnection, error) {
@@ -342,7 +342,7 @@ func TestTransportRequestContextCancellation(t *testing.T) {
 	conn := mockquic.NewMockEarlyConnection(mockCtrl)
 	handshakeChan := make(chan struct{})
 	close(handshakeChan)
-	conn.EXPECT().HandshakeComplete().Return(handshakeChan).MaxTimes(3)
+	conn.EXPECT().HandshakeComplete().Return(handshakeChan).AnyTimes()
 	var dialCount int
 	tr := &Transport{
 		Dial: func(context.Context, string, *tls.Config, *quic.Config) (quic.EarlyConnection, error) {
@@ -389,7 +389,7 @@ func TestTransportConnetionRedialHandshakeError(t *testing.T) {
 	conn := mockquic.NewMockEarlyConnection(mockCtrl)
 	handshakeChan := make(chan struct{})
 	close(handshakeChan)
-	conn.EXPECT().HandshakeComplete().Return(handshakeChan).MaxTimes(2)
+	conn.EXPECT().HandshakeComplete().Return(handshakeChan).AnyTimes()
 	var dialCount int
 	testErr := errors.New("handshake error")
 	tr := &Transport{