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