Dragonfly2

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

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

16
	dfdaemon "d7y.io/api/v2/pkg/apis/dfdaemon/v1"
17
	scheduler "d7y.io/api/v2/pkg/apis/scheduler/v1"
18
	storage "d7y.io/dragonfly/v2/client/daemon/storage"
19
	http "d7y.io/dragonfly/v2/pkg/net/http"
20
	gomock "go.uber.org/mock/gomock"
21
)
22

23
// MockPieceManager is a mock of PieceManager interface.
24
type MockPieceManager struct {
25
	ctrl     *gomock.Controller
26
	recorder *MockPieceManagerMockRecorder
27
}
28

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

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

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

46
// DownloadPiece mocks base method.
47
func (m *MockPieceManager) DownloadPiece(ctx context.Context, request *DownloadPieceRequest) (*DownloadPieceResult, error) {
48
	m.ctrl.T.Helper()
49
	ret := m.ctrl.Call(m, "DownloadPiece", ctx, request)
50
	ret0, _ := ret[0].(*DownloadPieceResult)
51
	ret1, _ := ret[1].(error)
52
	return ret0, ret1
53
}
54

55
// DownloadPiece indicates an expected call of DownloadPiece.
56
func (mr *MockPieceManagerMockRecorder) DownloadPiece(ctx, request any) *gomock.Call {
57
	mr.mock.ctrl.T.Helper()
58
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadPiece", reflect.TypeOf((*MockPieceManager)(nil).DownloadPiece), ctx, request)
59
}
60

61
// DownloadSource mocks base method.
62
func (m *MockPieceManager) DownloadSource(ctx context.Context, pt Task, request *scheduler.PeerTaskRequest, parsedRange *http.Range) error {
63
	m.ctrl.T.Helper()
64
	ret := m.ctrl.Call(m, "DownloadSource", ctx, pt, request, parsedRange)
65
	ret0, _ := ret[0].(error)
66
	return ret0
67
}
68

69
// DownloadSource indicates an expected call of DownloadSource.
70
func (mr *MockPieceManagerMockRecorder) DownloadSource(ctx, pt, request, parsedRange any) *gomock.Call {
71
	mr.mock.ctrl.T.Helper()
72
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadSource", reflect.TypeOf((*MockPieceManager)(nil).DownloadSource), ctx, pt, request, parsedRange)
73
}
74

75
// Import mocks base method.
76
func (m *MockPieceManager) Import(ctx context.Context, ptm storage.PeerTaskMetadata, tsd storage.TaskStorageDriver, contentLength int64, reader io.Reader) error {
77
	m.ctrl.T.Helper()
78
	ret := m.ctrl.Call(m, "Import", ctx, ptm, tsd, contentLength, reader)
79
	ret0, _ := ret[0].(error)
80
	return ret0
81
}
82

83
// Import indicates an expected call of Import.
84
func (mr *MockPieceManagerMockRecorder) Import(ctx, ptm, tsd, contentLength, reader any) *gomock.Call {
85
	mr.mock.ctrl.T.Helper()
86
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Import", reflect.TypeOf((*MockPieceManager)(nil).Import), ctx, ptm, tsd, contentLength, reader)
87
}
88

89
// ImportFile mocks base method.
90
func (m *MockPieceManager) ImportFile(ctx context.Context, ptm storage.PeerTaskMetadata, tsd storage.TaskStorageDriver, req *dfdaemon.ImportTaskRequest) error {
91
	m.ctrl.T.Helper()
92
	ret := m.ctrl.Call(m, "ImportFile", ctx, ptm, tsd, req)
93
	ret0, _ := ret[0].(error)
94
	return ret0
95
}
96

97
// ImportFile indicates an expected call of ImportFile.
98
func (mr *MockPieceManagerMockRecorder) ImportFile(ctx, ptm, tsd, req any) *gomock.Call {
99
	mr.mock.ctrl.T.Helper()
100
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ImportFile", reflect.TypeOf((*MockPieceManager)(nil).ImportFile), ctx, ptm, tsd, req)
101
}
102