github.com/aporeto-inc/trireme-lib@v10.358.0+incompatible/controller/pkg/flowtracking/mockflowclient/mockflowclient.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: controller/pkg/flowtracking/interfaces.go
     3  
     4  // Package mockflowclient is a generated GoMock package.
     5  package mockflowclient
     6  
     7  import (
     8  	net "net"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  )
    13  
    14  // MockFlowClient is a mock of FlowClient interface
    15  // nolint
    16  type MockFlowClient struct {
    17  	ctrl     *gomock.Controller
    18  	recorder *MockFlowClientMockRecorder
    19  }
    20  
    21  // MockFlowClientMockRecorder is the mock recorder for MockFlowClient
    22  // nolint
    23  type MockFlowClientMockRecorder struct {
    24  	mock *MockFlowClient
    25  }
    26  
    27  // NewMockFlowClient creates a new mock instance
    28  // nolint
    29  func NewMockFlowClient(ctrl *gomock.Controller) *MockFlowClient {
    30  	mock := &MockFlowClient{ctrl: ctrl}
    31  	mock.recorder = &MockFlowClientMockRecorder{mock}
    32  	return mock
    33  }
    34  
    35  // EXPECT returns an object that allows the caller to indicate expected use
    36  // nolint
    37  func (m *MockFlowClient) EXPECT() *MockFlowClientMockRecorder {
    38  	return m.recorder
    39  }
    40  
    41  // Close mocks base method
    42  // nolint
    43  func (m *MockFlowClient) Close() error {
    44  	m.ctrl.T.Helper()
    45  	ret := m.ctrl.Call(m, "Close")
    46  	ret0, _ := ret[0].(error)
    47  	return ret0
    48  }
    49  
    50  // Close indicates an expected call of Close
    51  // nolint
    52  func (mr *MockFlowClientMockRecorder) Close() *gomock.Call {
    53  	mr.mock.ctrl.T.Helper()
    54  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockFlowClient)(nil).Close))
    55  }
    56  
    57  // UpdateMark mocks base method
    58  // nolint
    59  func (m *MockFlowClient) UpdateMark(ipSrc, ipDst net.IP, protonum uint8, srcport, dstport uint16, newmark uint32, network bool) error {
    60  	m.ctrl.T.Helper()
    61  	ret := m.ctrl.Call(m, "UpdateMark", ipSrc, ipDst, protonum, srcport, dstport, newmark, network)
    62  	ret0, _ := ret[0].(error)
    63  	return ret0
    64  }
    65  
    66  // UpdateMark indicates an expected call of UpdateMark
    67  // nolint
    68  func (mr *MockFlowClientMockRecorder) UpdateMark(ipSrc, ipDst, protonum, srcport, dstport, newmark, network interface{}) *gomock.Call {
    69  	mr.mock.ctrl.T.Helper()
    70  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateMark", reflect.TypeOf((*MockFlowClient)(nil).UpdateMark), ipSrc, ipDst, protonum, srcport, dstport, newmark, network)
    71  }
    72  
    73  // GetOriginalDest mocks base method
    74  // nolint
    75  func (m *MockFlowClient) GetOriginalDest(ipSrc, ipDst net.IP, srcport, dstport uint16, protonum uint8) (net.IP, uint16, uint32, error) {
    76  	m.ctrl.T.Helper()
    77  	ret := m.ctrl.Call(m, "GetOriginalDest", ipSrc, ipDst, srcport, dstport, protonum)
    78  	ret0, _ := ret[0].(net.IP)
    79  	ret1, _ := ret[1].(uint16)
    80  	ret2, _ := ret[2].(uint32)
    81  	ret3, _ := ret[3].(error)
    82  	return ret0, ret1, ret2, ret3
    83  }
    84  
    85  // GetOriginalDest indicates an expected call of GetOriginalDest
    86  // nolint
    87  func (mr *MockFlowClientMockRecorder) GetOriginalDest(ipSrc, ipDst, srcport, dstport, protonum interface{}) *gomock.Call {
    88  	mr.mock.ctrl.T.Helper()
    89  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOriginalDest", reflect.TypeOf((*MockFlowClient)(nil).GetOriginalDest), ipSrc, ipDst, srcport, dstport, protonum)
    90  }
    91  
    92  // UpdateNetworkFlowMark mocks base method
    93  // nolint
    94  func (m *MockFlowClient) UpdateNetworkFlowMark(ipSrc, ipDst net.IP, protonum uint8, srcport, dstport uint16, newmark uint32) error {
    95  	m.ctrl.T.Helper()
    96  	ret := m.ctrl.Call(m, "UpdateNetworkFlowMark", ipSrc, ipDst, protonum, srcport, dstport, newmark)
    97  	ret0, _ := ret[0].(error)
    98  	return ret0
    99  }
   100  
   101  // UpdateNetworkFlowMark indicates an expected call of UpdateNetworkFlowMark
   102  // nolint
   103  func (mr *MockFlowClientMockRecorder) UpdateNetworkFlowMark(ipSrc, ipDst, protonum, srcport, dstport, newmark interface{}) *gomock.Call {
   104  	mr.mock.ctrl.T.Helper()
   105  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNetworkFlowMark", reflect.TypeOf((*MockFlowClient)(nil).UpdateNetworkFlowMark), ipSrc, ipDst, protonum, srcport, dstport, newmark)
   106  }
   107  
   108  // UpdateApplicationFlowMark mocks base method
   109  // nolint
   110  func (m *MockFlowClient) UpdateApplicationFlowMark(ipSrc, ipDst net.IP, protonum uint8, srcport, dstport uint16, newmark uint32) error {
   111  	m.ctrl.T.Helper()
   112  	ret := m.ctrl.Call(m, "UpdateApplicationFlowMark", ipSrc, ipDst, protonum, srcport, dstport, newmark)
   113  	ret0, _ := ret[0].(error)
   114  	return ret0
   115  }
   116  
   117  // UpdateApplicationFlowMark indicates an expected call of UpdateApplicationFlowMark
   118  // nolint
   119  func (mr *MockFlowClientMockRecorder) UpdateApplicationFlowMark(ipSrc, ipDst, protonum, srcport, dstport, newmark interface{}) *gomock.Call {
   120  	mr.mock.ctrl.T.Helper()
   121  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateApplicationFlowMark", reflect.TypeOf((*MockFlowClient)(nil).UpdateApplicationFlowMark), ipSrc, ipDst, protonum, srcport, dstport, newmark)
   122  }