// Code generated by MockGen. DO NOT EDIT. // Source: seed_peer.go // Package resource is a generated GoMock package. package resource import ( context "context" reflect "reflect" v1 "d7y.io/api/pkg/apis/scheduler/v1" http "d7y.io/dragonfly/v2/pkg/net/http" types "d7y.io/dragonfly/v2/pkg/types" gomock "github.com/golang/mock/gomock" ) // MockSeedPeer is a mock of SeedPeer interface. type MockSeedPeer struct { ctrl *gomock.Controller recorder *MockSeedPeerMockRecorder } // MockSeedPeerMockRecorder is the mock recorder for MockSeedPeer. type MockSeedPeerMockRecorder struct { mock *MockSeedPeer } // NewMockSeedPeer creates a new mock instance. func NewMockSeedPeer(ctrl *gomock.Controller) *MockSeedPeer { mock := &MockSeedPeer{ctrl: ctrl} mock.recorder = &MockSeedPeerMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockSeedPeer) EXPECT() *MockSeedPeerMockRecorder { return m.recorder } // Client mocks base method. func (m *MockSeedPeer) Client() SeedPeerClient { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Client") ret0, _ := ret[0].(SeedPeerClient) return ret0 } // Client indicates an expected call of Client. func (mr *MockSeedPeerMockRecorder) Client() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Client", reflect.TypeOf((*MockSeedPeer)(nil).Client)) } // DownloadTask mocks base method. func (m *MockSeedPeer) DownloadTask(arg0 context.Context, arg1 *Task, arg2 types.HostType) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DownloadTask", arg0, arg1, arg2) ret0, _ := ret[0].(error) return ret0 } // DownloadTask indicates an expected call of DownloadTask. func (mr *MockSeedPeerMockRecorder) DownloadTask(arg0, arg1, arg2 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadTask", reflect.TypeOf((*MockSeedPeer)(nil).DownloadTask), arg0, arg1, arg2) } // Stop mocks base method. func (m *MockSeedPeer) Stop() error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Stop") ret0, _ := ret[0].(error) return ret0 } // Stop indicates an expected call of Stop. func (mr *MockSeedPeerMockRecorder) Stop() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockSeedPeer)(nil).Stop)) } // TriggerTask mocks base method. func (m *MockSeedPeer) TriggerTask(arg0 context.Context, arg1 *http.Range, arg2 *Task) (*Peer, *v1.PeerResult, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "TriggerTask", arg0, arg1, arg2) ret0, _ := ret[0].(*Peer) ret1, _ := ret[1].(*v1.PeerResult) ret2, _ := ret[2].(error) return ret0, ret1, ret2 } // TriggerTask indicates an expected call of TriggerTask. func (mr *MockSeedPeerMockRecorder) TriggerTask(arg0, arg1, arg2 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TriggerTask", reflect.TypeOf((*MockSeedPeer)(nil).TriggerTask), arg0, arg1, arg2) }