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

http3: fix flaky RoundTripper test (#4442)

Browse Source
This commit is contained in:
Marten Seemann
2024-04-17 09:44:20 -07:00
committed by GitHub
parent 457ac2c4dc
commit 788b6f4409
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
http3/roundtrip_test.go
View File

@@ -277,14 +277,15 @@ var _ = Describe("RoundTripper", func() {
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
conn := mockquic.NewMockEarlyConnection(mockCtrl) conn := mockquic.NewMockEarlyConnection(mockCtrl)
handshakeChan := make(chan struct{})
close(handshakeChan)
var count int var count int
rt.Dial = func(context.Context, string, *tls.Config, *quic.Config) (quic.EarlyConnection, error) { rt.Dial = func(context.Context, string, *tls.Config, *quic.Config) (quic.EarlyConnection, error) {
count++ count++
return conn, nil return conn, nil
} }
testErr := errors.New("test err") testErr := errors.New("test err")
handshakeChan := make(chan struct{})
close(handshakeChan)
conn.EXPECT().HandshakeComplete().Return(handshakeChan).MaxTimes(2)
cl1.EXPECT().RoundTripOpt(req1, gomock.Any()).Return(nil, testErr) cl1.EXPECT().RoundTripOpt(req1, gomock.Any()).Return(nil, testErr)
cl2.EXPECT().RoundTripOpt(req2, gomock.Any()).Return(&http.Response{Request: req2}, nil) cl2.EXPECT().RoundTripOpt(req2, gomock.Any()).Return(&http.Response{Request: req2}, nil)
_, err = rt.RoundTrip(req1) _, err = rt.RoundTrip(req1)