Dragonfly2

1
// Code generated by MockGen. DO NOT EDIT.
2
// Source: piece_downloader.go
3
//
4
// Generated by this command:
5
//
6
//	mockgen -destination piece_downloader_mock.go -source piece_downloader.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
	gomock "go.uber.org/mock/gomock"
17
)
18

19
// MockPieceDownloader is a mock of PieceDownloader interface.
20
type MockPieceDownloader struct {
21
	ctrl     *gomock.Controller
22
	recorder *MockPieceDownloaderMockRecorder
23
}
24

25
// MockPieceDownloaderMockRecorder is the mock recorder for MockPieceDownloader.
26
type MockPieceDownloaderMockRecorder struct {
27
	mock *MockPieceDownloader
28
}
29

30
// NewMockPieceDownloader creates a new mock instance.
31
func NewMockPieceDownloader(ctrl *gomock.Controller) *MockPieceDownloader {
32
	mock := &MockPieceDownloader{ctrl: ctrl}
33
	mock.recorder = &MockPieceDownloaderMockRecorder{mock}
34
	return mock
35
}
36

37
// EXPECT returns an object that allows the caller to indicate expected use.
38
func (m *MockPieceDownloader) EXPECT() *MockPieceDownloaderMockRecorder {
39
	return m.recorder
40
}
41

42
// DownloadPiece mocks base method.
43
func (m *MockPieceDownloader) DownloadPiece(arg0 context.Context, arg1 *DownloadPieceRequest) (io.Reader, io.Closer, error) {
44
	m.ctrl.T.Helper()
45
	ret := m.ctrl.Call(m, "DownloadPiece", arg0, arg1)
46
	ret0, _ := ret[0].(io.Reader)
47
	ret1, _ := ret[1].(io.Closer)
48
	ret2, _ := ret[2].(error)
49
	return ret0, ret1, ret2
50
}
51

52
// DownloadPiece indicates an expected call of DownloadPiece.
53
func (mr *MockPieceDownloaderMockRecorder) DownloadPiece(arg0, arg1 any) *gomock.Call {
54
	mr.mock.ctrl.T.Helper()
55
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadPiece", reflect.TypeOf((*MockPieceDownloader)(nil).DownloadPiece), arg0, arg1)
56
}
57