github.com/apernet/quic-go@v0.43.1-0.20240515053213-5e9e635fd9f0/mock_batch_conn_test.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: sys_conn_oob.go
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -typed -package quic -self_package github.com/apernet/quic-go -source sys_conn_oob.go -destination mock_batch_conn_test.go -mock_names batchConn=MockBatchConn
     7  //
     8  
     9  // Package quic is a generated GoMock package.
    10  package quic
    11  
    12  import (
    13  	reflect "reflect"
    14  
    15  	gomock "go.uber.org/mock/gomock"
    16  	ipv4 "golang.org/x/net/ipv4"
    17  )
    18  
    19  // MockBatchConn is a mock of batchConn interface.
    20  type MockBatchConn struct {
    21  	ctrl     *gomock.Controller
    22  	recorder *MockBatchConnMockRecorder
    23  }
    24  
    25  // MockBatchConnMockRecorder is the mock recorder for MockBatchConn.
    26  type MockBatchConnMockRecorder struct {
    27  	mock *MockBatchConn
    28  }
    29  
    30  // NewMockBatchConn creates a new mock instance.
    31  func NewMockBatchConn(ctrl *gomock.Controller) *MockBatchConn {
    32  	mock := &MockBatchConn{ctrl: ctrl}
    33  	mock.recorder = &MockBatchConnMockRecorder{mock}
    34  	return mock
    35  }
    36  
    37  // EXPECT returns an object that allows the caller to indicate expected use.
    38  func (m *MockBatchConn) EXPECT() *MockBatchConnMockRecorder {
    39  	return m.recorder
    40  }
    41  
    42  // ReadBatch mocks base method.
    43  func (m *MockBatchConn) ReadBatch(ms []ipv4.Message, flags int) (int, error) {
    44  	m.ctrl.T.Helper()
    45  	ret := m.ctrl.Call(m, "ReadBatch", ms, flags)
    46  	ret0, _ := ret[0].(int)
    47  	ret1, _ := ret[1].(error)
    48  	return ret0, ret1
    49  }
    50  
    51  // ReadBatch indicates an expected call of ReadBatch.
    52  func (mr *MockBatchConnMockRecorder) ReadBatch(ms, flags any) *MockBatchConnReadBatchCall {
    53  	mr.mock.ctrl.T.Helper()
    54  	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadBatch", reflect.TypeOf((*MockBatchConn)(nil).ReadBatch), ms, flags)
    55  	return &MockBatchConnReadBatchCall{Call: call}
    56  }
    57  
    58  // MockBatchConnReadBatchCall wrap *gomock.Call
    59  type MockBatchConnReadBatchCall struct {
    60  	*gomock.Call
    61  }
    62  
    63  // Return rewrite *gomock.Call.Return
    64  func (c *MockBatchConnReadBatchCall) Return(arg0 int, arg1 error) *MockBatchConnReadBatchCall {
    65  	c.Call = c.Call.Return(arg0, arg1)
    66  	return c
    67  }
    68  
    69  // Do rewrite *gomock.Call.Do
    70  func (c *MockBatchConnReadBatchCall) Do(f func([]ipv4.Message, int) (int, error)) *MockBatchConnReadBatchCall {
    71  	c.Call = c.Call.Do(f)
    72  	return c
    73  }
    74  
    75  // DoAndReturn rewrite *gomock.Call.DoAndReturn
    76  func (c *MockBatchConnReadBatchCall) DoAndReturn(f func([]ipv4.Message, int) (int, error)) *MockBatchConnReadBatchCall {
    77  	c.Call = c.Call.DoAndReturn(f)
    78  	return c
    79  }