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

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: controller/internal/enforcer/dnsproxy/dnsproxy.go
     3  
     4  // Package mockdnsproxy is a generated GoMock package.
     5  package mockdnsproxy
     6  
     7  import (
     8  	context "context"
     9  	net "net"
    10  	reflect "reflect"
    11  
    12  	gomock "github.com/golang/mock/gomock"
    13  	pucontext "go.aporeto.io/enforcerd/trireme-lib/controller/pkg/pucontext"
    14  	policy "go.aporeto.io/enforcerd/trireme-lib/policy"
    15  )
    16  
    17  // MockDNSProxy is a mock of DNSProxy interface
    18  // nolint
    19  type MockDNSProxy struct {
    20  	ctrl     *gomock.Controller
    21  	recorder *MockDNSProxyMockRecorder
    22  }
    23  
    24  // MockDNSProxyMockRecorder is the mock recorder for MockDNSProxy
    25  // nolint
    26  type MockDNSProxyMockRecorder struct {
    27  	mock *MockDNSProxy
    28  }
    29  
    30  // NewMockDNSProxy creates a new mock instance
    31  // nolint
    32  func NewMockDNSProxy(ctrl *gomock.Controller) *MockDNSProxy {
    33  	mock := &MockDNSProxy{ctrl: ctrl}
    34  	mock.recorder = &MockDNSProxyMockRecorder{mock}
    35  	return mock
    36  }
    37  
    38  // EXPECT returns an object that allows the caller to indicate expected use
    39  // nolint
    40  func (m *MockDNSProxy) EXPECT() *MockDNSProxyMockRecorder {
    41  	return m.recorder
    42  }
    43  
    44  // StartDNSServer mocks base method
    45  // nolint
    46  func (m *MockDNSProxy) StartDNSServer(ctx context.Context, contextID, port string) error {
    47  	m.ctrl.T.Helper()
    48  	ret := m.ctrl.Call(m, "StartDNSServer", ctx, contextID, port)
    49  	ret0, _ := ret[0].(error)
    50  	return ret0
    51  }
    52  
    53  // StartDNSServer indicates an expected call of StartDNSServer
    54  // nolint
    55  func (mr *MockDNSProxyMockRecorder) StartDNSServer(ctx, contextID, port interface{}) *gomock.Call {
    56  	mr.mock.ctrl.T.Helper()
    57  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StartDNSServer", reflect.TypeOf((*MockDNSProxy)(nil).StartDNSServer), ctx, contextID, port)
    58  }
    59  
    60  // Enforce mocks base method
    61  // nolint
    62  func (m *MockDNSProxy) Enforce(ctx context.Context, contextID string, puInfo *policy.PUInfo) error {
    63  	m.ctrl.T.Helper()
    64  	ret := m.ctrl.Call(m, "Enforce", ctx, contextID, puInfo)
    65  	ret0, _ := ret[0].(error)
    66  	return ret0
    67  }
    68  
    69  // Enforce indicates an expected call of Enforce
    70  // nolint
    71  func (mr *MockDNSProxyMockRecorder) Enforce(ctx, contextID, puInfo interface{}) *gomock.Call {
    72  	mr.mock.ctrl.T.Helper()
    73  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Enforce", reflect.TypeOf((*MockDNSProxy)(nil).Enforce), ctx, contextID, puInfo)
    74  }
    75  
    76  // Unenforce mocks base method
    77  // nolint
    78  func (m *MockDNSProxy) Unenforce(ctx context.Context, contextID string) error {
    79  	m.ctrl.T.Helper()
    80  	ret := m.ctrl.Call(m, "Unenforce", ctx, contextID)
    81  	ret0, _ := ret[0].(error)
    82  	return ret0
    83  }
    84  
    85  // Unenforce indicates an expected call of Unenforce
    86  // nolint
    87  func (mr *MockDNSProxyMockRecorder) Unenforce(ctx, contextID interface{}) *gomock.Call {
    88  	mr.mock.ctrl.T.Helper()
    89  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Unenforce", reflect.TypeOf((*MockDNSProxy)(nil).Unenforce), ctx, contextID)
    90  }
    91  
    92  // SyncWithPlatformCache mocks base method
    93  // nolint
    94  func (m *MockDNSProxy) SyncWithPlatformCache(ctx context.Context, pctx *pucontext.PUContext) error {
    95  	m.ctrl.T.Helper()
    96  	ret := m.ctrl.Call(m, "SyncWithPlatformCache", ctx, pctx)
    97  	ret0, _ := ret[0].(error)
    98  	return ret0
    99  }
   100  
   101  // SyncWithPlatformCache indicates an expected call of SyncWithPlatformCache
   102  // nolint
   103  func (mr *MockDNSProxyMockRecorder) SyncWithPlatformCache(ctx, pctx interface{}) *gomock.Call {
   104  	mr.mock.ctrl.T.Helper()
   105  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncWithPlatformCache", reflect.TypeOf((*MockDNSProxy)(nil).SyncWithPlatformCache), ctx, pctx)
   106  }
   107  
   108  // HandleDNSResponsePacket mocks base method
   109  // nolint
   110  func (m *MockDNSProxy) HandleDNSResponsePacket(dnsPacketData []byte, sourceIP net.IP, sourcePort uint16, destIP net.IP, destPort uint16, puFromContextID func(string) (*pucontext.PUContext, error)) error {
   111  	m.ctrl.T.Helper()
   112  	ret := m.ctrl.Call(m, "HandleDNSResponsePacket", dnsPacketData, sourceIP, sourcePort, destIP, destPort, puFromContextID)
   113  	ret0, _ := ret[0].(error)
   114  	return ret0
   115  }
   116  
   117  // HandleDNSResponsePacket indicates an expected call of HandleDNSResponsePacket
   118  // nolint
   119  func (mr *MockDNSProxyMockRecorder) HandleDNSResponsePacket(dnsPacketData, sourceIP, sourcePort, destIP, destPort, puFromContextID interface{}) *gomock.Call {
   120  	mr.mock.ctrl.T.Helper()
   121  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleDNSResponsePacket", reflect.TypeOf((*MockDNSProxy)(nil).HandleDNSResponsePacket), dnsPacketData, sourceIP, sourcePort, destIP, destPort, puFromContextID)
   122  }