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

add context to EarlyConnection.NextConnection, handle handshake failures (#4551)

Browse Source
This commit is contained in:
Marten Seemann
2024-06-05 11:51:54 +08:00
committed by GitHub
parent 0db354456a
commit 07acaad2f7
5 changed files with 34 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
internal/mocks/quic/early_conn.go
View File

@@ -311,17 +311,18 @@ func (c *MockEarlyConnectionLocalAddrCall) DoAndReturn(f func() net.Addr) *MockE
}
// NextConnection mocks base method.
func (m *MockEarlyConnection) NextConnection() quic.Connection {
func (m *MockEarlyConnection) NextConnection(arg0 context.Context) (quic.Connection, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "NextConnection")
ret := m.ctrl.Call(m, "NextConnection", arg0)
ret0, _ := ret[0].(quic.Connection)
return ret0
ret1, _ := ret[1].(error)
return ret0, ret1
}
// NextConnection indicates an expected call of NextConnection.
func (mr *MockEarlyConnectionMockRecorder) NextConnection() *MockEarlyConnectionNextConnectionCall {
func (mr *MockEarlyConnectionMockRecorder) NextConnection(arg0 any) *MockEarlyConnectionNextConnectionCall {
mr.mock.ctrl.T.Helper()
call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NextConnection", reflect.TypeOf((*MockEarlyConnection)(nil).NextConnection))
call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NextConnection", reflect.TypeOf((*MockEarlyConnection)(nil).NextConnection), arg0)
return &MockEarlyConnectionNextConnectionCall{Call: call}
}
@@ -331,19 +332,19 @@ type MockEarlyConnectionNextConnectionCall struct {
}
// Return rewrite *gomock.Call.Return
func (c *MockEarlyConnectionNextConnectionCall) Return(arg0 quic.Connection) *MockEarlyConnectionNextConnectionCall {
c.Call = c.Call.Return(arg0)
func (c *MockEarlyConnectionNextConnectionCall) Return(arg0 quic.Connection, arg1 error) *MockEarlyConnectionNextConnectionCall {
c.Call = c.Call.Return(arg0, arg1)
return c
}
// Do rewrite *gomock.Call.Do
func (c *MockEarlyConnectionNextConnectionCall) Do(f func() quic.Connection) *MockEarlyConnectionNextConnectionCall {
func (c *MockEarlyConnectionNextConnectionCall) Do(f func(context.Context) (quic.Connection, error)) *MockEarlyConnectionNextConnectionCall {
c.Call = c.Call.Do(f)
return c
}
// DoAndReturn rewrite *gomock.Call.DoAndReturn
func (c *MockEarlyConnectionNextConnectionCall) DoAndReturn(f func() quic.Connection) *MockEarlyConnectionNextConnectionCall {
func (c *MockEarlyConnectionNextConnectionCall) DoAndReturn(f func(context.Context) (quic.Connection, error)) *MockEarlyConnectionNextConnectionCall {
c.Call = c.Call.DoAndReturn(f)
return c
}