github.com/mikelsr/quic-go@v0.36.1-0.20230701132136-1d9415b66898/mock_quic_conn_test.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/mikelsr/quic-go (interfaces: QUICConn)
     3  
     4  // Package quic is a generated GoMock package.
     5  package quic
     6  
     7  import (
     8  	context "context"
     9  	net "net"
    10  	reflect "reflect"
    11  
    12  	gomock "github.com/golang/mock/gomock"
    13  	protocol "github.com/mikelsr/quic-go/internal/protocol"
    14  	qerr "github.com/mikelsr/quic-go/internal/qerr"
    15  )
    16  
    17  // MockQUICConn is a mock of QUICConn interface.
    18  type MockQUICConn struct {
    19  	ctrl     *gomock.Controller
    20  	recorder *MockQUICConnMockRecorder
    21  }
    22  
    23  // MockQUICConnMockRecorder is the mock recorder for MockQUICConn.
    24  type MockQUICConnMockRecorder struct {
    25  	mock *MockQUICConn
    26  }
    27  
    28  // NewMockQUICConn creates a new mock instance.
    29  func NewMockQUICConn(ctrl *gomock.Controller) *MockQUICConn {
    30  	mock := &MockQUICConn{ctrl: ctrl}
    31  	mock.recorder = &MockQUICConnMockRecorder{mock}
    32  	return mock
    33  }
    34  
    35  // EXPECT returns an object that allows the caller to indicate expected use.
    36  func (m *MockQUICConn) EXPECT() *MockQUICConnMockRecorder {
    37  	return m.recorder
    38  }
    39  
    40  // AcceptStream mocks base method.
    41  func (m *MockQUICConn) AcceptStream(arg0 context.Context) (Stream, error) {
    42  	m.ctrl.T.Helper()
    43  	ret := m.ctrl.Call(m, "AcceptStream", arg0)
    44  	ret0, _ := ret[0].(Stream)
    45  	ret1, _ := ret[1].(error)
    46  	return ret0, ret1
    47  }
    48  
    49  // AcceptStream indicates an expected call of AcceptStream.
    50  func (mr *MockQUICConnMockRecorder) AcceptStream(arg0 interface{}) *gomock.Call {
    51  	mr.mock.ctrl.T.Helper()
    52  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AcceptStream", reflect.TypeOf((*MockQUICConn)(nil).AcceptStream), arg0)
    53  }
    54  
    55  // AcceptUniStream mocks base method.
    56  func (m *MockQUICConn) AcceptUniStream(arg0 context.Context) (ReceiveStream, error) {
    57  	m.ctrl.T.Helper()
    58  	ret := m.ctrl.Call(m, "AcceptUniStream", arg0)
    59  	ret0, _ := ret[0].(ReceiveStream)
    60  	ret1, _ := ret[1].(error)
    61  	return ret0, ret1
    62  }
    63  
    64  // AcceptUniStream indicates an expected call of AcceptUniStream.
    65  func (mr *MockQUICConnMockRecorder) AcceptUniStream(arg0 interface{}) *gomock.Call {
    66  	mr.mock.ctrl.T.Helper()
    67  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AcceptUniStream", reflect.TypeOf((*MockQUICConn)(nil).AcceptUniStream), arg0)
    68  }
    69  
    70  // CloseWithError mocks base method.
    71  func (m *MockQUICConn) CloseWithError(arg0 qerr.ApplicationErrorCode, arg1 string) error {
    72  	m.ctrl.T.Helper()
    73  	ret := m.ctrl.Call(m, "CloseWithError", arg0, arg1)
    74  	ret0, _ := ret[0].(error)
    75  	return ret0
    76  }
    77  
    78  // CloseWithError indicates an expected call of CloseWithError.
    79  func (mr *MockQUICConnMockRecorder) CloseWithError(arg0, arg1 interface{}) *gomock.Call {
    80  	mr.mock.ctrl.T.Helper()
    81  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseWithError", reflect.TypeOf((*MockQUICConn)(nil).CloseWithError), arg0, arg1)
    82  }
    83  
    84  // ConnectionState mocks base method.
    85  func (m *MockQUICConn) ConnectionState() ConnectionState {
    86  	m.ctrl.T.Helper()
    87  	ret := m.ctrl.Call(m, "ConnectionState")
    88  	ret0, _ := ret[0].(ConnectionState)
    89  	return ret0
    90  }
    91  
    92  // ConnectionState indicates an expected call of ConnectionState.
    93  func (mr *MockQUICConnMockRecorder) ConnectionState() *gomock.Call {
    94  	mr.mock.ctrl.T.Helper()
    95  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConnectionState", reflect.TypeOf((*MockQUICConn)(nil).ConnectionState))
    96  }
    97  
    98  // Context mocks base method.
    99  func (m *MockQUICConn) Context() context.Context {
   100  	m.ctrl.T.Helper()
   101  	ret := m.ctrl.Call(m, "Context")
   102  	ret0, _ := ret[0].(context.Context)
   103  	return ret0
   104  }
   105  
   106  // Context indicates an expected call of Context.
   107  func (mr *MockQUICConnMockRecorder) Context() *gomock.Call {
   108  	mr.mock.ctrl.T.Helper()
   109  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockQUICConn)(nil).Context))
   110  }
   111  
   112  // GetVersion mocks base method.
   113  func (m *MockQUICConn) GetVersion() protocol.VersionNumber {
   114  	m.ctrl.T.Helper()
   115  	ret := m.ctrl.Call(m, "GetVersion")
   116  	ret0, _ := ret[0].(protocol.VersionNumber)
   117  	return ret0
   118  }
   119  
   120  // GetVersion indicates an expected call of GetVersion.
   121  func (mr *MockQUICConnMockRecorder) GetVersion() *gomock.Call {
   122  	mr.mock.ctrl.T.Helper()
   123  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockQUICConn)(nil).GetVersion))
   124  }
   125  
   126  // HandshakeComplete mocks base method.
   127  func (m *MockQUICConn) HandshakeComplete() <-chan struct{} {
   128  	m.ctrl.T.Helper()
   129  	ret := m.ctrl.Call(m, "HandshakeComplete")
   130  	ret0, _ := ret[0].(<-chan struct{})
   131  	return ret0
   132  }
   133  
   134  // HandshakeComplete indicates an expected call of HandshakeComplete.
   135  func (mr *MockQUICConnMockRecorder) HandshakeComplete() *gomock.Call {
   136  	mr.mock.ctrl.T.Helper()
   137  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandshakeComplete", reflect.TypeOf((*MockQUICConn)(nil).HandshakeComplete))
   138  }
   139  
   140  // LocalAddr mocks base method.
   141  func (m *MockQUICConn) LocalAddr() net.Addr {
   142  	m.ctrl.T.Helper()
   143  	ret := m.ctrl.Call(m, "LocalAddr")
   144  	ret0, _ := ret[0].(net.Addr)
   145  	return ret0
   146  }
   147  
   148  // LocalAddr indicates an expected call of LocalAddr.
   149  func (mr *MockQUICConnMockRecorder) LocalAddr() *gomock.Call {
   150  	mr.mock.ctrl.T.Helper()
   151  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LocalAddr", reflect.TypeOf((*MockQUICConn)(nil).LocalAddr))
   152  }
   153  
   154  // NextConnection mocks base method.
   155  func (m *MockQUICConn) NextConnection() Connection {
   156  	m.ctrl.T.Helper()
   157  	ret := m.ctrl.Call(m, "NextConnection")
   158  	ret0, _ := ret[0].(Connection)
   159  	return ret0
   160  }
   161  
   162  // NextConnection indicates an expected call of NextConnection.
   163  func (mr *MockQUICConnMockRecorder) NextConnection() *gomock.Call {
   164  	mr.mock.ctrl.T.Helper()
   165  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NextConnection", reflect.TypeOf((*MockQUICConn)(nil).NextConnection))
   166  }
   167  
   168  // OpenStream mocks base method.
   169  func (m *MockQUICConn) OpenStream() (Stream, error) {
   170  	m.ctrl.T.Helper()
   171  	ret := m.ctrl.Call(m, "OpenStream")
   172  	ret0, _ := ret[0].(Stream)
   173  	ret1, _ := ret[1].(error)
   174  	return ret0, ret1
   175  }
   176  
   177  // OpenStream indicates an expected call of OpenStream.
   178  func (mr *MockQUICConnMockRecorder) OpenStream() *gomock.Call {
   179  	mr.mock.ctrl.T.Helper()
   180  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenStream", reflect.TypeOf((*MockQUICConn)(nil).OpenStream))
   181  }
   182  
   183  // OpenStreamSync mocks base method.
   184  func (m *MockQUICConn) OpenStreamSync(arg0 context.Context) (Stream, error) {
   185  	m.ctrl.T.Helper()
   186  	ret := m.ctrl.Call(m, "OpenStreamSync", arg0)
   187  	ret0, _ := ret[0].(Stream)
   188  	ret1, _ := ret[1].(error)
   189  	return ret0, ret1
   190  }
   191  
   192  // OpenStreamSync indicates an expected call of OpenStreamSync.
   193  func (mr *MockQUICConnMockRecorder) OpenStreamSync(arg0 interface{}) *gomock.Call {
   194  	mr.mock.ctrl.T.Helper()
   195  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenStreamSync", reflect.TypeOf((*MockQUICConn)(nil).OpenStreamSync), arg0)
   196  }
   197  
   198  // OpenUniStream mocks base method.
   199  func (m *MockQUICConn) OpenUniStream() (SendStream, error) {
   200  	m.ctrl.T.Helper()
   201  	ret := m.ctrl.Call(m, "OpenUniStream")
   202  	ret0, _ := ret[0].(SendStream)
   203  	ret1, _ := ret[1].(error)
   204  	return ret0, ret1
   205  }
   206  
   207  // OpenUniStream indicates an expected call of OpenUniStream.
   208  func (mr *MockQUICConnMockRecorder) OpenUniStream() *gomock.Call {
   209  	mr.mock.ctrl.T.Helper()
   210  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenUniStream", reflect.TypeOf((*MockQUICConn)(nil).OpenUniStream))
   211  }
   212  
   213  // OpenUniStreamSync mocks base method.
   214  func (m *MockQUICConn) OpenUniStreamSync(arg0 context.Context) (SendStream, error) {
   215  	m.ctrl.T.Helper()
   216  	ret := m.ctrl.Call(m, "OpenUniStreamSync", arg0)
   217  	ret0, _ := ret[0].(SendStream)
   218  	ret1, _ := ret[1].(error)
   219  	return ret0, ret1
   220  }
   221  
   222  // OpenUniStreamSync indicates an expected call of OpenUniStreamSync.
   223  func (mr *MockQUICConnMockRecorder) OpenUniStreamSync(arg0 interface{}) *gomock.Call {
   224  	mr.mock.ctrl.T.Helper()
   225  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenUniStreamSync", reflect.TypeOf((*MockQUICConn)(nil).OpenUniStreamSync), arg0)
   226  }
   227  
   228  // ReceiveMessage mocks base method.
   229  func (m *MockQUICConn) ReceiveMessage(arg0 context.Context) ([]byte, error) {
   230  	m.ctrl.T.Helper()
   231  	ret := m.ctrl.Call(m, "ReceiveMessage", arg0)
   232  	ret0, _ := ret[0].([]byte)
   233  	ret1, _ := ret[1].(error)
   234  	return ret0, ret1
   235  }
   236  
   237  // ReceiveMessage indicates an expected call of ReceiveMessage.
   238  func (mr *MockQUICConnMockRecorder) ReceiveMessage(arg0 interface{}) *gomock.Call {
   239  	mr.mock.ctrl.T.Helper()
   240  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReceiveMessage", reflect.TypeOf((*MockQUICConn)(nil).ReceiveMessage), arg0)
   241  }
   242  
   243  // RemoteAddr mocks base method.
   244  func (m *MockQUICConn) RemoteAddr() net.Addr {
   245  	m.ctrl.T.Helper()
   246  	ret := m.ctrl.Call(m, "RemoteAddr")
   247  	ret0, _ := ret[0].(net.Addr)
   248  	return ret0
   249  }
   250  
   251  // RemoteAddr indicates an expected call of RemoteAddr.
   252  func (mr *MockQUICConnMockRecorder) RemoteAddr() *gomock.Call {
   253  	mr.mock.ctrl.T.Helper()
   254  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoteAddr", reflect.TypeOf((*MockQUICConn)(nil).RemoteAddr))
   255  }
   256  
   257  // SendMessage mocks base method.
   258  func (m *MockQUICConn) SendMessage(arg0 []byte) error {
   259  	m.ctrl.T.Helper()
   260  	ret := m.ctrl.Call(m, "SendMessage", arg0)
   261  	ret0, _ := ret[0].(error)
   262  	return ret0
   263  }
   264  
   265  // SendMessage indicates an expected call of SendMessage.
   266  func (mr *MockQUICConnMockRecorder) SendMessage(arg0 interface{}) *gomock.Call {
   267  	mr.mock.ctrl.T.Helper()
   268  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMessage", reflect.TypeOf((*MockQUICConn)(nil).SendMessage), arg0)
   269  }
   270  
   271  // destroy mocks base method.
   272  func (m *MockQUICConn) destroy(arg0 error) {
   273  	m.ctrl.T.Helper()
   274  	m.ctrl.Call(m, "destroy", arg0)
   275  }
   276  
   277  // destroy indicates an expected call of destroy.
   278  func (mr *MockQUICConnMockRecorder) destroy(arg0 interface{}) *gomock.Call {
   279  	mr.mock.ctrl.T.Helper()
   280  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "destroy", reflect.TypeOf((*MockQUICConn)(nil).destroy), arg0)
   281  }
   282  
   283  // earlyConnReady mocks base method.
   284  func (m *MockQUICConn) earlyConnReady() <-chan struct{} {
   285  	m.ctrl.T.Helper()
   286  	ret := m.ctrl.Call(m, "earlyConnReady")
   287  	ret0, _ := ret[0].(<-chan struct{})
   288  	return ret0
   289  }
   290  
   291  // earlyConnReady indicates an expected call of earlyConnReady.
   292  func (mr *MockQUICConnMockRecorder) earlyConnReady() *gomock.Call {
   293  	mr.mock.ctrl.T.Helper()
   294  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "earlyConnReady", reflect.TypeOf((*MockQUICConn)(nil).earlyConnReady))
   295  }
   296  
   297  // getPerspective mocks base method.
   298  func (m *MockQUICConn) getPerspective() protocol.Perspective {
   299  	m.ctrl.T.Helper()
   300  	ret := m.ctrl.Call(m, "getPerspective")
   301  	ret0, _ := ret[0].(protocol.Perspective)
   302  	return ret0
   303  }
   304  
   305  // getPerspective indicates an expected call of getPerspective.
   306  func (mr *MockQUICConnMockRecorder) getPerspective() *gomock.Call {
   307  	mr.mock.ctrl.T.Helper()
   308  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "getPerspective", reflect.TypeOf((*MockQUICConn)(nil).getPerspective))
   309  }
   310  
   311  // handlePacket mocks base method.
   312  func (m *MockQUICConn) handlePacket(arg0 receivedPacket) {
   313  	m.ctrl.T.Helper()
   314  	m.ctrl.Call(m, "handlePacket", arg0)
   315  }
   316  
   317  // handlePacket indicates an expected call of handlePacket.
   318  func (mr *MockQUICConnMockRecorder) handlePacket(arg0 interface{}) *gomock.Call {
   319  	mr.mock.ctrl.T.Helper()
   320  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "handlePacket", reflect.TypeOf((*MockQUICConn)(nil).handlePacket), arg0)
   321  }
   322  
   323  // run mocks base method.
   324  func (m *MockQUICConn) run() error {
   325  	m.ctrl.T.Helper()
   326  	ret := m.ctrl.Call(m, "run")
   327  	ret0, _ := ret[0].(error)
   328  	return ret0
   329  }
   330  
   331  // run indicates an expected call of run.
   332  func (mr *MockQUICConnMockRecorder) run() *gomock.Call {
   333  	mr.mock.ctrl.T.Helper()
   334  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "run", reflect.TypeOf((*MockQUICConn)(nil).run))
   335  }
   336  
   337  // shutdown mocks base method.
   338  func (m *MockQUICConn) shutdown() {
   339  	m.ctrl.T.Helper()
   340  	m.ctrl.Call(m, "shutdown")
   341  }
   342  
   343  // shutdown indicates an expected call of shutdown.
   344  func (mr *MockQUICConnMockRecorder) shutdown() *gomock.Call {
   345  	mr.mock.ctrl.T.Helper()
   346  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "shutdown", reflect.TypeOf((*MockQUICConn)(nil).shutdown))
   347  }