github.com/aporeto-inc/trireme-lib@v10.358.0+incompatible/controller/internal/supervisor/mocksupervisor/mocksupervisor.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: controller/internal/supervisor/interfaces.go
     3  
     4  // Package mocksupervisor is a generated GoMock package.
     5  package mocksupervisor
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  	time "time"
    11  
    12  	gomock "github.com/golang/mock/gomock"
    13  	provider "go.aporeto.io/enforcerd/trireme-lib/controller/pkg/aclprovider"
    14  	runtime "go.aporeto.io/enforcerd/trireme-lib/controller/runtime"
    15  	policy "go.aporeto.io/enforcerd/trireme-lib/policy"
    16  )
    17  
    18  // MockSupervisor is a mock of Supervisor interface
    19  // nolint
    20  type MockSupervisor struct {
    21  	ctrl     *gomock.Controller
    22  	recorder *MockSupervisorMockRecorder
    23  }
    24  
    25  // MockSupervisorMockRecorder is the mock recorder for MockSupervisor
    26  // nolint
    27  type MockSupervisorMockRecorder struct {
    28  	mock *MockSupervisor
    29  }
    30  
    31  // NewMockSupervisor creates a new mock instance
    32  // nolint
    33  func NewMockSupervisor(ctrl *gomock.Controller) *MockSupervisor {
    34  	mock := &MockSupervisor{ctrl: ctrl}
    35  	mock.recorder = &MockSupervisorMockRecorder{mock}
    36  	return mock
    37  }
    38  
    39  // EXPECT returns an object that allows the caller to indicate expected use
    40  // nolint
    41  func (m *MockSupervisor) EXPECT() *MockSupervisorMockRecorder {
    42  	return m.recorder
    43  }
    44  
    45  // Supervise mocks base method
    46  // nolint
    47  func (m *MockSupervisor) Supervise(contextID string, puInfo *policy.PUInfo) error {
    48  	m.ctrl.T.Helper()
    49  	ret := m.ctrl.Call(m, "Supervise", contextID, puInfo)
    50  	ret0, _ := ret[0].(error)
    51  	return ret0
    52  }
    53  
    54  // Supervise indicates an expected call of Supervise
    55  // nolint
    56  func (mr *MockSupervisorMockRecorder) Supervise(contextID, puInfo interface{}) *gomock.Call {
    57  	mr.mock.ctrl.T.Helper()
    58  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Supervise", reflect.TypeOf((*MockSupervisor)(nil).Supervise), contextID, puInfo)
    59  }
    60  
    61  // Unsupervise mocks base method
    62  // nolint
    63  func (m *MockSupervisor) Unsupervise(contextID string) error {
    64  	m.ctrl.T.Helper()
    65  	ret := m.ctrl.Call(m, "Unsupervise", contextID)
    66  	ret0, _ := ret[0].(error)
    67  	return ret0
    68  }
    69  
    70  // Unsupervise indicates an expected call of Unsupervise
    71  // nolint
    72  func (mr *MockSupervisorMockRecorder) Unsupervise(contextID interface{}) *gomock.Call {
    73  	mr.mock.ctrl.T.Helper()
    74  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Unsupervise", reflect.TypeOf((*MockSupervisor)(nil).Unsupervise), contextID)
    75  }
    76  
    77  // Run mocks base method
    78  // nolint
    79  func (m *MockSupervisor) Run(ctx context.Context) error {
    80  	m.ctrl.T.Helper()
    81  	ret := m.ctrl.Call(m, "Run", ctx)
    82  	ret0, _ := ret[0].(error)
    83  	return ret0
    84  }
    85  
    86  // Run indicates an expected call of Run
    87  // nolint
    88  func (mr *MockSupervisorMockRecorder) Run(ctx interface{}) *gomock.Call {
    89  	mr.mock.ctrl.T.Helper()
    90  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Run", reflect.TypeOf((*MockSupervisor)(nil).Run), ctx)
    91  }
    92  
    93  // SetTargetNetworks mocks base method
    94  // nolint
    95  func (m *MockSupervisor) SetTargetNetworks(cfg *runtime.Configuration) error {
    96  	m.ctrl.T.Helper()
    97  	ret := m.ctrl.Call(m, "SetTargetNetworks", cfg)
    98  	ret0, _ := ret[0].(error)
    99  	return ret0
   100  }
   101  
   102  // SetTargetNetworks indicates an expected call of SetTargetNetworks
   103  // nolint
   104  func (mr *MockSupervisorMockRecorder) SetTargetNetworks(cfg interface{}) *gomock.Call {
   105  	mr.mock.ctrl.T.Helper()
   106  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTargetNetworks", reflect.TypeOf((*MockSupervisor)(nil).SetTargetNetworks), cfg)
   107  }
   108  
   109  // CleanUp mocks base method
   110  // nolint
   111  func (m *MockSupervisor) CleanUp() error {
   112  	m.ctrl.T.Helper()
   113  	ret := m.ctrl.Call(m, "CleanUp")
   114  	ret0, _ := ret[0].(error)
   115  	return ret0
   116  }
   117  
   118  // CleanUp indicates an expected call of CleanUp
   119  // nolint
   120  func (mr *MockSupervisorMockRecorder) CleanUp() *gomock.Call {
   121  	mr.mock.ctrl.T.Helper()
   122  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanUp", reflect.TypeOf((*MockSupervisor)(nil).CleanUp))
   123  }
   124  
   125  // EnableIPTablesPacketTracing mocks base method
   126  // nolint
   127  func (m *MockSupervisor) EnableIPTablesPacketTracing(ctx context.Context, contextID string, interval time.Duration) error {
   128  	m.ctrl.T.Helper()
   129  	ret := m.ctrl.Call(m, "EnableIPTablesPacketTracing", ctx, contextID, interval)
   130  	ret0, _ := ret[0].(error)
   131  	return ret0
   132  }
   133  
   134  // EnableIPTablesPacketTracing indicates an expected call of EnableIPTablesPacketTracing
   135  // nolint
   136  func (mr *MockSupervisorMockRecorder) EnableIPTablesPacketTracing(ctx, contextID, interval interface{}) *gomock.Call {
   137  	mr.mock.ctrl.T.Helper()
   138  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnableIPTablesPacketTracing", reflect.TypeOf((*MockSupervisor)(nil).EnableIPTablesPacketTracing), ctx, contextID, interval)
   139  }
   140  
   141  // MockImplementor is a mock of Implementor interface
   142  // nolint
   143  type MockImplementor struct {
   144  	ctrl     *gomock.Controller
   145  	recorder *MockImplementorMockRecorder
   146  }
   147  
   148  // MockImplementorMockRecorder is the mock recorder for MockImplementor
   149  // nolint
   150  type MockImplementorMockRecorder struct {
   151  	mock *MockImplementor
   152  }
   153  
   154  // NewMockImplementor creates a new mock instance
   155  // nolint
   156  func NewMockImplementor(ctrl *gomock.Controller) *MockImplementor {
   157  	mock := &MockImplementor{ctrl: ctrl}
   158  	mock.recorder = &MockImplementorMockRecorder{mock}
   159  	return mock
   160  }
   161  
   162  // EXPECT returns an object that allows the caller to indicate expected use
   163  // nolint
   164  func (m *MockImplementor) EXPECT() *MockImplementorMockRecorder {
   165  	return m.recorder
   166  }
   167  
   168  // ConfigureRules mocks base method
   169  // nolint
   170  func (m *MockImplementor) ConfigureRules(version int, contextID string, containerInfo *policy.PUInfo) error {
   171  	m.ctrl.T.Helper()
   172  	ret := m.ctrl.Call(m, "ConfigureRules", version, contextID, containerInfo)
   173  	ret0, _ := ret[0].(error)
   174  	return ret0
   175  }
   176  
   177  // ConfigureRules indicates an expected call of ConfigureRules
   178  // nolint
   179  func (mr *MockImplementorMockRecorder) ConfigureRules(version, contextID, containerInfo interface{}) *gomock.Call {
   180  	mr.mock.ctrl.T.Helper()
   181  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConfigureRules", reflect.TypeOf((*MockImplementor)(nil).ConfigureRules), version, contextID, containerInfo)
   182  }
   183  
   184  // UpdateRules mocks base method
   185  // nolint
   186  func (m *MockImplementor) UpdateRules(version int, contextID string, containerInfo, oldContainerInfo *policy.PUInfo) error {
   187  	m.ctrl.T.Helper()
   188  	ret := m.ctrl.Call(m, "UpdateRules", version, contextID, containerInfo, oldContainerInfo)
   189  	ret0, _ := ret[0].(error)
   190  	return ret0
   191  }
   192  
   193  // UpdateRules indicates an expected call of UpdateRules
   194  // nolint
   195  func (mr *MockImplementorMockRecorder) UpdateRules(version, contextID, containerInfo, oldContainerInfo interface{}) *gomock.Call {
   196  	mr.mock.ctrl.T.Helper()
   197  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRules", reflect.TypeOf((*MockImplementor)(nil).UpdateRules), version, contextID, containerInfo, oldContainerInfo)
   198  }
   199  
   200  // DeleteRules mocks base method
   201  // nolint
   202  func (m *MockImplementor) DeleteRules(version int, context, tcpPorts, udpPorts, mark, uid string, containerInfo *policy.PUInfo) error {
   203  	m.ctrl.T.Helper()
   204  	ret := m.ctrl.Call(m, "DeleteRules", version, context, tcpPorts, udpPorts, mark, uid, containerInfo)
   205  	ret0, _ := ret[0].(error)
   206  	return ret0
   207  }
   208  
   209  // DeleteRules indicates an expected call of DeleteRules
   210  // nolint
   211  func (mr *MockImplementorMockRecorder) DeleteRules(version, context, tcpPorts, udpPorts, mark, uid, containerInfo interface{}) *gomock.Call {
   212  	mr.mock.ctrl.T.Helper()
   213  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRules", reflect.TypeOf((*MockImplementor)(nil).DeleteRules), version, context, tcpPorts, udpPorts, mark, uid, containerInfo)
   214  }
   215  
   216  // SetTargetNetworks mocks base method
   217  // nolint
   218  func (m *MockImplementor) SetTargetNetworks(cfg *runtime.Configuration) error {
   219  	m.ctrl.T.Helper()
   220  	ret := m.ctrl.Call(m, "SetTargetNetworks", cfg)
   221  	ret0, _ := ret[0].(error)
   222  	return ret0
   223  }
   224  
   225  // SetTargetNetworks indicates an expected call of SetTargetNetworks
   226  // nolint
   227  func (mr *MockImplementorMockRecorder) SetTargetNetworks(cfg interface{}) *gomock.Call {
   228  	mr.mock.ctrl.T.Helper()
   229  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTargetNetworks", reflect.TypeOf((*MockImplementor)(nil).SetTargetNetworks), cfg)
   230  }
   231  
   232  // Run mocks base method
   233  // nolint
   234  func (m *MockImplementor) Run(ctx context.Context) error {
   235  	m.ctrl.T.Helper()
   236  	ret := m.ctrl.Call(m, "Run", ctx)
   237  	ret0, _ := ret[0].(error)
   238  	return ret0
   239  }
   240  
   241  // Run indicates an expected call of Run
   242  // nolint
   243  func (mr *MockImplementorMockRecorder) Run(ctx interface{}) *gomock.Call {
   244  	mr.mock.ctrl.T.Helper()
   245  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Run", reflect.TypeOf((*MockImplementor)(nil).Run), ctx)
   246  }
   247  
   248  // CleanUp mocks base method
   249  // nolint
   250  func (m *MockImplementor) CleanUp() error {
   251  	m.ctrl.T.Helper()
   252  	ret := m.ctrl.Call(m, "CleanUp")
   253  	ret0, _ := ret[0].(error)
   254  	return ret0
   255  }
   256  
   257  // CleanUp indicates an expected call of CleanUp
   258  // nolint
   259  func (mr *MockImplementorMockRecorder) CleanUp() *gomock.Call {
   260  	mr.mock.ctrl.T.Helper()
   261  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanUp", reflect.TypeOf((*MockImplementor)(nil).CleanUp))
   262  }
   263  
   264  // ACLProvider mocks base method
   265  // nolint
   266  func (m *MockImplementor) ACLProvider() []provider.IptablesProvider {
   267  	m.ctrl.T.Helper()
   268  	ret := m.ctrl.Call(m, "ACLProvider")
   269  	ret0, _ := ret[0].([]provider.IptablesProvider)
   270  	return ret0
   271  }
   272  
   273  // ACLProvider indicates an expected call of ACLProvider
   274  // nolint
   275  func (mr *MockImplementorMockRecorder) ACLProvider() *gomock.Call {
   276  	mr.mock.ctrl.T.Helper()
   277  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ACLProvider", reflect.TypeOf((*MockImplementor)(nil).ACLProvider))
   278  }
   279  
   280  // CreateCustomRulesChain mocks base method
   281  // nolint
   282  func (m *MockImplementor) CreateCustomRulesChain() error {
   283  	m.ctrl.T.Helper()
   284  	ret := m.ctrl.Call(m, "CreateCustomRulesChain")
   285  	ret0, _ := ret[0].(error)
   286  	return ret0
   287  }
   288  
   289  // CreateCustomRulesChain indicates an expected call of CreateCustomRulesChain
   290  // nolint
   291  func (mr *MockImplementorMockRecorder) CreateCustomRulesChain() *gomock.Call {
   292  	mr.mock.ctrl.T.Helper()
   293  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateCustomRulesChain", reflect.TypeOf((*MockImplementor)(nil).CreateCustomRulesChain))
   294  }