// Code generated by MockGen. DO NOT EDIT. // Source: storage.go // // Generated by this command: // // mockgen -destination mocks/storage_mock.go -source storage.go -package mocks // // Package mocks is a generated GoMock package. package mocks import ( io "io" reflect "reflect" storage "d7y.io/dragonfly/v2/scheduler/storage" gomock "go.uber.org/mock/gomock" ) // MockStorage is a mock of Storage interface. type MockStorage struct { ctrl *gomock.Controller recorder *MockStorageMockRecorder isgomock struct{} } // MockStorageMockRecorder is the mock recorder for MockStorage. type MockStorageMockRecorder struct { mock *MockStorage } // NewMockStorage creates a new mock instance. func NewMockStorage(ctrl *gomock.Controller) *MockStorage { mock := &MockStorage{ctrl: ctrl} mock.recorder = &MockStorageMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockStorage) EXPECT() *MockStorageMockRecorder { return m.recorder } // ClearDownload mocks base method. func (m *MockStorage) ClearDownload() error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ClearDownload") ret0, _ := ret[0].(error) return ret0 } // ClearDownload indicates an expected call of ClearDownload. func (mr *MockStorageMockRecorder) ClearDownload() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearDownload", reflect.TypeOf((*MockStorage)(nil).ClearDownload)) } // CreateDownload mocks base method. func (m *MockStorage) CreateDownload(arg0 storage.Download) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "CreateDownload", arg0) ret0, _ := ret[0].(error) return ret0 } // CreateDownload indicates an expected call of CreateDownload. func (mr *MockStorageMockRecorder) CreateDownload(arg0 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDownload", reflect.TypeOf((*MockStorage)(nil).CreateDownload), arg0) } // DownloadCount mocks base method. func (m *MockStorage) DownloadCount() int64 { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DownloadCount") ret0, _ := ret[0].(int64) return ret0 } // DownloadCount indicates an expected call of DownloadCount. func (mr *MockStorageMockRecorder) DownloadCount() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadCount", reflect.TypeOf((*MockStorage)(nil).DownloadCount)) } // ListDownload mocks base method. func (m *MockStorage) ListDownload() ([]storage.Download, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "ListDownload") ret0, _ := ret[0].([]storage.Download) ret1, _ := ret[1].(error) return ret0, ret1 } // ListDownload indicates an expected call of ListDownload. func (mr *MockStorageMockRecorder) ListDownload() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDownload", reflect.TypeOf((*MockStorage)(nil).ListDownload)) } // OpenDownload mocks base method. func (m *MockStorage) OpenDownload() (io.ReadCloser, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "OpenDownload") ret0, _ := ret[0].(io.ReadCloser) ret1, _ := ret[1].(error) return ret0, ret1 } // OpenDownload indicates an expected call of OpenDownload. func (mr *MockStorageMockRecorder) OpenDownload() *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenDownload", reflect.TypeOf((*MockStorage)(nil).OpenDownload)) }