http3: fix flaky Extended CONNECT unit test (#4537)

http3/client_test.go

@@ -419,9 +419,12 @@ var _ = Describe("Client", func() {
 		It("rejects Extended CONNECT requests if the server doesn't enable it", func() {
 			sendSettings()
 			done := make(chan struct{})
+			var wg sync.WaitGroup
+			wg.Add(2)
 			conn := mockquic.NewMockEarlyConnection(mockCtrl)
 			conn.EXPECT().OpenUniStream().DoAndReturn(func() (quic.SendStream, error) {
 				<-done
+				wg.Done()
 				return nil, errors.New("test done")
 			}).MaxTimes(1)
 			b := quicvarint.Append(nil, streamTypeControlStream)
@@ -432,6 +435,7 @@ var _ = Describe("Client", func() {
 			conn.EXPECT().AcceptUniStream(gomock.Any()).Return(controlStr, nil)
 			conn.EXPECT().AcceptUniStream(gomock.Any()).DoAndReturn(func(context.Context) (quic.ReceiveStream, error) {
 				<-done
+				wg.Done()
 				return nil, errors.New("test done")
 			})
 			conn.EXPECT().HandshakeComplete().Return(handshakeChan)
@@ -448,6 +452,7 @@ var _ = Describe("Client", func() {
 			// test shutdown
 			conn.EXPECT().CloseWithError(gomock.Any(), gomock.Any()).MaxTimes(1)
 			close(done)
+			wg.Wait()
 		})
 	})