Dragonfly2

1
// Code generated by MockGen. DO NOT EDIT.
2
// Source: seed_peer_client.go
3
//
4
// Generated by this command:
5
//
6
//	mockgen -destination seed_peer_client_mock.go -source seed_peer_client.go -package resource
7
//
8
// Package resource is a generated GoMock package.
9
package resource
10

11
import (
12
	context "context"
13
	reflect "reflect"
14

15
	cdnsystem "d7y.io/api/v2/pkg/apis/cdnsystem/v1"
16
	common "d7y.io/api/v2/pkg/apis/common/v1"
17
	dfdaemon "d7y.io/api/v2/pkg/apis/dfdaemon/v2"
18
	config "d7y.io/dragonfly/v2/scheduler/config"
19
	gomock "go.uber.org/mock/gomock"
20
	grpc "google.golang.org/grpc"
21
)
22

23
// MockSeedPeerClient is a mock of SeedPeerClient interface.
24
type MockSeedPeerClient struct {
25
	ctrl     *gomock.Controller
26
	recorder *MockSeedPeerClientMockRecorder
27
}
28

29
// MockSeedPeerClientMockRecorder is the mock recorder for MockSeedPeerClient.
30
type MockSeedPeerClientMockRecorder struct {
31
	mock *MockSeedPeerClient
32
}
33

34
// NewMockSeedPeerClient creates a new mock instance.
35
func NewMockSeedPeerClient(ctrl *gomock.Controller) *MockSeedPeerClient {
36
	mock := &MockSeedPeerClient{ctrl: ctrl}
37
	mock.recorder = &MockSeedPeerClientMockRecorder{mock}
38
	return mock
39
}
40

41
// EXPECT returns an object that allows the caller to indicate expected use.
42
func (m *MockSeedPeerClient) EXPECT() *MockSeedPeerClientMockRecorder {
43
	return m.recorder
44
}
45

46
// Addrs mocks base method.
47
func (m *MockSeedPeerClient) Addrs() []string {
48
	m.ctrl.T.Helper()
49
	ret := m.ctrl.Call(m, "Addrs")
50
	ret0, _ := ret[0].([]string)
51
	return ret0
52
}
53

54
// Addrs indicates an expected call of Addrs.
55
func (mr *MockSeedPeerClientMockRecorder) Addrs() *gomock.Call {
56
	mr.mock.ctrl.T.Helper()
57
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Addrs", reflect.TypeOf((*MockSeedPeerClient)(nil).Addrs))
58
}
59

60
// Close mocks base method.
61
func (m *MockSeedPeerClient) Close() error {
62
	m.ctrl.T.Helper()
63
	ret := m.ctrl.Call(m, "Close")
64
	ret0, _ := ret[0].(error)
65
	return ret0
66
}
67

68
// Close indicates an expected call of Close.
69
func (mr *MockSeedPeerClientMockRecorder) Close() *gomock.Call {
70
	mr.mock.ctrl.T.Helper()
71
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockSeedPeerClient)(nil).Close))
72
}
73

74
// DownloadPiece mocks base method.
75
func (m *MockSeedPeerClient) DownloadPiece(arg0 context.Context, arg1 *dfdaemon.DownloadPieceRequest, arg2 ...grpc.CallOption) (*dfdaemon.DownloadPieceResponse, error) {
76
	m.ctrl.T.Helper()
77
	varargs := []any{arg0, arg1}
78
	for _, a := range arg2 {
79
		varargs = append(varargs, a)
80
	}
81
	ret := m.ctrl.Call(m, "DownloadPiece", varargs...)
82
	ret0, _ := ret[0].(*dfdaemon.DownloadPieceResponse)
83
	ret1, _ := ret[1].(error)
84
	return ret0, ret1
85
}
86

87
// DownloadPiece indicates an expected call of DownloadPiece.
88
func (mr *MockSeedPeerClientMockRecorder) DownloadPiece(arg0, arg1 any, arg2 ...any) *gomock.Call {
89
	mr.mock.ctrl.T.Helper()
90
	varargs := append([]any{arg0, arg1}, arg2...)
91
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadPiece", reflect.TypeOf((*MockSeedPeerClient)(nil).DownloadPiece), varargs...)
92
}
93

94
// GetPieceTasks mocks base method.
95
func (m *MockSeedPeerClient) GetPieceTasks(arg0 context.Context, arg1 *common.PieceTaskRequest, arg2 ...grpc.CallOption) (*common.PiecePacket, error) {
96
	m.ctrl.T.Helper()
97
	varargs := []any{arg0, arg1}
98
	for _, a := range arg2 {
99
		varargs = append(varargs, a)
100
	}
101
	ret := m.ctrl.Call(m, "GetPieceTasks", varargs...)
102
	ret0, _ := ret[0].(*common.PiecePacket)
103
	ret1, _ := ret[1].(error)
104
	return ret0, ret1
105
}
106

107
// GetPieceTasks indicates an expected call of GetPieceTasks.
108
func (mr *MockSeedPeerClientMockRecorder) GetPieceTasks(arg0, arg1 any, arg2 ...any) *gomock.Call {
109
	mr.mock.ctrl.T.Helper()
110
	varargs := append([]any{arg0, arg1}, arg2...)
111
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceTasks", reflect.TypeOf((*MockSeedPeerClient)(nil).GetPieceTasks), varargs...)
112
}
113

114
// ObtainSeeds mocks base method.
115
func (m *MockSeedPeerClient) ObtainSeeds(arg0 context.Context, arg1 *cdnsystem.SeedRequest, arg2 ...grpc.CallOption) (cdnsystem.Seeder_ObtainSeedsClient, error) {
116
	m.ctrl.T.Helper()
117
	varargs := []any{arg0, arg1}
118
	for _, a := range arg2 {
119
		varargs = append(varargs, a)
120
	}
121
	ret := m.ctrl.Call(m, "ObtainSeeds", varargs...)
122
	ret0, _ := ret[0].(cdnsystem.Seeder_ObtainSeedsClient)
123
	ret1, _ := ret[1].(error)
124
	return ret0, ret1
125
}
126

127
// ObtainSeeds indicates an expected call of ObtainSeeds.
128
func (mr *MockSeedPeerClientMockRecorder) ObtainSeeds(arg0, arg1 any, arg2 ...any) *gomock.Call {
129
	mr.mock.ctrl.T.Helper()
130
	varargs := append([]any{arg0, arg1}, arg2...)
131
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ObtainSeeds", reflect.TypeOf((*MockSeedPeerClient)(nil).ObtainSeeds), varargs...)
132
}
133

134
// OnNotify mocks base method.
135
func (m *MockSeedPeerClient) OnNotify(arg0 *config.DynconfigData) {
136
	m.ctrl.T.Helper()
137
	m.ctrl.Call(m, "OnNotify", arg0)
138
}
139

140
// OnNotify indicates an expected call of OnNotify.
141
func (mr *MockSeedPeerClientMockRecorder) OnNotify(arg0 any) *gomock.Call {
142
	mr.mock.ctrl.T.Helper()
143
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnNotify", reflect.TypeOf((*MockSeedPeerClient)(nil).OnNotify), arg0)
144
}
145

146
// SyncPieceTasks mocks base method.
147
func (m *MockSeedPeerClient) SyncPieceTasks(arg0 context.Context, arg1 *common.PieceTaskRequest, arg2 ...grpc.CallOption) (cdnsystem.Seeder_SyncPieceTasksClient, error) {
148
	m.ctrl.T.Helper()
149
	varargs := []any{arg0, arg1}
150
	for _, a := range arg2 {
151
		varargs = append(varargs, a)
152
	}
153
	ret := m.ctrl.Call(m, "SyncPieceTasks", varargs...)
154
	ret0, _ := ret[0].(cdnsystem.Seeder_SyncPieceTasksClient)
155
	ret1, _ := ret[1].(error)
156
	return ret0, ret1
157
}
158

159
// SyncPieceTasks indicates an expected call of SyncPieceTasks.
160
func (mr *MockSeedPeerClientMockRecorder) SyncPieceTasks(arg0, arg1 any, arg2 ...any) *gomock.Call {
161
	mr.mock.ctrl.T.Helper()
162
	varargs := append([]any{arg0, arg1}, arg2...)
163
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieceTasks", reflect.TypeOf((*MockSeedPeerClient)(nil).SyncPieceTasks), varargs...)
164
}
165

166
// SyncPieces mocks base method.
167
func (m *MockSeedPeerClient) SyncPieces(arg0 context.Context, arg1 *dfdaemon.SyncPiecesRequest, arg2 ...grpc.CallOption) (dfdaemon.DfdaemonUpload_SyncPiecesClient, error) {
168
	m.ctrl.T.Helper()
169
	varargs := []any{arg0, arg1}
170
	for _, a := range arg2 {
171
		varargs = append(varargs, a)
172
	}
173
	ret := m.ctrl.Call(m, "SyncPieces", varargs...)
174
	ret0, _ := ret[0].(dfdaemon.DfdaemonUpload_SyncPiecesClient)
175
	ret1, _ := ret[1].(error)
176
	return ret0, ret1
177
}
178

179
// SyncPieces indicates an expected call of SyncPieces.
180
func (mr *MockSeedPeerClientMockRecorder) SyncPieces(arg0, arg1 any, arg2 ...any) *gomock.Call {
181
	mr.mock.ctrl.T.Helper()
182
	varargs := append([]any{arg0, arg1}, arg2...)
183
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieces", reflect.TypeOf((*MockSeedPeerClient)(nil).SyncPieces), varargs...)
184
}
185

186
// TriggerDownloadTask mocks base method.
187
func (m *MockSeedPeerClient) TriggerDownloadTask(arg0 context.Context, arg1 string, arg2 *dfdaemon.TriggerDownloadTaskRequest, arg3 ...grpc.CallOption) error {
188
	m.ctrl.T.Helper()
189
	varargs := []any{arg0, arg1, arg2}
190
	for _, a := range arg3 {
191
		varargs = append(varargs, a)
192
	}
193
	ret := m.ctrl.Call(m, "TriggerDownloadTask", varargs...)
194
	ret0, _ := ret[0].(error)
195
	return ret0
196
}
197

198
// TriggerDownloadTask indicates an expected call of TriggerDownloadTask.
199
func (mr *MockSeedPeerClientMockRecorder) TriggerDownloadTask(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call {
200
	mr.mock.ctrl.T.Helper()
201
	varargs := append([]any{arg0, arg1, arg2}, arg3...)
202
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TriggerDownloadTask", reflect.TypeOf((*MockSeedPeerClient)(nil).TriggerDownloadTask), varargs...)
203
}
204