dragonfly/trainer/storage/mocks/storage_mock.go

139 lines
4.6 KiB
Go

// Code generated by MockGen. DO NOT EDIT.
// Source: storage.go
// Package mocks is a generated GoMock package.
package mocks
import (
os "os"
reflect "reflect"
storage "d7y.io/dragonfly/v2/scheduler/storage"
gomock "github.com/golang/mock/gomock"
)
// MockStorage is a mock of Storage interface.
type MockStorage struct {
ctrl *gomock.Controller
recorder *MockStorageMockRecorder
}
// 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
}
// Clear mocks base method.
func (m *MockStorage) Clear() error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Clear")
ret0, _ := ret[0].(error)
return ret0
}
// Clear indicates an expected call of Clear.
func (mr *MockStorageMockRecorder) Clear() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Clear", reflect.TypeOf((*MockStorage)(nil).Clear))
}
// ClearDownload mocks base method.
func (m *MockStorage) ClearDownload(arg0 string) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ClearDownload", arg0)
ret0, _ := ret[0].(error)
return ret0
}
// ClearDownload indicates an expected call of ClearDownload.
func (mr *MockStorageMockRecorder) ClearDownload(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearDownload", reflect.TypeOf((*MockStorage)(nil).ClearDownload), arg0)
}
// ClearNetworkTopology mocks base method.
func (m *MockStorage) ClearNetworkTopology(arg0 string) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ClearNetworkTopology", arg0)
ret0, _ := ret[0].(error)
return ret0
}
// ClearNetworkTopology indicates an expected call of ClearNetworkTopology.
func (mr *MockStorageMockRecorder) ClearNetworkTopology(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearNetworkTopology", reflect.TypeOf((*MockStorage)(nil).ClearNetworkTopology), arg0)
}
// ListDownload mocks base method.
func (m *MockStorage) ListDownload(arg0 string) ([]storage.Download, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListDownload", arg0)
ret0, _ := ret[0].([]storage.Download)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListDownload indicates an expected call of ListDownload.
func (mr *MockStorageMockRecorder) ListDownload(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDownload", reflect.TypeOf((*MockStorage)(nil).ListDownload), arg0)
}
// ListNetworkTopology mocks base method.
func (m *MockStorage) ListNetworkTopology(arg0 string) ([]storage.NetworkTopology, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ListNetworkTopology", arg0)
ret0, _ := ret[0].([]storage.NetworkTopology)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ListNetworkTopology indicates an expected call of ListNetworkTopology.
func (mr *MockStorageMockRecorder) ListNetworkTopology(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListNetworkTopology", reflect.TypeOf((*MockStorage)(nil).ListNetworkTopology), arg0)
}
// OpenDownload mocks base method.
func (m *MockStorage) OpenDownload(arg0 string) (*os.File, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "OpenDownload", arg0)
ret0, _ := ret[0].(*os.File)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// OpenDownload indicates an expected call of OpenDownload.
func (mr *MockStorageMockRecorder) OpenDownload(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenDownload", reflect.TypeOf((*MockStorage)(nil).OpenDownload), arg0)
}
// OpenNetworkTopology mocks base method.
func (m *MockStorage) OpenNetworkTopology(arg0 string) (*os.File, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "OpenNetworkTopology", arg0)
ret0, _ := ret[0].(*os.File)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// OpenNetworkTopology indicates an expected call of OpenNetworkTopology.
func (mr *MockStorageMockRecorder) OpenNetworkTopology(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenNetworkTopology", reflect.TypeOf((*MockStorage)(nil).OpenNetworkTopology), arg0)
}