// Code generated by mockery v2.42.2. DO NOT EDIT. package task import ( context "context" q "github.com/goharbor/harbor/src/lib/q" mock "github.com/stretchr/testify/mock" task "github.com/goharbor/harbor/src/pkg/task" time "time" ) // ExecutionManager is an autogenerated mock type for the ExecutionManager type type ExecutionManager struct { mock.Mock } // Count provides a mock function with given fields: ctx, query func (_m *ExecutionManager) Count(ctx context.Context, query *q.Query) (int64, error) { ret := _m.Called(ctx, query) if len(ret) == 0 { panic("no return value specified for Count") } var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, *q.Query) (int64, error)); ok { return rf(ctx, query) } if rf, ok := ret.Get(0).(func(context.Context, *q.Query) int64); ok { r0 = rf(ctx, query) } else { r0 = ret.Get(0).(int64) } if rf, ok := ret.Get(1).(func(context.Context, *q.Query) error); ok { r1 = rf(ctx, query) } else { r1 = ret.Error(1) } return r0, r1 } // Create provides a mock function with given fields: ctx, vendorType, vendorID, trigger, extraAttrs func (_m *ExecutionManager) Create(ctx context.Context, vendorType string, vendorID int64, trigger string, extraAttrs ...map[string]interface{}) (int64, error) { _va := make([]interface{}, len(extraAttrs)) for _i := range extraAttrs { _va[_i] = extraAttrs[_i] } var _ca []interface{} _ca = append(_ca, ctx, vendorType, vendorID, trigger) _ca = append(_ca, _va...) ret := _m.Called(_ca...) if len(ret) == 0 { panic("no return value specified for Create") } var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, int64, string, ...map[string]interface{}) (int64, error)); ok { return rf(ctx, vendorType, vendorID, trigger, extraAttrs...) } if rf, ok := ret.Get(0).(func(context.Context, string, int64, string, ...map[string]interface{}) int64); ok { r0 = rf(ctx, vendorType, vendorID, trigger, extraAttrs...) } else { r0 = ret.Get(0).(int64) } if rf, ok := ret.Get(1).(func(context.Context, string, int64, string, ...map[string]interface{}) error); ok { r1 = rf(ctx, vendorType, vendorID, trigger, extraAttrs...) } else { r1 = ret.Error(1) } return r0, r1 } // Delete provides a mock function with given fields: ctx, id func (_m *ExecutionManager) Delete(ctx context.Context, id int64) error { ret := _m.Called(ctx, id) if len(ret) == 0 { panic("no return value specified for Delete") } var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64) error); ok { r0 = rf(ctx, id) } else { r0 = ret.Error(0) } return r0 } // DeleteByVendor provides a mock function with given fields: ctx, vendorType, vendorID func (_m *ExecutionManager) DeleteByVendor(ctx context.Context, vendorType string, vendorID int64) error { ret := _m.Called(ctx, vendorType, vendorID) if len(ret) == 0 { panic("no return value specified for DeleteByVendor") } var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, int64) error); ok { r0 = rf(ctx, vendorType, vendorID) } else { r0 = ret.Error(0) } return r0 } // Get provides a mock function with given fields: ctx, id func (_m *ExecutionManager) Get(ctx context.Context, id int64) (*task.Execution, error) { ret := _m.Called(ctx, id) if len(ret) == 0 { panic("no return value specified for Get") } var r0 *task.Execution var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (*task.Execution, error)); ok { return rf(ctx, id) } if rf, ok := ret.Get(0).(func(context.Context, int64) *task.Execution); ok { r0 = rf(ctx, id) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*task.Execution) } } if rf, ok := ret.Get(1).(func(context.Context, int64) error); ok { r1 = rf(ctx, id) } else { r1 = ret.Error(1) } return r0, r1 } // List provides a mock function with given fields: ctx, query func (_m *ExecutionManager) List(ctx context.Context, query *q.Query) ([]*task.Execution, error) { ret := _m.Called(ctx, query) if len(ret) == 0 { panic("no return value specified for List") } var r0 []*task.Execution var r1 error if rf, ok := ret.Get(0).(func(context.Context, *q.Query) ([]*task.Execution, error)); ok { return rf(ctx, query) } if rf, ok := ret.Get(0).(func(context.Context, *q.Query) []*task.Execution); ok { r0 = rf(ctx, query) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*task.Execution) } } if rf, ok := ret.Get(1).(func(context.Context, *q.Query) error); ok { r1 = rf(ctx, query) } else { r1 = ret.Error(1) } return r0, r1 } // MarkDone provides a mock function with given fields: ctx, id, message func (_m *ExecutionManager) MarkDone(ctx context.Context, id int64, message string) error { ret := _m.Called(ctx, id, message) if len(ret) == 0 { panic("no return value specified for MarkDone") } var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, string) error); ok { r0 = rf(ctx, id, message) } else { r0 = ret.Error(0) } return r0 } // MarkError provides a mock function with given fields: ctx, id, message func (_m *ExecutionManager) MarkError(ctx context.Context, id int64, message string) error { ret := _m.Called(ctx, id, message) if len(ret) == 0 { panic("no return value specified for MarkError") } var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, string) error); ok { r0 = rf(ctx, id, message) } else { r0 = ret.Error(0) } return r0 } // Stop provides a mock function with given fields: ctx, id func (_m *ExecutionManager) Stop(ctx context.Context, id int64) error { ret := _m.Called(ctx, id) if len(ret) == 0 { panic("no return value specified for Stop") } var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64) error); ok { r0 = rf(ctx, id) } else { r0 = ret.Error(0) } return r0 } // StopAndWait provides a mock function with given fields: ctx, id, timeout func (_m *ExecutionManager) StopAndWait(ctx context.Context, id int64, timeout time.Duration) error { ret := _m.Called(ctx, id, timeout) if len(ret) == 0 { panic("no return value specified for StopAndWait") } var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, time.Duration) error); ok { r0 = rf(ctx, id, timeout) } else { r0 = ret.Error(0) } return r0 } // StopAndWaitWithError provides a mock function with given fields: ctx, id, timeout, origError func (_m *ExecutionManager) StopAndWaitWithError(ctx context.Context, id int64, timeout time.Duration, origError error) error { ret := _m.Called(ctx, id, timeout, origError) if len(ret) == 0 { panic("no return value specified for StopAndWaitWithError") } var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, time.Duration, error) error); ok { r0 = rf(ctx, id, timeout, origError) } else { r0 = ret.Error(0) } return r0 } // UpdateExtraAttrs provides a mock function with given fields: ctx, id, extraAttrs func (_m *ExecutionManager) UpdateExtraAttrs(ctx context.Context, id int64, extraAttrs map[string]interface{}) error { ret := _m.Called(ctx, id, extraAttrs) if len(ret) == 0 { panic("no return value specified for UpdateExtraAttrs") } var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, map[string]interface{}) error); ok { r0 = rf(ctx, id, extraAttrs) } else { r0 = ret.Error(0) } return r0 } // NewExecutionManager creates a new instance of ExecutionManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. // The first argument is typically a *testing.T value. func NewExecutionManager(t interface { mock.TestingT Cleanup(func()) }) *ExecutionManager { mock := &ExecutionManager{} mock.Mock.Test(t) t.Cleanup(func() { mock.AssertExpectations(t) }) return mock }