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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
// Code generated by MockGen. DO NOT EDIT.
// Source: gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/reverse_tunnel/tracker (interfaces: Registerer,Querier,PollingQuerier)
// Package mock_reverse_tunnel_tracker is a generated GoMock package.
package mock_reverse_tunnel_tracker
import (
context "context"
reflect "reflect"
gomock "github.com/golang/mock/gomock"
tracker "gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/reverse_tunnel/tracker"
)
// MockRegisterer is a mock of Registerer interface.
type MockRegisterer struct {
ctrl *gomock.Controller
recorder *MockRegistererMockRecorder
}
// MockRegistererMockRecorder is the mock recorder for MockRegisterer.
type MockRegistererMockRecorder struct {
mock *MockRegisterer
}
// NewMockRegisterer creates a new mock instance.
func NewMockRegisterer(ctrl *gomock.Controller) *MockRegisterer {
mock := &MockRegisterer{ctrl: ctrl}
mock.recorder = &MockRegistererMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockRegisterer) EXPECT() *MockRegistererMockRecorder {
return m.recorder
}
// RegisterTunnel mocks base method.
func (m *MockRegisterer) RegisterTunnel(arg0 context.Context, arg1 int64) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "RegisterTunnel", arg0, arg1)
ret0, _ := ret[0].(error)
return ret0
}
// RegisterTunnel indicates an expected call of RegisterTunnel.
func (mr *MockRegistererMockRecorder) RegisterTunnel(arg0, arg1 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterTunnel", reflect.TypeOf((*MockRegisterer)(nil).RegisterTunnel), arg0, arg1)
}
// UnregisterTunnel mocks base method.
func (m *MockRegisterer) UnregisterTunnel(arg0 context.Context, arg1 int64) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UnregisterTunnel", arg0, arg1)
ret0, _ := ret[0].(error)
return ret0
}
// UnregisterTunnel indicates an expected call of UnregisterTunnel.
func (mr *MockRegistererMockRecorder) UnregisterTunnel(arg0, arg1 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnregisterTunnel", reflect.TypeOf((*MockRegisterer)(nil).UnregisterTunnel), arg0, arg1)
}
// MockQuerier is a mock of Querier interface.
type MockQuerier struct {
ctrl *gomock.Controller
recorder *MockQuerierMockRecorder
}
// MockQuerierMockRecorder is the mock recorder for MockQuerier.
type MockQuerierMockRecorder struct {
mock *MockQuerier
}
// NewMockQuerier creates a new mock instance.
func NewMockQuerier(ctrl *gomock.Controller) *MockQuerier {
mock := &MockQuerier{ctrl: ctrl}
mock.recorder = &MockQuerierMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockQuerier) EXPECT() *MockQuerierMockRecorder {
return m.recorder
}
// KasUrlsByAgentId mocks base method.
func (m *MockQuerier) KasUrlsByAgentId(arg0 context.Context, arg1 int64, arg2 tracker.KasUrlsByAgentIdCallback) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "KasUrlsByAgentId", arg0, arg1, arg2)
ret0, _ := ret[0].(error)
return ret0
}
// KasUrlsByAgentId indicates an expected call of KasUrlsByAgentId.
func (mr *MockQuerierMockRecorder) KasUrlsByAgentId(arg0, arg1, arg2 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "KasUrlsByAgentId", reflect.TypeOf((*MockQuerier)(nil).KasUrlsByAgentId), arg0, arg1, arg2)
}
// MockPollingQuerier is a mock of PollingQuerier interface.
type MockPollingQuerier struct {
ctrl *gomock.Controller
recorder *MockPollingQuerierMockRecorder
}
// MockPollingQuerierMockRecorder is the mock recorder for MockPollingQuerier.
type MockPollingQuerierMockRecorder struct {
mock *MockPollingQuerier
}
// NewMockPollingQuerier creates a new mock instance.
func NewMockPollingQuerier(ctrl *gomock.Controller) *MockPollingQuerier {
mock := &MockPollingQuerier{ctrl: ctrl}
mock.recorder = &MockPollingQuerierMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockPollingQuerier) EXPECT() *MockPollingQuerierMockRecorder {
return m.recorder
}
// CachedKasUrlsByAgentId mocks base method.
func (m *MockPollingQuerier) CachedKasUrlsByAgentId(arg0 int64) []string {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CachedKasUrlsByAgentId", arg0)
ret0, _ := ret[0].([]string)
return ret0
}
// CachedKasUrlsByAgentId indicates an expected call of CachedKasUrlsByAgentId.
func (mr *MockPollingQuerierMockRecorder) CachedKasUrlsByAgentId(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CachedKasUrlsByAgentId", reflect.TypeOf((*MockPollingQuerier)(nil).CachedKasUrlsByAgentId), arg0)
}
// PollKasUrlsByAgentId mocks base method.
func (m *MockPollingQuerier) PollKasUrlsByAgentId(arg0 context.Context, arg1 int64, arg2 tracker.PollKasUrlsByAgentIdCallback) {
m.ctrl.T.Helper()
m.ctrl.Call(m, "PollKasUrlsByAgentId", arg0, arg1, arg2)
}
// PollKasUrlsByAgentId indicates an expected call of PollKasUrlsByAgentId.
func (mr *MockPollingQuerierMockRecorder) PollKasUrlsByAgentId(arg0, arg1, arg2 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PollKasUrlsByAgentId", reflect.TypeOf((*MockPollingQuerier)(nil).PollKasUrlsByAgentId), arg0, arg1, arg2)
}
|
