Browse Source

update mocks

Darien Raymond 7 years ago
parent
commit
5ba41e50ec
6 changed files with 38 additions and 0 deletions
  1. 8 0
      testing/mocks/dns.go
  2. 4 0
      testing/mocks/io.go
  3. 2 0
      testing/mocks/log.go
  4. 2 0
      testing/mocks/mux.go
  5. 16 0
      testing/mocks/outbound.go
  6. 6 0
      testing/mocks/proxy.go

+ 8 - 0
testing/mocks/dns.go

@@ -35,6 +35,7 @@ func (m *DNSClient) EXPECT() *DNSClientMockRecorder {
 
 // Close mocks base method
 func (m *DNSClient) Close() error {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Close")
 	ret0, _ := ret[0].(error)
 	return ret0
@@ -42,11 +43,13 @@ func (m *DNSClient) Close() error {
 
 // Close indicates an expected call of Close
 func (mr *DNSClientMockRecorder) Close() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*DNSClient)(nil).Close))
 }
 
 // LookupIP mocks base method
 func (m *DNSClient) LookupIP(arg0 string) ([]net.IP, error) {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "LookupIP", arg0)
 	ret0, _ := ret[0].([]net.IP)
 	ret1, _ := ret[1].(error)
@@ -55,11 +58,13 @@ func (m *DNSClient) LookupIP(arg0 string) ([]net.IP, error) {
 
 // LookupIP indicates an expected call of LookupIP
 func (mr *DNSClientMockRecorder) LookupIP(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LookupIP", reflect.TypeOf((*DNSClient)(nil).LookupIP), arg0)
 }
 
 // Start mocks base method
 func (m *DNSClient) Start() error {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Start")
 	ret0, _ := ret[0].(error)
 	return ret0
@@ -67,11 +72,13 @@ func (m *DNSClient) Start() error {
 
 // Start indicates an expected call of Start
 func (mr *DNSClientMockRecorder) Start() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*DNSClient)(nil).Start))
 }
 
 // Type mocks base method
 func (m *DNSClient) Type() interface{} {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Type")
 	ret0, _ := ret[0].(interface{})
 	return ret0
@@ -79,5 +86,6 @@ func (m *DNSClient) Type() interface{} {
 
 // Type indicates an expected call of Type
 func (mr *DNSClientMockRecorder) Type() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Type", reflect.TypeOf((*DNSClient)(nil).Type))
 }

+ 4 - 0
testing/mocks/io.go

@@ -34,6 +34,7 @@ func (m *Reader) EXPECT() *ReaderMockRecorder {
 
 // Read mocks base method
 func (m *Reader) Read(arg0 []byte) (int, error) {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Read", arg0)
 	ret0, _ := ret[0].(int)
 	ret1, _ := ret[1].(error)
@@ -42,6 +43,7 @@ func (m *Reader) Read(arg0 []byte) (int, error) {
 
 // Read indicates an expected call of Read
 func (mr *ReaderMockRecorder) Read(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*Reader)(nil).Read), arg0)
 }
 
@@ -70,6 +72,7 @@ func (m *Writer) EXPECT() *WriterMockRecorder {
 
 // Write mocks base method
 func (m *Writer) Write(arg0 []byte) (int, error) {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Write", arg0)
 	ret0, _ := ret[0].(int)
 	ret1, _ := ret[1].(error)
@@ -78,5 +81,6 @@ func (m *Writer) Write(arg0 []byte) (int, error) {
 
 // Write indicates an expected call of Write
 func (mr *WriterMockRecorder) Write(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Write", reflect.TypeOf((*Writer)(nil).Write), arg0)
 }

+ 2 - 0
testing/mocks/log.go

@@ -35,10 +35,12 @@ func (m *LogHandler) EXPECT() *LogHandlerMockRecorder {
 
 // Handle mocks base method
 func (m *LogHandler) Handle(arg0 log.Message) {
+	m.ctrl.T.Helper()
 	m.ctrl.Call(m, "Handle", arg0)
 }
 
 // Handle indicates an expected call of Handle
 func (mr *LogHandlerMockRecorder) Handle(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Handle", reflect.TypeOf((*LogHandler)(nil).Handle), arg0)
 }

+ 2 - 0
testing/mocks/mux.go

@@ -35,6 +35,7 @@ func (m *MuxClientWorkerFactory) EXPECT() *MuxClientWorkerFactoryMockRecorder {
 
 // Create mocks base method
 func (m *MuxClientWorkerFactory) Create() (*mux.ClientWorker, error) {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Create")
 	ret0, _ := ret[0].(*mux.ClientWorker)
 	ret1, _ := ret[1].(error)
@@ -43,5 +44,6 @@ func (m *MuxClientWorkerFactory) Create() (*mux.ClientWorker, error) {
 
 // Create indicates an expected call of Create
 func (mr *MuxClientWorkerFactoryMockRecorder) Create() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MuxClientWorkerFactory)(nil).Create))
 }

+ 16 - 0
testing/mocks/outbound.go

@@ -36,6 +36,7 @@ func (m *OutboundManager) EXPECT() *OutboundManagerMockRecorder {
 
 // AddHandler mocks base method
 func (m *OutboundManager) AddHandler(arg0 context.Context, arg1 outbound.Handler) error {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "AddHandler", arg0, arg1)
 	ret0, _ := ret[0].(error)
 	return ret0
@@ -43,11 +44,13 @@ func (m *OutboundManager) AddHandler(arg0 context.Context, arg1 outbound.Handler
 
 // AddHandler indicates an expected call of AddHandler
 func (mr *OutboundManagerMockRecorder) AddHandler(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddHandler", reflect.TypeOf((*OutboundManager)(nil).AddHandler), arg0, arg1)
 }
 
 // Close mocks base method
 func (m *OutboundManager) Close() error {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Close")
 	ret0, _ := ret[0].(error)
 	return ret0
@@ -55,11 +58,13 @@ func (m *OutboundManager) Close() error {
 
 // Close indicates an expected call of Close
 func (mr *OutboundManagerMockRecorder) Close() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*OutboundManager)(nil).Close))
 }
 
 // GetDefaultHandler mocks base method
 func (m *OutboundManager) GetDefaultHandler() outbound.Handler {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "GetDefaultHandler")
 	ret0, _ := ret[0].(outbound.Handler)
 	return ret0
@@ -67,11 +72,13 @@ func (m *OutboundManager) GetDefaultHandler() outbound.Handler {
 
 // GetDefaultHandler indicates an expected call of GetDefaultHandler
 func (mr *OutboundManagerMockRecorder) GetDefaultHandler() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDefaultHandler", reflect.TypeOf((*OutboundManager)(nil).GetDefaultHandler))
 }
 
 // GetHandler mocks base method
 func (m *OutboundManager) GetHandler(arg0 string) outbound.Handler {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "GetHandler", arg0)
 	ret0, _ := ret[0].(outbound.Handler)
 	return ret0
@@ -79,11 +86,13 @@ func (m *OutboundManager) GetHandler(arg0 string) outbound.Handler {
 
 // GetHandler indicates an expected call of GetHandler
 func (mr *OutboundManagerMockRecorder) GetHandler(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHandler", reflect.TypeOf((*OutboundManager)(nil).GetHandler), arg0)
 }
 
 // RemoveHandler mocks base method
 func (m *OutboundManager) RemoveHandler(arg0 context.Context, arg1 string) error {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "RemoveHandler", arg0, arg1)
 	ret0, _ := ret[0].(error)
 	return ret0
@@ -91,11 +100,13 @@ func (m *OutboundManager) RemoveHandler(arg0 context.Context, arg1 string) error
 
 // RemoveHandler indicates an expected call of RemoveHandler
 func (mr *OutboundManagerMockRecorder) RemoveHandler(arg0, arg1 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveHandler", reflect.TypeOf((*OutboundManager)(nil).RemoveHandler), arg0, arg1)
 }
 
 // Start mocks base method
 func (m *OutboundManager) Start() error {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Start")
 	ret0, _ := ret[0].(error)
 	return ret0
@@ -103,11 +114,13 @@ func (m *OutboundManager) Start() error {
 
 // Start indicates an expected call of Start
 func (mr *OutboundManagerMockRecorder) Start() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*OutboundManager)(nil).Start))
 }
 
 // Type mocks base method
 func (m *OutboundManager) Type() interface{} {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Type")
 	ret0, _ := ret[0].(interface{})
 	return ret0
@@ -115,6 +128,7 @@ func (m *OutboundManager) Type() interface{} {
 
 // Type indicates an expected call of Type
 func (mr *OutboundManagerMockRecorder) Type() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Type", reflect.TypeOf((*OutboundManager)(nil).Type))
 }
 
@@ -143,6 +157,7 @@ func (m *OutboundHandlerSelector) EXPECT() *OutboundHandlerSelectorMockRecorder
 
 // Select mocks base method
 func (m *OutboundHandlerSelector) Select(arg0 []string) []string {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Select", arg0)
 	ret0, _ := ret[0].([]string)
 	return ret0
@@ -150,5 +165,6 @@ func (m *OutboundHandlerSelector) Select(arg0 []string) []string {
 
 // Select indicates an expected call of Select
 func (mr *OutboundHandlerSelectorMockRecorder) Select(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Select", reflect.TypeOf((*OutboundHandlerSelector)(nil).Select), arg0)
 }

+ 6 - 0
testing/mocks/proxy.go

@@ -39,6 +39,7 @@ func (m *ProxyInbound) EXPECT() *ProxyInboundMockRecorder {
 
 // Network mocks base method
 func (m *ProxyInbound) Network() []net.Network {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Network")
 	ret0, _ := ret[0].([]net.Network)
 	return ret0
@@ -46,11 +47,13 @@ func (m *ProxyInbound) Network() []net.Network {
 
 // Network indicates an expected call of Network
 func (mr *ProxyInboundMockRecorder) Network() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Network", reflect.TypeOf((*ProxyInbound)(nil).Network))
 }
 
 // Process mocks base method
 func (m *ProxyInbound) Process(arg0 context.Context, arg1 net.Network, arg2 internet.Connection, arg3 routing.Dispatcher) error {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Process", arg0, arg1, arg2, arg3)
 	ret0, _ := ret[0].(error)
 	return ret0
@@ -58,6 +61,7 @@ func (m *ProxyInbound) Process(arg0 context.Context, arg1 net.Network, arg2 inte
 
 // Process indicates an expected call of Process
 func (mr *ProxyInboundMockRecorder) Process(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Process", reflect.TypeOf((*ProxyInbound)(nil).Process), arg0, arg1, arg2, arg3)
 }
 
@@ -86,6 +90,7 @@ func (m *ProxyOutbound) EXPECT() *ProxyOutboundMockRecorder {
 
 // Process mocks base method
 func (m *ProxyOutbound) Process(arg0 context.Context, arg1 *transport.Link, arg2 internet.Dialer) error {
+	m.ctrl.T.Helper()
 	ret := m.ctrl.Call(m, "Process", arg0, arg1, arg2)
 	ret0, _ := ret[0].(error)
 	return ret0
@@ -93,5 +98,6 @@ func (m *ProxyOutbound) Process(arg0 context.Context, arg1 *transport.Link, arg2
 
 // Process indicates an expected call of Process
 func (mr *ProxyOutboundMockRecorder) Process(arg0, arg1, arg2 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Process", reflect.TypeOf((*ProxyOutbound)(nil).Process), arg0, arg1, arg2)
 }