547 lines
20 KiB
Go
547 lines
20 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: storage_manager.go
|
|
|
|
// Package mocks is a generated GoMock package.
|
|
package mocks
|
|
|
|
import (
|
|
context "context"
|
|
io "io"
|
|
reflect "reflect"
|
|
time "time"
|
|
|
|
common "d7y.io/api/v2/pkg/apis/common/v1"
|
|
storage "d7y.io/dragonfly/v2/client/daemon/storage"
|
|
http "d7y.io/dragonfly/v2/pkg/net/http"
|
|
gomock "github.com/golang/mock/gomock"
|
|
)
|
|
|
|
// MockTaskStorageDriver is a mock of TaskStorageDriver interface.
|
|
type MockTaskStorageDriver struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockTaskStorageDriverMockRecorder
|
|
}
|
|
|
|
// MockTaskStorageDriverMockRecorder is the mock recorder for MockTaskStorageDriver.
|
|
type MockTaskStorageDriverMockRecorder struct {
|
|
mock *MockTaskStorageDriver
|
|
}
|
|
|
|
// NewMockTaskStorageDriver creates a new mock instance.
|
|
func NewMockTaskStorageDriver(ctrl *gomock.Controller) *MockTaskStorageDriver {
|
|
mock := &MockTaskStorageDriver{ctrl: ctrl}
|
|
mock.recorder = &MockTaskStorageDriverMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockTaskStorageDriver) EXPECT() *MockTaskStorageDriverMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// GetExtendAttribute mocks base method.
|
|
func (m *MockTaskStorageDriver) GetExtendAttribute(ctx context.Context, req *storage.PeerTaskMetadata) (*common.ExtendAttribute, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetExtendAttribute", ctx, req)
|
|
ret0, _ := ret[0].(*common.ExtendAttribute)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetExtendAttribute indicates an expected call of GetExtendAttribute.
|
|
func (mr *MockTaskStorageDriverMockRecorder) GetExtendAttribute(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExtendAttribute", reflect.TypeOf((*MockTaskStorageDriver)(nil).GetExtendAttribute), ctx, req)
|
|
}
|
|
|
|
// GetPieces mocks base method.
|
|
func (m *MockTaskStorageDriver) GetPieces(ctx context.Context, req *common.PieceTaskRequest) (*common.PiecePacket, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetPieces", ctx, req)
|
|
ret0, _ := ret[0].(*common.PiecePacket)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetPieces indicates an expected call of GetPieces.
|
|
func (mr *MockTaskStorageDriverMockRecorder) GetPieces(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieces", reflect.TypeOf((*MockTaskStorageDriver)(nil).GetPieces), ctx, req)
|
|
}
|
|
|
|
// GetTotalPieces mocks base method.
|
|
func (m *MockTaskStorageDriver) GetTotalPieces(ctx context.Context, req *storage.PeerTaskMetadata) (int32, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetTotalPieces", ctx, req)
|
|
ret0, _ := ret[0].(int32)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetTotalPieces indicates an expected call of GetTotalPieces.
|
|
func (mr *MockTaskStorageDriverMockRecorder) GetTotalPieces(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTotalPieces", reflect.TypeOf((*MockTaskStorageDriver)(nil).GetTotalPieces), ctx, req)
|
|
}
|
|
|
|
// IsInvalid mocks base method.
|
|
func (m *MockTaskStorageDriver) IsInvalid(req *storage.PeerTaskMetadata) (bool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "IsInvalid", req)
|
|
ret0, _ := ret[0].(bool)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// IsInvalid indicates an expected call of IsInvalid.
|
|
func (mr *MockTaskStorageDriverMockRecorder) IsInvalid(req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsInvalid", reflect.TypeOf((*MockTaskStorageDriver)(nil).IsInvalid), req)
|
|
}
|
|
|
|
// ReadAllPieces mocks base method.
|
|
func (m *MockTaskStorageDriver) ReadAllPieces(ctx context.Context, req *storage.ReadAllPiecesRequest) (io.ReadCloser, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ReadAllPieces", ctx, req)
|
|
ret0, _ := ret[0].(io.ReadCloser)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ReadAllPieces indicates an expected call of ReadAllPieces.
|
|
func (mr *MockTaskStorageDriverMockRecorder) ReadAllPieces(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadAllPieces", reflect.TypeOf((*MockTaskStorageDriver)(nil).ReadAllPieces), ctx, req)
|
|
}
|
|
|
|
// ReadPiece mocks base method.
|
|
func (m *MockTaskStorageDriver) ReadPiece(ctx context.Context, req *storage.ReadPieceRequest) (io.Reader, io.Closer, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ReadPiece", ctx, req)
|
|
ret0, _ := ret[0].(io.Reader)
|
|
ret1, _ := ret[1].(io.Closer)
|
|
ret2, _ := ret[2].(error)
|
|
return ret0, ret1, ret2
|
|
}
|
|
|
|
// ReadPiece indicates an expected call of ReadPiece.
|
|
func (mr *MockTaskStorageDriverMockRecorder) ReadPiece(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadPiece", reflect.TypeOf((*MockTaskStorageDriver)(nil).ReadPiece), ctx, req)
|
|
}
|
|
|
|
// Store mocks base method.
|
|
func (m *MockTaskStorageDriver) Store(ctx context.Context, req *storage.StoreRequest) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Store", ctx, req)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Store indicates an expected call of Store.
|
|
func (mr *MockTaskStorageDriverMockRecorder) Store(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Store", reflect.TypeOf((*MockTaskStorageDriver)(nil).Store), ctx, req)
|
|
}
|
|
|
|
// UpdateTask mocks base method.
|
|
func (m *MockTaskStorageDriver) UpdateTask(ctx context.Context, req *storage.UpdateTaskRequest) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "UpdateTask", ctx, req)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// UpdateTask indicates an expected call of UpdateTask.
|
|
func (mr *MockTaskStorageDriverMockRecorder) UpdateTask(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTask", reflect.TypeOf((*MockTaskStorageDriver)(nil).UpdateTask), ctx, req)
|
|
}
|
|
|
|
// ValidateDigest mocks base method.
|
|
func (m *MockTaskStorageDriver) ValidateDigest(req *storage.PeerTaskMetadata) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ValidateDigest", req)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// ValidateDigest indicates an expected call of ValidateDigest.
|
|
func (mr *MockTaskStorageDriverMockRecorder) ValidateDigest(req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateDigest", reflect.TypeOf((*MockTaskStorageDriver)(nil).ValidateDigest), req)
|
|
}
|
|
|
|
// WritePiece mocks base method.
|
|
func (m *MockTaskStorageDriver) WritePiece(ctx context.Context, req *storage.WritePieceRequest) (int64, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "WritePiece", ctx, req)
|
|
ret0, _ := ret[0].(int64)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// WritePiece indicates an expected call of WritePiece.
|
|
func (mr *MockTaskStorageDriverMockRecorder) WritePiece(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WritePiece", reflect.TypeOf((*MockTaskStorageDriver)(nil).WritePiece), ctx, req)
|
|
}
|
|
|
|
// MockReclaimer is a mock of Reclaimer interface.
|
|
type MockReclaimer struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockReclaimerMockRecorder
|
|
}
|
|
|
|
// MockReclaimerMockRecorder is the mock recorder for MockReclaimer.
|
|
type MockReclaimerMockRecorder struct {
|
|
mock *MockReclaimer
|
|
}
|
|
|
|
// NewMockReclaimer creates a new mock instance.
|
|
func NewMockReclaimer(ctrl *gomock.Controller) *MockReclaimer {
|
|
mock := &MockReclaimer{ctrl: ctrl}
|
|
mock.recorder = &MockReclaimerMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockReclaimer) EXPECT() *MockReclaimerMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// CanReclaim mocks base method.
|
|
func (m *MockReclaimer) CanReclaim() bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CanReclaim")
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// CanReclaim indicates an expected call of CanReclaim.
|
|
func (mr *MockReclaimerMockRecorder) CanReclaim() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CanReclaim", reflect.TypeOf((*MockReclaimer)(nil).CanReclaim))
|
|
}
|
|
|
|
// MarkReclaim mocks base method.
|
|
func (m *MockReclaimer) MarkReclaim() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "MarkReclaim")
|
|
}
|
|
|
|
// MarkReclaim indicates an expected call of MarkReclaim.
|
|
func (mr *MockReclaimerMockRecorder) MarkReclaim() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MarkReclaim", reflect.TypeOf((*MockReclaimer)(nil).MarkReclaim))
|
|
}
|
|
|
|
// Reclaim mocks base method.
|
|
func (m *MockReclaimer) Reclaim() error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Reclaim")
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Reclaim indicates an expected call of Reclaim.
|
|
func (mr *MockReclaimerMockRecorder) Reclaim() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Reclaim", reflect.TypeOf((*MockReclaimer)(nil).Reclaim))
|
|
}
|
|
|
|
// MockManager is a mock of Manager interface.
|
|
type MockManager struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockManagerMockRecorder
|
|
}
|
|
|
|
// MockManagerMockRecorder is the mock recorder for MockManager.
|
|
type MockManagerMockRecorder struct {
|
|
mock *MockManager
|
|
}
|
|
|
|
// NewMockManager creates a new mock instance.
|
|
func NewMockManager(ctrl *gomock.Controller) *MockManager {
|
|
mock := &MockManager{ctrl: ctrl}
|
|
mock.recorder = &MockManagerMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockManager) EXPECT() *MockManagerMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Alive mocks base method.
|
|
func (m *MockManager) Alive(alive time.Duration) bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Alive", alive)
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// Alive indicates an expected call of Alive.
|
|
func (mr *MockManagerMockRecorder) Alive(alive interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Alive", reflect.TypeOf((*MockManager)(nil).Alive), alive)
|
|
}
|
|
|
|
// CleanUp mocks base method.
|
|
func (m *MockManager) CleanUp() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "CleanUp")
|
|
}
|
|
|
|
// CleanUp indicates an expected call of CleanUp.
|
|
func (mr *MockManagerMockRecorder) CleanUp() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanUp", reflect.TypeOf((*MockManager)(nil).CleanUp))
|
|
}
|
|
|
|
// FindCompletedSubTask mocks base method.
|
|
func (m *MockManager) FindCompletedSubTask(taskID string) *storage.ReusePeerTask {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FindCompletedSubTask", taskID)
|
|
ret0, _ := ret[0].(*storage.ReusePeerTask)
|
|
return ret0
|
|
}
|
|
|
|
// FindCompletedSubTask indicates an expected call of FindCompletedSubTask.
|
|
func (mr *MockManagerMockRecorder) FindCompletedSubTask(taskID interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FindCompletedSubTask", reflect.TypeOf((*MockManager)(nil).FindCompletedSubTask), taskID)
|
|
}
|
|
|
|
// FindCompletedTask mocks base method.
|
|
func (m *MockManager) FindCompletedTask(taskID string) *storage.ReusePeerTask {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FindCompletedTask", taskID)
|
|
ret0, _ := ret[0].(*storage.ReusePeerTask)
|
|
return ret0
|
|
}
|
|
|
|
// FindCompletedTask indicates an expected call of FindCompletedTask.
|
|
func (mr *MockManagerMockRecorder) FindCompletedTask(taskID interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FindCompletedTask", reflect.TypeOf((*MockManager)(nil).FindCompletedTask), taskID)
|
|
}
|
|
|
|
// FindPartialCompletedTask mocks base method.
|
|
func (m *MockManager) FindPartialCompletedTask(taskID string, rg *http.Range) *storage.ReusePeerTask {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FindPartialCompletedTask", taskID, rg)
|
|
ret0, _ := ret[0].(*storage.ReusePeerTask)
|
|
return ret0
|
|
}
|
|
|
|
// FindPartialCompletedTask indicates an expected call of FindPartialCompletedTask.
|
|
func (mr *MockManagerMockRecorder) FindPartialCompletedTask(taskID, rg interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FindPartialCompletedTask", reflect.TypeOf((*MockManager)(nil).FindPartialCompletedTask), taskID, rg)
|
|
}
|
|
|
|
// GetExtendAttribute mocks base method.
|
|
func (m *MockManager) GetExtendAttribute(ctx context.Context, req *storage.PeerTaskMetadata) (*common.ExtendAttribute, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetExtendAttribute", ctx, req)
|
|
ret0, _ := ret[0].(*common.ExtendAttribute)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetExtendAttribute indicates an expected call of GetExtendAttribute.
|
|
func (mr *MockManagerMockRecorder) GetExtendAttribute(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExtendAttribute", reflect.TypeOf((*MockManager)(nil).GetExtendAttribute), ctx, req)
|
|
}
|
|
|
|
// GetPieces mocks base method.
|
|
func (m *MockManager) GetPieces(ctx context.Context, req *common.PieceTaskRequest) (*common.PiecePacket, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetPieces", ctx, req)
|
|
ret0, _ := ret[0].(*common.PiecePacket)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetPieces indicates an expected call of GetPieces.
|
|
func (mr *MockManagerMockRecorder) GetPieces(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieces", reflect.TypeOf((*MockManager)(nil).GetPieces), ctx, req)
|
|
}
|
|
|
|
// GetTotalPieces mocks base method.
|
|
func (m *MockManager) GetTotalPieces(ctx context.Context, req *storage.PeerTaskMetadata) (int32, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetTotalPieces", ctx, req)
|
|
ret0, _ := ret[0].(int32)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetTotalPieces indicates an expected call of GetTotalPieces.
|
|
func (mr *MockManagerMockRecorder) GetTotalPieces(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTotalPieces", reflect.TypeOf((*MockManager)(nil).GetTotalPieces), ctx, req)
|
|
}
|
|
|
|
// IsInvalid mocks base method.
|
|
func (m *MockManager) IsInvalid(req *storage.PeerTaskMetadata) (bool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "IsInvalid", req)
|
|
ret0, _ := ret[0].(bool)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// IsInvalid indicates an expected call of IsInvalid.
|
|
func (mr *MockManagerMockRecorder) IsInvalid(req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsInvalid", reflect.TypeOf((*MockManager)(nil).IsInvalid), req)
|
|
}
|
|
|
|
// Keep mocks base method.
|
|
func (m *MockManager) Keep() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "Keep")
|
|
}
|
|
|
|
// Keep indicates an expected call of Keep.
|
|
func (mr *MockManagerMockRecorder) Keep() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Keep", reflect.TypeOf((*MockManager)(nil).Keep))
|
|
}
|
|
|
|
// ReadAllPieces mocks base method.
|
|
func (m *MockManager) ReadAllPieces(ctx context.Context, req *storage.ReadAllPiecesRequest) (io.ReadCloser, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ReadAllPieces", ctx, req)
|
|
ret0, _ := ret[0].(io.ReadCloser)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ReadAllPieces indicates an expected call of ReadAllPieces.
|
|
func (mr *MockManagerMockRecorder) ReadAllPieces(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadAllPieces", reflect.TypeOf((*MockManager)(nil).ReadAllPieces), ctx, req)
|
|
}
|
|
|
|
// ReadPiece mocks base method.
|
|
func (m *MockManager) ReadPiece(ctx context.Context, req *storage.ReadPieceRequest) (io.Reader, io.Closer, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ReadPiece", ctx, req)
|
|
ret0, _ := ret[0].(io.Reader)
|
|
ret1, _ := ret[1].(io.Closer)
|
|
ret2, _ := ret[2].(error)
|
|
return ret0, ret1, ret2
|
|
}
|
|
|
|
// ReadPiece indicates an expected call of ReadPiece.
|
|
func (mr *MockManagerMockRecorder) ReadPiece(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReadPiece", reflect.TypeOf((*MockManager)(nil).ReadPiece), ctx, req)
|
|
}
|
|
|
|
// RegisterSubTask mocks base method.
|
|
func (m *MockManager) RegisterSubTask(ctx context.Context, req *storage.RegisterSubTaskRequest) (storage.TaskStorageDriver, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RegisterSubTask", ctx, req)
|
|
ret0, _ := ret[0].(storage.TaskStorageDriver)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RegisterSubTask indicates an expected call of RegisterSubTask.
|
|
func (mr *MockManagerMockRecorder) RegisterSubTask(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterSubTask", reflect.TypeOf((*MockManager)(nil).RegisterSubTask), ctx, req)
|
|
}
|
|
|
|
// RegisterTask mocks base method.
|
|
func (m *MockManager) RegisterTask(ctx context.Context, req *storage.RegisterTaskRequest) (storage.TaskStorageDriver, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RegisterTask", ctx, req)
|
|
ret0, _ := ret[0].(storage.TaskStorageDriver)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RegisterTask indicates an expected call of RegisterTask.
|
|
func (mr *MockManagerMockRecorder) RegisterTask(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterTask", reflect.TypeOf((*MockManager)(nil).RegisterTask), ctx, req)
|
|
}
|
|
|
|
// Store mocks base method.
|
|
func (m *MockManager) Store(ctx context.Context, req *storage.StoreRequest) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Store", ctx, req)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Store indicates an expected call of Store.
|
|
func (mr *MockManagerMockRecorder) Store(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Store", reflect.TypeOf((*MockManager)(nil).Store), ctx, req)
|
|
}
|
|
|
|
// UnregisterTask mocks base method.
|
|
func (m *MockManager) UnregisterTask(ctx context.Context, req storage.CommonTaskRequest) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "UnregisterTask", ctx, req)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// UnregisterTask indicates an expected call of UnregisterTask.
|
|
func (mr *MockManagerMockRecorder) UnregisterTask(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnregisterTask", reflect.TypeOf((*MockManager)(nil).UnregisterTask), ctx, req)
|
|
}
|
|
|
|
// UpdateTask mocks base method.
|
|
func (m *MockManager) UpdateTask(ctx context.Context, req *storage.UpdateTaskRequest) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "UpdateTask", ctx, req)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// UpdateTask indicates an expected call of UpdateTask.
|
|
func (mr *MockManagerMockRecorder) UpdateTask(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTask", reflect.TypeOf((*MockManager)(nil).UpdateTask), ctx, req)
|
|
}
|
|
|
|
// ValidateDigest mocks base method.
|
|
func (m *MockManager) ValidateDigest(req *storage.PeerTaskMetadata) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ValidateDigest", req)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// ValidateDigest indicates an expected call of ValidateDigest.
|
|
func (mr *MockManagerMockRecorder) ValidateDigest(req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateDigest", reflect.TypeOf((*MockManager)(nil).ValidateDigest), req)
|
|
}
|
|
|
|
// WritePiece mocks base method.
|
|
func (m *MockManager) WritePiece(ctx context.Context, req *storage.WritePieceRequest) (int64, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "WritePiece", ctx, req)
|
|
ret0, _ := ret[0].(int64)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// WritePiece indicates an expected call of WritePiece.
|
|
func (mr *MockManagerMockRecorder) WritePiece(ctx, req interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WritePiece", reflect.TypeOf((*MockManager)(nil).WritePiece), ctx, req)
|
|
}
|