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

http3: use the connection, not the stream context, on the server side (#4510)

Browse Source
This commit is contained in:
Marten Seemann
2024-06-03 10:23:35 +08:00
committed by GitHub
parent 375fc59084
commit 5446b5f912
8 changed files with 96 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
http3/client_test.go
View File

@@ -326,6 +326,7 @@ var _ = Describe("Client", func() {
return len(b), nil
})
conn := mockquic.NewMockEarlyConnection(mockCtrl)
conn.EXPECT().Context().Return(context.Background())
conn.EXPECT().OpenUniStream().Return(controlStr, nil)
conn.EXPECT().OpenStreamSync(gomock.Any()).DoAndReturn(func(context.Context) (quic.Stream, error) {
<-settingsFrameWritten
@@ -360,6 +361,7 @@ var _ = Describe("Client", func() {
<-done
return nil, errors.New("test done")
}).MaxTimes(1)
conn.EXPECT().Context().Return(context.Background())
b := quicvarint.Append(nil, streamTypeControlStream)
b = (&settingsFrame{ExtendedConnect: true}).Append(b)
r := bytes.NewReader(b)
@@ -392,6 +394,7 @@ var _ = Describe("Client", func() {
wg.Done()
return nil, errors.New("test done")
}).MaxTimes(1)
conn.EXPECT().Context().Return(context.Background())
b := quicvarint.Append(nil, streamTypeControlStream)
b = (&settingsFrame{ExtendedConnect: true}).Append(b)
r := bytes.NewReader(b)
@@ -427,6 +430,7 @@ var _ = Describe("Client", func() {
var wg sync.WaitGroup
wg.Add(2)
conn := mockquic.NewMockEarlyConnection(mockCtrl)
conn.EXPECT().Context().Return(context.Background())
conn.EXPECT().OpenUniStream().DoAndReturn(func() (quic.SendStream, error) {
<-done
wg.Done()
@@ -507,6 +511,7 @@ var _ = Describe("Client", func() {
str.EXPECT().Context().Return(context.Background()).AnyTimes()
str.EXPECT().StreamID().AnyTimes()
conn = mockquic.NewMockEarlyConnection(mockCtrl)
conn.EXPECT().Context().Return(context.Background())
conn.EXPECT().OpenUniStream().Return(controlStr, nil)
conn.EXPECT().AcceptUniStream(gomock.Any()).DoAndReturn(func(context.Context) (quic.ReceiveStream, error) {
<-testDone