diff --git a/Makefile b/Makefile index 664c2d6f3e7..89447443681 100644 --- a/Makefile +++ b/Makefile @@ -239,12 +239,12 @@ build-e2e-sha256sum: # Run unittests test: - @go test -v -gcflags "all=-l" -race -short ${PKG_LIST} + @go test -v -race -short ${PKG_LIST} .PHONY: test # Run tests with coverage test-coverage: - @go test -v -gcflags "all=-l" -race -short ${PKG_LIST} -coverprofile cover.out -covermode=atomic + @go test -v -race -short ${PKG_LIST} -coverprofile cover.out -covermode=atomic @cat cover.out >> coverage.txt .PHONY: test-coverage diff --git a/client/clientutil/keepalive.go b/client/clientutil/keepalive.go index bf5cb22afab..57ecd8866a3 100644 --- a/client/clientutil/keepalive.go +++ b/client/clientutil/keepalive.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/keepalive_mock.go -source keepalive.go -package mocks + package clientutil import ( @@ -24,8 +26,6 @@ import ( logger "d7y.io/dragonfly/v2/internal/dflog" ) -var _ *logger.SugaredLoggerOnWith // pin this package for no log code generation - type KeepAlive interface { Keep() Alive(alive time.Duration) bool @@ -36,8 +36,6 @@ type keepAlive struct { access atomic.Int64 } -var _ KeepAlive = (*keepAlive)(nil) - func NewKeepAlive(name string) KeepAlive { return &keepAlive{ name: name, diff --git a/client/clientutil/mocks/keepalive_mock.go b/client/clientutil/mocks/keepalive_mock.go new file mode 100644 index 00000000000..71067fe21b0 --- /dev/null +++ b/client/clientutil/mocks/keepalive_mock.go @@ -0,0 +1,61 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: keepalive.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + time "time" + + gomock "github.com/golang/mock/gomock" +) + +// MockKeepAlive is a mock of KeepAlive interface. +type MockKeepAlive struct { + ctrl *gomock.Controller + recorder *MockKeepAliveMockRecorder +} + +// MockKeepAliveMockRecorder is the mock recorder for MockKeepAlive. +type MockKeepAliveMockRecorder struct { + mock *MockKeepAlive +} + +// NewMockKeepAlive creates a new mock instance. +func NewMockKeepAlive(ctrl *gomock.Controller) *MockKeepAlive { + mock := &MockKeepAlive{ctrl: ctrl} + mock.recorder = &MockKeepAliveMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockKeepAlive) EXPECT() *MockKeepAliveMockRecorder { + return m.recorder +} + +// Alive mocks base method. +func (m *MockKeepAlive) 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 *MockKeepAliveMockRecorder) Alive(alive interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Alive", reflect.TypeOf((*MockKeepAlive)(nil).Alive), alive) +} + +// Keep mocks base method. +func (m *MockKeepAlive) Keep() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Keep") +} + +// Keep indicates an expected call of Keep. +func (mr *MockKeepAliveMockRecorder) Keep() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Keep", reflect.TypeOf((*MockKeepAlive)(nil).Keep)) +} diff --git a/client/config/dynconfig.go b/client/config/dynconfig.go index 75ade12f399..fa1fd6c0d55 100644 --- a/client/config/dynconfig.go +++ b/client/config/dynconfig.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/dynconfig_mock.go -source dynconfig.go -package mocks + package config import ( diff --git a/client/config/mocks/dynconfig_mock.go b/client/config/mocks/dynconfig_mock.go new file mode 100644 index 00000000000..65a79d402c0 --- /dev/null +++ b/client/config/mocks/dynconfig_mock.go @@ -0,0 +1,197 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: dynconfig.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + config "d7y.io/dragonfly/v2/client/config" + manager "d7y.io/dragonfly/v2/pkg/rpc/manager" + gomock "github.com/golang/mock/gomock" +) + +// MockDynconfig is a mock of Dynconfig interface. +type MockDynconfig struct { + ctrl *gomock.Controller + recorder *MockDynconfigMockRecorder +} + +// MockDynconfigMockRecorder is the mock recorder for MockDynconfig. +type MockDynconfigMockRecorder struct { + mock *MockDynconfig +} + +// NewMockDynconfig creates a new mock instance. +func NewMockDynconfig(ctrl *gomock.Controller) *MockDynconfig { + mock := &MockDynconfig{ctrl: ctrl} + mock.recorder = &MockDynconfigMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockDynconfig) EXPECT() *MockDynconfigMockRecorder { + return m.recorder +} + +// Deregister mocks base method. +func (m *MockDynconfig) Deregister(arg0 config.Observer) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Deregister", arg0) +} + +// Deregister indicates an expected call of Deregister. +func (mr *MockDynconfigMockRecorder) Deregister(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Deregister", reflect.TypeOf((*MockDynconfig)(nil).Deregister), arg0) +} + +// Get mocks base method. +func (m *MockDynconfig) Get() (*config.DynconfigData, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Get") + ret0, _ := ret[0].(*config.DynconfigData) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Get indicates an expected call of Get. +func (mr *MockDynconfigMockRecorder) Get() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockDynconfig)(nil).Get)) +} + +// GetBuckets mocks base method. +func (m *MockDynconfig) GetBuckets() ([]*manager.Bucket, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetBuckets") + ret0, _ := ret[0].([]*manager.Bucket) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetBuckets indicates an expected call of GetBuckets. +func (mr *MockDynconfigMockRecorder) GetBuckets() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBuckets", reflect.TypeOf((*MockDynconfig)(nil).GetBuckets)) +} + +// GetObjectStorage mocks base method. +func (m *MockDynconfig) GetObjectStorage() (*manager.ObjectStorage, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetObjectStorage") + ret0, _ := ret[0].(*manager.ObjectStorage) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetObjectStorage indicates an expected call of GetObjectStorage. +func (mr *MockDynconfigMockRecorder) GetObjectStorage() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetObjectStorage", reflect.TypeOf((*MockDynconfig)(nil).GetObjectStorage)) +} + +// GetSchedulers mocks base method. +func (m *MockDynconfig) GetSchedulers() ([]*manager.Scheduler, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSchedulers") + ret0, _ := ret[0].([]*manager.Scheduler) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetSchedulers indicates an expected call of GetSchedulers. +func (mr *MockDynconfigMockRecorder) GetSchedulers() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSchedulers", reflect.TypeOf((*MockDynconfig)(nil).GetSchedulers)) +} + +// Notify mocks base method. +func (m *MockDynconfig) Notify() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Notify") + ret0, _ := ret[0].(error) + return ret0 +} + +// Notify indicates an expected call of Notify. +func (mr *MockDynconfigMockRecorder) Notify() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Notify", reflect.TypeOf((*MockDynconfig)(nil).Notify)) +} + +// Register mocks base method. +func (m *MockDynconfig) Register(arg0 config.Observer) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Register", arg0) +} + +// Register indicates an expected call of Register. +func (mr *MockDynconfigMockRecorder) Register(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Register", reflect.TypeOf((*MockDynconfig)(nil).Register), arg0) +} + +// Serve mocks base method. +func (m *MockDynconfig) Serve() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Serve") + ret0, _ := ret[0].(error) + return ret0 +} + +// Serve indicates an expected call of Serve. +func (mr *MockDynconfigMockRecorder) Serve() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Serve", reflect.TypeOf((*MockDynconfig)(nil).Serve)) +} + +// Stop mocks base method. +func (m *MockDynconfig) 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 *MockDynconfigMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockDynconfig)(nil).Stop)) +} + +// MockObserver is a mock of Observer interface. +type MockObserver struct { + ctrl *gomock.Controller + recorder *MockObserverMockRecorder +} + +// MockObserverMockRecorder is the mock recorder for MockObserver. +type MockObserverMockRecorder struct { + mock *MockObserver +} + +// NewMockObserver creates a new mock instance. +func NewMockObserver(ctrl *gomock.Controller) *MockObserver { + mock := &MockObserver{ctrl: ctrl} + mock.recorder = &MockObserverMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockObserver) EXPECT() *MockObserverMockRecorder { + return m.recorder +} + +// OnNotify mocks base method. +func (m *MockObserver) OnNotify(arg0 *config.DynconfigData) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "OnNotify", arg0) +} + +// OnNotify indicates an expected call of OnNotify. +func (mr *MockObserverMockRecorder) OnNotify(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnNotify", reflect.TypeOf((*MockObserver)(nil).OnNotify), arg0) +} diff --git a/client/daemon/objectstorage/mocks/objectstorage_mock.go b/client/daemon/objectstorage/mocks/objectstorage_mock.go new file mode 100644 index 00000000000..982288444af --- /dev/null +++ b/client/daemon/objectstorage/mocks/objectstorage_mock.go @@ -0,0 +1,63 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: objectstorage.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + net "net" + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockObjectStorage is a mock of ObjectStorage interface. +type MockObjectStorage struct { + ctrl *gomock.Controller + recorder *MockObjectStorageMockRecorder +} + +// MockObjectStorageMockRecorder is the mock recorder for MockObjectStorage. +type MockObjectStorageMockRecorder struct { + mock *MockObjectStorage +} + +// NewMockObjectStorage creates a new mock instance. +func NewMockObjectStorage(ctrl *gomock.Controller) *MockObjectStorage { + mock := &MockObjectStorage{ctrl: ctrl} + mock.recorder = &MockObjectStorageMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockObjectStorage) EXPECT() *MockObjectStorageMockRecorder { + return m.recorder +} + +// Serve mocks base method. +func (m *MockObjectStorage) Serve(lis net.Listener) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Serve", lis) + ret0, _ := ret[0].(error) + return ret0 +} + +// Serve indicates an expected call of Serve. +func (mr *MockObjectStorageMockRecorder) Serve(lis interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Serve", reflect.TypeOf((*MockObjectStorage)(nil).Serve), lis) +} + +// Stop mocks base method. +func (m *MockObjectStorage) 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 *MockObjectStorageMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockObjectStorage)(nil).Stop)) +} diff --git a/client/daemon/objectstorage/objectstorage.go b/client/daemon/objectstorage/objectstorage.go index 468fa712597..7a72a92bee5 100644 --- a/client/daemon/objectstorage/objectstorage.go +++ b/client/daemon/objectstorage/objectstorage.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/objectstorage_mock.go -source objectstorage.go -package mocks + package objectstorage import ( diff --git a/client/daemon/peer/peertask_manager.go b/client/daemon/peer/peertask_manager.go index 34c3c8f81bc..24f943b121b 100644 --- a/client/daemon/peer/peertask_manager.go +++ b/client/daemon/peer/peertask_manager.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination peertask_manager_mock.go -source peertask_manager.go -package peer + package peer import ( @@ -70,8 +72,6 @@ type TaskManager interface { Stop(ctx context.Context) error } -//go:generate mockgen -source peertask_manager.go -package peer -self_package d7y.io/dragonfly/v2/client/daemon/peer -destination peertask_manager_mock_test.go -//go:generate mockgen -source peertask_manager.go -destination ../test/mock/peer/peertask_manager.go // Task represents common interface to operate a peer task type Task interface { Logger diff --git a/client/daemon/peer/peertask_manager_mock_test.go b/client/daemon/peer/peertask_manager_mock.go similarity index 100% rename from client/daemon/peer/peertask_manager_mock_test.go rename to client/daemon/peer/peertask_manager_mock.go diff --git a/client/daemon/peer/peertask_manager_test.go b/client/daemon/peer/peertask_manager_test.go index 2ea3dc7fc24..43b94ecd378 100644 --- a/client/daemon/peer/peertask_manager_test.go +++ b/client/daemon/peer/peertask_manager_test.go @@ -46,7 +46,6 @@ import ( "d7y.io/dragonfly/v2/client/config" "d7y.io/dragonfly/v2/client/daemon/storage" "d7y.io/dragonfly/v2/client/daemon/test" - mock_daemon "d7y.io/dragonfly/v2/client/daemon/test/mock/daemon" "d7y.io/dragonfly/v2/internal/dferrors" logger "d7y.io/dragonfly/v2/internal/dflog" "d7y.io/dragonfly/v2/pkg/dfnet" @@ -56,13 +55,14 @@ import ( "d7y.io/dragonfly/v2/pkg/rpc/base" "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon" daemonserver "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon/server" + servermocks "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon/server/mocks" "d7y.io/dragonfly/v2/pkg/rpc/scheduler" schedulerclient "d7y.io/dragonfly/v2/pkg/rpc/scheduler/client" mock_scheduler_client "d7y.io/dragonfly/v2/pkg/rpc/scheduler/client/mocks" mock_scheduler "d7y.io/dragonfly/v2/pkg/rpc/scheduler/mocks" "d7y.io/dragonfly/v2/pkg/source" "d7y.io/dragonfly/v2/pkg/source/clients/httpprotocol" - sourceMock "d7y.io/dragonfly/v2/pkg/source/mock" + sourcemocks "d7y.io/dragonfly/v2/pkg/source/mocks" ) func TestMain(m *testing.M) { @@ -84,12 +84,11 @@ type componentsOption struct { getPieceTasks bool } -//go:generate mockgen -package mock_server_grpc -source ../../../pkg/rpc/dfdaemon/dfdaemon.pb.go -destination ../test/mock/daemongrpc/daemon_server_grpc.go func setupPeerTaskManagerComponents(ctrl *gomock.Controller, opt componentsOption) ( schedulerclient.Client, storage.Manager) { port := int32(freeport.GetPort()) // 1. set up a mock daemon server for uploading pieces info - var daemon = mock_daemon.NewMockDaemonServer(ctrl) + var daemon = servermocks.NewMockDaemonServer(ctrl) // 1.1 calculate piece digest and total digest r := bytes.NewBuffer(opt.content) @@ -425,7 +424,7 @@ func TestPeerTaskManager_TaskSuite(t *testing.T) { sizeScope: base.SizeScope_NORMAL, mockPieceDownloader: nil, mockHTTPSourceClient: func(t *testing.T, ctrl *gomock.Controller, rg *clientutil.Range, taskData []byte, url string) source.ResourceClient { - sourceClient := sourceMock.NewMockResourceClient(ctrl) + sourceClient := sourcemocks.NewMockResourceClient(ctrl) sourceClient.EXPECT().GetContentLength(source.RequestEq(url)).AnyTimes().DoAndReturn( func(request *source.Request) (int64, error) { return int64(len(taskData)), nil @@ -453,7 +452,7 @@ func TestPeerTaskManager_TaskSuite(t *testing.T) { }, mockPieceDownloader: nil, mockHTTPSourceClient: func(t *testing.T, ctrl *gomock.Controller, rg *clientutil.Range, taskData []byte, url string) source.ResourceClient { - sourceClient := sourceMock.NewMockResourceClient(ctrl) + sourceClient := sourcemocks.NewMockResourceClient(ctrl) sourceClient.EXPECT().GetContentLength(source.RequestEq(url)).AnyTimes().DoAndReturn( func(request *source.Request) (int64, error) { assert := testifyassert.New(t) @@ -491,7 +490,7 @@ func TestPeerTaskManager_TaskSuite(t *testing.T) { sizeScope: base.SizeScope_NORMAL, mockPieceDownloader: nil, mockHTTPSourceClient: func(t *testing.T, ctrl *gomock.Controller, rg *clientutil.Range, taskData []byte, url string) source.ResourceClient { - sourceClient := sourceMock.NewMockResourceClient(ctrl) + sourceClient := sourcemocks.NewMockResourceClient(ctrl) sourceClient.EXPECT().GetContentLength(source.RequestEq(url)).AnyTimes().DoAndReturn( func(request *source.Request) (int64, error) { return -1, nil @@ -515,7 +514,7 @@ func TestPeerTaskManager_TaskSuite(t *testing.T) { sizeScope: base.SizeScope_NORMAL, mockPieceDownloader: nil, mockHTTPSourceClient: func(t *testing.T, ctrl *gomock.Controller, rg *clientutil.Range, taskData []byte, url string) source.ResourceClient { - sourceClient := sourceMock.NewMockResourceClient(ctrl) + sourceClient := sourcemocks.NewMockResourceClient(ctrl) sourceClient.EXPECT().GetContentLength(source.RequestEq(url)).AnyTimes().DoAndReturn( func(request *source.Request) (int64, error) { return -1, nil diff --git a/client/daemon/peer/peertask_reuse_test.go b/client/daemon/peer/peertask_reuse_test.go index 8875d12aa08..9c117608656 100644 --- a/client/daemon/peer/peertask_reuse_test.go +++ b/client/daemon/peer/peertask_reuse_test.go @@ -31,8 +31,8 @@ import ( "d7y.io/dragonfly/v2/client/clientutil" "d7y.io/dragonfly/v2/client/daemon/storage" + "d7y.io/dragonfly/v2/client/daemon/storage/mocks" "d7y.io/dragonfly/v2/client/daemon/test" - ms "d7y.io/dragonfly/v2/client/daemon/test/mock/storage" "d7y.io/dragonfly/v2/pkg/rpc/base" "d7y.io/dragonfly/v2/pkg/rpc/scheduler" ) @@ -50,7 +50,7 @@ func TestReuseFilePeerTask(t *testing.T) { name string request *FileTaskRequest enablePrefetch bool - storageManager func(sm *ms.MockManager) + storageManager func(sm *mocks.MockManager) verify func(pg chan *FileTaskProgress, ok bool) }{ { @@ -71,7 +71,7 @@ func TestReuseFilePeerTask(t *testing.T) { Range: nil, }, enablePrefetch: false, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -115,7 +115,7 @@ func TestReuseFilePeerTask(t *testing.T) { Range: nil, }, enablePrefetch: false, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { //sm.EXPECT().FindPartialCompletedTask(gomock.Any(), gomock.Any()).DoAndReturn( // func(taskID string, rg *clientutil.Range) *storage.ReusePeerTask { // return nil @@ -152,7 +152,7 @@ func TestReuseFilePeerTask(t *testing.T) { Range: &clientutil.Range{Start: 200, Length: 100}, }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedSubTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -196,7 +196,7 @@ func TestReuseFilePeerTask(t *testing.T) { Range: &clientutil.Range{Start: 0, Length: 10}, }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { sm.EXPECT().FindPartialCompletedTask(gomock.Any(), gomock.Any()).DoAndReturn( func(taskID string, rg *clientutil.Range) *storage.ReusePeerTask { return nil @@ -229,7 +229,7 @@ func TestReuseFilePeerTask(t *testing.T) { Range: &clientutil.Range{Start: 300, Length: 100}, }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedSubTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -278,7 +278,7 @@ func TestReuseFilePeerTask(t *testing.T) { Range: &clientutil.Range{Start: 300, Length: 100000}, }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedSubTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -314,7 +314,7 @@ func TestReuseFilePeerTask(t *testing.T) { for _, tc := range testCases { t.Run(tc.name, func(t *testing.T) { defer os.Remove(testOutput) - sm := ms.NewMockManager(ctrl) + sm := mocks.NewMockManager(ctrl) tc.storageManager(sm) ptm := &peerTaskManager{ host: &scheduler.PeerHost{}, @@ -337,7 +337,7 @@ func TestReuseStreamPeerTask(t *testing.T) { name string request *StreamTaskRequest enablePrefetch bool - storageManager func(sm *ms.MockManager) + storageManager func(sm *mocks.MockManager) verify func(rc io.ReadCloser, attr map[string]string, ok bool) }{ { @@ -355,7 +355,7 @@ func TestReuseStreamPeerTask(t *testing.T) { PeerID: "", }, enablePrefetch: false, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -408,7 +408,7 @@ func TestReuseStreamPeerTask(t *testing.T) { PeerID: "", }, enablePrefetch: false, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { //sm.EXPECT().FindPartialCompletedTask(gomock.Any(), gomock.Any()).DoAndReturn( // func(taskID string, rg *clientutil.Range) *storage.ReusePeerTask { // return nil @@ -443,7 +443,7 @@ func TestReuseStreamPeerTask(t *testing.T) { PeerID: "", }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedSubTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -495,7 +495,7 @@ func TestReuseStreamPeerTask(t *testing.T) { PeerID: "", }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { sm.EXPECT().FindPartialCompletedTask(gomock.Any(), gomock.Any()).DoAndReturn( func(taskID string, rg *clientutil.Range) *storage.ReusePeerTask { return nil @@ -526,7 +526,7 @@ func TestReuseStreamPeerTask(t *testing.T) { PeerID: "", }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedSubTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -582,7 +582,7 @@ func TestReuseStreamPeerTask(t *testing.T) { PeerID: "", }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedSubTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -642,7 +642,7 @@ func TestReuseStreamPeerTask(t *testing.T) { PeerID: "", }, enablePrefetch: true, - storageManager: func(sm *ms.MockManager) { + storageManager: func(sm *mocks.MockManager) { var taskID string sm.EXPECT().FindCompletedSubTask(gomock.Any()).DoAndReturn( func(id string) *storage.ReusePeerTask { @@ -691,7 +691,7 @@ func TestReuseStreamPeerTask(t *testing.T) { for _, tc := range testCases { t.Run(tc.name, func(t *testing.T) { - sm := ms.NewMockManager(ctrl) + sm := mocks.NewMockManager(ctrl) tc.storageManager(sm) ptm := &peerTaskManager{ host: &scheduler.PeerHost{}, diff --git a/client/daemon/peer/peertask_stream_backsource_partial_test.go b/client/daemon/peer/peertask_stream_backsource_partial_test.go index 7eeb3285fc6..a4777fd1950 100644 --- a/client/daemon/peer/peertask_stream_backsource_partial_test.go +++ b/client/daemon/peer/peertask_stream_backsource_partial_test.go @@ -42,7 +42,6 @@ import ( "d7y.io/dragonfly/v2/client/config" "d7y.io/dragonfly/v2/client/daemon/storage" "d7y.io/dragonfly/v2/client/daemon/test" - mock_daemon "d7y.io/dragonfly/v2/client/daemon/test/mock/daemon" "d7y.io/dragonfly/v2/internal/dferrors" "d7y.io/dragonfly/v2/pkg/dfnet" "d7y.io/dragonfly/v2/pkg/digest" @@ -50,20 +49,21 @@ import ( "d7y.io/dragonfly/v2/pkg/rpc/base" "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon" daemonserver "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon/server" + servermocks "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon/server/mocks" "d7y.io/dragonfly/v2/pkg/rpc/scheduler" schedulerclient "d7y.io/dragonfly/v2/pkg/rpc/scheduler/client" mock_scheduler_client "d7y.io/dragonfly/v2/pkg/rpc/scheduler/client/mocks" mock_scheduler "d7y.io/dragonfly/v2/pkg/rpc/scheduler/mocks" "d7y.io/dragonfly/v2/pkg/source" "d7y.io/dragonfly/v2/pkg/source/clients/httpprotocol" - sourceMock "d7y.io/dragonfly/v2/pkg/source/mock" + sourcemocks "d7y.io/dragonfly/v2/pkg/source/mocks" ) func setupBackSourcePartialComponents(ctrl *gomock.Controller, testBytes []byte, opt componentsOption) ( schedulerclient.Client, storage.Manager) { port := int32(freeport.GetPort()) // 1. set up a mock daemon server for uploading pieces info - var daemon = mock_daemon.NewMockDaemonServer(ctrl) + var daemon = servermocks.NewMockDaemonServer(ctrl) var piecesMd5 []string pieceCount := int32(math.Ceil(float64(opt.contentLength) / float64(opt.pieceSize))) @@ -104,7 +104,7 @@ func setupBackSourcePartialComponents(ctrl *gomock.Controller, testBytes []byte, Type: "tcp", Addr: fmt.Sprintf("0.0.0.0:%d", port), }) - go func(daemon *mock_daemon.MockDaemonServer, ln net.Listener) { + go func(daemon *servermocks.MockDaemonServer, ln net.Listener) { if err := daemonserver.New(daemon).Serve(ln); err != nil { log.Fatal(err) } @@ -230,7 +230,7 @@ func TestStreamPeerTask_BackSource_Partial_WithContentLength(t *testing.T) { return rc, rc, nil }) - sourceClient := sourceMock.NewMockResourceClient(ctrl) + sourceClient := sourcemocks.NewMockResourceClient(ctrl) source.UnRegister("http") require.Nil(t, source.Register("http", sourceClient, httpprotocol.Adapter)) defer source.UnRegister("http") diff --git a/client/daemon/peer/piece_downloader.go b/client/daemon/peer/piece_downloader.go index 45e21c0407b..1f808ab73b2 100644 --- a/client/daemon/peer/piece_downloader.go +++ b/client/daemon/peer/piece_downloader.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination piece_downloader_mock.go -source piece_downloader.go -package peer + package peer import ( @@ -52,7 +54,6 @@ type DownloadPieceResult struct { FinishTime int64 } -//go:generate mockgen -source piece_downloader.go -destination ../test/mock/peer/piece_downloader.go type PieceDownloader interface { DownloadPiece(context.Context, *DownloadPieceRequest) (io.Reader, io.Closer, error) } diff --git a/client/daemon/test/mock/peer/piece_downloader.go b/client/daemon/peer/piece_downloader_mock.go similarity index 89% rename from client/daemon/test/mock/peer/piece_downloader.go rename to client/daemon/peer/piece_downloader_mock.go index d79f0531917..bacd6a89b89 100644 --- a/client/daemon/test/mock/peer/piece_downloader.go +++ b/client/daemon/peer/piece_downloader_mock.go @@ -1,15 +1,14 @@ // Code generated by MockGen. DO NOT EDIT. // Source: piece_downloader.go -// Package mock_peer is a generated GoMock package. -package mock_peer +// Package peer is a generated GoMock package. +package peer import ( context "context" io "io" reflect "reflect" - peer "d7y.io/dragonfly/v2/client/daemon/peer" gomock "github.com/golang/mock/gomock" ) @@ -37,7 +36,7 @@ func (m *MockPieceDownloader) EXPECT() *MockPieceDownloaderMockRecorder { } // DownloadPiece mocks base method. -func (m *MockPieceDownloader) DownloadPiece(arg0 context.Context, arg1 *peer.DownloadPieceRequest) (io.Reader, io.Closer, error) { +func (m *MockPieceDownloader) DownloadPiece(arg0 context.Context, arg1 *DownloadPieceRequest) (io.Reader, io.Closer, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "DownloadPiece", arg0, arg1) ret0, _ := ret[0].(io.Reader) diff --git a/client/daemon/peer/piece_downloader_mock_test.go b/client/daemon/peer/piece_downloader_mock_test.go deleted file mode 100644 index a5cac8f64b0..00000000000 --- a/client/daemon/peer/piece_downloader_mock_test.go +++ /dev/null @@ -1,68 +0,0 @@ -/* - * Copyright 2020 The Dragonfly Authors - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -// Code generated by MockGen. DO NOT EDIT. -// Source: piece_downloader.go - -// Package mock_peer is a generated GoMock package. -package peer - -import ( - context "context" - io "io" - reflect "reflect" - - gomock "github.com/golang/mock/gomock" -) - -// MockPieceDownloader is a mock of PieceDownloader interface. -type MockPieceDownloader struct { - ctrl *gomock.Controller - recorder *MockPieceDownloaderMockRecorder -} - -// MockPieceDownloaderMockRecorder is the mock recorder for MockPieceDownloader. -type MockPieceDownloaderMockRecorder struct { - mock *MockPieceDownloader -} - -// NewMockPieceDownloader creates a new mock instance. -func NewMockPieceDownloader(ctrl *gomock.Controller) *MockPieceDownloader { - mock := &MockPieceDownloader{ctrl: ctrl} - mock.recorder = &MockPieceDownloaderMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockPieceDownloader) EXPECT() *MockPieceDownloaderMockRecorder { - return m.recorder -} - -// DownloadPiece mocks base method. -func (m *MockPieceDownloader) DownloadPiece(arg0 context.Context, arg1 *DownloadPieceRequest) (io.Reader, io.Closer, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DownloadPiece", arg0, arg1) - ret0, _ := ret[0].(io.Reader) - ret1, _ := ret[1].(io.Closer) - ret2, _ := ret[2].(error) - return ret0, ret1, ret2 -} - -// DownloadPiece indicates an expected call of DownloadPiece. -func (mr *MockPieceDownloaderMockRecorder) DownloadPiece(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadPiece", reflect.TypeOf((*MockPieceDownloader)(nil).DownloadPiece), arg0, arg1) -} diff --git a/client/daemon/proxy/mocks/proxy_manager_mock.go b/client/daemon/proxy/mocks/proxy_manager_mock.go new file mode 100644 index 00000000000..63acd19cb37 --- /dev/null +++ b/client/daemon/proxy/mocks/proxy_manager_mock.go @@ -0,0 +1,91 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: proxy_manager.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + net "net" + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// 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 +} + +// IsEnabled mocks base method. +func (m *MockManager) IsEnabled() bool { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "IsEnabled") + ret0, _ := ret[0].(bool) + return ret0 +} + +// IsEnabled indicates an expected call of IsEnabled. +func (mr *MockManagerMockRecorder) IsEnabled() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsEnabled", reflect.TypeOf((*MockManager)(nil).IsEnabled)) +} + +// Serve mocks base method. +func (m *MockManager) Serve(arg0 net.Listener) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Serve", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// Serve indicates an expected call of Serve. +func (mr *MockManagerMockRecorder) Serve(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Serve", reflect.TypeOf((*MockManager)(nil).Serve), arg0) +} + +// ServeSNI mocks base method. +func (m *MockManager) ServeSNI(arg0 net.Listener) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ServeSNI", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// ServeSNI indicates an expected call of ServeSNI. +func (mr *MockManagerMockRecorder) ServeSNI(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServeSNI", reflect.TypeOf((*MockManager)(nil).ServeSNI), arg0) +} + +// Stop mocks base method. +func (m *MockManager) 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 *MockManagerMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockManager)(nil).Stop)) +} diff --git a/client/daemon/proxy/proxy_manager.go b/client/daemon/proxy/proxy_manager.go index 95ca563cfff..17ba3ddcb37 100644 --- a/client/daemon/proxy/proxy_manager.go +++ b/client/daemon/proxy/proxy_manager.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/proxy_manager_mock.go -source proxy_manager.go -package mocks + package proxy import ( diff --git a/client/daemon/rpcserver/mocks/rpcserver_mock.go b/client/daemon/rpcserver/mocks/rpcserver_mock.go new file mode 100644 index 00000000000..20b132d25ae --- /dev/null +++ b/client/daemon/rpcserver/mocks/rpcserver_mock.go @@ -0,0 +1,102 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: rpcserver.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + net "net" + reflect "reflect" + time "time" + + gomock "github.com/golang/mock/gomock" +) + +// MockServer is a mock of Server interface. +type MockServer struct { + ctrl *gomock.Controller + recorder *MockServerMockRecorder +} + +// MockServerMockRecorder is the mock recorder for MockServer. +type MockServerMockRecorder struct { + mock *MockServer +} + +// NewMockServer creates a new mock instance. +func NewMockServer(ctrl *gomock.Controller) *MockServer { + mock := &MockServer{ctrl: ctrl} + mock.recorder = &MockServerMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockServer) EXPECT() *MockServerMockRecorder { + return m.recorder +} + +// Alive mocks base method. +func (m *MockServer) 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 *MockServerMockRecorder) Alive(alive interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Alive", reflect.TypeOf((*MockServer)(nil).Alive), alive) +} + +// Keep mocks base method. +func (m *MockServer) Keep() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Keep") +} + +// Keep indicates an expected call of Keep. +func (mr *MockServerMockRecorder) Keep() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Keep", reflect.TypeOf((*MockServer)(nil).Keep)) +} + +// ServeDownload mocks base method. +func (m *MockServer) ServeDownload(listener net.Listener) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ServeDownload", listener) + ret0, _ := ret[0].(error) + return ret0 +} + +// ServeDownload indicates an expected call of ServeDownload. +func (mr *MockServerMockRecorder) ServeDownload(listener interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServeDownload", reflect.TypeOf((*MockServer)(nil).ServeDownload), listener) +} + +// ServePeer mocks base method. +func (m *MockServer) ServePeer(listener net.Listener) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ServePeer", listener) + ret0, _ := ret[0].(error) + return ret0 +} + +// ServePeer indicates an expected call of ServePeer. +func (mr *MockServerMockRecorder) ServePeer(listener interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServePeer", reflect.TypeOf((*MockServer)(nil).ServePeer), listener) +} + +// Stop mocks base method. +func (m *MockServer) Stop() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Stop") +} + +// Stop indicates an expected call of Stop. +func (mr *MockServerMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockServer)(nil).Stop)) +} diff --git a/client/daemon/rpcserver/rpcserver.go b/client/daemon/rpcserver/rpcserver.go index 935d9a039d8..6988eb9dac1 100644 --- a/client/daemon/rpcserver/rpcserver.go +++ b/client/daemon/rpcserver/rpcserver.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/rpcserver_mock.go -source rpcserver.go -package mocks + package rpcserver import ( diff --git a/client/daemon/rpcserver/rpcserver_test.go b/client/daemon/rpcserver/rpcserver_test.go index 9129fd81ad1..0e4bafb8987 100644 --- a/client/daemon/rpcserver/rpcserver_test.go +++ b/client/daemon/rpcserver/rpcserver_test.go @@ -33,8 +33,7 @@ import ( "d7y.io/dragonfly/v2/client/clientutil" "d7y.io/dragonfly/v2/client/daemon/peer" "d7y.io/dragonfly/v2/client/daemon/storage" - mock_peer "d7y.io/dragonfly/v2/client/daemon/test/mock/peer" - mock_storage "d7y.io/dragonfly/v2/client/daemon/test/mock/storage" + "d7y.io/dragonfly/v2/client/daemon/storage/mocks" "d7y.io/dragonfly/v2/pkg/dfnet" "d7y.io/dragonfly/v2/pkg/idgen" "d7y.io/dragonfly/v2/pkg/net/ip" @@ -54,7 +53,7 @@ func Test_ServeDownload(t *testing.T) { ctrl := gomock.NewController(t) defer ctrl.Finish() - mockPeerTaskManager := mock_peer.NewMockTaskManager(ctrl) + mockPeerTaskManager := peer.NewMockTaskManager(ctrl) mockPeerTaskManager.EXPECT().StartFileTask(gomock.Any(), gomock.Any()).DoAndReturn( func(ctx context.Context, req *peer.FileTaskRequest) (chan *peer.FileTaskProgress, bool, error) { ch := make(chan *peer.FileTaskProgress) @@ -119,7 +118,7 @@ func Test_ServePeer(t *testing.T) { defer ctrl.Finish() var maxPieceNum uint32 = 10 - mockStorageManger := mock_storage.NewMockManager(ctrl) + mockStorageManger := mocks.NewMockManager(ctrl) mockStorageManger.EXPECT().GetPieces(gomock.Any(), gomock.Any()).AnyTimes().DoAndReturn(func(ctx context.Context, req *base.PieceTaskRequest) (*base.PiecePacket, error) { var ( pieces []*base.PieceInfo @@ -367,7 +366,7 @@ func Test_SyncPieceTasks(t *testing.T) { t.Run(tc.name, func(t *testing.T) { for _, delay := range []bool{false, true} { delay := delay - mockStorageManger := mock_storage.NewMockManager(ctrl) + mockStorageManger := mocks.NewMockManager(ctrl) if tc.limit == 0 { tc.limit = 1024 @@ -431,7 +430,7 @@ func Test_SyncPieceTasks(t *testing.T) { }, }, nil }) - mockTaskManager := mock_peer.NewMockTaskManager(ctrl) + mockTaskManager := peer.NewMockTaskManager(ctrl) mockTaskManager.EXPECT().Subscribe(gomock.Any()).AnyTimes().DoAndReturn( func(request *base.PieceTaskRequest) (*peer.SubscribeResponse, bool) { ch := make(chan *peer.PieceInfo) diff --git a/client/daemon/rpcserver/seeder_test.go b/client/daemon/rpcserver/seeder_test.go index b222f48795c..346061ed227 100644 --- a/client/daemon/rpcserver/seeder_test.go +++ b/client/daemon/rpcserver/seeder_test.go @@ -34,8 +34,7 @@ import ( "d7y.io/dragonfly/v2/client/config" "d7y.io/dragonfly/v2/client/daemon/peer" "d7y.io/dragonfly/v2/client/daemon/storage" - mock_peer "d7y.io/dragonfly/v2/client/daemon/test/mock/peer" - mock_storage "d7y.io/dragonfly/v2/client/daemon/test/mock/storage" + "d7y.io/dragonfly/v2/client/daemon/storage/mocks" "d7y.io/dragonfly/v2/pkg/dfnet" "d7y.io/dragonfly/v2/pkg/rpc/base" "d7y.io/dragonfly/v2/pkg/rpc/base/common" @@ -190,7 +189,7 @@ func Test_ObtainSeeds(t *testing.T) { t.Run(tc.name, func(t *testing.T) { for _, delay := range []bool{false, true} { delay := delay - mockStorageManger := mock_storage.NewMockManager(ctrl) + mockStorageManger := mocks.NewMockManager(ctrl) if tc.limit == 0 { tc.limit = 1024 @@ -254,7 +253,7 @@ func Test_ObtainSeeds(t *testing.T) { }, }, nil }) - mockTaskManager := mock_peer.NewMockTaskManager(ctrl) + mockTaskManager := peer.NewMockTaskManager(ctrl) mockTaskManager.EXPECT().StartSeedTask(gomock.Any(), gomock.Any()).DoAndReturn( func(ctx context.Context, req *peer.SeedTaskRequest) (*peer.SeedTaskResponse, bool, error) { ch := make(chan *peer.PieceInfo) diff --git a/client/daemon/test/mock/storage/manager.go b/client/daemon/storage/mocks/stroage_manager_mock.go similarity index 99% rename from client/daemon/test/mock/storage/manager.go rename to client/daemon/storage/mocks/stroage_manager_mock.go index 7cdfe8a94f5..f9a7f9ce74e 100644 --- a/client/daemon/test/mock/storage/manager.go +++ b/client/daemon/storage/mocks/stroage_manager_mock.go @@ -1,8 +1,8 @@ // Code generated by MockGen. DO NOT EDIT. // Source: storage_manager.go -// Package mock_storage is a generated GoMock package. -package mock_storage +// Package mocks is a generated GoMock package. +package mocks import ( context "context" diff --git a/client/daemon/storage/storage_manager.go b/client/daemon/storage/storage_manager.go index b737f558866..176122223b0 100644 --- a/client/daemon/storage/storage_manager.go +++ b/client/daemon/storage/storage_manager.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/stroage_manager_mock.go -source storage_manager.go -package mocks + package storage import ( @@ -43,7 +45,6 @@ import ( "d7y.io/dragonfly/v2/pkg/rpc/base" ) -//go:generate mockgen -source storage_manager.go -destination ../test/mock/storage/manager.go type TaskStorageDriver interface { // WritePiece put a piece of a task to storage WritePiece(ctx context.Context, req *WritePieceRequest) (int64, error) diff --git a/client/daemon/test/mock/daemon/daemon_server.go b/client/daemon/test/mock/daemon/daemon_server.go deleted file mode 100644 index b87794372cd..00000000000 --- a/client/daemon/test/mock/daemon/daemon_server.go +++ /dev/null @@ -1,150 +0,0 @@ -// Code generated by MockGen. DO NOT EDIT. -// Source: server.go - -// Package mock_server is a generated GoMock package. -package mock_server - -import ( - context "context" - reflect "reflect" - - base "d7y.io/dragonfly/v2/pkg/rpc/base" - dfdaemon "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon" - gomock "github.com/golang/mock/gomock" -) - -// MockDaemonServer is a mock of DaemonServer interface. -type MockDaemonServer struct { - ctrl *gomock.Controller - recorder *MockDaemonServerMockRecorder -} - -// MockDaemonServerMockRecorder is the mock recorder for MockDaemonServer. -type MockDaemonServerMockRecorder struct { - mock *MockDaemonServer -} - -// NewMockDaemonServer creates a new mock instance. -func NewMockDaemonServer(ctrl *gomock.Controller) *MockDaemonServer { - mock := &MockDaemonServer{ctrl: ctrl} - mock.recorder = &MockDaemonServerMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockDaemonServer) EXPECT() *MockDaemonServerMockRecorder { - return m.recorder -} - -// CheckHealth mocks base method. -func (m *MockDaemonServer) CheckHealth(arg0 context.Context) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CheckHealth", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// CheckHealth indicates an expected call of CheckHealth. -func (mr *MockDaemonServerMockRecorder) CheckHealth(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckHealth", reflect.TypeOf((*MockDaemonServer)(nil).CheckHealth), arg0) -} - -// DeleteTask mocks base method. -func (m *MockDaemonServer) DeleteTask(arg0 context.Context, arg1 *dfdaemon.DeleteTaskRequest) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteTask", arg0, arg1) - ret0, _ := ret[0].(error) - return ret0 -} - -// DeleteTask indicates an expected call of DeleteTask. -func (mr *MockDaemonServerMockRecorder) DeleteTask(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTask", reflect.TypeOf((*MockDaemonServer)(nil).DeleteTask), arg0, arg1) -} - -// Download mocks base method. -func (m *MockDaemonServer) Download(arg0 context.Context, arg1 *dfdaemon.DownRequest, arg2 chan<- *dfdaemon.DownResult) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Download", arg0, arg1, arg2) - ret0, _ := ret[0].(error) - return ret0 -} - -// Download indicates an expected call of Download. -func (mr *MockDaemonServerMockRecorder) Download(arg0, arg1, arg2 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Download", reflect.TypeOf((*MockDaemonServer)(nil).Download), arg0, arg1, arg2) -} - -// ExportTask mocks base method. -func (m *MockDaemonServer) ExportTask(arg0 context.Context, arg1 *dfdaemon.ExportTaskRequest) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ExportTask", arg0, arg1) - ret0, _ := ret[0].(error) - return ret0 -} - -// ExportTask indicates an expected call of ExportTask. -func (mr *MockDaemonServerMockRecorder) ExportTask(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExportTask", reflect.TypeOf((*MockDaemonServer)(nil).ExportTask), arg0, arg1) -} - -// GetPieceTasks mocks base method. -func (m *MockDaemonServer) GetPieceTasks(arg0 context.Context, arg1 *base.PieceTaskRequest) (*base.PiecePacket, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPieceTasks", arg0, arg1) - ret0, _ := ret[0].(*base.PiecePacket) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetPieceTasks indicates an expected call of GetPieceTasks. -func (mr *MockDaemonServerMockRecorder) GetPieceTasks(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceTasks", reflect.TypeOf((*MockDaemonServer)(nil).GetPieceTasks), arg0, arg1) -} - -// ImportTask mocks base method. -func (m *MockDaemonServer) ImportTask(arg0 context.Context, arg1 *dfdaemon.ImportTaskRequest) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ImportTask", arg0, arg1) - ret0, _ := ret[0].(error) - return ret0 -} - -// ImportTask indicates an expected call of ImportTask. -func (mr *MockDaemonServerMockRecorder) ImportTask(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ImportTask", reflect.TypeOf((*MockDaemonServer)(nil).ImportTask), arg0, arg1) -} - -// StatTask mocks base method. -func (m *MockDaemonServer) StatTask(arg0 context.Context, arg1 *dfdaemon.StatTaskRequest) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "StatTask", arg0, arg1) - ret0, _ := ret[0].(error) - return ret0 -} - -// StatTask indicates an expected call of StatTask. -func (mr *MockDaemonServerMockRecorder) StatTask(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StatTask", reflect.TypeOf((*MockDaemonServer)(nil).StatTask), arg0, arg1) -} - -// SyncPieceTasks mocks base method. -func (m *MockDaemonServer) SyncPieceTasks(arg0 dfdaemon.Daemon_SyncPieceTasksServer) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SyncPieceTasks", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// SyncPieceTasks indicates an expected call of SyncPieceTasks. -func (mr *MockDaemonServerMockRecorder) SyncPieceTasks(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieceTasks", reflect.TypeOf((*MockDaemonServer)(nil).SyncPieceTasks), arg0) -} diff --git a/client/daemon/test/mock/daemongrpc/daemon_server_grpc.go b/client/daemon/test/mock/daemongrpc/daemon_server_grpc.go deleted file mode 100644 index df0c61b9980..00000000000 --- a/client/daemon/test/mock/daemongrpc/daemon_server_grpc.go +++ /dev/null @@ -1,854 +0,0 @@ -// Code generated by MockGen. DO NOT EDIT. -// Source: ../../../pkg/rpc/dfdaemon/dfdaemon.pb.go - -// Package mock_server_grpc is a generated GoMock package. -package mock_server_grpc - -import ( - context "context" - reflect "reflect" - - base "d7y.io/dragonfly/v2/pkg/rpc/base" - dfdaemon "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon" - gomock "github.com/golang/mock/gomock" - grpc "google.golang.org/grpc" - metadata "google.golang.org/grpc/metadata" - emptypb "google.golang.org/protobuf/types/known/emptypb" -) - -// MockDaemonClient is a mock of DaemonClient interface. -type MockDaemonClient struct { - ctrl *gomock.Controller - recorder *MockDaemonClientMockRecorder -} - -// MockDaemonClientMockRecorder is the mock recorder for MockDaemonClient. -type MockDaemonClientMockRecorder struct { - mock *MockDaemonClient -} - -// NewMockDaemonClient creates a new mock instance. -func NewMockDaemonClient(ctrl *gomock.Controller) *MockDaemonClient { - mock := &MockDaemonClient{ctrl: ctrl} - mock.recorder = &MockDaemonClientMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockDaemonClient) EXPECT() *MockDaemonClientMockRecorder { - return m.recorder -} - -// CheckHealth mocks base method. -func (m *MockDaemonClient) CheckHealth(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx, in} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "CheckHealth", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// CheckHealth indicates an expected call of CheckHealth. -func (mr *MockDaemonClientMockRecorder) CheckHealth(ctx, in interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckHealth", reflect.TypeOf((*MockDaemonClient)(nil).CheckHealth), varargs...) -} - -// DeleteTask mocks base method. -func (m *MockDaemonClient) DeleteTask(ctx context.Context, in *dfdaemon.DeleteTaskRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx, in} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "DeleteTask", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// DeleteTask indicates an expected call of DeleteTask. -func (mr *MockDaemonClientMockRecorder) DeleteTask(ctx, in interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTask", reflect.TypeOf((*MockDaemonClient)(nil).DeleteTask), varargs...) -} - -// Download mocks base method. -func (m *MockDaemonClient) Download(ctx context.Context, in *dfdaemon.DownRequest, opts ...grpc.CallOption) (dfdaemon.Daemon_DownloadClient, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx, in} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "Download", varargs...) - ret0, _ := ret[0].(dfdaemon.Daemon_DownloadClient) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// Download indicates an expected call of Download. -func (mr *MockDaemonClientMockRecorder) Download(ctx, in interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Download", reflect.TypeOf((*MockDaemonClient)(nil).Download), varargs...) -} - -// ExportTask mocks base method. -func (m *MockDaemonClient) ExportTask(ctx context.Context, in *dfdaemon.ExportTaskRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx, in} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "ExportTask", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// ExportTask indicates an expected call of ExportTask. -func (mr *MockDaemonClientMockRecorder) ExportTask(ctx, in interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExportTask", reflect.TypeOf((*MockDaemonClient)(nil).ExportTask), varargs...) -} - -// GetPieceTasks mocks base method. -func (m *MockDaemonClient) GetPieceTasks(ctx context.Context, in *base.PieceTaskRequest, opts ...grpc.CallOption) (*base.PiecePacket, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx, in} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "GetPieceTasks", varargs...) - ret0, _ := ret[0].(*base.PiecePacket) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetPieceTasks indicates an expected call of GetPieceTasks. -func (mr *MockDaemonClientMockRecorder) GetPieceTasks(ctx, in interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceTasks", reflect.TypeOf((*MockDaemonClient)(nil).GetPieceTasks), varargs...) -} - -// ImportTask mocks base method. -func (m *MockDaemonClient) ImportTask(ctx context.Context, in *dfdaemon.ImportTaskRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx, in} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "ImportTask", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// ImportTask indicates an expected call of ImportTask. -func (mr *MockDaemonClientMockRecorder) ImportTask(ctx, in interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ImportTask", reflect.TypeOf((*MockDaemonClient)(nil).ImportTask), varargs...) -} - -// StatTask mocks base method. -func (m *MockDaemonClient) StatTask(ctx context.Context, in *dfdaemon.StatTaskRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx, in} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "StatTask", varargs...) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// StatTask indicates an expected call of StatTask. -func (mr *MockDaemonClientMockRecorder) StatTask(ctx, in interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx, in}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StatTask", reflect.TypeOf((*MockDaemonClient)(nil).StatTask), varargs...) -} - -// SyncPieceTasks mocks base method. -func (m *MockDaemonClient) SyncPieceTasks(ctx context.Context, opts ...grpc.CallOption) (dfdaemon.Daemon_SyncPieceTasksClient, error) { - m.ctrl.T.Helper() - varargs := []interface{}{ctx} - for _, a := range opts { - varargs = append(varargs, a) - } - ret := m.ctrl.Call(m, "SyncPieceTasks", varargs...) - ret0, _ := ret[0].(dfdaemon.Daemon_SyncPieceTasksClient) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// SyncPieceTasks indicates an expected call of SyncPieceTasks. -func (mr *MockDaemonClientMockRecorder) SyncPieceTasks(ctx interface{}, opts ...interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{ctx}, opts...) - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieceTasks", reflect.TypeOf((*MockDaemonClient)(nil).SyncPieceTasks), varargs...) -} - -// MockDaemon_DownloadClient is a mock of Daemon_DownloadClient interface. -type MockDaemon_DownloadClient struct { - ctrl *gomock.Controller - recorder *MockDaemon_DownloadClientMockRecorder -} - -// MockDaemon_DownloadClientMockRecorder is the mock recorder for MockDaemon_DownloadClient. -type MockDaemon_DownloadClientMockRecorder struct { - mock *MockDaemon_DownloadClient -} - -// NewMockDaemon_DownloadClient creates a new mock instance. -func NewMockDaemon_DownloadClient(ctrl *gomock.Controller) *MockDaemon_DownloadClient { - mock := &MockDaemon_DownloadClient{ctrl: ctrl} - mock.recorder = &MockDaemon_DownloadClientMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockDaemon_DownloadClient) EXPECT() *MockDaemon_DownloadClientMockRecorder { - return m.recorder -} - -// CloseSend mocks base method. -func (m *MockDaemon_DownloadClient) CloseSend() error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CloseSend") - ret0, _ := ret[0].(error) - return ret0 -} - -// CloseSend indicates an expected call of CloseSend. -func (mr *MockDaemon_DownloadClientMockRecorder) CloseSend() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockDaemon_DownloadClient)(nil).CloseSend)) -} - -// Context mocks base method. -func (m *MockDaemon_DownloadClient) Context() context.Context { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Context") - ret0, _ := ret[0].(context.Context) - return ret0 -} - -// Context indicates an expected call of Context. -func (mr *MockDaemon_DownloadClientMockRecorder) Context() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockDaemon_DownloadClient)(nil).Context)) -} - -// Header mocks base method. -func (m *MockDaemon_DownloadClient) Header() (metadata.MD, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Header") - ret0, _ := ret[0].(metadata.MD) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// Header indicates an expected call of Header. -func (mr *MockDaemon_DownloadClientMockRecorder) Header() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockDaemon_DownloadClient)(nil).Header)) -} - -// Recv mocks base method. -func (m *MockDaemon_DownloadClient) Recv() (*dfdaemon.DownResult, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Recv") - ret0, _ := ret[0].(*dfdaemon.DownResult) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// Recv indicates an expected call of Recv. -func (mr *MockDaemon_DownloadClientMockRecorder) Recv() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockDaemon_DownloadClient)(nil).Recv)) -} - -// RecvMsg mocks base method. -func (m_2 *MockDaemon_DownloadClient) RecvMsg(m interface{}) error { - m_2.ctrl.T.Helper() - ret := m_2.ctrl.Call(m_2, "RecvMsg", m) - ret0, _ := ret[0].(error) - return ret0 -} - -// RecvMsg indicates an expected call of RecvMsg. -func (mr *MockDaemon_DownloadClientMockRecorder) RecvMsg(m interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockDaemon_DownloadClient)(nil).RecvMsg), m) -} - -// SendMsg mocks base method. -func (m_2 *MockDaemon_DownloadClient) SendMsg(m interface{}) error { - m_2.ctrl.T.Helper() - ret := m_2.ctrl.Call(m_2, "SendMsg", m) - ret0, _ := ret[0].(error) - return ret0 -} - -// SendMsg indicates an expected call of SendMsg. -func (mr *MockDaemon_DownloadClientMockRecorder) SendMsg(m interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockDaemon_DownloadClient)(nil).SendMsg), m) -} - -// Trailer mocks base method. -func (m *MockDaemon_DownloadClient) Trailer() metadata.MD { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Trailer") - ret0, _ := ret[0].(metadata.MD) - return ret0 -} - -// Trailer indicates an expected call of Trailer. -func (mr *MockDaemon_DownloadClientMockRecorder) Trailer() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockDaemon_DownloadClient)(nil).Trailer)) -} - -// MockDaemon_SyncPieceTasksClient is a mock of Daemon_SyncPieceTasksClient interface. -type MockDaemon_SyncPieceTasksClient struct { - ctrl *gomock.Controller - recorder *MockDaemon_SyncPieceTasksClientMockRecorder -} - -// MockDaemon_SyncPieceTasksClientMockRecorder is the mock recorder for MockDaemon_SyncPieceTasksClient. -type MockDaemon_SyncPieceTasksClientMockRecorder struct { - mock *MockDaemon_SyncPieceTasksClient -} - -// NewMockDaemon_SyncPieceTasksClient creates a new mock instance. -func NewMockDaemon_SyncPieceTasksClient(ctrl *gomock.Controller) *MockDaemon_SyncPieceTasksClient { - mock := &MockDaemon_SyncPieceTasksClient{ctrl: ctrl} - mock.recorder = &MockDaemon_SyncPieceTasksClientMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockDaemon_SyncPieceTasksClient) EXPECT() *MockDaemon_SyncPieceTasksClientMockRecorder { - return m.recorder -} - -// CloseSend mocks base method. -func (m *MockDaemon_SyncPieceTasksClient) CloseSend() error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CloseSend") - ret0, _ := ret[0].(error) - return ret0 -} - -// CloseSend indicates an expected call of CloseSend. -func (mr *MockDaemon_SyncPieceTasksClientMockRecorder) CloseSend() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockDaemon_SyncPieceTasksClient)(nil).CloseSend)) -} - -// Context mocks base method. -func (m *MockDaemon_SyncPieceTasksClient) Context() context.Context { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Context") - ret0, _ := ret[0].(context.Context) - return ret0 -} - -// Context indicates an expected call of Context. -func (mr *MockDaemon_SyncPieceTasksClientMockRecorder) Context() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockDaemon_SyncPieceTasksClient)(nil).Context)) -} - -// Header mocks base method. -func (m *MockDaemon_SyncPieceTasksClient) Header() (metadata.MD, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Header") - ret0, _ := ret[0].(metadata.MD) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// Header indicates an expected call of Header. -func (mr *MockDaemon_SyncPieceTasksClientMockRecorder) Header() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockDaemon_SyncPieceTasksClient)(nil).Header)) -} - -// Recv mocks base method. -func (m *MockDaemon_SyncPieceTasksClient) Recv() (*base.PiecePacket, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Recv") - ret0, _ := ret[0].(*base.PiecePacket) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// Recv indicates an expected call of Recv. -func (mr *MockDaemon_SyncPieceTasksClientMockRecorder) Recv() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockDaemon_SyncPieceTasksClient)(nil).Recv)) -} - -// RecvMsg mocks base method. -func (m_2 *MockDaemon_SyncPieceTasksClient) RecvMsg(m interface{}) error { - m_2.ctrl.T.Helper() - ret := m_2.ctrl.Call(m_2, "RecvMsg", m) - ret0, _ := ret[0].(error) - return ret0 -} - -// RecvMsg indicates an expected call of RecvMsg. -func (mr *MockDaemon_SyncPieceTasksClientMockRecorder) RecvMsg(m interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockDaemon_SyncPieceTasksClient)(nil).RecvMsg), m) -} - -// Send mocks base method. -func (m *MockDaemon_SyncPieceTasksClient) Send(arg0 *base.PieceTaskRequest) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Send", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// Send indicates an expected call of Send. -func (mr *MockDaemon_SyncPieceTasksClientMockRecorder) Send(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockDaemon_SyncPieceTasksClient)(nil).Send), arg0) -} - -// SendMsg mocks base method. -func (m_2 *MockDaemon_SyncPieceTasksClient) SendMsg(m interface{}) error { - m_2.ctrl.T.Helper() - ret := m_2.ctrl.Call(m_2, "SendMsg", m) - ret0, _ := ret[0].(error) - return ret0 -} - -// SendMsg indicates an expected call of SendMsg. -func (mr *MockDaemon_SyncPieceTasksClientMockRecorder) SendMsg(m interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockDaemon_SyncPieceTasksClient)(nil).SendMsg), m) -} - -// Trailer mocks base method. -func (m *MockDaemon_SyncPieceTasksClient) Trailer() metadata.MD { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Trailer") - ret0, _ := ret[0].(metadata.MD) - return ret0 -} - -// Trailer indicates an expected call of Trailer. -func (mr *MockDaemon_SyncPieceTasksClientMockRecorder) Trailer() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockDaemon_SyncPieceTasksClient)(nil).Trailer)) -} - -// MockDaemonServer is a mock of DaemonServer interface. -type MockDaemonServer struct { - ctrl *gomock.Controller - recorder *MockDaemonServerMockRecorder -} - -// MockDaemonServerMockRecorder is the mock recorder for MockDaemonServer. -type MockDaemonServerMockRecorder struct { - mock *MockDaemonServer -} - -// NewMockDaemonServer creates a new mock instance. -func NewMockDaemonServer(ctrl *gomock.Controller) *MockDaemonServer { - mock := &MockDaemonServer{ctrl: ctrl} - mock.recorder = &MockDaemonServerMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockDaemonServer) EXPECT() *MockDaemonServerMockRecorder { - return m.recorder -} - -// CheckHealth mocks base method. -func (m *MockDaemonServer) CheckHealth(arg0 context.Context, arg1 *emptypb.Empty) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CheckHealth", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// CheckHealth indicates an expected call of CheckHealth. -func (mr *MockDaemonServerMockRecorder) CheckHealth(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckHealth", reflect.TypeOf((*MockDaemonServer)(nil).CheckHealth), arg0, arg1) -} - -// DeleteTask mocks base method. -func (m *MockDaemonServer) DeleteTask(arg0 context.Context, arg1 *dfdaemon.DeleteTaskRequest) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteTask", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// DeleteTask indicates an expected call of DeleteTask. -func (mr *MockDaemonServerMockRecorder) DeleteTask(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTask", reflect.TypeOf((*MockDaemonServer)(nil).DeleteTask), arg0, arg1) -} - -// Download mocks base method. -func (m *MockDaemonServer) Download(arg0 *dfdaemon.DownRequest, arg1 dfdaemon.Daemon_DownloadServer) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Download", arg0, arg1) - ret0, _ := ret[0].(error) - return ret0 -} - -// Download indicates an expected call of Download. -func (mr *MockDaemonServerMockRecorder) Download(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Download", reflect.TypeOf((*MockDaemonServer)(nil).Download), arg0, arg1) -} - -// ExportTask mocks base method. -func (m *MockDaemonServer) ExportTask(arg0 context.Context, arg1 *dfdaemon.ExportTaskRequest) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ExportTask", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// ExportTask indicates an expected call of ExportTask. -func (mr *MockDaemonServerMockRecorder) ExportTask(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExportTask", reflect.TypeOf((*MockDaemonServer)(nil).ExportTask), arg0, arg1) -} - -// GetPieceTasks mocks base method. -func (m *MockDaemonServer) GetPieceTasks(arg0 context.Context, arg1 *base.PieceTaskRequest) (*base.PiecePacket, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPieceTasks", arg0, arg1) - ret0, _ := ret[0].(*base.PiecePacket) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetPieceTasks indicates an expected call of GetPieceTasks. -func (mr *MockDaemonServerMockRecorder) GetPieceTasks(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceTasks", reflect.TypeOf((*MockDaemonServer)(nil).GetPieceTasks), arg0, arg1) -} - -// ImportTask mocks base method. -func (m *MockDaemonServer) ImportTask(arg0 context.Context, arg1 *dfdaemon.ImportTaskRequest) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ImportTask", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// ImportTask indicates an expected call of ImportTask. -func (mr *MockDaemonServerMockRecorder) ImportTask(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ImportTask", reflect.TypeOf((*MockDaemonServer)(nil).ImportTask), arg0, arg1) -} - -// StatTask mocks base method. -func (m *MockDaemonServer) StatTask(arg0 context.Context, arg1 *dfdaemon.StatTaskRequest) (*emptypb.Empty, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "StatTask", arg0, arg1) - ret0, _ := ret[0].(*emptypb.Empty) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// StatTask indicates an expected call of StatTask. -func (mr *MockDaemonServerMockRecorder) StatTask(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StatTask", reflect.TypeOf((*MockDaemonServer)(nil).StatTask), arg0, arg1) -} - -// SyncPieceTasks mocks base method. -func (m *MockDaemonServer) SyncPieceTasks(arg0 dfdaemon.Daemon_SyncPieceTasksServer) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SyncPieceTasks", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// SyncPieceTasks indicates an expected call of SyncPieceTasks. -func (mr *MockDaemonServerMockRecorder) SyncPieceTasks(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieceTasks", reflect.TypeOf((*MockDaemonServer)(nil).SyncPieceTasks), arg0) -} - -// MockDaemon_DownloadServer is a mock of Daemon_DownloadServer interface. -type MockDaemon_DownloadServer struct { - ctrl *gomock.Controller - recorder *MockDaemon_DownloadServerMockRecorder -} - -// MockDaemon_DownloadServerMockRecorder is the mock recorder for MockDaemon_DownloadServer. -type MockDaemon_DownloadServerMockRecorder struct { - mock *MockDaemon_DownloadServer -} - -// NewMockDaemon_DownloadServer creates a new mock instance. -func NewMockDaemon_DownloadServer(ctrl *gomock.Controller) *MockDaemon_DownloadServer { - mock := &MockDaemon_DownloadServer{ctrl: ctrl} - mock.recorder = &MockDaemon_DownloadServerMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockDaemon_DownloadServer) EXPECT() *MockDaemon_DownloadServerMockRecorder { - return m.recorder -} - -// Context mocks base method. -func (m *MockDaemon_DownloadServer) Context() context.Context { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Context") - ret0, _ := ret[0].(context.Context) - return ret0 -} - -// Context indicates an expected call of Context. -func (mr *MockDaemon_DownloadServerMockRecorder) Context() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockDaemon_DownloadServer)(nil).Context)) -} - -// RecvMsg mocks base method. -func (m_2 *MockDaemon_DownloadServer) RecvMsg(m interface{}) error { - m_2.ctrl.T.Helper() - ret := m_2.ctrl.Call(m_2, "RecvMsg", m) - ret0, _ := ret[0].(error) - return ret0 -} - -// RecvMsg indicates an expected call of RecvMsg. -func (mr *MockDaemon_DownloadServerMockRecorder) RecvMsg(m interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockDaemon_DownloadServer)(nil).RecvMsg), m) -} - -// Send mocks base method. -func (m *MockDaemon_DownloadServer) Send(arg0 *dfdaemon.DownResult) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Send", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// Send indicates an expected call of Send. -func (mr *MockDaemon_DownloadServerMockRecorder) Send(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockDaemon_DownloadServer)(nil).Send), arg0) -} - -// SendHeader mocks base method. -func (m *MockDaemon_DownloadServer) SendHeader(arg0 metadata.MD) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SendHeader", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// SendHeader indicates an expected call of SendHeader. -func (mr *MockDaemon_DownloadServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockDaemon_DownloadServer)(nil).SendHeader), arg0) -} - -// SendMsg mocks base method. -func (m_2 *MockDaemon_DownloadServer) SendMsg(m interface{}) error { - m_2.ctrl.T.Helper() - ret := m_2.ctrl.Call(m_2, "SendMsg", m) - ret0, _ := ret[0].(error) - return ret0 -} - -// SendMsg indicates an expected call of SendMsg. -func (mr *MockDaemon_DownloadServerMockRecorder) SendMsg(m interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockDaemon_DownloadServer)(nil).SendMsg), m) -} - -// SetHeader mocks base method. -func (m *MockDaemon_DownloadServer) SetHeader(arg0 metadata.MD) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SetHeader", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// SetHeader indicates an expected call of SetHeader. -func (mr *MockDaemon_DownloadServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockDaemon_DownloadServer)(nil).SetHeader), arg0) -} - -// SetTrailer mocks base method. -func (m *MockDaemon_DownloadServer) SetTrailer(arg0 metadata.MD) { - m.ctrl.T.Helper() - m.ctrl.Call(m, "SetTrailer", arg0) -} - -// SetTrailer indicates an expected call of SetTrailer. -func (mr *MockDaemon_DownloadServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockDaemon_DownloadServer)(nil).SetTrailer), arg0) -} - -// MockDaemon_SyncPieceTasksServer is a mock of Daemon_SyncPieceTasksServer interface. -type MockDaemon_SyncPieceTasksServer struct { - ctrl *gomock.Controller - recorder *MockDaemon_SyncPieceTasksServerMockRecorder -} - -// MockDaemon_SyncPieceTasksServerMockRecorder is the mock recorder for MockDaemon_SyncPieceTasksServer. -type MockDaemon_SyncPieceTasksServerMockRecorder struct { - mock *MockDaemon_SyncPieceTasksServer -} - -// NewMockDaemon_SyncPieceTasksServer creates a new mock instance. -func NewMockDaemon_SyncPieceTasksServer(ctrl *gomock.Controller) *MockDaemon_SyncPieceTasksServer { - mock := &MockDaemon_SyncPieceTasksServer{ctrl: ctrl} - mock.recorder = &MockDaemon_SyncPieceTasksServerMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockDaemon_SyncPieceTasksServer) EXPECT() *MockDaemon_SyncPieceTasksServerMockRecorder { - return m.recorder -} - -// Context mocks base method. -func (m *MockDaemon_SyncPieceTasksServer) Context() context.Context { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Context") - ret0, _ := ret[0].(context.Context) - return ret0 -} - -// Context indicates an expected call of Context. -func (mr *MockDaemon_SyncPieceTasksServerMockRecorder) Context() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockDaemon_SyncPieceTasksServer)(nil).Context)) -} - -// Recv mocks base method. -func (m *MockDaemon_SyncPieceTasksServer) Recv() (*base.PieceTaskRequest, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Recv") - ret0, _ := ret[0].(*base.PieceTaskRequest) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// Recv indicates an expected call of Recv. -func (mr *MockDaemon_SyncPieceTasksServerMockRecorder) Recv() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockDaemon_SyncPieceTasksServer)(nil).Recv)) -} - -// RecvMsg mocks base method. -func (m_2 *MockDaemon_SyncPieceTasksServer) RecvMsg(m interface{}) error { - m_2.ctrl.T.Helper() - ret := m_2.ctrl.Call(m_2, "RecvMsg", m) - ret0, _ := ret[0].(error) - return ret0 -} - -// RecvMsg indicates an expected call of RecvMsg. -func (mr *MockDaemon_SyncPieceTasksServerMockRecorder) RecvMsg(m interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockDaemon_SyncPieceTasksServer)(nil).RecvMsg), m) -} - -// Send mocks base method. -func (m *MockDaemon_SyncPieceTasksServer) Send(arg0 *base.PiecePacket) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Send", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// Send indicates an expected call of Send. -func (mr *MockDaemon_SyncPieceTasksServerMockRecorder) Send(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockDaemon_SyncPieceTasksServer)(nil).Send), arg0) -} - -// SendHeader mocks base method. -func (m *MockDaemon_SyncPieceTasksServer) SendHeader(arg0 metadata.MD) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SendHeader", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// SendHeader indicates an expected call of SendHeader. -func (mr *MockDaemon_SyncPieceTasksServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockDaemon_SyncPieceTasksServer)(nil).SendHeader), arg0) -} - -// SendMsg mocks base method. -func (m_2 *MockDaemon_SyncPieceTasksServer) SendMsg(m interface{}) error { - m_2.ctrl.T.Helper() - ret := m_2.ctrl.Call(m_2, "SendMsg", m) - ret0, _ := ret[0].(error) - return ret0 -} - -// SendMsg indicates an expected call of SendMsg. -func (mr *MockDaemon_SyncPieceTasksServerMockRecorder) SendMsg(m interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockDaemon_SyncPieceTasksServer)(nil).SendMsg), m) -} - -// SetHeader mocks base method. -func (m *MockDaemon_SyncPieceTasksServer) SetHeader(arg0 metadata.MD) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SetHeader", arg0) - ret0, _ := ret[0].(error) - return ret0 -} - -// SetHeader indicates an expected call of SetHeader. -func (mr *MockDaemon_SyncPieceTasksServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockDaemon_SyncPieceTasksServer)(nil).SetHeader), arg0) -} - -// SetTrailer mocks base method. -func (m *MockDaemon_SyncPieceTasksServer) SetTrailer(arg0 metadata.MD) { - m.ctrl.T.Helper() - m.ctrl.Call(m, "SetTrailer", arg0) -} - -// SetTrailer indicates an expected call of SetTrailer. -func (mr *MockDaemon_SyncPieceTasksServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockDaemon_SyncPieceTasksServer)(nil).SetTrailer), arg0) -} diff --git a/client/daemon/test/mock/peer/peertask_manager.go b/client/daemon/test/mock/peer/peertask_manager.go deleted file mode 100644 index 3d8ea69de3d..00000000000 --- a/client/daemon/test/mock/peer/peertask_manager.go +++ /dev/null @@ -1,434 +0,0 @@ -// Code generated by MockGen. DO NOT EDIT. -// Source: peertask_manager.go - -// Package mock_peer is a generated GoMock package. -package mock_peer - -import ( - context "context" - io "io" - reflect "reflect" - - peer "d7y.io/dragonfly/v2/client/daemon/peer" - storage "d7y.io/dragonfly/v2/client/daemon/storage" - logger "d7y.io/dragonfly/v2/internal/dflog" - base "d7y.io/dragonfly/v2/pkg/rpc/base" - scheduler "d7y.io/dragonfly/v2/pkg/rpc/scheduler" - gomock "github.com/golang/mock/gomock" -) - -// MockTaskManager is a mock of TaskManager interface. -type MockTaskManager struct { - ctrl *gomock.Controller - recorder *MockTaskManagerMockRecorder -} - -// MockTaskManagerMockRecorder is the mock recorder for MockTaskManager. -type MockTaskManagerMockRecorder struct { - mock *MockTaskManager -} - -// NewMockTaskManager creates a new mock instance. -func NewMockTaskManager(ctrl *gomock.Controller) *MockTaskManager { - mock := &MockTaskManager{ctrl: ctrl} - mock.recorder = &MockTaskManagerMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockTaskManager) EXPECT() *MockTaskManagerMockRecorder { - return m.recorder -} - -// AnnouncePeerTask mocks base method. -func (m *MockTaskManager) AnnouncePeerTask(ctx context.Context, meta storage.PeerTaskMetadata, url string, taskType base.TaskType, urlMeta *base.UrlMeta) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "AnnouncePeerTask", ctx, meta, url, taskType, urlMeta) - ret0, _ := ret[0].(error) - return ret0 -} - -// AnnouncePeerTask indicates an expected call of AnnouncePeerTask. -func (mr *MockTaskManagerMockRecorder) AnnouncePeerTask(ctx, meta, url, taskType, urlMeta interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AnnouncePeerTask", reflect.TypeOf((*MockTaskManager)(nil).AnnouncePeerTask), ctx, meta, url, taskType, urlMeta) -} - -// GetPieceManager mocks base method. -func (m *MockTaskManager) GetPieceManager() peer.PieceManager { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPieceManager") - ret0, _ := ret[0].(peer.PieceManager) - return ret0 -} - -// GetPieceManager indicates an expected call of GetPieceManager. -func (mr *MockTaskManagerMockRecorder) GetPieceManager() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceManager", reflect.TypeOf((*MockTaskManager)(nil).GetPieceManager)) -} - -// IsPeerTaskRunning mocks base method. -func (m *MockTaskManager) IsPeerTaskRunning(taskID string) (peer.Task, bool) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "IsPeerTaskRunning", taskID) - ret0, _ := ret[0].(peer.Task) - ret1, _ := ret[1].(bool) - return ret0, ret1 -} - -// IsPeerTaskRunning indicates an expected call of IsPeerTaskRunning. -func (mr *MockTaskManagerMockRecorder) IsPeerTaskRunning(taskID interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsPeerTaskRunning", reflect.TypeOf((*MockTaskManager)(nil).IsPeerTaskRunning), taskID) -} - -// StartFileTask mocks base method. -func (m *MockTaskManager) StartFileTask(ctx context.Context, req *peer.FileTaskRequest) (chan *peer.FileTaskProgress, *peer.TinyData, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "StartFileTask", ctx, req) - ret0, _ := ret[0].(chan *peer.FileTaskProgress) - ret1, _ := ret[1].(*peer.TinyData) - ret2, _ := ret[2].(error) - return ret0, ret1, ret2 -} - -// StartFileTask indicates an expected call of StartFileTask. -func (mr *MockTaskManagerMockRecorder) StartFileTask(ctx, req interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StartFileTask", reflect.TypeOf((*MockTaskManager)(nil).StartFileTask), ctx, req) -} - -// StartSeedTask mocks base method. -func (m *MockTaskManager) StartSeedTask(ctx context.Context, req *peer.SeedTaskRequest) (*peer.SeedTaskResponse, bool, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "StartSeedTask", ctx, req) - ret0, _ := ret[0].(*peer.SeedTaskResponse) - ret1, _ := ret[1].(bool) - ret2, _ := ret[2].(error) - return ret0, ret1, ret2 -} - -// StartSeedTask indicates an expected call of StartSeedTask. -func (mr *MockTaskManagerMockRecorder) StartSeedTask(ctx, req interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StartSeedTask", reflect.TypeOf((*MockTaskManager)(nil).StartSeedTask), ctx, req) -} - -// StartStreamTask mocks base method. -func (m *MockTaskManager) StartStreamTask(ctx context.Context, req *peer.StreamTaskRequest) (io.ReadCloser, map[string]string, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "StartStreamTask", ctx, req) - ret0, _ := ret[0].(io.ReadCloser) - ret1, _ := ret[1].(map[string]string) - ret2, _ := ret[2].(error) - return ret0, ret1, ret2 -} - -// StartStreamTask indicates an expected call of StartStreamTask. -func (mr *MockTaskManagerMockRecorder) StartStreamTask(ctx, req interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StartStreamTask", reflect.TypeOf((*MockTaskManager)(nil).StartStreamTask), ctx, req) -} - -// StatTask mocks base method. -func (m *MockTaskManager) StatTask(ctx context.Context, taskID string) (*scheduler.Task, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "StatTask", ctx, taskID) - ret0, _ := ret[0].(*scheduler.Task) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// StatTask indicates an expected call of StatTask. -func (mr *MockTaskManagerMockRecorder) StatTask(ctx, taskID interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StatTask", reflect.TypeOf((*MockTaskManager)(nil).StatTask), ctx, taskID) -} - -// Stop mocks base method. -func (m *MockTaskManager) Stop(ctx context.Context) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Stop", ctx) - ret0, _ := ret[0].(error) - return ret0 -} - -// Stop indicates an expected call of Stop. -func (mr *MockTaskManagerMockRecorder) Stop(ctx interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockTaskManager)(nil).Stop), ctx) -} - -// Subscribe mocks base method. -func (m *MockTaskManager) Subscribe(request *base.PieceTaskRequest) (*peer.SubscribeResponse, bool) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Subscribe", request) - ret0, _ := ret[0].(*peer.SubscribeResponse) - ret1, _ := ret[1].(bool) - return ret0, ret1 -} - -// Subscribe indicates an expected call of Subscribe. -func (mr *MockTaskManagerMockRecorder) Subscribe(request interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Subscribe", reflect.TypeOf((*MockTaskManager)(nil).Subscribe), request) -} - -// MockTask is a mock of Task interface. -type MockTask struct { - ctrl *gomock.Controller - recorder *MockTaskMockRecorder -} - -// MockTaskMockRecorder is the mock recorder for MockTask. -type MockTaskMockRecorder struct { - mock *MockTask -} - -// NewMockTask creates a new mock instance. -func NewMockTask(ctrl *gomock.Controller) *MockTask { - mock := &MockTask{ctrl: ctrl} - mock.recorder = &MockTaskMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockTask) EXPECT() *MockTaskMockRecorder { - return m.recorder -} - -// AddTraffic mocks base method. -func (m *MockTask) AddTraffic(arg0 uint64) { - m.ctrl.T.Helper() - m.ctrl.Call(m, "AddTraffic", arg0) -} - -// AddTraffic indicates an expected call of AddTraffic. -func (mr *MockTaskMockRecorder) AddTraffic(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTraffic", reflect.TypeOf((*MockTask)(nil).AddTraffic), arg0) -} - -// Context mocks base method. -func (m *MockTask) Context() context.Context { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Context") - ret0, _ := ret[0].(context.Context) - return ret0 -} - -// Context indicates an expected call of Context. -func (mr *MockTaskMockRecorder) Context() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockTask)(nil).Context)) -} - -// GetContentLength mocks base method. -func (m *MockTask) GetContentLength() int64 { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetContentLength") - ret0, _ := ret[0].(int64) - return ret0 -} - -// GetContentLength indicates an expected call of GetContentLength. -func (mr *MockTaskMockRecorder) GetContentLength() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetContentLength", reflect.TypeOf((*MockTask)(nil).GetContentLength)) -} - -// GetPeerID mocks base method. -func (m *MockTask) GetPeerID() string { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPeerID") - ret0, _ := ret[0].(string) - return ret0 -} - -// GetPeerID indicates an expected call of GetPeerID. -func (mr *MockTaskMockRecorder) GetPeerID() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPeerID", reflect.TypeOf((*MockTask)(nil).GetPeerID)) -} - -// GetPieceMd5Sign mocks base method. -func (m *MockTask) GetPieceMd5Sign() string { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPieceMd5Sign") - ret0, _ := ret[0].(string) - return ret0 -} - -// GetPieceMd5Sign indicates an expected call of GetPieceMd5Sign. -func (mr *MockTaskMockRecorder) GetPieceMd5Sign() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceMd5Sign", reflect.TypeOf((*MockTask)(nil).GetPieceMd5Sign)) -} - -// GetStorage mocks base method. -func (m *MockTask) GetStorage() storage.TaskStorageDriver { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetStorage") - ret0, _ := ret[0].(storage.TaskStorageDriver) - return ret0 -} - -// GetStorage indicates an expected call of GetStorage. -func (mr *MockTaskMockRecorder) GetStorage() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStorage", reflect.TypeOf((*MockTask)(nil).GetStorage)) -} - -// GetTaskID mocks base method. -func (m *MockTask) GetTaskID() string { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTaskID") - ret0, _ := ret[0].(string) - return ret0 -} - -// GetTaskID indicates an expected call of GetTaskID. -func (mr *MockTaskMockRecorder) GetTaskID() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskID", reflect.TypeOf((*MockTask)(nil).GetTaskID)) -} - -// GetTotalPieces mocks base method. -func (m *MockTask) GetTotalPieces() int32 { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTotalPieces") - ret0, _ := ret[0].(int32) - return ret0 -} - -// GetTotalPieces indicates an expected call of GetTotalPieces. -func (mr *MockTaskMockRecorder) GetTotalPieces() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTotalPieces", reflect.TypeOf((*MockTask)(nil).GetTotalPieces)) -} - -// GetTraffic mocks base method. -func (m *MockTask) GetTraffic() uint64 { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTraffic") - ret0, _ := ret[0].(uint64) - return ret0 -} - -// GetTraffic indicates an expected call of GetTraffic. -func (mr *MockTaskMockRecorder) GetTraffic() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTraffic", reflect.TypeOf((*MockTask)(nil).GetTraffic)) -} - -// Log mocks base method. -func (m *MockTask) Log() *logger.SugaredLoggerOnWith { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Log") - ret0, _ := ret[0].(*logger.SugaredLoggerOnWith) - return ret0 -} - -// Log indicates an expected call of Log. -func (mr *MockTaskMockRecorder) Log() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Log", reflect.TypeOf((*MockTask)(nil).Log)) -} - -// PublishPieceInfo mocks base method. -func (m *MockTask) PublishPieceInfo(pieceNum int32, size uint32) { - m.ctrl.T.Helper() - m.ctrl.Call(m, "PublishPieceInfo", pieceNum, size) -} - -// PublishPieceInfo indicates an expected call of PublishPieceInfo. -func (mr *MockTaskMockRecorder) PublishPieceInfo(pieceNum, size interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishPieceInfo", reflect.TypeOf((*MockTask)(nil).PublishPieceInfo), pieceNum, size) -} - -// ReportPieceResult mocks base method. -func (m *MockTask) ReportPieceResult(request *peer.DownloadPieceRequest, result *peer.DownloadPieceResult, err error) { - m.ctrl.T.Helper() - m.ctrl.Call(m, "ReportPieceResult", request, result, err) -} - -// ReportPieceResult indicates an expected call of ReportPieceResult. -func (mr *MockTaskMockRecorder) ReportPieceResult(request, result, err interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReportPieceResult", reflect.TypeOf((*MockTask)(nil).ReportPieceResult), request, result, err) -} - -// SetContentLength mocks base method. -func (m *MockTask) SetContentLength(arg0 int64) { - m.ctrl.T.Helper() - m.ctrl.Call(m, "SetContentLength", arg0) -} - -// SetContentLength indicates an expected call of SetContentLength. -func (mr *MockTaskMockRecorder) SetContentLength(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetContentLength", reflect.TypeOf((*MockTask)(nil).SetContentLength), arg0) -} - -// SetPieceMd5Sign mocks base method. -func (m *MockTask) SetPieceMd5Sign(arg0 string) { - m.ctrl.T.Helper() - m.ctrl.Call(m, "SetPieceMd5Sign", arg0) -} - -// SetPieceMd5Sign indicates an expected call of SetPieceMd5Sign. -func (mr *MockTaskMockRecorder) SetPieceMd5Sign(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetPieceMd5Sign", reflect.TypeOf((*MockTask)(nil).SetPieceMd5Sign), arg0) -} - -// SetTotalPieces mocks base method. -func (m *MockTask) SetTotalPieces(arg0 int32) { - m.ctrl.T.Helper() - m.ctrl.Call(m, "SetTotalPieces", arg0) -} - -// SetTotalPieces indicates an expected call of SetTotalPieces. -func (mr *MockTaskMockRecorder) SetTotalPieces(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTotalPieces", reflect.TypeOf((*MockTask)(nil).SetTotalPieces), arg0) -} - -// MockLogger is a mock of Logger interface. -type MockLogger struct { - ctrl *gomock.Controller - recorder *MockLoggerMockRecorder -} - -// MockLoggerMockRecorder is the mock recorder for MockLogger. -type MockLoggerMockRecorder struct { - mock *MockLogger -} - -// NewMockLogger creates a new mock instance. -func NewMockLogger(ctrl *gomock.Controller) *MockLogger { - mock := &MockLogger{ctrl: ctrl} - mock.recorder = &MockLoggerMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockLogger) EXPECT() *MockLoggerMockRecorder { - return m.recorder -} - -// Log mocks base method. -func (m *MockLogger) Log() *logger.SugaredLoggerOnWith { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Log") - ret0, _ := ret[0].(*logger.SugaredLoggerOnWith) - return ret0 -} - -// Log indicates an expected call of Log. -func (mr *MockLoggerMockRecorder) Log() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Log", reflect.TypeOf((*MockLogger)(nil).Log)) -} diff --git a/client/daemon/test/mock/service/manager.go b/client/daemon/test/mock/service/manager.go deleted file mode 100644 index 57226da0576..00000000000 --- a/client/daemon/test/mock/service/manager.go +++ /dev/null @@ -1,116 +0,0 @@ -// Code generated by MockGen. DO NOT EDIT. -// Source: ../../../service/manager.go - -// Package mock_service is a generated GoMock package. -package mock_service - -import ( - net "net" - reflect "reflect" - time "time" - - gomock "github.com/golang/mock/gomock" -) - -// 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 -} - -// 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)) -} - -// 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) -} - -// ServeDownload mocks base method -func (m *MockManager) ServeDownload(listener net.Listener) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ServeDownload", listener) - ret0, _ := ret[0].(error) - return ret0 -} - -// ServeDownload indicates an expected call of ServeDownload -func (mr *MockManagerMockRecorder) ServeDownload(listener interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServeDownload", reflect.TypeOf((*MockManager)(nil).ServeDownload), listener) -} - -// ServePeer mocks base method -func (m *MockManager) ServePeer(listener net.Listener) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ServePeer", listener) - ret0, _ := ret[0].(error) - return ret0 -} - -// ServePeer indicates an expected call of ServePeer -func (mr *MockManagerMockRecorder) ServePeer(listener interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServePeer", reflect.TypeOf((*MockManager)(nil).ServePeer), listener) -} - -// ServeProxy mocks base method -func (m *MockManager) ServeProxy(listener net.Listener) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ServeProxy", listener) - ret0, _ := ret[0].(error) - return ret0 -} - -// ServeProxy indicates an expected call of ServeProxy -func (mr *MockManagerMockRecorder) ServeProxy(listener interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServeProxy", reflect.TypeOf((*MockManager)(nil).ServeProxy), listener) -} - -// Stop mocks base method -func (m *MockManager) Stop() { - m.ctrl.T.Helper() - m.ctrl.Call(m, "Stop") -} - -// Stop indicates an expected call of Stop -func (mr *MockManagerMockRecorder) Stop() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockManager)(nil).Stop)) -} diff --git a/client/daemon/transport/transport_test.go b/client/daemon/transport/transport_test.go index 84a05e27e13..48b139c865a 100644 --- a/client/daemon/transport/transport_test.go +++ b/client/daemon/transport/transport_test.go @@ -29,7 +29,6 @@ import ( "d7y.io/dragonfly/v2/client/daemon/peer" "d7y.io/dragonfly/v2/client/daemon/test" - mock_peer "d7y.io/dragonfly/v2/client/daemon/test/mock/peer" ) func TestMain(m *testing.M) { @@ -43,7 +42,7 @@ func TestTransport_RoundTrip(t *testing.T) { assert.Nil(err, "load test file") var url = "http://x/y" - peerTaskManager := mock_peer.NewMockTaskManager(ctrl) + peerTaskManager := peer.NewMockTaskManager(ctrl) peerTaskManager.EXPECT().StartStreamTask(gomock.Any(), gomock.Any()).DoAndReturn( func(ctx context.Context, req *peer.StreamTaskRequest) (io.ReadCloser, map[string]string, error) { assert.Equal(req.URL, url) diff --git a/client/daemon/upload/mocks/upload_manager_mock.go b/client/daemon/upload/mocks/upload_manager_mock.go new file mode 100644 index 00000000000..79908cd4348 --- /dev/null +++ b/client/daemon/upload/mocks/upload_manager_mock.go @@ -0,0 +1,63 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: upload_manager.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + net "net" + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// 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 +} + +// Serve mocks base method. +func (m *MockManager) Serve(lis net.Listener) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Serve", lis) + ret0, _ := ret[0].(error) + return ret0 +} + +// Serve indicates an expected call of Serve. +func (mr *MockManagerMockRecorder) Serve(lis interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Serve", reflect.TypeOf((*MockManager)(nil).Serve), lis) +} + +// Stop mocks base method. +func (m *MockManager) 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 *MockManagerMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockManager)(nil).Stop)) +} diff --git a/client/daemon/upload/upload_manager.go b/client/daemon/upload/upload_manager.go index 477153e5271..7c633260a8d 100644 --- a/client/daemon/upload/upload_manager.go +++ b/client/daemon/upload/upload_manager.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/upload_manager_mock.go -source upload_manager.go -package mocks + package upload import ( diff --git a/client/daemon/upload/upload_manager_test.go b/client/daemon/upload/upload_manager_test.go index 56957e07100..74121bce7b9 100644 --- a/client/daemon/upload/upload_manager_test.go +++ b/client/daemon/upload/upload_manager_test.go @@ -32,8 +32,8 @@ import ( "d7y.io/dragonfly/v2/client/config" "d7y.io/dragonfly/v2/client/daemon/storage" + "d7y.io/dragonfly/v2/client/daemon/storage/mocks" "d7y.io/dragonfly/v2/client/daemon/test" - mock_storage "d7y.io/dragonfly/v2/client/daemon/test/mock/storage" _ "d7y.io/dragonfly/v2/pkg/rpc/dfdaemon/server" ) @@ -49,7 +49,7 @@ func TestUploadManager_Serve(t *testing.T) { testData, err := os.ReadFile(test.File) assert.Nil(err, "load test file") - mockStorageManager := mock_storage.NewMockManager(ctrl) + mockStorageManager := mocks.NewMockManager(ctrl) mockStorageManager.EXPECT().ReadPiece(gomock.Any(), gomock.Any()).AnyTimes(). DoAndReturn(func(ctx context.Context, req *storage.ReadPieceRequest) (io.Reader, io.Closer, error) { return bytes.NewBuffer(testData[req.Range.Start : req.Range.Start+req.Range.Length]), diff --git a/client/dfget/dfget_test.go b/client/dfget/dfget_test.go index 4a066d9d232..7ab480c9cea 100644 --- a/client/dfget/dfget_test.go +++ b/client/dfget/dfget_test.go @@ -33,7 +33,7 @@ import ( "d7y.io/dragonfly/v2/pkg/digest" "d7y.io/dragonfly/v2/pkg/idgen" "d7y.io/dragonfly/v2/pkg/source" - sourcemock "d7y.io/dragonfly/v2/pkg/source/mock" + "d7y.io/dragonfly/v2/pkg/source/mocks" ) func Test_downloadFromSource(t *testing.T) { @@ -44,7 +44,7 @@ func Test_downloadFromSource(t *testing.T) { content := idgen.UUIDString() - sourceClient := sourcemock.NewMockResourceClient(gomock.NewController(t)) + sourceClient := mocks.NewMockResourceClient(gomock.NewController(t)) require.Nil(t, source.Register("http", sourceClient, func(request *source.Request) *source.Request { return request })) diff --git a/go.mod b/go.mod index ef32f34413c..6e4c3de331c 100644 --- a/go.mod +++ b/go.mod @@ -6,7 +6,6 @@ require ( github.com/RichardKnop/machinery v1.10.6 github.com/Showmax/go-fqdn v1.0.0 github.com/VividCortex/mysqlerr v1.0.0 - github.com/agiledragon/gomonkey/v2 v2.3.1 github.com/aliyun/aliyun-oss-go-sdk v2.2.4+incompatible github.com/appleboy/gin-jwt/v2 v2.8.0 github.com/aws/aws-sdk-go v1.44.44 diff --git a/go.sum b/go.sum index 1c93221bfd7..8a70d1518e7 100644 --- a/go.sum +++ b/go.sum @@ -108,7 +108,6 @@ github.com/VividCortex/gohistogram v1.0.0/go.mod h1:Pf5mBqqDxYaXu3hDrrU+w6nw50o/ github.com/VividCortex/mysqlerr v1.0.0 h1:5pZ2TZA+YnzPgzBfiUWGqWmKDVNBdrkf9g+DNe1Tiq8= github.com/VividCortex/mysqlerr v1.0.0/go.mod h1:xERx8E4tBhLvpjzdUyQiSfUxeMcATEQrflDAfXsqcAE= github.com/afex/hystrix-go v0.0.0-20180502004556-fa1af6a1f4f5/go.mod h1:SkGFH1ia65gfNATL8TAiHDNxPzPdmEL5uirI2Uyuz6c= -github.com/agiledragon/gomonkey/v2 v2.3.1 h1:k+UnUY0EMNYUFUAQVETGY9uUTxjMdnUkP0ARyJS1zzs= github.com/agiledragon/gomonkey/v2 v2.3.1/go.mod h1:ap1AmDzcVOAz1YpeJ3TCzIgstoaWLA6jbbgxfB4w2iY= github.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc= github.com/alecthomas/template v0.0.0-20190718012654-fb15b899a751/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc= diff --git a/hack/protoc.sh b/hack/protoc.sh index fe551dd86e8..6199c6adb41 100755 --- a/hack/protoc.sh +++ b/hack/protoc.sh @@ -1,6 +1,6 @@ #!/bin/bash -PROTOC_ALL_IMAGE=${PROTOC_ALL_IMAGE:-"namely/protoc-all:1.42_1"} +PROTOC_ALL_IMAGE=${PROTOC_ALL_IMAGE:-"namely/protoc-all:1.47_0"} PROTO_PATH=pkg/rpc LANGUAGE=go diff --git a/internal/dynconfig/dynconfig.go b/internal/dynconfig/dynconfig.go index ca72c671bf9..7fa888ba778 100644 --- a/internal/dynconfig/dynconfig.go +++ b/internal/dynconfig/dynconfig.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/dynconfig_mock.go -source dynconfig.go -package mocks + package dynconfig import ( diff --git a/internal/dynconfig/dynconfig_test.go b/internal/dynconfig/dynconfig_test.go index 9e808488eb8..167c87f1741 100644 --- a/internal/dynconfig/dynconfig_test.go +++ b/internal/dynconfig/dynconfig_test.go @@ -27,7 +27,7 @@ import ( "github.com/mitchellh/mapstructure" "github.com/stretchr/testify/assert" - mock_manager_client "d7y.io/dragonfly/v2/internal/dynconfig/mocks" + "d7y.io/dragonfly/v2/internal/dynconfig/mocks" ) type TestDynconfig struct { @@ -51,7 +51,7 @@ func TestDynconfigUnmarshal_ManagerSourceType(t *testing.T) { dynconfig TestDynconfig sleep func() cleanFileCache func(t *testing.T) - mock func(m *mock_manager_client.MockmanagerClientMockRecorder) + mock func(m *mocks.MockManagerClientMockRecorder) expect func(t *testing.T, data interface{}) }{ { @@ -64,7 +64,7 @@ func TestDynconfigUnmarshal_ManagerSourceType(t *testing.T) { }, sleep: func() {}, cleanFileCache: func(t *testing.T) {}, - mock: func(m *mock_manager_client.MockmanagerClientMockRecorder) { + mock: func(m *mocks.MockManagerClientMockRecorder) { var d map[string]interface{} if err := mapstructure.Decode(TestDynconfig{ Scheduler: SchedulerOption{ @@ -106,7 +106,7 @@ func TestDynconfigUnmarshal_ManagerSourceType(t *testing.T) { t.Fatal(err) } }, - mock: func(m *mock_manager_client.MockmanagerClientMockRecorder) { + mock: func(m *mocks.MockManagerClientMockRecorder) { var d map[string]interface{} if err := mapstructure.Decode(TestDynconfig{ Scheduler: SchedulerOption{ @@ -143,7 +143,7 @@ func TestDynconfigUnmarshal_ManagerSourceType(t *testing.T) { t.Fatal(err) } }, - mock: func(m *mock_manager_client.MockmanagerClientMockRecorder) { + mock: func(m *mocks.MockManagerClientMockRecorder) { var d map[string]interface{} if err := mapstructure.Decode(TestDynconfig{ Scheduler: SchedulerOption{ @@ -171,7 +171,7 @@ func TestDynconfigUnmarshal_ManagerSourceType(t *testing.T) { t.Run(tc.name, func(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockManagerClient := mock_manager_client.NewMockmanagerClient(ctl) + mockManagerClient := mocks.NewMockManagerClient(ctl) tc.mock(mockManagerClient.EXPECT()) d, err := New(ManagerSourceType, []Option{ diff --git a/internal/dynconfig/manager_client.go b/internal/dynconfig/manager_client.go index 6675b89ea6f..b175c6cab18 100644 --- a/internal/dynconfig/manager_client.go +++ b/internal/dynconfig/manager_client.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/manager_client_mock.go -source manager_client.go -package mocks + package dynconfig // managerClient is a client of manager diff --git a/internal/dynconfig/mocks/dynconfig_mock.go b/internal/dynconfig/mocks/dynconfig_mock.go new file mode 100644 index 00000000000..0f620935c8d --- /dev/null +++ b/internal/dynconfig/mocks/dynconfig_mock.go @@ -0,0 +1,48 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: dynconfig.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// Mockstrategy is a mock of strategy interface. +type Mockstrategy struct { + ctrl *gomock.Controller + recorder *MockstrategyMockRecorder +} + +// MockstrategyMockRecorder is the mock recorder for Mockstrategy. +type MockstrategyMockRecorder struct { + mock *Mockstrategy +} + +// NewMockstrategy creates a new mock instance. +func NewMockstrategy(ctrl *gomock.Controller) *Mockstrategy { + mock := &Mockstrategy{ctrl: ctrl} + mock.recorder = &MockstrategyMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *Mockstrategy) EXPECT() *MockstrategyMockRecorder { + return m.recorder +} + +// Unmarshal mocks base method. +func (m *Mockstrategy) Unmarshal(rawVal interface{}) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Unmarshal", rawVal) + ret0, _ := ret[0].(error) + return ret0 +} + +// Unmarshal indicates an expected call of Unmarshal. +func (mr *MockstrategyMockRecorder) Unmarshal(rawVal interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Unmarshal", reflect.TypeOf((*Mockstrategy)(nil).Unmarshal), rawVal) +} diff --git a/internal/dynconfig/mocks/manager_client_mock.go b/internal/dynconfig/mocks/manager_client_mock.go index b3b64b07996..c403a6e9178 100644 --- a/internal/dynconfig/mocks/manager_client_mock.go +++ b/internal/dynconfig/mocks/manager_client_mock.go @@ -1,8 +1,8 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: ./manager_client.go +// Source: manager_client.go -// Package mock_manager_client is a generated GoMock package. -package mock_manager_client +// Package mocks is a generated GoMock package. +package mocks import ( reflect "reflect" @@ -10,31 +10,31 @@ import ( gomock "github.com/golang/mock/gomock" ) -// MockmanagerClient is a mock of managerClient interface. -type MockmanagerClient struct { +// MockManagerClient is a mock of ManagerClient interface. +type MockManagerClient struct { ctrl *gomock.Controller - recorder *MockmanagerClientMockRecorder + recorder *MockManagerClientMockRecorder } -// MockmanagerClientMockRecorder is the mock recorder for MockmanagerClient. -type MockmanagerClientMockRecorder struct { - mock *MockmanagerClient +// MockManagerClientMockRecorder is the mock recorder for MockManagerClient. +type MockManagerClientMockRecorder struct { + mock *MockManagerClient } -// NewMockmanagerClient creates a new mock instance. -func NewMockmanagerClient(ctrl *gomock.Controller) *MockmanagerClient { - mock := &MockmanagerClient{ctrl: ctrl} - mock.recorder = &MockmanagerClientMockRecorder{mock} +// NewMockManagerClient creates a new mock instance. +func NewMockManagerClient(ctrl *gomock.Controller) *MockManagerClient { + mock := &MockManagerClient{ctrl: ctrl} + mock.recorder = &MockManagerClientMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockmanagerClient) EXPECT() *MockmanagerClientMockRecorder { +func (m *MockManagerClient) EXPECT() *MockManagerClientMockRecorder { return m.recorder } // Get mocks base method. -func (m *MockmanagerClient) Get() (interface{}, error) { +func (m *MockManagerClient) Get() (interface{}, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "Get") ret0, _ := ret[0].(interface{}) @@ -43,7 +43,7 @@ func (m *MockmanagerClient) Get() (interface{}, error) { } // Get indicates an expected call of Get. -func (mr *MockmanagerClientMockRecorder) Get() *gomock.Call { +func (mr *MockManagerClientMockRecorder) Get() *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockmanagerClient)(nil).Get)) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockManagerClient)(nil).Get)) } diff --git a/manager/auth/oauth/mocks/oauth_mock.go b/manager/auth/oauth/mocks/oauth_mock.go new file mode 100644 index 00000000000..3f046349b2d --- /dev/null +++ b/manager/auth/oauth/mocks/oauth_mock.go @@ -0,0 +1,81 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: oauth.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + oauth "d7y.io/dragonfly/v2/manager/auth/oauth" + gomock "github.com/golang/mock/gomock" + oauth2 "golang.org/x/oauth2" +) + +// MockOauth is a mock of Oauth interface. +type MockOauth struct { + ctrl *gomock.Controller + recorder *MockOauthMockRecorder +} + +// MockOauthMockRecorder is the mock recorder for MockOauth. +type MockOauthMockRecorder struct { + mock *MockOauth +} + +// NewMockOauth creates a new mock instance. +func NewMockOauth(ctrl *gomock.Controller) *MockOauth { + mock := &MockOauth{ctrl: ctrl} + mock.recorder = &MockOauthMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockOauth) EXPECT() *MockOauthMockRecorder { + return m.recorder +} + +// AuthCodeURL mocks base method. +func (m *MockOauth) AuthCodeURL() (string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AuthCodeURL") + ret0, _ := ret[0].(string) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// AuthCodeURL indicates an expected call of AuthCodeURL. +func (mr *MockOauthMockRecorder) AuthCodeURL() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AuthCodeURL", reflect.TypeOf((*MockOauth)(nil).AuthCodeURL)) +} + +// Exchange mocks base method. +func (m *MockOauth) Exchange(arg0 string) (*oauth2.Token, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Exchange", arg0) + ret0, _ := ret[0].(*oauth2.Token) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Exchange indicates an expected call of Exchange. +func (mr *MockOauthMockRecorder) Exchange(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Exchange", reflect.TypeOf((*MockOauth)(nil).Exchange), arg0) +} + +// GetUser mocks base method. +func (m *MockOauth) GetUser(arg0 *oauth2.Token) (*oauth.User, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetUser", arg0) + ret0, _ := ret[0].(*oauth.User) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetUser indicates an expected call of GetUser. +func (mr *MockOauthMockRecorder) GetUser(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUser", reflect.TypeOf((*MockOauth)(nil).GetUser), arg0) +} diff --git a/manager/auth/oauth/oauth.go b/manager/auth/oauth/oauth.go index 4ef87d6d02e..149312148fc 100644 --- a/manager/auth/oauth/oauth.go +++ b/manager/auth/oauth/oauth.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/oauth_mock.go -source oauth.go -package mocks + package oauth import ( diff --git a/manager/job/mocks/preheat_mock.go b/manager/job/mocks/preheat_mock.go new file mode 100644 index 00000000000..5e2965f0723 --- /dev/null +++ b/manager/job/mocks/preheat_mock.go @@ -0,0 +1,53 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: preheat.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + job "d7y.io/dragonfly/v2/internal/job" + model "d7y.io/dragonfly/v2/manager/model" + types "d7y.io/dragonfly/v2/manager/types" + gomock "github.com/golang/mock/gomock" +) + +// MockPreheat is a mock of Preheat interface. +type MockPreheat struct { + ctrl *gomock.Controller + recorder *MockPreheatMockRecorder +} + +// MockPreheatMockRecorder is the mock recorder for MockPreheat. +type MockPreheatMockRecorder struct { + mock *MockPreheat +} + +// NewMockPreheat creates a new mock instance. +func NewMockPreheat(ctrl *gomock.Controller) *MockPreheat { + mock := &MockPreheat{ctrl: ctrl} + mock.recorder = &MockPreheatMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockPreheat) EXPECT() *MockPreheatMockRecorder { + return m.recorder +} + +// CreatePreheat mocks base method. +func (m *MockPreheat) CreatePreheat(arg0 context.Context, arg1 []model.Scheduler, arg2 types.PreheatArgs) (*job.GroupJobState, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreatePreheat", arg0, arg1, arg2) + ret0, _ := ret[0].(*job.GroupJobState) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreatePreheat indicates an expected call of CreatePreheat. +func (mr *MockPreheatMockRecorder) CreatePreheat(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreatePreheat", reflect.TypeOf((*MockPreheat)(nil).CreatePreheat), arg0, arg1, arg2) +} diff --git a/manager/job/preheat.go b/manager/job/preheat.go index 74dccca8e43..7db03f08a42 100644 --- a/manager/job/preheat.go +++ b/manager/job/preheat.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/preheat_mock.go -source preheat.go -package mocks + package job import ( diff --git a/manager/searcher/mocks/searcher_mock.go b/manager/searcher/mocks/searcher_mock.go new file mode 100644 index 00000000000..47a9df07de9 --- /dev/null +++ b/manager/searcher/mocks/searcher_mock.go @@ -0,0 +1,52 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: searcher.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + model "d7y.io/dragonfly/v2/manager/model" + manager "d7y.io/dragonfly/v2/pkg/rpc/manager" + gomock "github.com/golang/mock/gomock" +) + +// MockSearcher is a mock of Searcher interface. +type MockSearcher struct { + ctrl *gomock.Controller + recorder *MockSearcherMockRecorder +} + +// MockSearcherMockRecorder is the mock recorder for MockSearcher. +type MockSearcherMockRecorder struct { + mock *MockSearcher +} + +// NewMockSearcher creates a new mock instance. +func NewMockSearcher(ctrl *gomock.Controller) *MockSearcher { + mock := &MockSearcher{ctrl: ctrl} + mock.recorder = &MockSearcherMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSearcher) EXPECT() *MockSearcherMockRecorder { + return m.recorder +} + +// FindSchedulerClusters mocks base method. +func (m *MockSearcher) FindSchedulerClusters(arg0 context.Context, arg1 []model.SchedulerCluster, arg2 *manager.ListSchedulersRequest) ([]model.SchedulerCluster, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "FindSchedulerClusters", arg0, arg1, arg2) + ret0, _ := ret[0].([]model.SchedulerCluster) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// FindSchedulerClusters indicates an expected call of FindSchedulerClusters. +func (mr *MockSearcherMockRecorder) FindSchedulerClusters(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FindSchedulerClusters", reflect.TypeOf((*MockSearcher)(nil).FindSchedulerClusters), arg0, arg1, arg2) +} diff --git a/manager/searcher/searcher.go b/manager/searcher/searcher.go index 54078243dab..d6deec19c8d 100644 --- a/manager/searcher/searcher.go +++ b/manager/searcher/searcher.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/searcher_mock.go -source searcher.go -package mocks + package searcher import ( diff --git a/manager/service/mocks/service_mock.go b/manager/service/mocks/service_mock.go new file mode 100644 index 00000000000..2723e9cab4e --- /dev/null +++ b/manager/service/mocks/service_mock.go @@ -0,0 +1,1313 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: service.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + model "d7y.io/dragonfly/v2/manager/model" + rbac "d7y.io/dragonfly/v2/manager/permission/rbac" + types "d7y.io/dragonfly/v2/manager/types" + objectstorage "d7y.io/dragonfly/v2/pkg/objectstorage" + gin "github.com/gin-gonic/gin" + gomock "github.com/golang/mock/gomock" +) + +// MockService is a mock of Service interface. +type MockService struct { + ctrl *gomock.Controller + recorder *MockServiceMockRecorder +} + +// MockServiceMockRecorder is the mock recorder for MockService. +type MockServiceMockRecorder struct { + mock *MockService +} + +// NewMockService creates a new mock instance. +func NewMockService(ctrl *gomock.Controller) *MockService { + mock := &MockService{ctrl: ctrl} + mock.recorder = &MockServiceMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockService) EXPECT() *MockServiceMockRecorder { + return m.recorder +} + +// AddPermissionForRole mocks base method. +func (m *MockService) AddPermissionForRole(arg0 context.Context, arg1 string, arg2 types.AddPermissionForRoleRequest) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddPermissionForRole", arg0, arg1, arg2) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// AddPermissionForRole indicates an expected call of AddPermissionForRole. +func (mr *MockServiceMockRecorder) AddPermissionForRole(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddPermissionForRole", reflect.TypeOf((*MockService)(nil).AddPermissionForRole), arg0, arg1, arg2) +} + +// AddRoleForUser mocks base method. +func (m *MockService) AddRoleForUser(arg0 context.Context, arg1 types.AddRoleForUserParams) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddRoleForUser", arg0, arg1) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// AddRoleForUser indicates an expected call of AddRoleForUser. +func (mr *MockServiceMockRecorder) AddRoleForUser(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddRoleForUser", reflect.TypeOf((*MockService)(nil).AddRoleForUser), arg0, arg1) +} + +// AddSchedulerClusterToApplication mocks base method. +func (m *MockService) AddSchedulerClusterToApplication(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddSchedulerClusterToApplication", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddSchedulerClusterToApplication indicates an expected call of AddSchedulerClusterToApplication. +func (mr *MockServiceMockRecorder) AddSchedulerClusterToApplication(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSchedulerClusterToApplication", reflect.TypeOf((*MockService)(nil).AddSchedulerClusterToApplication), arg0, arg1, arg2) +} + +// AddSchedulerClusterToSecurityGroup mocks base method. +func (m *MockService) AddSchedulerClusterToSecurityGroup(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddSchedulerClusterToSecurityGroup", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddSchedulerClusterToSecurityGroup indicates an expected call of AddSchedulerClusterToSecurityGroup. +func (mr *MockServiceMockRecorder) AddSchedulerClusterToSecurityGroup(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSchedulerClusterToSecurityGroup", reflect.TypeOf((*MockService)(nil).AddSchedulerClusterToSecurityGroup), arg0, arg1, arg2) +} + +// AddSchedulerClusterToSeedPeerCluster mocks base method. +func (m *MockService) AddSchedulerClusterToSeedPeerCluster(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddSchedulerClusterToSeedPeerCluster", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddSchedulerClusterToSeedPeerCluster indicates an expected call of AddSchedulerClusterToSeedPeerCluster. +func (mr *MockServiceMockRecorder) AddSchedulerClusterToSeedPeerCluster(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSchedulerClusterToSeedPeerCluster", reflect.TypeOf((*MockService)(nil).AddSchedulerClusterToSeedPeerCluster), arg0, arg1, arg2) +} + +// AddSchedulerToSchedulerCluster mocks base method. +func (m *MockService) AddSchedulerToSchedulerCluster(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddSchedulerToSchedulerCluster", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddSchedulerToSchedulerCluster indicates an expected call of AddSchedulerToSchedulerCluster. +func (mr *MockServiceMockRecorder) AddSchedulerToSchedulerCluster(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSchedulerToSchedulerCluster", reflect.TypeOf((*MockService)(nil).AddSchedulerToSchedulerCluster), arg0, arg1, arg2) +} + +// AddSecurityRuleToSecurityGroup mocks base method. +func (m *MockService) AddSecurityRuleToSecurityGroup(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddSecurityRuleToSecurityGroup", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddSecurityRuleToSecurityGroup indicates an expected call of AddSecurityRuleToSecurityGroup. +func (mr *MockServiceMockRecorder) AddSecurityRuleToSecurityGroup(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSecurityRuleToSecurityGroup", reflect.TypeOf((*MockService)(nil).AddSecurityRuleToSecurityGroup), arg0, arg1, arg2) +} + +// AddSeedPeerClusterToApplication mocks base method. +func (m *MockService) AddSeedPeerClusterToApplication(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddSeedPeerClusterToApplication", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddSeedPeerClusterToApplication indicates an expected call of AddSeedPeerClusterToApplication. +func (mr *MockServiceMockRecorder) AddSeedPeerClusterToApplication(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSeedPeerClusterToApplication", reflect.TypeOf((*MockService)(nil).AddSeedPeerClusterToApplication), arg0, arg1, arg2) +} + +// AddSeedPeerClusterToSecurityGroup mocks base method. +func (m *MockService) AddSeedPeerClusterToSecurityGroup(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddSeedPeerClusterToSecurityGroup", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddSeedPeerClusterToSecurityGroup indicates an expected call of AddSeedPeerClusterToSecurityGroup. +func (mr *MockServiceMockRecorder) AddSeedPeerClusterToSecurityGroup(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSeedPeerClusterToSecurityGroup", reflect.TypeOf((*MockService)(nil).AddSeedPeerClusterToSecurityGroup), arg0, arg1, arg2) +} + +// AddSeedPeerToSeedPeerCluster mocks base method. +func (m *MockService) AddSeedPeerToSeedPeerCluster(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddSeedPeerToSeedPeerCluster", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddSeedPeerToSeedPeerCluster indicates an expected call of AddSeedPeerToSeedPeerCluster. +func (mr *MockServiceMockRecorder) AddSeedPeerToSeedPeerCluster(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddSeedPeerToSeedPeerCluster", reflect.TypeOf((*MockService)(nil).AddSeedPeerToSeedPeerCluster), arg0, arg1, arg2) +} + +// CreateApplication mocks base method. +func (m *MockService) CreateApplication(arg0 context.Context, arg1 types.CreateApplicationRequest) (*model.Application, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateApplication", arg0, arg1) + ret0, _ := ret[0].(*model.Application) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateApplication indicates an expected call of CreateApplication. +func (mr *MockServiceMockRecorder) CreateApplication(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateApplication", reflect.TypeOf((*MockService)(nil).CreateApplication), arg0, arg1) +} + +// CreateBucket mocks base method. +func (m *MockService) CreateBucket(arg0 context.Context, arg1 types.CreateBucketRequest) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateBucket", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// CreateBucket indicates an expected call of CreateBucket. +func (mr *MockServiceMockRecorder) CreateBucket(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateBucket", reflect.TypeOf((*MockService)(nil).CreateBucket), arg0, arg1) +} + +// CreateConfig mocks base method. +func (m *MockService) CreateConfig(arg0 context.Context, arg1 types.CreateConfigRequest) (*model.Config, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateConfig", arg0, arg1) + ret0, _ := ret[0].(*model.Config) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateConfig indicates an expected call of CreateConfig. +func (mr *MockServiceMockRecorder) CreateConfig(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateConfig", reflect.TypeOf((*MockService)(nil).CreateConfig), arg0, arg1) +} + +// CreateOauth mocks base method. +func (m *MockService) CreateOauth(arg0 context.Context, arg1 types.CreateOauthRequest) (*model.Oauth, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateOauth", arg0, arg1) + ret0, _ := ret[0].(*model.Oauth) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateOauth indicates an expected call of CreateOauth. +func (mr *MockServiceMockRecorder) CreateOauth(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOauth", reflect.TypeOf((*MockService)(nil).CreateOauth), arg0, arg1) +} + +// CreatePreheatJob mocks base method. +func (m *MockService) CreatePreheatJob(arg0 context.Context, arg1 types.CreatePreheatJobRequest) (*model.Job, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreatePreheatJob", arg0, arg1) + ret0, _ := ret[0].(*model.Job) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreatePreheatJob indicates an expected call of CreatePreheatJob. +func (mr *MockServiceMockRecorder) CreatePreheatJob(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreatePreheatJob", reflect.TypeOf((*MockService)(nil).CreatePreheatJob), arg0, arg1) +} + +// CreateRole mocks base method. +func (m *MockService) CreateRole(arg0 context.Context, arg1 types.CreateRoleRequest) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateRole", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// CreateRole indicates an expected call of CreateRole. +func (mr *MockServiceMockRecorder) CreateRole(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateRole", reflect.TypeOf((*MockService)(nil).CreateRole), arg0, arg1) +} + +// CreateScheduler mocks base method. +func (m *MockService) CreateScheduler(arg0 context.Context, arg1 types.CreateSchedulerRequest) (*model.Scheduler, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateScheduler", arg0, arg1) + ret0, _ := ret[0].(*model.Scheduler) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateScheduler indicates an expected call of CreateScheduler. +func (mr *MockServiceMockRecorder) CreateScheduler(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateScheduler", reflect.TypeOf((*MockService)(nil).CreateScheduler), arg0, arg1) +} + +// CreateSchedulerCluster mocks base method. +func (m *MockService) CreateSchedulerCluster(arg0 context.Context, arg1 types.CreateSchedulerClusterRequest) (*model.SchedulerCluster, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateSchedulerCluster", arg0, arg1) + ret0, _ := ret[0].(*model.SchedulerCluster) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateSchedulerCluster indicates an expected call of CreateSchedulerCluster. +func (mr *MockServiceMockRecorder) CreateSchedulerCluster(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSchedulerCluster", reflect.TypeOf((*MockService)(nil).CreateSchedulerCluster), arg0, arg1) +} + +// CreateSecurityGroup mocks base method. +func (m *MockService) CreateSecurityGroup(arg0 context.Context, arg1 types.CreateSecurityGroupRequest) (*model.SecurityGroup, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateSecurityGroup", arg0, arg1) + ret0, _ := ret[0].(*model.SecurityGroup) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateSecurityGroup indicates an expected call of CreateSecurityGroup. +func (mr *MockServiceMockRecorder) CreateSecurityGroup(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSecurityGroup", reflect.TypeOf((*MockService)(nil).CreateSecurityGroup), arg0, arg1) +} + +// CreateSecurityRule mocks base method. +func (m *MockService) CreateSecurityRule(arg0 context.Context, arg1 types.CreateSecurityRuleRequest) (*model.SecurityRule, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateSecurityRule", arg0, arg1) + ret0, _ := ret[0].(*model.SecurityRule) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateSecurityRule indicates an expected call of CreateSecurityRule. +func (mr *MockServiceMockRecorder) CreateSecurityRule(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSecurityRule", reflect.TypeOf((*MockService)(nil).CreateSecurityRule), arg0, arg1) +} + +// CreateSeedPeer mocks base method. +func (m *MockService) CreateSeedPeer(arg0 context.Context, arg1 types.CreateSeedPeerRequest) (*model.SeedPeer, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateSeedPeer", arg0, arg1) + ret0, _ := ret[0].(*model.SeedPeer) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateSeedPeer indicates an expected call of CreateSeedPeer. +func (mr *MockServiceMockRecorder) CreateSeedPeer(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSeedPeer", reflect.TypeOf((*MockService)(nil).CreateSeedPeer), arg0, arg1) +} + +// CreateSeedPeerCluster mocks base method. +func (m *MockService) CreateSeedPeerCluster(arg0 context.Context, arg1 types.CreateSeedPeerClusterRequest) (*model.SeedPeerCluster, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateSeedPeerCluster", arg0, arg1) + ret0, _ := ret[0].(*model.SeedPeerCluster) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateSeedPeerCluster indicates an expected call of CreateSeedPeerCluster. +func (mr *MockServiceMockRecorder) CreateSeedPeerCluster(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSeedPeerCluster", reflect.TypeOf((*MockService)(nil).CreateSeedPeerCluster), arg0, arg1) +} + +// CreateV1Preheat mocks base method. +func (m *MockService) CreateV1Preheat(arg0 context.Context, arg1 types.CreateV1PreheatRequest) (*types.CreateV1PreheatResponse, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateV1Preheat", arg0, arg1) + ret0, _ := ret[0].(*types.CreateV1PreheatResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateV1Preheat indicates an expected call of CreateV1Preheat. +func (mr *MockServiceMockRecorder) CreateV1Preheat(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateV1Preheat", reflect.TypeOf((*MockService)(nil).CreateV1Preheat), arg0, arg1) +} + +// DeletePermissionForRole mocks base method. +func (m *MockService) DeletePermissionForRole(arg0 context.Context, arg1 string, arg2 types.DeletePermissionForRoleRequest) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DeletePermissionForRole", arg0, arg1, arg2) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DeletePermissionForRole indicates an expected call of DeletePermissionForRole. +func (mr *MockServiceMockRecorder) DeletePermissionForRole(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePermissionForRole", reflect.TypeOf((*MockService)(nil).DeletePermissionForRole), arg0, arg1, arg2) +} + +// DeleteRoleForUser mocks base method. +func (m *MockService) DeleteRoleForUser(arg0 context.Context, arg1 types.DeleteRoleForUserParams) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DeleteRoleForUser", arg0, arg1) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DeleteRoleForUser indicates an expected call of DeleteRoleForUser. +func (mr *MockServiceMockRecorder) DeleteRoleForUser(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRoleForUser", reflect.TypeOf((*MockService)(nil).DeleteRoleForUser), arg0, arg1) +} + +// DeleteSchedulerClusterToApplication mocks base method. +func (m *MockService) DeleteSchedulerClusterToApplication(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DeleteSchedulerClusterToApplication", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// DeleteSchedulerClusterToApplication indicates an expected call of DeleteSchedulerClusterToApplication. +func (mr *MockServiceMockRecorder) DeleteSchedulerClusterToApplication(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteSchedulerClusterToApplication", reflect.TypeOf((*MockService)(nil).DeleteSchedulerClusterToApplication), arg0, arg1, arg2) +} + +// DeleteSeedPeerClusterToApplication mocks base method. +func (m *MockService) DeleteSeedPeerClusterToApplication(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DeleteSeedPeerClusterToApplication", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// DeleteSeedPeerClusterToApplication indicates an expected call of DeleteSeedPeerClusterToApplication. +func (mr *MockServiceMockRecorder) DeleteSeedPeerClusterToApplication(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteSeedPeerClusterToApplication", reflect.TypeOf((*MockService)(nil).DeleteSeedPeerClusterToApplication), arg0, arg1, arg2) +} + +// DestroyApplication mocks base method. +func (m *MockService) DestroyApplication(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroyApplication", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroyApplication indicates an expected call of DestroyApplication. +func (mr *MockServiceMockRecorder) DestroyApplication(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroyApplication", reflect.TypeOf((*MockService)(nil).DestroyApplication), arg0, arg1) +} + +// DestroyBucket mocks base method. +func (m *MockService) DestroyBucket(arg0 context.Context, arg1 string) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroyBucket", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroyBucket indicates an expected call of DestroyBucket. +func (mr *MockServiceMockRecorder) DestroyBucket(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroyBucket", reflect.TypeOf((*MockService)(nil).DestroyBucket), arg0, arg1) +} + +// DestroyConfig mocks base method. +func (m *MockService) DestroyConfig(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroyConfig", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroyConfig indicates an expected call of DestroyConfig. +func (mr *MockServiceMockRecorder) DestroyConfig(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroyConfig", reflect.TypeOf((*MockService)(nil).DestroyConfig), arg0, arg1) +} + +// DestroyJob mocks base method. +func (m *MockService) DestroyJob(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroyJob", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroyJob indicates an expected call of DestroyJob. +func (mr *MockServiceMockRecorder) DestroyJob(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroyJob", reflect.TypeOf((*MockService)(nil).DestroyJob), arg0, arg1) +} + +// DestroyOauth mocks base method. +func (m *MockService) DestroyOauth(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroyOauth", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroyOauth indicates an expected call of DestroyOauth. +func (mr *MockServiceMockRecorder) DestroyOauth(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroyOauth", reflect.TypeOf((*MockService)(nil).DestroyOauth), arg0, arg1) +} + +// DestroyRole mocks base method. +func (m *MockService) DestroyRole(arg0 context.Context, arg1 string) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroyRole", arg0, arg1) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DestroyRole indicates an expected call of DestroyRole. +func (mr *MockServiceMockRecorder) DestroyRole(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroyRole", reflect.TypeOf((*MockService)(nil).DestroyRole), arg0, arg1) +} + +// DestroyScheduler mocks base method. +func (m *MockService) DestroyScheduler(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroyScheduler", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroyScheduler indicates an expected call of DestroyScheduler. +func (mr *MockServiceMockRecorder) DestroyScheduler(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroyScheduler", reflect.TypeOf((*MockService)(nil).DestroyScheduler), arg0, arg1) +} + +// DestroySchedulerCluster mocks base method. +func (m *MockService) DestroySchedulerCluster(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroySchedulerCluster", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroySchedulerCluster indicates an expected call of DestroySchedulerCluster. +func (mr *MockServiceMockRecorder) DestroySchedulerCluster(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroySchedulerCluster", reflect.TypeOf((*MockService)(nil).DestroySchedulerCluster), arg0, arg1) +} + +// DestroySecurityGroup mocks base method. +func (m *MockService) DestroySecurityGroup(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroySecurityGroup", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroySecurityGroup indicates an expected call of DestroySecurityGroup. +func (mr *MockServiceMockRecorder) DestroySecurityGroup(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroySecurityGroup", reflect.TypeOf((*MockService)(nil).DestroySecurityGroup), arg0, arg1) +} + +// DestroySecurityRule mocks base method. +func (m *MockService) DestroySecurityRule(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroySecurityRule", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroySecurityRule indicates an expected call of DestroySecurityRule. +func (mr *MockServiceMockRecorder) DestroySecurityRule(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroySecurityRule", reflect.TypeOf((*MockService)(nil).DestroySecurityRule), arg0, arg1) +} + +// DestroySecurityRuleToSecurityGroup mocks base method. +func (m *MockService) DestroySecurityRuleToSecurityGroup(arg0 context.Context, arg1, arg2 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroySecurityRuleToSecurityGroup", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroySecurityRuleToSecurityGroup indicates an expected call of DestroySecurityRuleToSecurityGroup. +func (mr *MockServiceMockRecorder) DestroySecurityRuleToSecurityGroup(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroySecurityRuleToSecurityGroup", reflect.TypeOf((*MockService)(nil).DestroySecurityRuleToSecurityGroup), arg0, arg1, arg2) +} + +// DestroySeedPeer mocks base method. +func (m *MockService) DestroySeedPeer(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroySeedPeer", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroySeedPeer indicates an expected call of DestroySeedPeer. +func (mr *MockServiceMockRecorder) DestroySeedPeer(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroySeedPeer", reflect.TypeOf((*MockService)(nil).DestroySeedPeer), arg0, arg1) +} + +// DestroySeedPeerCluster mocks base method. +func (m *MockService) DestroySeedPeerCluster(arg0 context.Context, arg1 uint) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DestroySeedPeerCluster", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DestroySeedPeerCluster indicates an expected call of DestroySeedPeerCluster. +func (mr *MockServiceMockRecorder) DestroySeedPeerCluster(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DestroySeedPeerCluster", reflect.TypeOf((*MockService)(nil).DestroySeedPeerCluster), arg0, arg1) +} + +// GetApplication mocks base method. +func (m *MockService) GetApplication(arg0 context.Context, arg1 uint) (*model.Application, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetApplication", arg0, arg1) + ret0, _ := ret[0].(*model.Application) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetApplication indicates an expected call of GetApplication. +func (mr *MockServiceMockRecorder) GetApplication(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetApplication", reflect.TypeOf((*MockService)(nil).GetApplication), arg0, arg1) +} + +// GetApplications mocks base method. +func (m *MockService) GetApplications(arg0 context.Context, arg1 types.GetApplicationsQuery) ([]model.Application, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetApplications", arg0, arg1) + ret0, _ := ret[0].([]model.Application) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetApplications indicates an expected call of GetApplications. +func (mr *MockServiceMockRecorder) GetApplications(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetApplications", reflect.TypeOf((*MockService)(nil).GetApplications), arg0, arg1) +} + +// GetBucket mocks base method. +func (m *MockService) GetBucket(arg0 context.Context, arg1 string) (*objectstorage.BucketMetadata, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetBucket", arg0, arg1) + ret0, _ := ret[0].(*objectstorage.BucketMetadata) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetBucket indicates an expected call of GetBucket. +func (mr *MockServiceMockRecorder) GetBucket(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBucket", reflect.TypeOf((*MockService)(nil).GetBucket), arg0, arg1) +} + +// GetBuckets mocks base method. +func (m *MockService) GetBuckets(arg0 context.Context) ([]*objectstorage.BucketMetadata, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetBuckets", arg0) + ret0, _ := ret[0].([]*objectstorage.BucketMetadata) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetBuckets indicates an expected call of GetBuckets. +func (mr *MockServiceMockRecorder) GetBuckets(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBuckets", reflect.TypeOf((*MockService)(nil).GetBuckets), arg0) +} + +// GetConfig mocks base method. +func (m *MockService) GetConfig(arg0 context.Context, arg1 uint) (*model.Config, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetConfig", arg0, arg1) + ret0, _ := ret[0].(*model.Config) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetConfig indicates an expected call of GetConfig. +func (mr *MockServiceMockRecorder) GetConfig(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConfig", reflect.TypeOf((*MockService)(nil).GetConfig), arg0, arg1) +} + +// GetConfigs mocks base method. +func (m *MockService) GetConfigs(arg0 context.Context, arg1 types.GetConfigsQuery) ([]model.Config, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetConfigs", arg0, arg1) + ret0, _ := ret[0].([]model.Config) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetConfigs indicates an expected call of GetConfigs. +func (mr *MockServiceMockRecorder) GetConfigs(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetConfigs", reflect.TypeOf((*MockService)(nil).GetConfigs), arg0, arg1) +} + +// GetJob mocks base method. +func (m *MockService) GetJob(arg0 context.Context, arg1 uint) (*model.Job, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetJob", arg0, arg1) + ret0, _ := ret[0].(*model.Job) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetJob indicates an expected call of GetJob. +func (mr *MockServiceMockRecorder) GetJob(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetJob", reflect.TypeOf((*MockService)(nil).GetJob), arg0, arg1) +} + +// GetJobs mocks base method. +func (m *MockService) GetJobs(arg0 context.Context, arg1 types.GetJobsQuery) ([]model.Job, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetJobs", arg0, arg1) + ret0, _ := ret[0].([]model.Job) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetJobs indicates an expected call of GetJobs. +func (mr *MockServiceMockRecorder) GetJobs(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetJobs", reflect.TypeOf((*MockService)(nil).GetJobs), arg0, arg1) +} + +// GetOauth mocks base method. +func (m *MockService) GetOauth(arg0 context.Context, arg1 uint) (*model.Oauth, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetOauth", arg0, arg1) + ret0, _ := ret[0].(*model.Oauth) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetOauth indicates an expected call of GetOauth. +func (mr *MockServiceMockRecorder) GetOauth(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOauth", reflect.TypeOf((*MockService)(nil).GetOauth), arg0, arg1) +} + +// GetOauths mocks base method. +func (m *MockService) GetOauths(arg0 context.Context, arg1 types.GetOauthsQuery) ([]model.Oauth, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetOauths", arg0, arg1) + ret0, _ := ret[0].([]model.Oauth) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetOauths indicates an expected call of GetOauths. +func (mr *MockServiceMockRecorder) GetOauths(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOauths", reflect.TypeOf((*MockService)(nil).GetOauths), arg0, arg1) +} + +// GetPeers mocks base method. +func (m *MockService) GetPeers(arg0 context.Context) ([]string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetPeers", arg0) + ret0, _ := ret[0].([]string) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetPeers indicates an expected call of GetPeers. +func (mr *MockServiceMockRecorder) GetPeers(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPeers", reflect.TypeOf((*MockService)(nil).GetPeers), arg0) +} + +// GetPermissions mocks base method. +func (m *MockService) GetPermissions(arg0 context.Context, arg1 *gin.Engine) []rbac.Permission { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetPermissions", arg0, arg1) + ret0, _ := ret[0].([]rbac.Permission) + return ret0 +} + +// GetPermissions indicates an expected call of GetPermissions. +func (mr *MockServiceMockRecorder) GetPermissions(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPermissions", reflect.TypeOf((*MockService)(nil).GetPermissions), arg0, arg1) +} + +// GetRole mocks base method. +func (m *MockService) GetRole(arg0 context.Context, arg1 string) [][]string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetRole", arg0, arg1) + ret0, _ := ret[0].([][]string) + return ret0 +} + +// GetRole indicates an expected call of GetRole. +func (mr *MockServiceMockRecorder) GetRole(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRole", reflect.TypeOf((*MockService)(nil).GetRole), arg0, arg1) +} + +// GetRoles mocks base method. +func (m *MockService) GetRoles(arg0 context.Context) []string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetRoles", arg0) + ret0, _ := ret[0].([]string) + return ret0 +} + +// GetRoles indicates an expected call of GetRoles. +func (mr *MockServiceMockRecorder) GetRoles(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRoles", reflect.TypeOf((*MockService)(nil).GetRoles), arg0) +} + +// GetRolesForUser mocks base method. +func (m *MockService) GetRolesForUser(arg0 context.Context, arg1 uint) ([]string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetRolesForUser", arg0, arg1) + ret0, _ := ret[0].([]string) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetRolesForUser indicates an expected call of GetRolesForUser. +func (mr *MockServiceMockRecorder) GetRolesForUser(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRolesForUser", reflect.TypeOf((*MockService)(nil).GetRolesForUser), arg0, arg1) +} + +// GetScheduler mocks base method. +func (m *MockService) GetScheduler(arg0 context.Context, arg1 uint) (*model.Scheduler, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetScheduler", arg0, arg1) + ret0, _ := ret[0].(*model.Scheduler) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetScheduler indicates an expected call of GetScheduler. +func (mr *MockServiceMockRecorder) GetScheduler(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetScheduler", reflect.TypeOf((*MockService)(nil).GetScheduler), arg0, arg1) +} + +// GetSchedulerCluster mocks base method. +func (m *MockService) GetSchedulerCluster(arg0 context.Context, arg1 uint) (*model.SchedulerCluster, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSchedulerCluster", arg0, arg1) + ret0, _ := ret[0].(*model.SchedulerCluster) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetSchedulerCluster indicates an expected call of GetSchedulerCluster. +func (mr *MockServiceMockRecorder) GetSchedulerCluster(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSchedulerCluster", reflect.TypeOf((*MockService)(nil).GetSchedulerCluster), arg0, arg1) +} + +// GetSchedulerClusters mocks base method. +func (m *MockService) GetSchedulerClusters(arg0 context.Context, arg1 types.GetSchedulerClustersQuery) ([]model.SchedulerCluster, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSchedulerClusters", arg0, arg1) + ret0, _ := ret[0].([]model.SchedulerCluster) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetSchedulerClusters indicates an expected call of GetSchedulerClusters. +func (mr *MockServiceMockRecorder) GetSchedulerClusters(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSchedulerClusters", reflect.TypeOf((*MockService)(nil).GetSchedulerClusters), arg0, arg1) +} + +// GetSchedulers mocks base method. +func (m *MockService) GetSchedulers(arg0 context.Context, arg1 types.GetSchedulersQuery) ([]model.Scheduler, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSchedulers", arg0, arg1) + ret0, _ := ret[0].([]model.Scheduler) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetSchedulers indicates an expected call of GetSchedulers. +func (mr *MockServiceMockRecorder) GetSchedulers(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSchedulers", reflect.TypeOf((*MockService)(nil).GetSchedulers), arg0, arg1) +} + +// GetSecurityGroup mocks base method. +func (m *MockService) GetSecurityGroup(arg0 context.Context, arg1 uint) (*model.SecurityGroup, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSecurityGroup", arg0, arg1) + ret0, _ := ret[0].(*model.SecurityGroup) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetSecurityGroup indicates an expected call of GetSecurityGroup. +func (mr *MockServiceMockRecorder) GetSecurityGroup(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSecurityGroup", reflect.TypeOf((*MockService)(nil).GetSecurityGroup), arg0, arg1) +} + +// GetSecurityGroups mocks base method. +func (m *MockService) GetSecurityGroups(arg0 context.Context, arg1 types.GetSecurityGroupsQuery) ([]model.SecurityGroup, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSecurityGroups", arg0, arg1) + ret0, _ := ret[0].([]model.SecurityGroup) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetSecurityGroups indicates an expected call of GetSecurityGroups. +func (mr *MockServiceMockRecorder) GetSecurityGroups(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSecurityGroups", reflect.TypeOf((*MockService)(nil).GetSecurityGroups), arg0, arg1) +} + +// GetSecurityRule mocks base method. +func (m *MockService) GetSecurityRule(arg0 context.Context, arg1 uint) (*model.SecurityRule, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSecurityRule", arg0, arg1) + ret0, _ := ret[0].(*model.SecurityRule) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetSecurityRule indicates an expected call of GetSecurityRule. +func (mr *MockServiceMockRecorder) GetSecurityRule(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSecurityRule", reflect.TypeOf((*MockService)(nil).GetSecurityRule), arg0, arg1) +} + +// GetSecurityRules mocks base method. +func (m *MockService) GetSecurityRules(arg0 context.Context, arg1 types.GetSecurityRulesQuery) ([]model.SecurityRule, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSecurityRules", arg0, arg1) + ret0, _ := ret[0].([]model.SecurityRule) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetSecurityRules indicates an expected call of GetSecurityRules. +func (mr *MockServiceMockRecorder) GetSecurityRules(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSecurityRules", reflect.TypeOf((*MockService)(nil).GetSecurityRules), arg0, arg1) +} + +// GetSeedPeer mocks base method. +func (m *MockService) GetSeedPeer(arg0 context.Context, arg1 uint) (*model.SeedPeer, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSeedPeer", arg0, arg1) + ret0, _ := ret[0].(*model.SeedPeer) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetSeedPeer indicates an expected call of GetSeedPeer. +func (mr *MockServiceMockRecorder) GetSeedPeer(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSeedPeer", reflect.TypeOf((*MockService)(nil).GetSeedPeer), arg0, arg1) +} + +// GetSeedPeerCluster mocks base method. +func (m *MockService) GetSeedPeerCluster(arg0 context.Context, arg1 uint) (*model.SeedPeerCluster, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSeedPeerCluster", arg0, arg1) + ret0, _ := ret[0].(*model.SeedPeerCluster) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetSeedPeerCluster indicates an expected call of GetSeedPeerCluster. +func (mr *MockServiceMockRecorder) GetSeedPeerCluster(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSeedPeerCluster", reflect.TypeOf((*MockService)(nil).GetSeedPeerCluster), arg0, arg1) +} + +// GetSeedPeerClusters mocks base method. +func (m *MockService) GetSeedPeerClusters(arg0 context.Context, arg1 types.GetSeedPeerClustersQuery) ([]model.SeedPeerCluster, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSeedPeerClusters", arg0, arg1) + ret0, _ := ret[0].([]model.SeedPeerCluster) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetSeedPeerClusters indicates an expected call of GetSeedPeerClusters. +func (mr *MockServiceMockRecorder) GetSeedPeerClusters(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSeedPeerClusters", reflect.TypeOf((*MockService)(nil).GetSeedPeerClusters), arg0, arg1) +} + +// GetSeedPeers mocks base method. +func (m *MockService) GetSeedPeers(arg0 context.Context, arg1 types.GetSeedPeersQuery) ([]model.SeedPeer, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSeedPeers", arg0, arg1) + ret0, _ := ret[0].([]model.SeedPeer) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetSeedPeers indicates an expected call of GetSeedPeers. +func (mr *MockServiceMockRecorder) GetSeedPeers(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSeedPeers", reflect.TypeOf((*MockService)(nil).GetSeedPeers), arg0, arg1) +} + +// GetUser mocks base method. +func (m *MockService) GetUser(arg0 context.Context, arg1 uint) (*model.User, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetUser", arg0, arg1) + ret0, _ := ret[0].(*model.User) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetUser indicates an expected call of GetUser. +func (mr *MockServiceMockRecorder) GetUser(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUser", reflect.TypeOf((*MockService)(nil).GetUser), arg0, arg1) +} + +// GetUsers mocks base method. +func (m *MockService) GetUsers(arg0 context.Context, arg1 types.GetUsersQuery) ([]model.User, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetUsers", arg0, arg1) + ret0, _ := ret[0].([]model.User) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetUsers indicates an expected call of GetUsers. +func (mr *MockServiceMockRecorder) GetUsers(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUsers", reflect.TypeOf((*MockService)(nil).GetUsers), arg0, arg1) +} + +// GetV1Preheat mocks base method. +func (m *MockService) GetV1Preheat(arg0 context.Context, arg1 string) (*types.GetV1PreheatResponse, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetV1Preheat", arg0, arg1) + ret0, _ := ret[0].(*types.GetV1PreheatResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetV1Preheat indicates an expected call of GetV1Preheat. +func (mr *MockServiceMockRecorder) GetV1Preheat(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetV1Preheat", reflect.TypeOf((*MockService)(nil).GetV1Preheat), arg0, arg1) +} + +// OauthSignin mocks base method. +func (m *MockService) OauthSignin(arg0 context.Context, arg1 string) (string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "OauthSignin", arg0, arg1) + ret0, _ := ret[0].(string) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// OauthSignin indicates an expected call of OauthSignin. +func (mr *MockServiceMockRecorder) OauthSignin(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OauthSignin", reflect.TypeOf((*MockService)(nil).OauthSignin), arg0, arg1) +} + +// OauthSigninCallback mocks base method. +func (m *MockService) OauthSigninCallback(arg0 context.Context, arg1, arg2 string) (*model.User, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "OauthSigninCallback", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.User) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// OauthSigninCallback indicates an expected call of OauthSigninCallback. +func (mr *MockServiceMockRecorder) OauthSigninCallback(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OauthSigninCallback", reflect.TypeOf((*MockService)(nil).OauthSigninCallback), arg0, arg1, arg2) +} + +// ResetPassword mocks base method. +func (m *MockService) ResetPassword(arg0 context.Context, arg1 uint, arg2 types.ResetPasswordRequest) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ResetPassword", arg0, arg1, arg2) + ret0, _ := ret[0].(error) + return ret0 +} + +// ResetPassword indicates an expected call of ResetPassword. +func (mr *MockServiceMockRecorder) ResetPassword(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetPassword", reflect.TypeOf((*MockService)(nil).ResetPassword), arg0, arg1, arg2) +} + +// SignIn mocks base method. +func (m *MockService) SignIn(arg0 context.Context, arg1 types.SignInRequest) (*model.User, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SignIn", arg0, arg1) + ret0, _ := ret[0].(*model.User) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// SignIn indicates an expected call of SignIn. +func (mr *MockServiceMockRecorder) SignIn(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SignIn", reflect.TypeOf((*MockService)(nil).SignIn), arg0, arg1) +} + +// SignUp mocks base method. +func (m *MockService) SignUp(arg0 context.Context, arg1 types.SignUpRequest) (*model.User, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SignUp", arg0, arg1) + ret0, _ := ret[0].(*model.User) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// SignUp indicates an expected call of SignUp. +func (mr *MockServiceMockRecorder) SignUp(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SignUp", reflect.TypeOf((*MockService)(nil).SignUp), arg0, arg1) +} + +// UpdateApplication mocks base method. +func (m *MockService) UpdateApplication(arg0 context.Context, arg1 uint, arg2 types.UpdateApplicationRequest) (*model.Application, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateApplication", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.Application) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateApplication indicates an expected call of UpdateApplication. +func (mr *MockServiceMockRecorder) UpdateApplication(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateApplication", reflect.TypeOf((*MockService)(nil).UpdateApplication), arg0, arg1, arg2) +} + +// UpdateConfig mocks base method. +func (m *MockService) UpdateConfig(arg0 context.Context, arg1 uint, arg2 types.UpdateConfigRequest) (*model.Config, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateConfig", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.Config) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateConfig indicates an expected call of UpdateConfig. +func (mr *MockServiceMockRecorder) UpdateConfig(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateConfig", reflect.TypeOf((*MockService)(nil).UpdateConfig), arg0, arg1, arg2) +} + +// UpdateJob mocks base method. +func (m *MockService) UpdateJob(arg0 context.Context, arg1 uint, arg2 types.UpdateJobRequest) (*model.Job, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateJob", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.Job) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateJob indicates an expected call of UpdateJob. +func (mr *MockServiceMockRecorder) UpdateJob(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateJob", reflect.TypeOf((*MockService)(nil).UpdateJob), arg0, arg1, arg2) +} + +// UpdateOauth mocks base method. +func (m *MockService) UpdateOauth(arg0 context.Context, arg1 uint, arg2 types.UpdateOauthRequest) (*model.Oauth, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateOauth", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.Oauth) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateOauth indicates an expected call of UpdateOauth. +func (mr *MockServiceMockRecorder) UpdateOauth(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOauth", reflect.TypeOf((*MockService)(nil).UpdateOauth), arg0, arg1, arg2) +} + +// UpdateScheduler mocks base method. +func (m *MockService) UpdateScheduler(arg0 context.Context, arg1 uint, arg2 types.UpdateSchedulerRequest) (*model.Scheduler, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateScheduler", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.Scheduler) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateScheduler indicates an expected call of UpdateScheduler. +func (mr *MockServiceMockRecorder) UpdateScheduler(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateScheduler", reflect.TypeOf((*MockService)(nil).UpdateScheduler), arg0, arg1, arg2) +} + +// UpdateSchedulerCluster mocks base method. +func (m *MockService) UpdateSchedulerCluster(arg0 context.Context, arg1 uint, arg2 types.UpdateSchedulerClusterRequest) (*model.SchedulerCluster, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateSchedulerCluster", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.SchedulerCluster) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateSchedulerCluster indicates an expected call of UpdateSchedulerCluster. +func (mr *MockServiceMockRecorder) UpdateSchedulerCluster(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSchedulerCluster", reflect.TypeOf((*MockService)(nil).UpdateSchedulerCluster), arg0, arg1, arg2) +} + +// UpdateSecurityGroup mocks base method. +func (m *MockService) UpdateSecurityGroup(arg0 context.Context, arg1 uint, arg2 types.UpdateSecurityGroupRequest) (*model.SecurityGroup, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateSecurityGroup", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.SecurityGroup) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateSecurityGroup indicates an expected call of UpdateSecurityGroup. +func (mr *MockServiceMockRecorder) UpdateSecurityGroup(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSecurityGroup", reflect.TypeOf((*MockService)(nil).UpdateSecurityGroup), arg0, arg1, arg2) +} + +// UpdateSecurityRule mocks base method. +func (m *MockService) UpdateSecurityRule(arg0 context.Context, arg1 uint, arg2 types.UpdateSecurityRuleRequest) (*model.SecurityRule, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateSecurityRule", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.SecurityRule) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateSecurityRule indicates an expected call of UpdateSecurityRule. +func (mr *MockServiceMockRecorder) UpdateSecurityRule(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSecurityRule", reflect.TypeOf((*MockService)(nil).UpdateSecurityRule), arg0, arg1, arg2) +} + +// UpdateSeedPeer mocks base method. +func (m *MockService) UpdateSeedPeer(arg0 context.Context, arg1 uint, arg2 types.UpdateSeedPeerRequest) (*model.SeedPeer, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateSeedPeer", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.SeedPeer) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateSeedPeer indicates an expected call of UpdateSeedPeer. +func (mr *MockServiceMockRecorder) UpdateSeedPeer(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSeedPeer", reflect.TypeOf((*MockService)(nil).UpdateSeedPeer), arg0, arg1, arg2) +} + +// UpdateSeedPeerCluster mocks base method. +func (m *MockService) UpdateSeedPeerCluster(arg0 context.Context, arg1 uint, arg2 types.UpdateSeedPeerClusterRequest) (*model.SeedPeerCluster, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateSeedPeerCluster", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.SeedPeerCluster) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateSeedPeerCluster indicates an expected call of UpdateSeedPeerCluster. +func (mr *MockServiceMockRecorder) UpdateSeedPeerCluster(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSeedPeerCluster", reflect.TypeOf((*MockService)(nil).UpdateSeedPeerCluster), arg0, arg1, arg2) +} + +// UpdateUser mocks base method. +func (m *MockService) UpdateUser(arg0 context.Context, arg1 uint, arg2 types.UpdateUserRequest) (*model.User, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateUser", arg0, arg1, arg2) + ret0, _ := ret[0].(*model.User) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateUser indicates an expected call of UpdateUser. +func (mr *MockServiceMockRecorder) UpdateUser(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUser", reflect.TypeOf((*MockService)(nil).UpdateUser), arg0, arg1, arg2) +} diff --git a/manager/service/service.go b/manager/service/service.go index f253dda3e58..6180ca381fd 100644 --- a/manager/service/service.go +++ b/manager/service/service.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/service_mock.go -source service.go -package mocks + package service import ( diff --git a/pkg/container/list/mocks/list_mock.go b/pkg/container/list/mocks/list_mock.go deleted file mode 100644 index 5453e83dfa4..00000000000 --- a/pkg/container/list/mocks/list_mock.go +++ /dev/null @@ -1,48 +0,0 @@ -// Code generated by MockGen. DO NOT EDIT. -// Source: d7y.io/dragonfly/v2/pkg/container/list (interfaces: Item) - -// Package mocks is a generated GoMock package. -package mocks - -import ( - reflect "reflect" - - gomock "github.com/golang/mock/gomock" -) - -// MockItem is a mock of Item interface. -type MockItem struct { - ctrl *gomock.Controller - recorder *MockItemMockRecorder -} - -// MockItemMockRecorder is the mock recorder for MockItem. -type MockItemMockRecorder struct { - mock *MockItem -} - -// NewMockItem creates a new mock instance. -func NewMockItem(ctrl *gomock.Controller) *MockItem { - mock := &MockItem{ctrl: ctrl} - mock.recorder = &MockItemMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockItem) EXPECT() *MockItemMockRecorder { - return m.recorder -} - -// SortedValue mocks base method. -func (m *MockItem) SortedValue() int { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SortedValue") - ret0, _ := ret[0].(int) - return ret0 -} - -// SortedValue indicates an expected call of SortedValue. -func (mr *MockItemMockRecorder) SortedValue() *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SortedValue", reflect.TypeOf((*MockItem)(nil).SortedValue)) -} diff --git a/pkg/container/list/sorted_list.go b/pkg/container/list/sorted_list.go index cb108c8f4a4..8b0cb0bb707 100644 --- a/pkg/container/list/sorted_list.go +++ b/pkg/container/list/sorted_list.go @@ -14,7 +14,7 @@ * limitations under the License. */ -//go:generate mockgen -destination ./mocks/list_mock.go -package mocks d7y.io/dragonfly/v2/pkg/container/list Item +//go:generate mockgen -destination sorted_list_mock.go -source sorted_list.go -package list package list diff --git a/pkg/container/list/sorted_list_mock.go b/pkg/container/list/sorted_list_mock.go new file mode 100644 index 00000000000..5eb6fc859d2 --- /dev/null +++ b/pkg/container/list/sorted_list_mock.go @@ -0,0 +1,147 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: sorted_list.go + +// Package list is a generated GoMock package. +package list + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockItem is a mock of Item interface. +type MockItem struct { + ctrl *gomock.Controller + recorder *MockItemMockRecorder +} + +// MockItemMockRecorder is the mock recorder for MockItem. +type MockItemMockRecorder struct { + mock *MockItem +} + +// NewMockItem creates a new mock instance. +func NewMockItem(ctrl *gomock.Controller) *MockItem { + mock := &MockItem{ctrl: ctrl} + mock.recorder = &MockItemMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockItem) EXPECT() *MockItemMockRecorder { + return m.recorder +} + +// SortedValue mocks base method. +func (m *MockItem) SortedValue() int { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SortedValue") + ret0, _ := ret[0].(int) + return ret0 +} + +// SortedValue indicates an expected call of SortedValue. +func (mr *MockItemMockRecorder) SortedValue() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SortedValue", reflect.TypeOf((*MockItem)(nil).SortedValue)) +} + +// MockSortedList is a mock of SortedList interface. +type MockSortedList struct { + ctrl *gomock.Controller + recorder *MockSortedListMockRecorder +} + +// MockSortedListMockRecorder is the mock recorder for MockSortedList. +type MockSortedListMockRecorder struct { + mock *MockSortedList +} + +// NewMockSortedList creates a new mock instance. +func NewMockSortedList(ctrl *gomock.Controller) *MockSortedList { + mock := &MockSortedList{ctrl: ctrl} + mock.recorder = &MockSortedListMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSortedList) EXPECT() *MockSortedListMockRecorder { + return m.recorder +} + +// Contains mocks base method. +func (m *MockSortedList) Contains(arg0 Item) bool { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Contains", arg0) + ret0, _ := ret[0].(bool) + return ret0 +} + +// Contains indicates an expected call of Contains. +func (mr *MockSortedListMockRecorder) Contains(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Contains", reflect.TypeOf((*MockSortedList)(nil).Contains), arg0) +} + +// Insert mocks base method. +func (m *MockSortedList) Insert(arg0 Item) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Insert", arg0) +} + +// Insert indicates an expected call of Insert. +func (mr *MockSortedListMockRecorder) Insert(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Insert", reflect.TypeOf((*MockSortedList)(nil).Insert), arg0) +} + +// Len mocks base method. +func (m *MockSortedList) Len() int { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Len") + ret0, _ := ret[0].(int) + return ret0 +} + +// Len indicates an expected call of Len. +func (mr *MockSortedListMockRecorder) Len() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Len", reflect.TypeOf((*MockSortedList)(nil).Len)) +} + +// Range mocks base method. +func (m *MockSortedList) Range(arg0 func(Item) bool) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Range", arg0) +} + +// Range indicates an expected call of Range. +func (mr *MockSortedListMockRecorder) Range(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Range", reflect.TypeOf((*MockSortedList)(nil).Range), arg0) +} + +// Remove mocks base method. +func (m *MockSortedList) Remove(arg0 Item) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Remove", arg0) +} + +// Remove indicates an expected call of Remove. +func (mr *MockSortedListMockRecorder) Remove(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Remove", reflect.TypeOf((*MockSortedList)(nil).Remove), arg0) +} + +// ReverseRange mocks base method. +func (m *MockSortedList) ReverseRange(fn func(Item) bool) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "ReverseRange", fn) +} + +// ReverseRange indicates an expected call of ReverseRange. +func (mr *MockSortedListMockRecorder) ReverseRange(fn interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReverseRange", reflect.TypeOf((*MockSortedList)(nil).ReverseRange), fn) +} diff --git a/pkg/container/list/sorted_list_test.go b/pkg/container/list/sorted_list_test.go index 9d411626bca..56bc2998879 100644 --- a/pkg/container/list/sorted_list_test.go +++ b/pkg/container/list/sorted_list_test.go @@ -24,8 +24,6 @@ import ( "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" - - "d7y.io/dragonfly/v2/pkg/container/list/mocks" ) const N = 1000 @@ -33,12 +31,12 @@ const N = 1000 func TestSortedListInsert(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, l SortedList, items ...Item) }{ { name: "insert values", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, l SortedList, items ...Item) { assert := assert.New(t) l.Insert(items[0]) @@ -48,7 +46,7 @@ func TestSortedListInsert(t *testing.T) { }, { name: "insert multi value", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -65,7 +63,7 @@ func TestSortedListInsert(t *testing.T) { }, { name: "insert same values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(2), ) @@ -85,7 +83,7 @@ func TestSortedListInsert(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItems := []*mocks.MockItem{mocks.NewMockItem(ctl), mocks.NewMockItem(ctl)} + mockItems := []*MockItem{NewMockItem(ctl), NewMockItem(ctl)} tc.mock(mockItems[0].EXPECT(), mockItems[1].EXPECT()) tc.expect(t, NewSortedList(), mockItems[0], mockItems[1]) }) @@ -97,7 +95,7 @@ func TestSortedListInsert_Concurrent(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return rand.Intn(N) }).AnyTimes() l := NewSortedList() @@ -126,12 +124,12 @@ func TestSortedListInsert_Concurrent(t *testing.T) { func TestSortedListRemove(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, l SortedList, items ...Item) }{ { name: "remove values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -154,7 +152,7 @@ func TestSortedListRemove(t *testing.T) { }, { name: "remove value dost not exits", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, l SortedList, items ...Item) { assert := assert.New(t) l.Insert(items[0]) @@ -172,7 +170,7 @@ func TestSortedListRemove(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItems := []*mocks.MockItem{mocks.NewMockItem(ctl), mocks.NewMockItem(ctl)} + mockItems := []*MockItem{NewMockItem(ctl), NewMockItem(ctl)} tc.mock(mockItems[0].EXPECT(), mockItems[1].EXPECT()) tc.expect(t, NewSortedList(), mockItems[0], mockItems[1]) }) @@ -184,7 +182,7 @@ func TestSortedListRemove_Concurrent(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return rand.Intn(N) }).AnyTimes() l := NewSortedList() @@ -212,12 +210,12 @@ func TestSortedListRemove_Concurrent(t *testing.T) { func TestSortedListContains(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, l SortedList, items ...Item) }{ { name: "contains values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -233,7 +231,7 @@ func TestSortedListContains(t *testing.T) { }, { name: "contains value dost not exits", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, l SortedList, items ...Item) { assert := assert.New(t) l.Insert(items[0]) @@ -247,7 +245,7 @@ func TestSortedListContains(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItems := []*mocks.MockItem{mocks.NewMockItem(ctl), mocks.NewMockItem(ctl)} + mockItems := []*MockItem{NewMockItem(ctl), NewMockItem(ctl)} tc.mock(mockItems[0].EXPECT(), mockItems[1].EXPECT()) tc.expect(t, NewSortedList(), mockItems[0], mockItems[1]) }) @@ -259,7 +257,7 @@ func TestSortedListContains_Concurrent(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return rand.Intn(N) }).AnyTimes() l := NewSortedList() @@ -284,12 +282,12 @@ func TestSortedListContains_Concurrent(t *testing.T) { func TestSortedListLen(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, l SortedList, items ...Item) }{ { name: "get length", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -304,7 +302,7 @@ func TestSortedListLen(t *testing.T) { }, { name: "get empty list length", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, l SortedList, items ...Item) { assert := assert.New(t) assert.Equal(l.Len(), 0) @@ -317,7 +315,7 @@ func TestSortedListLen(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItems := []*mocks.MockItem{mocks.NewMockItem(ctl), mocks.NewMockItem(ctl)} + mockItems := []*MockItem{NewMockItem(ctl), NewMockItem(ctl)} tc.mock(mockItems[0].EXPECT(), mockItems[1].EXPECT()) tc.expect(t, NewSortedList(), mockItems[0], mockItems[1]) }) @@ -329,7 +327,7 @@ func TestSortedListLen_Concurrent(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return rand.Intn(N) }).AnyTimes() l := NewSortedList() @@ -356,12 +354,12 @@ func TestSortedListLen_Concurrent(t *testing.T) { func TestSortedListRange(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, l SortedList, items ...Item) }{ { name: "range values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -383,7 +381,7 @@ func TestSortedListRange(t *testing.T) { }, { name: "range multi values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { for i := range m { m[i].SortedValue().Return(i).AnyTimes() } @@ -405,7 +403,7 @@ func TestSortedListRange(t *testing.T) { }, { name: "range stoped", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -425,7 +423,7 @@ func TestSortedListRange(t *testing.T) { }, { name: "range same values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).AnyTimes(), ) @@ -448,7 +446,7 @@ func TestSortedListRange(t *testing.T) { }, { name: "range empty list", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { }, expect: func(t *testing.T, l SortedList, items ...Item) { assert := assert.New(t) @@ -468,9 +466,9 @@ func TestSortedListRange(t *testing.T) { defer ctl.Finish() var mockItems []Item - var mockItemRecorders []*mocks.MockItemMockRecorder + var mockItemRecorders []*MockItemMockRecorder for i := 0; i < 10; i++ { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItemRecorders = append(mockItemRecorders, mockItem.EXPECT()) mockItems = append(mockItems, mockItem) } @@ -486,7 +484,7 @@ func TestSortedListRange_Concurrent(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return rand.Intn(N) }).AnyTimes() l := NewSortedList() @@ -519,12 +517,12 @@ func TestSortedListRange_Concurrent(t *testing.T) { func TestSortedListReverseRange(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, l SortedList, items ...Item) }{ { name: "reverse range values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -546,7 +544,7 @@ func TestSortedListReverseRange(t *testing.T) { }, { name: "reverse range multi values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { for i := range m { m[i].SortedValue().Return(i).AnyTimes() } @@ -568,7 +566,7 @@ func TestSortedListReverseRange(t *testing.T) { }, { name: "reverse range stoped", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -588,7 +586,7 @@ func TestSortedListReverseRange(t *testing.T) { }, { name: "reverse range same values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).AnyTimes(), ) @@ -611,7 +609,7 @@ func TestSortedListReverseRange(t *testing.T) { }, { name: "reverse range empty list", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { }, expect: func(t *testing.T, l SortedList, items ...Item) { assert := assert.New(t) @@ -631,9 +629,9 @@ func TestSortedListReverseRange(t *testing.T) { defer ctl.Finish() var mockItems []Item - var mockItemRecorders []*mocks.MockItemMockRecorder + var mockItemRecorders []*MockItemMockRecorder for i := 0; i < 10; i++ { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItemRecorders = append(mockItemRecorders, mockItem.EXPECT()) mockItems = append(mockItems, mockItem) } @@ -649,7 +647,7 @@ func TestSortedListReverseRange_Concurrent(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return rand.Intn(N) }).AnyTimes() l := NewSortedList() diff --git a/pkg/container/list/sorted_unique_list.go b/pkg/container/list/sorted_unique_list.go index 5e0232423b6..32ce6d9b716 100644 --- a/pkg/container/list/sorted_unique_list.go +++ b/pkg/container/list/sorted_unique_list.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination sorted_unique_list_mock.go -source sorted_unique_list.go -package list + package list import ( diff --git a/pkg/container/list/sorted_unique_list_mock.go b/pkg/container/list/sorted_unique_list_mock.go new file mode 100644 index 00000000000..58e96a7aac7 --- /dev/null +++ b/pkg/container/list/sorted_unique_list_mock.go @@ -0,0 +1,110 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: sorted_unique_list.go + +// Package list is a generated GoMock package. +package list + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockSortedUniqueList is a mock of SortedUniqueList interface. +type MockSortedUniqueList struct { + ctrl *gomock.Controller + recorder *MockSortedUniqueListMockRecorder +} + +// MockSortedUniqueListMockRecorder is the mock recorder for MockSortedUniqueList. +type MockSortedUniqueListMockRecorder struct { + mock *MockSortedUniqueList +} + +// NewMockSortedUniqueList creates a new mock instance. +func NewMockSortedUniqueList(ctrl *gomock.Controller) *MockSortedUniqueList { + mock := &MockSortedUniqueList{ctrl: ctrl} + mock.recorder = &MockSortedUniqueListMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSortedUniqueList) EXPECT() *MockSortedUniqueListMockRecorder { + return m.recorder +} + +// Contains mocks base method. +func (m *MockSortedUniqueList) Contains(arg0 Item) bool { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Contains", arg0) + ret0, _ := ret[0].(bool) + return ret0 +} + +// Contains indicates an expected call of Contains. +func (mr *MockSortedUniqueListMockRecorder) Contains(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Contains", reflect.TypeOf((*MockSortedUniqueList)(nil).Contains), arg0) +} + +// Insert mocks base method. +func (m *MockSortedUniqueList) Insert(arg0 Item) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Insert", arg0) +} + +// Insert indicates an expected call of Insert. +func (mr *MockSortedUniqueListMockRecorder) Insert(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Insert", reflect.TypeOf((*MockSortedUniqueList)(nil).Insert), arg0) +} + +// Len mocks base method. +func (m *MockSortedUniqueList) Len() int { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Len") + ret0, _ := ret[0].(int) + return ret0 +} + +// Len indicates an expected call of Len. +func (mr *MockSortedUniqueListMockRecorder) Len() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Len", reflect.TypeOf((*MockSortedUniqueList)(nil).Len)) +} + +// Range mocks base method. +func (m *MockSortedUniqueList) Range(arg0 func(Item) bool) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Range", arg0) +} + +// Range indicates an expected call of Range. +func (mr *MockSortedUniqueListMockRecorder) Range(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Range", reflect.TypeOf((*MockSortedUniqueList)(nil).Range), arg0) +} + +// Remove mocks base method. +func (m *MockSortedUniqueList) Remove(arg0 Item) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Remove", arg0) +} + +// Remove indicates an expected call of Remove. +func (mr *MockSortedUniqueListMockRecorder) Remove(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Remove", reflect.TypeOf((*MockSortedUniqueList)(nil).Remove), arg0) +} + +// ReverseRange mocks base method. +func (m *MockSortedUniqueList) ReverseRange(fn func(Item) bool) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "ReverseRange", fn) +} + +// ReverseRange indicates an expected call of ReverseRange. +func (mr *MockSortedUniqueListMockRecorder) ReverseRange(fn interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReverseRange", reflect.TypeOf((*MockSortedUniqueList)(nil).ReverseRange), fn) +} diff --git a/pkg/container/list/sorted_unique_list_test.go b/pkg/container/list/sorted_unique_list_test.go index 3622d5ca3ce..4c5f50ecd9e 100644 --- a/pkg/container/list/sorted_unique_list_test.go +++ b/pkg/container/list/sorted_unique_list_test.go @@ -24,19 +24,17 @@ import ( "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" - - "d7y.io/dragonfly/v2/pkg/container/list/mocks" ) func TestSortedUniqueListInsert(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, ul SortedUniqueList, items ...Item) }{ { name: "insert values", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, ul SortedUniqueList, items ...Item) { assert := assert.New(t) ul.Insert(items[0]) @@ -46,7 +44,7 @@ func TestSortedUniqueListInsert(t *testing.T) { }, { name: "insert multi values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -63,7 +61,7 @@ func TestSortedUniqueListInsert(t *testing.T) { }, { name: "insert same values", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, ul SortedUniqueList, items ...Item) { assert := assert.New(t) ul.Insert(items[0]) @@ -79,7 +77,7 @@ func TestSortedUniqueListInsert(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItems := []*mocks.MockItem{mocks.NewMockItem(ctl), mocks.NewMockItem(ctl)} + mockItems := []*MockItem{NewMockItem(ctl), NewMockItem(ctl)} tc.mock(mockItems[0].EXPECT(), mockItems[1].EXPECT()) tc.expect(t, NewSortedUniqueList(), mockItems[0], mockItems[1]) }) @@ -97,7 +95,7 @@ func TestSortedUniqueListInsert_Concurrent(t *testing.T) { var mockItems []Item for _, v := range nums { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return v }).AnyTimes() mockItems = append(mockItems, mockItem) } @@ -125,12 +123,12 @@ func TestSortedUniqueListInsert_Concurrent(t *testing.T) { func TestSortedUniqueListRemove(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, ul SortedUniqueList, items ...Item) }{ { name: "remove values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -153,7 +151,7 @@ func TestSortedUniqueListRemove(t *testing.T) { }, { name: "remove value dost not exits", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, ul SortedUniqueList, items ...Item) { assert := assert.New(t) ul.Insert(items[0]) @@ -171,7 +169,7 @@ func TestSortedUniqueListRemove(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItems := []*mocks.MockItem{mocks.NewMockItem(ctl), mocks.NewMockItem(ctl)} + mockItems := []*MockItem{NewMockItem(ctl), NewMockItem(ctl)} tc.mock(mockItems[0].EXPECT(), mockItems[1].EXPECT()) tc.expect(t, NewSortedUniqueList(), mockItems[0], mockItems[1]) }) @@ -189,7 +187,7 @@ func TestSortedUniqueListRemove_Concurrent(t *testing.T) { var mockItems []Item for _, v := range nums { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return v }).AnyTimes() mockItems = append(mockItems, mockItem) ul.Insert(mockItem) @@ -213,12 +211,12 @@ func TestSortedUniqueListRemove_Concurrent(t *testing.T) { func TestSortedUniqueListContains(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, ul SortedUniqueList, items ...Item) }{ { name: "contains values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -234,7 +232,7 @@ func TestSortedUniqueListContains(t *testing.T) { }, { name: "contains value dost not exits", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, ul SortedUniqueList, items ...Item) { assert := assert.New(t) ul.Insert(items[0]) @@ -248,7 +246,7 @@ func TestSortedUniqueListContains(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItems := []*mocks.MockItem{mocks.NewMockItem(ctl), mocks.NewMockItem(ctl)} + mockItems := []*MockItem{NewMockItem(ctl), NewMockItem(ctl)} tc.mock(mockItems[0].EXPECT(), mockItems[1].EXPECT()) tc.expect(t, NewSortedUniqueList(), mockItems[0], mockItems[1]) }) @@ -266,7 +264,7 @@ func TestSortedUniqueListContains_Concurrent(t *testing.T) { var mockItems []Item for _, v := range nums { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return v }).AnyTimes() mockItems = append(mockItems, mockItem) ul.Insert(mockItem) @@ -288,12 +286,12 @@ func TestSortedUniqueListContains_Concurrent(t *testing.T) { func TestSortedUniqueListLen(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, ul SortedUniqueList, items ...Item) }{ { name: "get length", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -308,7 +306,7 @@ func TestSortedUniqueListLen(t *testing.T) { }, { name: "get empty list length", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, ul SortedUniqueList, items ...Item) { assert := assert.New(t) assert.Equal(ul.Len(), 0) @@ -316,7 +314,7 @@ func TestSortedUniqueListLen(t *testing.T) { }, { name: "get same values length", - mock: func(m ...*mocks.MockItemMockRecorder) {}, + mock: func(m ...*MockItemMockRecorder) {}, expect: func(t *testing.T, ul SortedUniqueList, items ...Item) { assert := assert.New(t) ul.Insert(items[0]) @@ -331,7 +329,7 @@ func TestSortedUniqueListLen(t *testing.T) { ctl := gomock.NewController(t) defer ctl.Finish() - mockItems := []*mocks.MockItem{mocks.NewMockItem(ctl), mocks.NewMockItem(ctl)} + mockItems := []*MockItem{NewMockItem(ctl), NewMockItem(ctl)} tc.mock(mockItems[0].EXPECT(), mockItems[1].EXPECT()) tc.expect(t, NewSortedUniqueList(), mockItems[0], mockItems[1]) }) @@ -349,7 +347,7 @@ func TestSortedUniqueListLen_Concurrent(t *testing.T) { var mockItems []Item for _, v := range nums { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return v }).AnyTimes() mockItems = append(mockItems, mockItem) } @@ -376,12 +374,12 @@ func TestSortedUniqueListLen_Concurrent(t *testing.T) { func TestSortedUniqueListRange(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, ul SortedUniqueList, items ...Item) }{ { name: "range values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -403,7 +401,7 @@ func TestSortedUniqueListRange(t *testing.T) { }, { name: "range multi values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { for i := range m { m[i].SortedValue().Return(i).AnyTimes() } @@ -426,7 +424,7 @@ func TestSortedUniqueListRange(t *testing.T) { }, { name: "range stoped", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -446,7 +444,7 @@ func TestSortedUniqueListRange(t *testing.T) { }, { name: "range same values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).AnyTimes(), ) @@ -469,7 +467,7 @@ func TestSortedUniqueListRange(t *testing.T) { }, { name: "range empty list", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { }, expect: func(t *testing.T, ul SortedUniqueList, items ...Item) { assert := assert.New(t) @@ -489,9 +487,9 @@ func TestSortedUniqueListRange(t *testing.T) { defer ctl.Finish() var mockItems []Item - var mockItemRecorders []*mocks.MockItemMockRecorder + var mockItemRecorders []*MockItemMockRecorder for i := 0; i < 10; i++ { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItemRecorders = append(mockItemRecorders, mockItem.EXPECT()) mockItems = append(mockItems, mockItem) } @@ -513,7 +511,7 @@ func TestSortedUniqueListRange_Concurrent(t *testing.T) { var mockItems []Item for _, v := range nums { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return v }).AnyTimes() mockItems = append(mockItems, mockItem) } @@ -547,12 +545,12 @@ func TestSortedUniqueListRange_Concurrent(t *testing.T) { func TestSortedUniqueListReverseRange(t *testing.T) { tests := []struct { name string - mock func(m ...*mocks.MockItemMockRecorder) + mock func(m ...*MockItemMockRecorder) expect func(t *testing.T, ul SortedUniqueList, items ...Item) }{ { name: "reverse range values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -574,7 +572,7 @@ func TestSortedUniqueListReverseRange(t *testing.T) { }, { name: "reverse range multi values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { for i := range m { m[i].SortedValue().Return(i).AnyTimes() } @@ -597,7 +595,7 @@ func TestSortedUniqueListReverseRange(t *testing.T) { }, { name: "reverse range stoped", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).Times(1), m[1].SortedValue().Return(1).Times(1), @@ -617,7 +615,7 @@ func TestSortedUniqueListReverseRange(t *testing.T) { }, { name: "reverse range same values", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { gomock.InOrder( m[0].SortedValue().Return(0).AnyTimes(), ) @@ -640,7 +638,7 @@ func TestSortedUniqueListReverseRange(t *testing.T) { }, { name: "reverse range empty list", - mock: func(m ...*mocks.MockItemMockRecorder) { + mock: func(m ...*MockItemMockRecorder) { }, expect: func(t *testing.T, ul SortedUniqueList, items ...Item) { assert := assert.New(t) @@ -660,9 +658,9 @@ func TestSortedUniqueListReverseRange(t *testing.T) { defer ctl.Finish() var mockItems []Item - var mockItemRecorders []*mocks.MockItemMockRecorder + var mockItemRecorders []*MockItemMockRecorder for i := 0; i < 10; i++ { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItemRecorders = append(mockItemRecorders, mockItem.EXPECT()) mockItems = append(mockItems, mockItem) } @@ -684,7 +682,7 @@ func TestSortedUniqueListReverseRange_Concurrent(t *testing.T) { var mockItems []Item for _, v := range nums { - mockItem := mocks.NewMockItem(ctl) + mockItem := NewMockItem(ctl) mockItem.EXPECT().SortedValue().DoAndReturn(func() int { return v }).AnyTimes() mockItems = append(mockItems, mockItem) } diff --git a/pkg/container/set/mocks/safe_set_mock.go b/pkg/container/set/mocks/safe_set_mock.go new file mode 100644 index 00000000000..dfa64823cd4 --- /dev/null +++ b/pkg/container/set/mocks/safe_set_mock.go @@ -0,0 +1,130 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: safe_set.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockSafeSet is a mock of SafeSet interface. +type MockSafeSet struct { + ctrl *gomock.Controller + recorder *MockSafeSetMockRecorder +} + +// MockSafeSetMockRecorder is the mock recorder for MockSafeSet. +type MockSafeSetMockRecorder struct { + mock *MockSafeSet +} + +// NewMockSafeSet creates a new mock instance. +func NewMockSafeSet(ctrl *gomock.Controller) *MockSafeSet { + mock := &MockSafeSet{ctrl: ctrl} + mock.recorder = &MockSafeSetMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSafeSet) EXPECT() *MockSafeSetMockRecorder { + return m.recorder +} + +// Add mocks base method. +func (m *MockSafeSet) Add(arg0 interface{}) bool { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Add", arg0) + ret0, _ := ret[0].(bool) + return ret0 +} + +// Add indicates an expected call of Add. +func (mr *MockSafeSetMockRecorder) Add(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Add", reflect.TypeOf((*MockSafeSet)(nil).Add), arg0) +} + +// Clear mocks base method. +func (m *MockSafeSet) Clear() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Clear") +} + +// Clear indicates an expected call of Clear. +func (mr *MockSafeSetMockRecorder) Clear() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Clear", reflect.TypeOf((*MockSafeSet)(nil).Clear)) +} + +// Contains mocks base method. +func (m *MockSafeSet) Contains(arg0 ...interface{}) bool { + m.ctrl.T.Helper() + varargs := []interface{}{} + for _, a := range arg0 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "Contains", varargs...) + ret0, _ := ret[0].(bool) + return ret0 +} + +// Contains indicates an expected call of Contains. +func (mr *MockSafeSetMockRecorder) Contains(arg0 ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Contains", reflect.TypeOf((*MockSafeSet)(nil).Contains), arg0...) +} + +// Delete mocks base method. +func (m *MockSafeSet) Delete(arg0 interface{}) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Delete", arg0) +} + +// Delete indicates an expected call of Delete. +func (mr *MockSafeSetMockRecorder) Delete(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockSafeSet)(nil).Delete), arg0) +} + +// Len mocks base method. +func (m *MockSafeSet) Len() uint { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Len") + ret0, _ := ret[0].(uint) + return ret0 +} + +// Len indicates an expected call of Len. +func (mr *MockSafeSetMockRecorder) Len() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Len", reflect.TypeOf((*MockSafeSet)(nil).Len)) +} + +// Range mocks base method. +func (m *MockSafeSet) Range(arg0 func(interface{}) bool) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Range", arg0) +} + +// Range indicates an expected call of Range. +func (mr *MockSafeSetMockRecorder) Range(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Range", reflect.TypeOf((*MockSafeSet)(nil).Range), arg0) +} + +// Values mocks base method. +func (m *MockSafeSet) Values() []interface{} { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Values") + ret0, _ := ret[0].([]interface{}) + return ret0 +} + +// Values indicates an expected call of Values. +func (mr *MockSafeSetMockRecorder) Values() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Values", reflect.TypeOf((*MockSafeSet)(nil).Values)) +} diff --git a/pkg/container/set/mocks/set_mock.go b/pkg/container/set/mocks/set_mock.go new file mode 100644 index 00000000000..5bea96f291c --- /dev/null +++ b/pkg/container/set/mocks/set_mock.go @@ -0,0 +1,130 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: set.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockSet is a mock of Set interface. +type MockSet struct { + ctrl *gomock.Controller + recorder *MockSetMockRecorder +} + +// MockSetMockRecorder is the mock recorder for MockSet. +type MockSetMockRecorder struct { + mock *MockSet +} + +// NewMockSet creates a new mock instance. +func NewMockSet(ctrl *gomock.Controller) *MockSet { + mock := &MockSet{ctrl: ctrl} + mock.recorder = &MockSetMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSet) EXPECT() *MockSetMockRecorder { + return m.recorder +} + +// Add mocks base method. +func (m *MockSet) Add(arg0 interface{}) bool { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Add", arg0) + ret0, _ := ret[0].(bool) + return ret0 +} + +// Add indicates an expected call of Add. +func (mr *MockSetMockRecorder) Add(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Add", reflect.TypeOf((*MockSet)(nil).Add), arg0) +} + +// Clear mocks base method. +func (m *MockSet) Clear() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Clear") +} + +// Clear indicates an expected call of Clear. +func (mr *MockSetMockRecorder) Clear() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Clear", reflect.TypeOf((*MockSet)(nil).Clear)) +} + +// Contains mocks base method. +func (m *MockSet) Contains(arg0 ...interface{}) bool { + m.ctrl.T.Helper() + varargs := []interface{}{} + for _, a := range arg0 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "Contains", varargs...) + ret0, _ := ret[0].(bool) + return ret0 +} + +// Contains indicates an expected call of Contains. +func (mr *MockSetMockRecorder) Contains(arg0 ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Contains", reflect.TypeOf((*MockSet)(nil).Contains), arg0...) +} + +// Delete mocks base method. +func (m *MockSet) Delete(arg0 interface{}) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Delete", arg0) +} + +// Delete indicates an expected call of Delete. +func (mr *MockSetMockRecorder) Delete(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockSet)(nil).Delete), arg0) +} + +// Len mocks base method. +func (m *MockSet) Len() uint { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Len") + ret0, _ := ret[0].(uint) + return ret0 +} + +// Len indicates an expected call of Len. +func (mr *MockSetMockRecorder) Len() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Len", reflect.TypeOf((*MockSet)(nil).Len)) +} + +// Range mocks base method. +func (m *MockSet) Range(arg0 func(interface{}) bool) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Range", arg0) +} + +// Range indicates an expected call of Range. +func (mr *MockSetMockRecorder) Range(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Range", reflect.TypeOf((*MockSet)(nil).Range), arg0) +} + +// Values mocks base method. +func (m *MockSet) Values() []interface{} { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Values") + ret0, _ := ret[0].([]interface{}) + return ret0 +} + +// Values indicates an expected call of Values. +func (mr *MockSetMockRecorder) Values() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Values", reflect.TypeOf((*MockSet)(nil).Values)) +} diff --git a/pkg/container/set/safe_set.go b/pkg/container/set/safe_set.go index 20d21debd7f..9b1a5316371 100644 --- a/pkg/container/set/safe_set.go +++ b/pkg/container/set/safe_set.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/safe_set_mock.go -source safe_set.go -package mocks + package set import ( diff --git a/pkg/container/set/set.go b/pkg/container/set/set.go index c3f2f65c033..b9f5c9b0828 100644 --- a/pkg/container/set/set.go +++ b/pkg/container/set/set.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/set_mock.go -source set.go -package mocks + package set type Set interface { diff --git a/pkg/dfpath/dfpath.go b/pkg/dfpath/dfpath.go index fc763ecb3ae..a5548529952 100644 --- a/pkg/dfpath/dfpath.go +++ b/pkg/dfpath/dfpath.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/dfpath_mock.go -source dfpath.go -package mocks + package dfpath import ( diff --git a/pkg/dfpath/mocks/dfpath_mock.go b/pkg/dfpath/mocks/dfpath_mock.go index 9ed3410b5ab..ceaafc4d8c7 100644 --- a/pkg/dfpath/mocks/dfpath_mock.go +++ b/pkg/dfpath/mocks/dfpath_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/dfpath/dfpath.go +// Source: dfpath.go // Package mocks is a generated GoMock package. package mocks diff --git a/pkg/gc/gc.go b/pkg/gc/gc.go index 1d195da43d1..5130f6b54ff 100644 --- a/pkg/gc/gc.go +++ b/pkg/gc/gc.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination gc_mock.go -source gc.go -package gc + package gc import ( @@ -24,42 +26,42 @@ import ( "github.com/sirupsen/logrus" ) -// GC is the interface used for release resource +// GC is the interface used for release resource. type GC interface { - // Add adds GC task + // Add adds GC task. Add(Task) error - // Run GC task + // Run GC task. Run(string) error - // Run all registered GC tasks + // Run all registered GC tasks. RunAll() - // Serve running the GC task + // Serve running the GC task. Serve() - // Stop running the GC task + // Stop running the GC task. Stop() } -// GC provides task release function +// GC provides task release function. type gc struct { tasks *sync.Map logger Logger done chan bool } -// Option is a functional option for configuring the GC +// Option is a functional option for configuring the GC. type Option func(g *gc) -// WithLogger set the logger for GC +// WithLogger set the logger for GC. func WithLogger(logger Logger) Option { return func(g *gc) { g.logger = logger } } -// New returns a new GC instence +// New returns a new GC instence. func New(options ...Option) GC { g := &gc{ tasks: &sync.Map{}, diff --git a/pkg/gc/gc_mock.go b/pkg/gc/gc_mock.go index 8661a50d44d..dfa45843e1b 100644 --- a/pkg/gc/gc_mock.go +++ b/pkg/gc/gc_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/gc/gc.go +// Source: gc.go // Package gc is a generated GoMock package. package gc diff --git a/pkg/gc/logger.go b/pkg/gc/logger.go index 41f07905af4..935b1bc2505 100644 --- a/pkg/gc/logger.go +++ b/pkg/gc/logger.go @@ -14,12 +14,14 @@ * limitations under the License. */ +//go:generate mockgen -destination logger_mock.go -source logger.go -package gc + package gc -// Logger is the interface used in GC for logging +// Logger is the interface used in GC for logging. type Logger interface { - // Infof logs routine messages for GC + // Infof logs routine messages for GC. Infof(template string, args ...interface{}) - // Error logs error messages for GC + // Error logs error messages for GC. Errorf(template string, args ...interface{}) } diff --git a/pkg/gc/logger_mock.go b/pkg/gc/logger_mock.go index 3ca8708f6ab..4a80a5de5cc 100644 --- a/pkg/gc/logger_mock.go +++ b/pkg/gc/logger_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/gc/logger.go +// Source: logger.go // Package gc is a generated GoMock package. package gc diff --git a/pkg/gc/runner_mock.go b/pkg/gc/runner_mock.go index 11fc07c5728..6f69f50846b 100644 --- a/pkg/gc/runner_mock.go +++ b/pkg/gc/runner_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/gc/task.go +// Source: task.go // Package gc is a generated GoMock package. package gc diff --git a/pkg/gc/task.go b/pkg/gc/task.go index 84c626c37d8..c94d6f62712 100644 --- a/pkg/gc/task.go +++ b/pkg/gc/task.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination runner_mock.go -source task.go -package gc + package gc import ( @@ -25,7 +27,7 @@ type Runner interface { RunGC() error } -// Task is an struct used to run GC instance +// Task is an struct used to run GC instance. type Task struct { ID string Interval time.Duration @@ -33,7 +35,7 @@ type Task struct { Runner Runner } -// Validate task params +// Validate task params. func (t *Task) validate() error { if t.ID == "" { return errors.New("empty ID is not specified") diff --git a/pkg/objectstorage/objectstorage.go b/pkg/objectstorage/objectstorage.go index 55064dbdd2f..2816fcd915b 100644 --- a/pkg/objectstorage/objectstorage.go +++ b/pkg/objectstorage/objectstorage.go @@ -14,7 +14,7 @@ * limitations under the License. */ -//go:generate mockgen -package mocks -source objectstorage.go -destination ./mocks/objectstorage_mock.go +//go:generate mockgen -destination mocks/objectstorage_mock.go -source objectstorage.go -package mocks package objectstorage diff --git a/pkg/reachable/mocks/reachable_mock.go b/pkg/reachable/mocks/reachable_mock.go new file mode 100644 index 00000000000..5f71e4b731c --- /dev/null +++ b/pkg/reachable/mocks/reachable_mock.go @@ -0,0 +1,48 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: reachable.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockReachable is a mock of Reachable interface. +type MockReachable struct { + ctrl *gomock.Controller + recorder *MockReachableMockRecorder +} + +// MockReachableMockRecorder is the mock recorder for MockReachable. +type MockReachableMockRecorder struct { + mock *MockReachable +} + +// NewMockReachable creates a new mock instance. +func NewMockReachable(ctrl *gomock.Controller) *MockReachable { + mock := &MockReachable{ctrl: ctrl} + mock.recorder = &MockReachableMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockReachable) EXPECT() *MockReachableMockRecorder { + return m.recorder +} + +// Check mocks base method. +func (m *MockReachable) Check() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Check") + ret0, _ := ret[0].(error) + return ret0 +} + +// Check indicates an expected call of Check. +func (mr *MockReachableMockRecorder) Check() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Check", reflect.TypeOf((*MockReachable)(nil).Check)) +} diff --git a/pkg/reachable/reachable.go b/pkg/reachable/reachable.go index 394f2275c52..01f9553c26a 100644 --- a/pkg/reachable/reachable.go +++ b/pkg/reachable/reachable.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/reachable_mock.go -source reachable.go -package mocks + package reachable import ( diff --git a/pkg/rpc/base/base.pb.go b/pkg/rpc/base/base.pb.go index cb779db977c..b7db816f85f 100644 --- a/pkg/rpc/base/base.pb.go +++ b/pkg/rpc/base/base.pb.go @@ -15,8 +15,8 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.27.1 -// protoc v3.18.1 +// protoc-gen-go v1.28.0 +// protoc v3.19.4 // source: pkg/rpc/base/base.proto package base diff --git a/pkg/rpc/base/mocks/base_mock.go b/pkg/rpc/base/mocks/base_mock.go new file mode 100644 index 00000000000..dc3272a3e1d --- /dev/null +++ b/pkg/rpc/base/mocks/base_mock.go @@ -0,0 +1,5 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: base/base.pb.go + +// Package mocks is a generated GoMock package. +package mocks diff --git a/pkg/rpc/cdnsystem/cdnsystem.pb.go b/pkg/rpc/cdnsystem/cdnsystem.pb.go index 75f1cfc6b21..eea9b9236ee 100644 --- a/pkg/rpc/cdnsystem/cdnsystem.pb.go +++ b/pkg/rpc/cdnsystem/cdnsystem.pb.go @@ -15,8 +15,8 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.27.1 -// protoc v3.18.1 +// protoc-gen-go v1.28.0 +// protoc v3.19.4 // source: pkg/rpc/cdnsystem/cdnsystem.proto package cdnsystem diff --git a/pkg/rpc/cdnsystem/client/client.go b/pkg/rpc/cdnsystem/client/client.go index b82b11a795c..e063ec24523 100644 --- a/pkg/rpc/cdnsystem/client/client.go +++ b/pkg/rpc/cdnsystem/client/client.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/client_mock.go -source client.go -package mocks + package client import ( diff --git a/pkg/rpc/cdnsystem/client/mocks/client_mock.go b/pkg/rpc/cdnsystem/client/mocks/client_mock.go new file mode 100644 index 00000000000..3b921d32a04 --- /dev/null +++ b/pkg/rpc/cdnsystem/client/mocks/client_mock.go @@ -0,0 +1,126 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: client.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + dfnet "d7y.io/dragonfly/v2/pkg/dfnet" + base "d7y.io/dragonfly/v2/pkg/rpc/base" + cdnsystem "d7y.io/dragonfly/v2/pkg/rpc/cdnsystem" + client "d7y.io/dragonfly/v2/pkg/rpc/cdnsystem/client" + gomock "github.com/golang/mock/gomock" + grpc "google.golang.org/grpc" +) + +// MockCdnClient is a mock of CdnClient interface. +type MockCdnClient struct { + ctrl *gomock.Controller + recorder *MockCdnClientMockRecorder +} + +// MockCdnClientMockRecorder is the mock recorder for MockCdnClient. +type MockCdnClientMockRecorder struct { + mock *MockCdnClient +} + +// NewMockCdnClient creates a new mock instance. +func NewMockCdnClient(ctrl *gomock.Controller) *MockCdnClient { + mock := &MockCdnClient{ctrl: ctrl} + mock.recorder = &MockCdnClientMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockCdnClient) EXPECT() *MockCdnClientMockRecorder { + return m.recorder +} + +// Close mocks base method. +func (m *MockCdnClient) Close() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Close") + ret0, _ := ret[0].(error) + return ret0 +} + +// Close indicates an expected call of Close. +func (mr *MockCdnClientMockRecorder) Close() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockCdnClient)(nil).Close)) +} + +// GetPieceTasks mocks base method. +func (m *MockCdnClient) GetPieceTasks(ctx context.Context, addr dfnet.NetAddr, req *base.PieceTaskRequest, opts ...grpc.CallOption) (*base.PiecePacket, error) { + m.ctrl.T.Helper() + varargs := []interface{}{ctx, addr, req} + for _, a := range opts { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetPieceTasks", varargs...) + ret0, _ := ret[0].(*base.PiecePacket) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetPieceTasks indicates an expected call of GetPieceTasks. +func (mr *MockCdnClientMockRecorder) GetPieceTasks(ctx, addr, req interface{}, opts ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{ctx, addr, req}, opts...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceTasks", reflect.TypeOf((*MockCdnClient)(nil).GetPieceTasks), varargs...) +} + +// ObtainSeeds mocks base method. +func (m *MockCdnClient) ObtainSeeds(ctx context.Context, sr *cdnsystem.SeedRequest, opts ...grpc.CallOption) (*client.PieceSeedStream, error) { + m.ctrl.T.Helper() + varargs := []interface{}{ctx, sr} + for _, a := range opts { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ObtainSeeds", varargs...) + ret0, _ := ret[0].(*client.PieceSeedStream) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ObtainSeeds indicates an expected call of ObtainSeeds. +func (mr *MockCdnClientMockRecorder) ObtainSeeds(ctx, sr interface{}, opts ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{ctx, sr}, opts...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ObtainSeeds", reflect.TypeOf((*MockCdnClient)(nil).ObtainSeeds), varargs...) +} + +// SyncPieceTasks mocks base method. +func (m *MockCdnClient) SyncPieceTasks(ctx context.Context, addr dfnet.NetAddr, ptr *base.PieceTaskRequest, opts ...grpc.CallOption) (cdnsystem.Seeder_SyncPieceTasksClient, error) { + m.ctrl.T.Helper() + varargs := []interface{}{ctx, addr, ptr} + for _, a := range opts { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "SyncPieceTasks", varargs...) + ret0, _ := ret[0].(cdnsystem.Seeder_SyncPieceTasksClient) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// SyncPieceTasks indicates an expected call of SyncPieceTasks. +func (mr *MockCdnClientMockRecorder) SyncPieceTasks(ctx, addr, ptr interface{}, opts ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{ctx, addr, ptr}, opts...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieceTasks", reflect.TypeOf((*MockCdnClient)(nil).SyncPieceTasks), varargs...) +} + +// UpdateState mocks base method. +func (m *MockCdnClient) UpdateState(addrs []dfnet.NetAddr) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "UpdateState", addrs) +} + +// UpdateState indicates an expected call of UpdateState. +func (mr *MockCdnClientMockRecorder) UpdateState(addrs interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateState", reflect.TypeOf((*MockCdnClient)(nil).UpdateState), addrs) +} diff --git a/pkg/rpc/cdnsystem/mocks/cdnsystem_mock.go b/pkg/rpc/cdnsystem/mocks/cdnsystem_mock.go new file mode 100644 index 00000000000..9b7c8f2d0f3 --- /dev/null +++ b/pkg/rpc/cdnsystem/mocks/cdnsystem_mock.go @@ -0,0 +1,678 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: cdnsystem/cdnsystem.pb.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + base "d7y.io/dragonfly/v2/pkg/rpc/base" + cdnsystem "d7y.io/dragonfly/v2/pkg/rpc/cdnsystem" + gomock "github.com/golang/mock/gomock" + grpc "google.golang.org/grpc" + metadata "google.golang.org/grpc/metadata" +) + +// MockSeederClient is a mock of SeederClient interface. +type MockSeederClient struct { + ctrl *gomock.Controller + recorder *MockSeederClientMockRecorder +} + +// MockSeederClientMockRecorder is the mock recorder for MockSeederClient. +type MockSeederClientMockRecorder struct { + mock *MockSeederClient +} + +// NewMockSeederClient creates a new mock instance. +func NewMockSeederClient(ctrl *gomock.Controller) *MockSeederClient { + mock := &MockSeederClient{ctrl: ctrl} + mock.recorder = &MockSeederClientMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSeederClient) EXPECT() *MockSeederClientMockRecorder { + return m.recorder +} + +// GetPieceTasks mocks base method. +func (m *MockSeederClient) GetPieceTasks(ctx context.Context, in *base.PieceTaskRequest, opts ...grpc.CallOption) (*base.PiecePacket, error) { + m.ctrl.T.Helper() + varargs := []interface{}{ctx, in} + for _, a := range opts { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetPieceTasks", varargs...) + ret0, _ := ret[0].(*base.PiecePacket) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetPieceTasks indicates an expected call of GetPieceTasks. +func (mr *MockSeederClientMockRecorder) GetPieceTasks(ctx, in interface{}, opts ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{ctx, in}, opts...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceTasks", reflect.TypeOf((*MockSeederClient)(nil).GetPieceTasks), varargs...) +} + +// ObtainSeeds mocks base method. +func (m *MockSeederClient) ObtainSeeds(ctx context.Context, in *cdnsystem.SeedRequest, opts ...grpc.CallOption) (cdnsystem.Seeder_ObtainSeedsClient, error) { + m.ctrl.T.Helper() + varargs := []interface{}{ctx, in} + for _, a := range opts { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ObtainSeeds", varargs...) + ret0, _ := ret[0].(cdnsystem.Seeder_ObtainSeedsClient) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ObtainSeeds indicates an expected call of ObtainSeeds. +func (mr *MockSeederClientMockRecorder) ObtainSeeds(ctx, in interface{}, opts ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{ctx, in}, opts...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ObtainSeeds", reflect.TypeOf((*MockSeederClient)(nil).ObtainSeeds), varargs...) +} + +// SyncPieceTasks mocks base method. +func (m *MockSeederClient) SyncPieceTasks(ctx context.Context, opts ...grpc.CallOption) (cdnsystem.Seeder_SyncPieceTasksClient, error) { + m.ctrl.T.Helper() + varargs := []interface{}{ctx} + for _, a := range opts { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "SyncPieceTasks", varargs...) + ret0, _ := ret[0].(cdnsystem.Seeder_SyncPieceTasksClient) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// SyncPieceTasks indicates an expected call of SyncPieceTasks. +func (mr *MockSeederClientMockRecorder) SyncPieceTasks(ctx interface{}, opts ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{ctx}, opts...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieceTasks", reflect.TypeOf((*MockSeederClient)(nil).SyncPieceTasks), varargs...) +} + +// MockSeeder_ObtainSeedsClient is a mock of Seeder_ObtainSeedsClient interface. +type MockSeeder_ObtainSeedsClient struct { + ctrl *gomock.Controller + recorder *MockSeeder_ObtainSeedsClientMockRecorder +} + +// MockSeeder_ObtainSeedsClientMockRecorder is the mock recorder for MockSeeder_ObtainSeedsClient. +type MockSeeder_ObtainSeedsClientMockRecorder struct { + mock *MockSeeder_ObtainSeedsClient +} + +// NewMockSeeder_ObtainSeedsClient creates a new mock instance. +func NewMockSeeder_ObtainSeedsClient(ctrl *gomock.Controller) *MockSeeder_ObtainSeedsClient { + mock := &MockSeeder_ObtainSeedsClient{ctrl: ctrl} + mock.recorder = &MockSeeder_ObtainSeedsClientMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSeeder_ObtainSeedsClient) EXPECT() *MockSeeder_ObtainSeedsClientMockRecorder { + return m.recorder +} + +// CloseSend mocks base method. +func (m *MockSeeder_ObtainSeedsClient) CloseSend() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CloseSend") + ret0, _ := ret[0].(error) + return ret0 +} + +// CloseSend indicates an expected call of CloseSend. +func (mr *MockSeeder_ObtainSeedsClientMockRecorder) CloseSend() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockSeeder_ObtainSeedsClient)(nil).CloseSend)) +} + +// Context mocks base method. +func (m *MockSeeder_ObtainSeedsClient) Context() context.Context { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Context") + ret0, _ := ret[0].(context.Context) + return ret0 +} + +// Context indicates an expected call of Context. +func (mr *MockSeeder_ObtainSeedsClientMockRecorder) Context() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockSeeder_ObtainSeedsClient)(nil).Context)) +} + +// Header mocks base method. +func (m *MockSeeder_ObtainSeedsClient) Header() (metadata.MD, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Header") + ret0, _ := ret[0].(metadata.MD) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Header indicates an expected call of Header. +func (mr *MockSeeder_ObtainSeedsClientMockRecorder) Header() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockSeeder_ObtainSeedsClient)(nil).Header)) +} + +// Recv mocks base method. +func (m *MockSeeder_ObtainSeedsClient) Recv() (*cdnsystem.PieceSeed, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Recv") + ret0, _ := ret[0].(*cdnsystem.PieceSeed) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Recv indicates an expected call of Recv. +func (mr *MockSeeder_ObtainSeedsClientMockRecorder) Recv() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockSeeder_ObtainSeedsClient)(nil).Recv)) +} + +// RecvMsg mocks base method. +func (m_2 *MockSeeder_ObtainSeedsClient) RecvMsg(m interface{}) error { + m_2.ctrl.T.Helper() + ret := m_2.ctrl.Call(m_2, "RecvMsg", m) + ret0, _ := ret[0].(error) + return ret0 +} + +// RecvMsg indicates an expected call of RecvMsg. +func (mr *MockSeeder_ObtainSeedsClientMockRecorder) RecvMsg(m interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockSeeder_ObtainSeedsClient)(nil).RecvMsg), m) +} + +// SendMsg mocks base method. +func (m_2 *MockSeeder_ObtainSeedsClient) SendMsg(m interface{}) error { + m_2.ctrl.T.Helper() + ret := m_2.ctrl.Call(m_2, "SendMsg", m) + ret0, _ := ret[0].(error) + return ret0 +} + +// SendMsg indicates an expected call of SendMsg. +func (mr *MockSeeder_ObtainSeedsClientMockRecorder) SendMsg(m interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockSeeder_ObtainSeedsClient)(nil).SendMsg), m) +} + +// Trailer mocks base method. +func (m *MockSeeder_ObtainSeedsClient) Trailer() metadata.MD { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Trailer") + ret0, _ := ret[0].(metadata.MD) + return ret0 +} + +// Trailer indicates an expected call of Trailer. +func (mr *MockSeeder_ObtainSeedsClientMockRecorder) Trailer() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockSeeder_ObtainSeedsClient)(nil).Trailer)) +} + +// MockSeeder_SyncPieceTasksClient is a mock of Seeder_SyncPieceTasksClient interface. +type MockSeeder_SyncPieceTasksClient struct { + ctrl *gomock.Controller + recorder *MockSeeder_SyncPieceTasksClientMockRecorder +} + +// MockSeeder_SyncPieceTasksClientMockRecorder is the mock recorder for MockSeeder_SyncPieceTasksClient. +type MockSeeder_SyncPieceTasksClientMockRecorder struct { + mock *MockSeeder_SyncPieceTasksClient +} + +// NewMockSeeder_SyncPieceTasksClient creates a new mock instance. +func NewMockSeeder_SyncPieceTasksClient(ctrl *gomock.Controller) *MockSeeder_SyncPieceTasksClient { + mock := &MockSeeder_SyncPieceTasksClient{ctrl: ctrl} + mock.recorder = &MockSeeder_SyncPieceTasksClientMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSeeder_SyncPieceTasksClient) EXPECT() *MockSeeder_SyncPieceTasksClientMockRecorder { + return m.recorder +} + +// CloseSend mocks base method. +func (m *MockSeeder_SyncPieceTasksClient) CloseSend() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CloseSend") + ret0, _ := ret[0].(error) + return ret0 +} + +// CloseSend indicates an expected call of CloseSend. +func (mr *MockSeeder_SyncPieceTasksClientMockRecorder) CloseSend() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloseSend", reflect.TypeOf((*MockSeeder_SyncPieceTasksClient)(nil).CloseSend)) +} + +// Context mocks base method. +func (m *MockSeeder_SyncPieceTasksClient) Context() context.Context { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Context") + ret0, _ := ret[0].(context.Context) + return ret0 +} + +// Context indicates an expected call of Context. +func (mr *MockSeeder_SyncPieceTasksClientMockRecorder) Context() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockSeeder_SyncPieceTasksClient)(nil).Context)) +} + +// Header mocks base method. +func (m *MockSeeder_SyncPieceTasksClient) Header() (metadata.MD, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Header") + ret0, _ := ret[0].(metadata.MD) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Header indicates an expected call of Header. +func (mr *MockSeeder_SyncPieceTasksClientMockRecorder) Header() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Header", reflect.TypeOf((*MockSeeder_SyncPieceTasksClient)(nil).Header)) +} + +// Recv mocks base method. +func (m *MockSeeder_SyncPieceTasksClient) Recv() (*base.PiecePacket, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Recv") + ret0, _ := ret[0].(*base.PiecePacket) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Recv indicates an expected call of Recv. +func (mr *MockSeeder_SyncPieceTasksClientMockRecorder) Recv() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockSeeder_SyncPieceTasksClient)(nil).Recv)) +} + +// RecvMsg mocks base method. +func (m_2 *MockSeeder_SyncPieceTasksClient) RecvMsg(m interface{}) error { + m_2.ctrl.T.Helper() + ret := m_2.ctrl.Call(m_2, "RecvMsg", m) + ret0, _ := ret[0].(error) + return ret0 +} + +// RecvMsg indicates an expected call of RecvMsg. +func (mr *MockSeeder_SyncPieceTasksClientMockRecorder) RecvMsg(m interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockSeeder_SyncPieceTasksClient)(nil).RecvMsg), m) +} + +// Send mocks base method. +func (m *MockSeeder_SyncPieceTasksClient) Send(arg0 *base.PieceTaskRequest) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Send", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// Send indicates an expected call of Send. +func (mr *MockSeeder_SyncPieceTasksClientMockRecorder) Send(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockSeeder_SyncPieceTasksClient)(nil).Send), arg0) +} + +// SendMsg mocks base method. +func (m_2 *MockSeeder_SyncPieceTasksClient) SendMsg(m interface{}) error { + m_2.ctrl.T.Helper() + ret := m_2.ctrl.Call(m_2, "SendMsg", m) + ret0, _ := ret[0].(error) + return ret0 +} + +// SendMsg indicates an expected call of SendMsg. +func (mr *MockSeeder_SyncPieceTasksClientMockRecorder) SendMsg(m interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockSeeder_SyncPieceTasksClient)(nil).SendMsg), m) +} + +// Trailer mocks base method. +func (m *MockSeeder_SyncPieceTasksClient) Trailer() metadata.MD { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Trailer") + ret0, _ := ret[0].(metadata.MD) + return ret0 +} + +// Trailer indicates an expected call of Trailer. +func (mr *MockSeeder_SyncPieceTasksClientMockRecorder) Trailer() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Trailer", reflect.TypeOf((*MockSeeder_SyncPieceTasksClient)(nil).Trailer)) +} + +// MockSeederServer is a mock of SeederServer interface. +type MockSeederServer struct { + ctrl *gomock.Controller + recorder *MockSeederServerMockRecorder +} + +// MockSeederServerMockRecorder is the mock recorder for MockSeederServer. +type MockSeederServerMockRecorder struct { + mock *MockSeederServer +} + +// NewMockSeederServer creates a new mock instance. +func NewMockSeederServer(ctrl *gomock.Controller) *MockSeederServer { + mock := &MockSeederServer{ctrl: ctrl} + mock.recorder = &MockSeederServerMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSeederServer) EXPECT() *MockSeederServerMockRecorder { + return m.recorder +} + +// GetPieceTasks mocks base method. +func (m *MockSeederServer) GetPieceTasks(arg0 context.Context, arg1 *base.PieceTaskRequest) (*base.PiecePacket, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetPieceTasks", arg0, arg1) + ret0, _ := ret[0].(*base.PiecePacket) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetPieceTasks indicates an expected call of GetPieceTasks. +func (mr *MockSeederServerMockRecorder) GetPieceTasks(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceTasks", reflect.TypeOf((*MockSeederServer)(nil).GetPieceTasks), arg0, arg1) +} + +// ObtainSeeds mocks base method. +func (m *MockSeederServer) ObtainSeeds(arg0 *cdnsystem.SeedRequest, arg1 cdnsystem.Seeder_ObtainSeedsServer) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ObtainSeeds", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// ObtainSeeds indicates an expected call of ObtainSeeds. +func (mr *MockSeederServerMockRecorder) ObtainSeeds(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ObtainSeeds", reflect.TypeOf((*MockSeederServer)(nil).ObtainSeeds), arg0, arg1) +} + +// SyncPieceTasks mocks base method. +func (m *MockSeederServer) SyncPieceTasks(arg0 cdnsystem.Seeder_SyncPieceTasksServer) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SyncPieceTasks", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// SyncPieceTasks indicates an expected call of SyncPieceTasks. +func (mr *MockSeederServerMockRecorder) SyncPieceTasks(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieceTasks", reflect.TypeOf((*MockSeederServer)(nil).SyncPieceTasks), arg0) +} + +// MockSeeder_ObtainSeedsServer is a mock of Seeder_ObtainSeedsServer interface. +type MockSeeder_ObtainSeedsServer struct { + ctrl *gomock.Controller + recorder *MockSeeder_ObtainSeedsServerMockRecorder +} + +// MockSeeder_ObtainSeedsServerMockRecorder is the mock recorder for MockSeeder_ObtainSeedsServer. +type MockSeeder_ObtainSeedsServerMockRecorder struct { + mock *MockSeeder_ObtainSeedsServer +} + +// NewMockSeeder_ObtainSeedsServer creates a new mock instance. +func NewMockSeeder_ObtainSeedsServer(ctrl *gomock.Controller) *MockSeeder_ObtainSeedsServer { + mock := &MockSeeder_ObtainSeedsServer{ctrl: ctrl} + mock.recorder = &MockSeeder_ObtainSeedsServerMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSeeder_ObtainSeedsServer) EXPECT() *MockSeeder_ObtainSeedsServerMockRecorder { + return m.recorder +} + +// Context mocks base method. +func (m *MockSeeder_ObtainSeedsServer) Context() context.Context { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Context") + ret0, _ := ret[0].(context.Context) + return ret0 +} + +// Context indicates an expected call of Context. +func (mr *MockSeeder_ObtainSeedsServerMockRecorder) Context() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockSeeder_ObtainSeedsServer)(nil).Context)) +} + +// RecvMsg mocks base method. +func (m_2 *MockSeeder_ObtainSeedsServer) RecvMsg(m interface{}) error { + m_2.ctrl.T.Helper() + ret := m_2.ctrl.Call(m_2, "RecvMsg", m) + ret0, _ := ret[0].(error) + return ret0 +} + +// RecvMsg indicates an expected call of RecvMsg. +func (mr *MockSeeder_ObtainSeedsServerMockRecorder) RecvMsg(m interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockSeeder_ObtainSeedsServer)(nil).RecvMsg), m) +} + +// Send mocks base method. +func (m *MockSeeder_ObtainSeedsServer) Send(arg0 *cdnsystem.PieceSeed) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Send", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// Send indicates an expected call of Send. +func (mr *MockSeeder_ObtainSeedsServerMockRecorder) Send(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockSeeder_ObtainSeedsServer)(nil).Send), arg0) +} + +// SendHeader mocks base method. +func (m *MockSeeder_ObtainSeedsServer) SendHeader(arg0 metadata.MD) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SendHeader", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// SendHeader indicates an expected call of SendHeader. +func (mr *MockSeeder_ObtainSeedsServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockSeeder_ObtainSeedsServer)(nil).SendHeader), arg0) +} + +// SendMsg mocks base method. +func (m_2 *MockSeeder_ObtainSeedsServer) SendMsg(m interface{}) error { + m_2.ctrl.T.Helper() + ret := m_2.ctrl.Call(m_2, "SendMsg", m) + ret0, _ := ret[0].(error) + return ret0 +} + +// SendMsg indicates an expected call of SendMsg. +func (mr *MockSeeder_ObtainSeedsServerMockRecorder) SendMsg(m interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockSeeder_ObtainSeedsServer)(nil).SendMsg), m) +} + +// SetHeader mocks base method. +func (m *MockSeeder_ObtainSeedsServer) SetHeader(arg0 metadata.MD) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SetHeader", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// SetHeader indicates an expected call of SetHeader. +func (mr *MockSeeder_ObtainSeedsServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockSeeder_ObtainSeedsServer)(nil).SetHeader), arg0) +} + +// SetTrailer mocks base method. +func (m *MockSeeder_ObtainSeedsServer) SetTrailer(arg0 metadata.MD) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "SetTrailer", arg0) +} + +// SetTrailer indicates an expected call of SetTrailer. +func (mr *MockSeeder_ObtainSeedsServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockSeeder_ObtainSeedsServer)(nil).SetTrailer), arg0) +} + +// MockSeeder_SyncPieceTasksServer is a mock of Seeder_SyncPieceTasksServer interface. +type MockSeeder_SyncPieceTasksServer struct { + ctrl *gomock.Controller + recorder *MockSeeder_SyncPieceTasksServerMockRecorder +} + +// MockSeeder_SyncPieceTasksServerMockRecorder is the mock recorder for MockSeeder_SyncPieceTasksServer. +type MockSeeder_SyncPieceTasksServerMockRecorder struct { + mock *MockSeeder_SyncPieceTasksServer +} + +// NewMockSeeder_SyncPieceTasksServer creates a new mock instance. +func NewMockSeeder_SyncPieceTasksServer(ctrl *gomock.Controller) *MockSeeder_SyncPieceTasksServer { + mock := &MockSeeder_SyncPieceTasksServer{ctrl: ctrl} + mock.recorder = &MockSeeder_SyncPieceTasksServerMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSeeder_SyncPieceTasksServer) EXPECT() *MockSeeder_SyncPieceTasksServerMockRecorder { + return m.recorder +} + +// Context mocks base method. +func (m *MockSeeder_SyncPieceTasksServer) Context() context.Context { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Context") + ret0, _ := ret[0].(context.Context) + return ret0 +} + +// Context indicates an expected call of Context. +func (mr *MockSeeder_SyncPieceTasksServerMockRecorder) Context() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Context", reflect.TypeOf((*MockSeeder_SyncPieceTasksServer)(nil).Context)) +} + +// Recv mocks base method. +func (m *MockSeeder_SyncPieceTasksServer) Recv() (*base.PieceTaskRequest, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Recv") + ret0, _ := ret[0].(*base.PieceTaskRequest) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Recv indicates an expected call of Recv. +func (mr *MockSeeder_SyncPieceTasksServerMockRecorder) Recv() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockSeeder_SyncPieceTasksServer)(nil).Recv)) +} + +// RecvMsg mocks base method. +func (m_2 *MockSeeder_SyncPieceTasksServer) RecvMsg(m interface{}) error { + m_2.ctrl.T.Helper() + ret := m_2.ctrl.Call(m_2, "RecvMsg", m) + ret0, _ := ret[0].(error) + return ret0 +} + +// RecvMsg indicates an expected call of RecvMsg. +func (mr *MockSeeder_SyncPieceTasksServerMockRecorder) RecvMsg(m interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecvMsg", reflect.TypeOf((*MockSeeder_SyncPieceTasksServer)(nil).RecvMsg), m) +} + +// Send mocks base method. +func (m *MockSeeder_SyncPieceTasksServer) Send(arg0 *base.PiecePacket) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Send", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// Send indicates an expected call of Send. +func (mr *MockSeeder_SyncPieceTasksServerMockRecorder) Send(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockSeeder_SyncPieceTasksServer)(nil).Send), arg0) +} + +// SendHeader mocks base method. +func (m *MockSeeder_SyncPieceTasksServer) SendHeader(arg0 metadata.MD) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SendHeader", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// SendHeader indicates an expected call of SendHeader. +func (mr *MockSeeder_SyncPieceTasksServerMockRecorder) SendHeader(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendHeader", reflect.TypeOf((*MockSeeder_SyncPieceTasksServer)(nil).SendHeader), arg0) +} + +// SendMsg mocks base method. +func (m_2 *MockSeeder_SyncPieceTasksServer) SendMsg(m interface{}) error { + m_2.ctrl.T.Helper() + ret := m_2.ctrl.Call(m_2, "SendMsg", m) + ret0, _ := ret[0].(error) + return ret0 +} + +// SendMsg indicates an expected call of SendMsg. +func (mr *MockSeeder_SyncPieceTasksServerMockRecorder) SendMsg(m interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendMsg", reflect.TypeOf((*MockSeeder_SyncPieceTasksServer)(nil).SendMsg), m) +} + +// SetHeader mocks base method. +func (m *MockSeeder_SyncPieceTasksServer) SetHeader(arg0 metadata.MD) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SetHeader", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// SetHeader indicates an expected call of SetHeader. +func (mr *MockSeeder_SyncPieceTasksServerMockRecorder) SetHeader(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeader", reflect.TypeOf((*MockSeeder_SyncPieceTasksServer)(nil).SetHeader), arg0) +} + +// SetTrailer mocks base method. +func (m *MockSeeder_SyncPieceTasksServer) SetTrailer(arg0 metadata.MD) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "SetTrailer", arg0) +} + +// SetTrailer indicates an expected call of SetTrailer. +func (mr *MockSeeder_SyncPieceTasksServerMockRecorder) SetTrailer(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTrailer", reflect.TypeOf((*MockSeeder_SyncPieceTasksServer)(nil).SetTrailer), arg0) +} diff --git a/pkg/rpc/dfdaemon/client/client.go b/pkg/rpc/dfdaemon/client/client.go index ccd9b8f9b35..0fd423bd137 100644 --- a/pkg/rpc/dfdaemon/client/client.go +++ b/pkg/rpc/dfdaemon/client/client.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/client_mock.go -source client.go -package mocks + package client import ( diff --git a/pkg/rpc/dfdaemon/client/mocks/client_mock.go b/pkg/rpc/dfdaemon/client/mocks/client_mock.go index 5d89874445b..01cb67d22fb 100644 --- a/pkg/rpc/dfdaemon/client/mocks/client_mock.go +++ b/pkg/rpc/dfdaemon/client/mocks/client_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/rpc/dfdaemon/client/client.go +// Source: client.go // Package mocks is a generated GoMock package. package mocks diff --git a/pkg/rpc/dfdaemon/dfdaemon.pb.go b/pkg/rpc/dfdaemon/dfdaemon.pb.go index a45bf09c5dd..dd522a131de 100644 --- a/pkg/rpc/dfdaemon/dfdaemon.pb.go +++ b/pkg/rpc/dfdaemon/dfdaemon.pb.go @@ -15,8 +15,8 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.27.1 -// protoc v3.18.1 +// protoc-gen-go v1.28.0 +// protoc v3.19.4 // source: pkg/rpc/dfdaemon/dfdaemon.proto package dfdaemon diff --git a/pkg/rpc/dfdaemon/mocks/dfdaemon_mock.go b/pkg/rpc/dfdaemon/mocks/dfdaemon_mock.go index d83d2b8c7e5..322ff28d317 100644 --- a/pkg/rpc/dfdaemon/mocks/dfdaemon_mock.go +++ b/pkg/rpc/dfdaemon/mocks/dfdaemon_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/rpc/dfdaemon/dfdaemon.pb.go +// Source: dfdaemon/dfdaemon.pb.go // Package mocks is a generated GoMock package. package mocks diff --git a/pkg/rpc/dfdaemon/server/mocks/server_mock.go b/pkg/rpc/dfdaemon/server/mocks/server_mock.go index 1392961b505..e8f8a23cd30 100644 --- a/pkg/rpc/dfdaemon/server/mocks/server_mock.go +++ b/pkg/rpc/dfdaemon/server/mocks/server_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/rpc/dfdaemon/server/server.go +// Source: server.go // Package mocks is a generated GoMock package. package mocks @@ -50,6 +50,20 @@ func (mr *MockDaemonServerMockRecorder) CheckHealth(arg0 interface{}) *gomock.Ca return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckHealth", reflect.TypeOf((*MockDaemonServer)(nil).CheckHealth), arg0) } +// DeleteTask mocks base method. +func (m *MockDaemonServer) DeleteTask(arg0 context.Context, arg1 *dfdaemon.DeleteTaskRequest) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DeleteTask", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// DeleteTask indicates an expected call of DeleteTask. +func (mr *MockDaemonServerMockRecorder) DeleteTask(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTask", reflect.TypeOf((*MockDaemonServer)(nil).DeleteTask), arg0, arg1) +} + // Download mocks base method. func (m *MockDaemonServer) Download(arg0 context.Context, arg1 *dfdaemon.DownRequest, arg2 chan<- *dfdaemon.DownResult) error { m.ctrl.T.Helper() @@ -64,6 +78,20 @@ func (mr *MockDaemonServerMockRecorder) Download(arg0, arg1, arg2 interface{}) * return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Download", reflect.TypeOf((*MockDaemonServer)(nil).Download), arg0, arg1, arg2) } +// ExportTask mocks base method. +func (m *MockDaemonServer) ExportTask(arg0 context.Context, arg1 *dfdaemon.ExportTaskRequest) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ExportTask", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// ExportTask indicates an expected call of ExportTask. +func (mr *MockDaemonServerMockRecorder) ExportTask(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExportTask", reflect.TypeOf((*MockDaemonServer)(nil).ExportTask), arg0, arg1) +} + // GetPieceTasks mocks base method. func (m *MockDaemonServer) GetPieceTasks(arg0 context.Context, arg1 *base.PieceTaskRequest) (*base.PiecePacket, error) { m.ctrl.T.Helper() @@ -78,3 +106,45 @@ func (mr *MockDaemonServerMockRecorder) GetPieceTasks(arg0, arg1 interface{}) *g mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPieceTasks", reflect.TypeOf((*MockDaemonServer)(nil).GetPieceTasks), arg0, arg1) } + +// ImportTask mocks base method. +func (m *MockDaemonServer) ImportTask(arg0 context.Context, arg1 *dfdaemon.ImportTaskRequest) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ImportTask", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// ImportTask indicates an expected call of ImportTask. +func (mr *MockDaemonServerMockRecorder) ImportTask(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ImportTask", reflect.TypeOf((*MockDaemonServer)(nil).ImportTask), arg0, arg1) +} + +// StatTask mocks base method. +func (m *MockDaemonServer) StatTask(arg0 context.Context, arg1 *dfdaemon.StatTaskRequest) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "StatTask", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// StatTask indicates an expected call of StatTask. +func (mr *MockDaemonServerMockRecorder) StatTask(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StatTask", reflect.TypeOf((*MockDaemonServer)(nil).StatTask), arg0, arg1) +} + +// SyncPieceTasks mocks base method. +func (m *MockDaemonServer) SyncPieceTasks(arg0 dfdaemon.Daemon_SyncPieceTasksServer) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SyncPieceTasks", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// SyncPieceTasks indicates an expected call of SyncPieceTasks. +func (mr *MockDaemonServerMockRecorder) SyncPieceTasks(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncPieceTasks", reflect.TypeOf((*MockDaemonServer)(nil).SyncPieceTasks), arg0) +} diff --git a/pkg/rpc/dfdaemon/server/server.go b/pkg/rpc/dfdaemon/server/server.go index c925122d668..5aa1146c84c 100644 --- a/pkg/rpc/dfdaemon/server/server.go +++ b/pkg/rpc/dfdaemon/server/server.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/server_mock.go -source server.go -package mocks + package server import ( @@ -32,7 +34,6 @@ import ( "d7y.io/dragonfly/v2/pkg/safe" ) -//go:generate mockgen -source server.go -destination ../../../../client/daemon/test/mock/daemon/daemon_server.go // DaemonServer refer to dfdaemon.DaemonServer type DaemonServer interface { // Download triggers client to download file diff --git a/pkg/rpc/manager/manager.pb.go b/pkg/rpc/manager/manager.pb.go index 3023eaf10f0..6b16ac5b025 100644 --- a/pkg/rpc/manager/manager.pb.go +++ b/pkg/rpc/manager/manager.pb.go @@ -15,8 +15,8 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.27.1 -// protoc v3.18.1 +// protoc-gen-go v1.28.0 +// protoc v3.19.4 // source: pkg/rpc/manager/manager.proto package manager @@ -1114,7 +1114,7 @@ type ListSchedulersRequest struct { HostName string `protobuf:"bytes,2,opt,name=host_name,json=hostName,proto3" json:"host_name,omitempty"` // Source service ip. Ip string `protobuf:"bytes,3,opt,name=ip,proto3" json:"ip,omitempty"` - // Source service host infomation. + // Source service host information. HostInfo map[string]string `protobuf:"bytes,5,rep,name=host_info,json=hostInfo,proto3" json:"host_info,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value,proto3"` } diff --git a/pkg/rpc/manager/mocks/manager_mock.go b/pkg/rpc/manager/mocks/manager_mock.go index d4fb6630df0..adfb1345d8c 100644 --- a/pkg/rpc/manager/mocks/manager_mock.go +++ b/pkg/rpc/manager/mocks/manager_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/rpc/manager/manager.pb.go +// Source: manager/manager.pb.go // Package mocks is a generated GoMock package. package mocks diff --git a/pkg/rpc/scheduler/client/client.go b/pkg/rpc/scheduler/client/client.go index b420bfeac2e..897228625b3 100644 --- a/pkg/rpc/scheduler/client/client.go +++ b/pkg/rpc/scheduler/client/client.go @@ -14,7 +14,7 @@ * limitations under the License. */ -//go:generate mockgen -package mocks -source client.go -destination ./mocks/client_mock.go +//go:generate mockgen -destination mocks/client_mock.go -source client.go -package mocks package client diff --git a/pkg/rpc/scheduler/mocks/scheduler_mock.go b/pkg/rpc/scheduler/mocks/scheduler_mock.go index 59a37bfe4e3..88baadb1958 100644 --- a/pkg/rpc/scheduler/mocks/scheduler_mock.go +++ b/pkg/rpc/scheduler/mocks/scheduler_mock.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: pkg/rpc/scheduler/scheduler.pb.go +// Source: scheduler/scheduler.pb.go // Package mocks is a generated GoMock package. package mocks diff --git a/pkg/rpc/scheduler/scheduler.pb.go b/pkg/rpc/scheduler/scheduler.pb.go index 71d5d472c5c..1c39941f094 100644 --- a/pkg/rpc/scheduler/scheduler.pb.go +++ b/pkg/rpc/scheduler/scheduler.pb.go @@ -15,8 +15,8 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.27.1 -// protoc v3.18.1 +// protoc-gen-go v1.28.0 +// protoc v3.19.4 // source: pkg/rpc/scheduler/scheduler.proto package scheduler @@ -276,7 +276,7 @@ func (*RegisterResult_SinglePiece) isRegisterResult_DirectPiece() {} func (*RegisterResult_PieceContent) isRegisterResult_DirectPiece() {} -// SinglePiece represents infomation of single piece. +// SinglePiece represents information of single piece. type SinglePiece struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache @@ -343,7 +343,7 @@ func (x *SinglePiece) GetPieceInfo() *base.PieceInfo { return nil } -// PeerHost represents infomation of peer host. +// PeerHost represents information of peer host. type PeerHost struct { state protoimpl.MessageState sizeCache protoimpl.SizeCache diff --git a/pkg/rpc/server.go b/pkg/rpc/server.go index 0ae7ad51aa0..0d2158f24f3 100644 --- a/pkg/rpc/server.go +++ b/pkg/rpc/server.go @@ -14,6 +14,12 @@ * limitations under the License. */ +//go:generate mockgen -destination base/mocks/base_mock.go -source base/base.pb.go -package mocks +//go:generate mockgen -destination cdnsystem/mocks/cdnsystem_mock.go -source cdnsystem/cdnsystem.pb.go -package mocks +//go:generate mockgen -destination dfdaemon/mocks/dfdaemon_mock.go -source dfdaemon/dfdaemon.pb.go -package mocks +//go:generate mockgen -destination manager/mocks/manager_mock.go -source manager/manager.pb.go -package mocks +//go:generate mockgen -destination scheduler/mocks/scheduler_mock.go -source scheduler/scheduler.pb.go -package mocks + package rpc import ( diff --git a/pkg/source/clients/hdfsprotocol/hdfs_source_client_test.go b/pkg/source/clients/hdfsprotocol/hdfs_source_client_test.go deleted file mode 100644 index 939a0393d37..00000000000 --- a/pkg/source/clients/hdfsprotocol/hdfs_source_client_test.go +++ /dev/null @@ -1,382 +0,0 @@ -//go:build linux -// +build linux - -/* - * Copyright 2020 The Dragonfly Authors - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -package hdfsprotocol - -import ( - "errors" - "io" - "os" - "reflect" - "testing" - "time" - - "github.com/agiledragon/gomonkey/v2" - "github.com/colinmarc/hdfs/v2" - "github.com/stretchr/testify/assert" - - "d7y.io/dragonfly/v2/pkg/net/http" - "d7y.io/dragonfly/v2/pkg/source" -) - -var sourceClient source.ResourceClient - -const ( - hdfsExistFileHost = "127.0.0.1:9000" - hdfsExistFilePath = "/user/root/input/f1.txt" - hdfsExistFileURL = "hdfs://" + hdfsExistFileHost + hdfsExistFilePath - hdfsExistFileContentLength int64 = 11 - hdfsExistFileContent = "Hello World" - hdfsExistFileLastModifiedMillis int64 = 1136214245000 - hdfsExistFileLastModified = "Mon, 02 Jan 2006 15:04:05 GMT" - hdfsExistFileRangeStart int64 = 3 - hdfsExistFileRangeEnd int64 = 10 -) - -const ( - hdfsNotExistFileURL = "hdfs://127.0.0.1:9000/user/root/input/f3.txt" - hdfsNotExistFileContentLength int64 = source.UnknownSourceFileLen - hdfsNotExistLastModified int64 = -1 -) - -var fakeHDFSClient = &hdfs.Client{} - -func testBefore() { - sourceClient = newHDFSSourceClient(func(p *hdfsSourceClient) { - p.clientMap[hdfsExistFileHost] = fakeHDFSClient - }) -} - -func TestMain(t *testing.M) { - testBefore() - t.Run() - os.Exit(0) -} - -// TestGetContentLength_OK function test exist file and return file length -func TestGetContentLength_OK(t *testing.T) { - - var info os.FileInfo = fakeHDFSFileInfo{ - contents: hdfsExistFileContent, - } - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{info, nil}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Stat", stubRet) - - defer patch.Reset() - - contentLengthRequest, err := source.NewRequest(hdfsExistFileURL) - assert.Nil(t, err) - contentLengthRequest.Header.Add(source.Range, "0-12") - // exist file - length, err := sourceClient.GetContentLength(contentLengthRequest) - assert.Equal(t, hdfsExistFileContentLength, length) - assert.Nil(t, err) - -} - -// TestGetContentLength_Fail test file not exist, return error -func TestGetContentLength_Fail(t *testing.T) { - - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{nil, errors.New("stat /user/root/input/f3.txt: file does not exist")}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Stat", stubRet) - - defer patch.Reset() - hdfsNotExistFileRequest, err := source.NewRequest(hdfsExistFileURL) - assert.Nil(t, err) - hdfsNotExistFileRequest.Header.Add(source.Range, "0-10") - // not exist file - length, err := sourceClient.GetContentLength(hdfsNotExistFileRequest) - assert.Equal(t, hdfsNotExistFileContentLength, length) - assert.EqualError(t, err, "stat /user/root/input/f3.txt: file does not exist") -} - -// TestIsSupportRange_FileExist test file exist, return file support range -func TestIsSupportRange_FileExist(t *testing.T) { - var info os.FileInfo = fakeHDFSFileInfo{ - contents: hdfsExistFileContent, - } - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{info, nil}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Stat", stubRet) - - defer patch.Reset() - request, err := source.NewRequest(hdfsExistFileURL) - assert.Nil(t, err) - supportRange, err := sourceClient.IsSupportRange(request) - assert.Equal(t, true, supportRange) - assert.Nil(t, err) -} - -// TestIsSupportRange_FileNotExist test file not exist, return error and not support range -func TestIsSupportRange_FileNotExist(t *testing.T) { - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{nil, errors.New("stat /user/root/input/f3.txt: file does not exist")}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Stat", stubRet) - - defer patch.Reset() - - request, err := source.NewRequest(hdfsNotExistFileURL) - assert.Nil(t, err) - supportRange, err := sourceClient.IsSupportRange(request) - assert.Equal(t, false, supportRange) - assert.EqualError(t, err, "stat /user/root/input/f3.txt: file does not exist") -} - -func TestIsExpired_LastModifiedExpired(t *testing.T) { - lastModified, _ := time.Parse(source.LastModifiedLayout, hdfsExistFileLastModified) - var info os.FileInfo = fakeHDFSFileInfo{ - modtime: lastModified, - } - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{info, nil}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Stat", stubRet) - - defer patch.Reset() - - request, err := source.NewRequest(hdfsExistFileURL) - assert.Nil(t, err) - // header have Last-Modified - expired, err := sourceClient.IsExpired(request, &source.ExpireInfo{ - LastModified: time.Date(2020, 01, 01, 0, 0, 0, 0, time.UTC).Format(source.TimeFormat), - ETag: "", - }) - assert.Equal(t, true, expired) - assert.Nil(t, err) -} - -func TestIsExpired_LastModifiedNotExpired(t *testing.T) { - lastModified, _ := time.Parse(source.LastModifiedLayout, hdfsExistFileLastModified) - var info os.FileInfo = fakeHDFSFileInfo{ - modtime: lastModified, - } - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{info, nil}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Stat", stubRet) - - defer patch.Reset() - - request, err := source.NewRequest(hdfsExistFileURL) - assert.Nil(t, err) - // header have Last-Modified - expired, err := sourceClient.IsExpired(request, &source.ExpireInfo{ - LastModified: hdfsExistFileLastModified, - ETag: "", - }) - assert.Equal(t, false, expired) - assert.Nil(t, err) -} - -func Test_Download_FileExist_ByRange(t *testing.T) { - var reader *hdfs.FileReader = &hdfs.FileReader{} - patch := gomonkey.ApplyMethod(reflect.TypeOf(fakeHDFSClient), "Open", func(*hdfs.Client, string) (*hdfs.FileReader, error) { - return reader, nil - }) - patch.ApplyMethod(reflect.TypeOf(reader), "Seek", func(_ *hdfs.FileReader, offset int64, whence int) (int64, error) { - return 0 - hdfsExistFileContentLength, nil - }) - patch.ApplyMethod(reflect.TypeOf(reader), "Read", func(_ *hdfs.FileReader, b []byte) (int, error) { - bytes := []byte(hdfsExistFileContent) - return copy(b, bytes), io.EOF - }) - patch.ApplyMethodSeq(reflect.TypeOf(reader), "Stat", []gomonkey.OutputCell{ - { - Values: gomonkey.Params{ - fakeHDFSFileInfo{ - contents: hdfsExistFileContent, - }, - }, - }, - }) - defer patch.Reset() - - rang := &http.Range{StartIndex: 0, EndIndex: uint64(hdfsExistFileContentLength) - 1} - // exist file - request, err := source.NewRequestWithHeader(hdfsExistFileURL, map[string]string{ - source.Range: rang.String(), - }) - assert.Nil(t, err) - - response, err := sourceClient.Download(request) - assert.Nil(t, err) - data, _ := io.ReadAll(response.Body) - - assert.Equal(t, hdfsExistFileContent, string(data)) -} - -func TestDownload_FileNotExist(t *testing.T) { - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{nil, errors.New("open /user/root/input/f3.txt: file does not exist")}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Open", stubRet) - - defer patch.Reset() - - rang := http.Range{StartIndex: 0, EndIndex: uint64(hdfsExistFileContentLength)} - - request, err := source.NewRequestWithHeader(hdfsNotExistFileURL, map[string]string{ - source.Range: rang.String(), - }) - assert.Nil(t, err) - // not exist file - download, err := sourceClient.Download(request) - assert.Nil(t, download) - assert.EqualError(t, err, "open /user/root/input/f3.txt: file does not exist") -} - -func Test_DownloadWithResponseHeader_FileExist_ByRange(t *testing.T) { - lastModified, _ := time.Parse(source.LastModifiedLayout, hdfsExistFileLastModified) - var reader *hdfs.FileReader = &hdfs.FileReader{} - patches := gomonkey.NewPatches() - defer patches.Reset() - patches.ApplyMethod(reflect.TypeOf(fakeHDFSClient), "Open", func(*hdfs.Client, string) (*hdfs.FileReader, error) { - return reader, nil - }) - patches.ApplyMethod(reflect.TypeOf(reader), "Stat", func(_ *hdfs.FileReader) os.FileInfo { - return fakeHDFSFileInfo{ - contents: hdfsExistFileContent, - modtime: lastModified, - } - }) - - patches.ApplyMethod(reflect.TypeOf(reader), "Seek", func(_ *hdfs.FileReader, offset int64, whence int) (int64, error) { - return hdfsExistFileRangeEnd - hdfsExistFileRangeStart, nil - }) - patches.ApplyMethod(reflect.TypeOf(reader), "Read", func(_ *hdfs.FileReader, b []byte) (int, error) { - b = b[0 : hdfsExistFileRangeEnd-hdfsExistFileRangeStart] - bytes := []byte(hdfsExistFileContent) - copy(b, bytes[hdfsExistFileRangeStart:hdfsExistFileRangeEnd]) - return len(b), io.EOF - }) - - rang := http.Range{StartIndex: uint64(hdfsExistFileRangeStart), EndIndex: uint64(hdfsExistFileRangeEnd)} - request, err := source.NewRequest(hdfsExistFileURL) - assert.Nil(t, err) - request.Header.Add(source.Range, rang.String()) - response, err := sourceClient.Download(request) - assert.Nil(t, err) - assert.Equal(t, hdfsExistFileLastModified, response.ExpireInfo().LastModified) - - data, _ := io.ReadAll(response.Body) - assert.Equal(t, string(data), string([]byte(hdfsExistFileContent)[hdfsExistFileRangeStart:hdfsExistFileRangeEnd])) -} - -func TestDownloadWithResponseHeader_FileNotExist(t *testing.T) { - patch := gomonkey.ApplyMethod(reflect.TypeOf(fakeHDFSClient), "Open", func(*hdfs.Client, string) (*hdfs.FileReader, error) { - return nil, errors.New("open /user/root/input/f3.txt: file does not exist") - }) - defer patch.Reset() - - rang := http.Range{StartIndex: 0, EndIndex: uint64(hdfsExistFileContentLength)} - request, err := source.NewRequest(hdfsNotExistFileURL) - assert.Nil(t, err) - request.Header.Add(source.Range, rang.String()) - response, err := sourceClient.Download(request) - assert.EqualError(t, err, "open /user/root/input/f3.txt: file does not exist") - assert.Nil(t, response) -} - -func TestGetLastModified_FileExist(t *testing.T) { - lastModified, _ := time.Parse(source.LastModifiedLayout, hdfsExistFileLastModified) - var info os.FileInfo = fakeHDFSFileInfo{ - modtime: lastModified, - } - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{info, nil}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Stat", stubRet) - - defer patch.Reset() - - request, err := source.NewRequest(hdfsExistFileURL) - assert.Nil(t, err) - - lastModifiedMillis, err := sourceClient.GetLastModified(request) - assert.Nil(t, err) - assert.Equal(t, hdfsExistFileLastModifiedMillis, lastModifiedMillis) -} - -func TestGetLastModified_FileNotExist(t *testing.T) { - stubRet := []gomonkey.OutputCell{ - {Values: gomonkey.Params{nil, errors.New("stat /user/root/input/f3.txt: file does not exist")}}, - } - - patch := gomonkey.ApplyMethodSeq(reflect.TypeOf(fakeHDFSClient), "Stat", stubRet) - - defer patch.Reset() - - request, err := source.NewRequest(hdfsNotExistFileURL) - assert.Nil(t, err) - - lastModifiedMillis, err := sourceClient.GetLastModified(request) - assert.EqualError(t, err, "stat /user/root/input/f3.txt: file does not exist") - assert.Equal(t, hdfsNotExistLastModified, lastModifiedMillis) -} - -func TestNewHDFSSourceClient(t *testing.T) { - client := newHDFSSourceClient() - assert.NotNil(t, client) - - options := make([]HDFSSourceClientOption, 0) - - option := func(p *hdfsSourceClient) { - c, _ := hdfs.New(hdfsExistFileHost) - p.clientMap[hdfsExistFileHost] = c - } - options = append(options, option) - - newHDFSSourceClient := newHDFSSourceClient(options...) - - assert.IsType(t, &hdfsSourceClient{}, newHDFSSourceClient) - -} - -type fakeHDFSFileInfo struct { - dir bool - basename string - modtime time.Time - contents string -} - -func (f fakeHDFSFileInfo) Name() string { return f.basename } -func (f fakeHDFSFileInfo) Sys() interface{} { return nil } -func (f fakeHDFSFileInfo) ModTime() time.Time { return f.modtime } -func (f fakeHDFSFileInfo) IsDir() bool { return f.dir } -func (f fakeHDFSFileInfo) Size() int64 { return int64(len(f.contents)) } -func (f fakeHDFSFileInfo) Mode() os.FileMode { - if f.dir { - return 0755 | os.ModeDir - } - return 0644 -} diff --git a/pkg/source/mock/mock_source_client.go b/pkg/source/mock/mock_source_client.go deleted file mode 100644 index 3096c53ac80..00000000000 --- a/pkg/source/mock/mock_source_client.go +++ /dev/null @@ -1,110 +0,0 @@ -// Code generated by MockGen. DO NOT EDIT. -// Source: d7y.io/dragonfly/v2/pkg/source (interfaces: ResourceClient) - -// Package mock is a generated GoMock package. -package mock - -import ( - reflect "reflect" - - source "d7y.io/dragonfly/v2/pkg/source" - gomock "github.com/golang/mock/gomock" -) - -// MockResourceClient is a mock of ResourceClient interface. -type MockResourceClient struct { - ctrl *gomock.Controller - recorder *MockResourceClientMockRecorder -} - -// MockResourceClientMockRecorder is the mock recorder for MockResourceClient. -type MockResourceClientMockRecorder struct { - mock *MockResourceClient -} - -// NewMockResourceClient creates a new mock instance. -func NewMockResourceClient(ctrl *gomock.Controller) *MockResourceClient { - mock := &MockResourceClient{ctrl: ctrl} - mock.recorder = &MockResourceClientMockRecorder{mock} - return mock -} - -// EXPECT returns an object that allows the caller to indicate expected use. -func (m *MockResourceClient) EXPECT() *MockResourceClientMockRecorder { - return m.recorder -} - -// Download mocks base method. -func (m *MockResourceClient) Download(arg0 *source.Request) (*source.Response, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Download", arg0) - ret0, _ := ret[0].(*source.Response) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// Download indicates an expected call of Download. -func (mr *MockResourceClientMockRecorder) Download(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Download", reflect.TypeOf((*MockResourceClient)(nil).Download), arg0) -} - -// GetContentLength mocks base method. -func (m *MockResourceClient) GetContentLength(arg0 *source.Request) (int64, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetContentLength", arg0) - ret0, _ := ret[0].(int64) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetContentLength indicates an expected call of GetContentLength. -func (mr *MockResourceClientMockRecorder) GetContentLength(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetContentLength", reflect.TypeOf((*MockResourceClient)(nil).GetContentLength), arg0) -} - -// GetLastModified mocks base method. -func (m *MockResourceClient) GetLastModified(arg0 *source.Request) (int64, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLastModified", arg0) - ret0, _ := ret[0].(int64) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetLastModified indicates an expected call of GetLastModified. -func (mr *MockResourceClientMockRecorder) GetLastModified(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLastModified", reflect.TypeOf((*MockResourceClient)(nil).GetLastModified), arg0) -} - -// IsExpired mocks base method. -func (m *MockResourceClient) IsExpired(arg0 *source.Request, arg1 *source.ExpireInfo) (bool, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "IsExpired", arg0, arg1) - ret0, _ := ret[0].(bool) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// IsExpired indicates an expected call of IsExpired. -func (mr *MockResourceClientMockRecorder) IsExpired(arg0, arg1 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsExpired", reflect.TypeOf((*MockResourceClient)(nil).IsExpired), arg0, arg1) -} - -// IsSupportRange mocks base method. -func (m *MockResourceClient) IsSupportRange(arg0 *source.Request) (bool, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "IsSupportRange", arg0) - ret0, _ := ret[0].(bool) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// IsSupportRange indicates an expected call of IsSupportRange. -func (mr *MockResourceClientMockRecorder) IsSupportRange(arg0 interface{}) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsSupportRange", reflect.TypeOf((*MockResourceClient)(nil).IsSupportRange), arg0) -} diff --git a/pkg/source/mocks/mock_source_client.go b/pkg/source/mocks/mock_source_client.go new file mode 100644 index 00000000000..cd8407914f4 --- /dev/null +++ b/pkg/source/mocks/mock_source_client.go @@ -0,0 +1,288 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: source_client.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + url "net/url" + reflect "reflect" + + source "d7y.io/dragonfly/v2/pkg/source" + gomock "github.com/golang/mock/gomock" +) + +// MockResourceClient is a mock of ResourceClient interface. +type MockResourceClient struct { + ctrl *gomock.Controller + recorder *MockResourceClientMockRecorder +} + +// MockResourceClientMockRecorder is the mock recorder for MockResourceClient. +type MockResourceClientMockRecorder struct { + mock *MockResourceClient +} + +// NewMockResourceClient creates a new mock instance. +func NewMockResourceClient(ctrl *gomock.Controller) *MockResourceClient { + mock := &MockResourceClient{ctrl: ctrl} + mock.recorder = &MockResourceClientMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockResourceClient) EXPECT() *MockResourceClientMockRecorder { + return m.recorder +} + +// Download mocks base method. +func (m *MockResourceClient) Download(request *source.Request) (*source.Response, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Download", request) + ret0, _ := ret[0].(*source.Response) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Download indicates an expected call of Download. +func (mr *MockResourceClientMockRecorder) Download(request interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Download", reflect.TypeOf((*MockResourceClient)(nil).Download), request) +} + +// GetContentLength mocks base method. +func (m *MockResourceClient) GetContentLength(request *source.Request) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetContentLength", request) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetContentLength indicates an expected call of GetContentLength. +func (mr *MockResourceClientMockRecorder) GetContentLength(request interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetContentLength", reflect.TypeOf((*MockResourceClient)(nil).GetContentLength), request) +} + +// GetLastModified mocks base method. +func (m *MockResourceClient) GetLastModified(request *source.Request) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetLastModified", request) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetLastModified indicates an expected call of GetLastModified. +func (mr *MockResourceClientMockRecorder) GetLastModified(request interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLastModified", reflect.TypeOf((*MockResourceClient)(nil).GetLastModified), request) +} + +// IsExpired mocks base method. +func (m *MockResourceClient) IsExpired(request *source.Request, info *source.ExpireInfo) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "IsExpired", request, info) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// IsExpired indicates an expected call of IsExpired. +func (mr *MockResourceClientMockRecorder) IsExpired(request, info interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsExpired", reflect.TypeOf((*MockResourceClient)(nil).IsExpired), request, info) +} + +// IsSupportRange mocks base method. +func (m *MockResourceClient) IsSupportRange(request *source.Request) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "IsSupportRange", request) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// IsSupportRange indicates an expected call of IsSupportRange. +func (mr *MockResourceClientMockRecorder) IsSupportRange(request interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsSupportRange", reflect.TypeOf((*MockResourceClient)(nil).IsSupportRange), request) +} + +// MockResourceLister is a mock of ResourceLister interface. +type MockResourceLister struct { + ctrl *gomock.Controller + recorder *MockResourceListerMockRecorder +} + +// MockResourceListerMockRecorder is the mock recorder for MockResourceLister. +type MockResourceListerMockRecorder struct { + mock *MockResourceLister +} + +// NewMockResourceLister creates a new mock instance. +func NewMockResourceLister(ctrl *gomock.Controller) *MockResourceLister { + mock := &MockResourceLister{ctrl: ctrl} + mock.recorder = &MockResourceListerMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockResourceLister) EXPECT() *MockResourceListerMockRecorder { + return m.recorder +} + +// List mocks base method. +func (m *MockResourceLister) List(request *source.Request) ([]*url.URL, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "List", request) + ret0, _ := ret[0].([]*url.URL) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// List indicates an expected call of List. +func (mr *MockResourceListerMockRecorder) List(request interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockResourceLister)(nil).List), request) +} + +// MockClientManager is a mock of ClientManager interface. +type MockClientManager struct { + ctrl *gomock.Controller + recorder *MockClientManagerMockRecorder +} + +// MockClientManagerMockRecorder is the mock recorder for MockClientManager. +type MockClientManagerMockRecorder struct { + mock *MockClientManager +} + +// NewMockClientManager creates a new mock instance. +func NewMockClientManager(ctrl *gomock.Controller) *MockClientManager { + mock := &MockClientManager{ctrl: ctrl} + mock.recorder = &MockClientManagerMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockClientManager) EXPECT() *MockClientManagerMockRecorder { + return m.recorder +} + +// GetClient mocks base method. +func (m *MockClientManager) GetClient(scheme string, options ...source.Option) (source.ResourceClient, bool) { + m.ctrl.T.Helper() + varargs := []interface{}{scheme} + for _, a := range options { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetClient", varargs...) + ret0, _ := ret[0].(source.ResourceClient) + ret1, _ := ret[1].(bool) + return ret0, ret1 +} + +// GetClient indicates an expected call of GetClient. +func (mr *MockClientManagerMockRecorder) GetClient(scheme interface{}, options ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{scheme}, options...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetClient", reflect.TypeOf((*MockClientManager)(nil).GetClient), varargs...) +} + +// ListClients mocks base method. +func (m *MockClientManager) ListClients() []string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ListClients") + ret0, _ := ret[0].([]string) + return ret0 +} + +// ListClients indicates an expected call of ListClients. +func (mr *MockClientManagerMockRecorder) ListClients() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListClients", reflect.TypeOf((*MockClientManager)(nil).ListClients)) +} + +// Register mocks base method. +func (m *MockClientManager) Register(scheme string, resourceClient source.ResourceClient, adapter source.RequestAdapter, hook ...source.Hook) error { + m.ctrl.T.Helper() + varargs := []interface{}{scheme, resourceClient, adapter} + for _, a := range hook { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "Register", varargs...) + ret0, _ := ret[0].(error) + return ret0 +} + +// Register indicates an expected call of Register. +func (mr *MockClientManagerMockRecorder) Register(scheme, resourceClient, adapter interface{}, hook ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{scheme, resourceClient, adapter}, hook...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Register", reflect.TypeOf((*MockClientManager)(nil).Register), varargs...) +} + +// UnRegister mocks base method. +func (m *MockClientManager) UnRegister(scheme string) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "UnRegister", scheme) +} + +// UnRegister indicates an expected call of UnRegister. +func (mr *MockClientManagerMockRecorder) UnRegister(scheme interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnRegister", reflect.TypeOf((*MockClientManager)(nil).UnRegister), scheme) +} + +// MockHook is a mock of Hook interface. +type MockHook struct { + ctrl *gomock.Controller + recorder *MockHookMockRecorder +} + +// MockHookMockRecorder is the mock recorder for MockHook. +type MockHookMockRecorder struct { + mock *MockHook +} + +// NewMockHook creates a new mock instance. +func NewMockHook(ctrl *gomock.Controller) *MockHook { + mock := &MockHook{ctrl: ctrl} + mock.recorder = &MockHookMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockHook) EXPECT() *MockHookMockRecorder { + return m.recorder +} + +// AfterResponse mocks base method. +func (m *MockHook) AfterResponse(response *source.Response) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AfterResponse", response) + ret0, _ := ret[0].(error) + return ret0 +} + +// AfterResponse indicates an expected call of AfterResponse. +func (mr *MockHookMockRecorder) AfterResponse(response interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AfterResponse", reflect.TypeOf((*MockHook)(nil).AfterResponse), response) +} + +// BeforeRequest mocks base method. +func (m *MockHook) BeforeRequest(request *source.Request) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "BeforeRequest", request) + ret0, _ := ret[0].(error) + return ret0 +} + +// BeforeRequest indicates an expected call of BeforeRequest. +func (mr *MockHookMockRecorder) BeforeRequest(request interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BeforeRequest", reflect.TypeOf((*MockHook)(nil).BeforeRequest), request) +} diff --git a/pkg/source/source_client.go b/pkg/source/source_client.go index ec79f98f156..f3fd8935b2e 100644 --- a/pkg/source/source_client.go +++ b/pkg/source/source_client.go @@ -13,7 +13,8 @@ * See the License for the specific language governing permissions and * limitations under the License. */ -//go:generate mockgen -destination ./mock/mock_source_client.go -package mock d7y.io/dragonfly/v2/pkg/source ResourceClient + +//go:generate mockgen -destination mocks/mock_source_client.go -source source_client.go -package mocks package source @@ -120,7 +121,7 @@ type ResourceLister interface { type ClientManager interface { // Register registers a source client with scheme - Register(scheme string, resourceClient ResourceClient, adapter requestAdapter, hook ...Hook) error + Register(scheme string, resourceClient ResourceClient, adapter RequestAdapter, hook ...Hook) error // UnRegister revoke a source client from manager UnRegister(scheme string) @@ -155,7 +156,7 @@ func UpdatePluginDir(pluginDir string) { _defaultManager.(*clientManager).pluginDir = pluginDir } -func (m *clientManager) Register(scheme string, resourceClient ResourceClient, adaptor requestAdapter, hooks ...Hook) error { +func (m *clientManager) Register(scheme string, resourceClient ResourceClient, adaptor RequestAdapter, hooks ...Hook) error { scheme = strings.ToLower(scheme) m.mu.Lock() defer m.mu.Unlock() @@ -230,7 +231,7 @@ func (m *clientManager) GetClient(scheme string, options ...Option) (ResourceCli return client, true } -func Register(scheme string, resourceClient ResourceClient, adaptor requestAdapter, hooks ...Hook) error { +func Register(scheme string, resourceClient ResourceClient, adaptor RequestAdapter, hooks ...Hook) error { return _defaultManager.Register(scheme, resourceClient, adaptor, hooks...) } @@ -242,7 +243,7 @@ func ListClients() []string { return _defaultManager.ListClients() } -type requestAdapter func(request *Request) *Request +type RequestAdapter func(request *Request) *Request // Hook TODO hook type Hook interface { @@ -251,7 +252,7 @@ type Hook interface { } type clientWrapper struct { - adapter requestAdapter + adapter RequestAdapter hooks []Hook rc ResourceClient } diff --git a/scheduler/job/job.go b/scheduler/job/job.go index 11e14ed084b..bc54a6bf1a5 100644 --- a/scheduler/job/job.go +++ b/scheduler/job/job.go @@ -14,6 +14,8 @@ * limitations under the License. */ +//go:generate mockgen -destination mocks/job_mock.go -source job.go -package mocks + package job import ( diff --git a/scheduler/job/mocks/job_mock.go b/scheduler/job/mocks/job_mock.go new file mode 100644 index 00000000000..63c1fcc99a0 --- /dev/null +++ b/scheduler/job/mocks/job_mock.go @@ -0,0 +1,58 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: job.go + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockJob is a mock of Job interface. +type MockJob struct { + ctrl *gomock.Controller + recorder *MockJobMockRecorder +} + +// MockJobMockRecorder is the mock recorder for MockJob. +type MockJobMockRecorder struct { + mock *MockJob +} + +// NewMockJob creates a new mock instance. +func NewMockJob(ctrl *gomock.Controller) *MockJob { + mock := &MockJob{ctrl: ctrl} + mock.recorder = &MockJobMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockJob) EXPECT() *MockJobMockRecorder { + return m.recorder +} + +// Serve mocks base method. +func (m *MockJob) Serve() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Serve") +} + +// Serve indicates an expected call of Serve. +func (mr *MockJobMockRecorder) Serve() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Serve", reflect.TypeOf((*MockJob)(nil).Serve)) +} + +// Stop mocks base method. +func (m *MockJob) Stop() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Stop") +} + +// Stop indicates an expected call of Stop. +func (mr *MockJobMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockJob)(nil).Stop)) +}