From 0f5fe2b369c85ae94d2603d056ee8c617ac07f42 Mon Sep 17 00:00:00 2001 From: "wangluming.wlm" Date: Wed, 5 Nov 2025 22:07:14 +0800 Subject: [PATCH] mockgen format --- backend/infra/limiter/mocks/rate_limiter.go | 11 +- .../limiter/mocks/rate_limiter_factory.go | 9 +- .../session/mocks/session_service.go | 72 ++++ .../mocks/latest_write_tracker.go | 35 +- .../data/domain/component/conf/mocks/conf.go | 1 + .../component/rpc/mocks/user_provider.go | 9 +- .../component/userinfo/mocks/userinfo.go | 9 +- .../data/domain/tag/repo/mocks/tag_mock.go | 131 +++---- .../tag/service/mocks/tag_service_mock.go | 129 ++++--- .../metrics/mocks/openapi_evaluation_set.go | 53 +++ .../entity/mocks/expt_scheduler_mock.go | 62 +-- .../mocks/evaluator_event_publisher_mock.go | 14 +- .../events/mocks/expt_event_publisher_mock.go | 65 ++-- .../domain/repo/mocks/evaluator_mock.go | 110 +++--- .../repo/mocks/evaluator_record_mock.go | 38 +- .../domain/repo/mocks/ratelimiter_mock.go | 14 +- .../evaluator/mysql/mocks/evaluator_mock.go | 96 ++--- .../mysql/mocks/evaluator_record_mock.go | 46 ++- .../mysql/mocks/evaluator_version_mock.go | 96 ++--- .../infra/repo/experiment/ck/mocks/expt.go | 30 +- .../experiment/mysql/mocks/annotate_record.go | 47 ++- .../infra/repo/experiment/mysql/mocks/expt.go | 70 ++-- .../mysql/mocks/expt_aggr_result.go | 89 +++-- .../mysql/mocks/expt_evaluator_ref.go | 22 +- .../expt_insight_analysis_feedback_comment.go | 52 +-- .../expt_insight_analysis_feedback_vote.go | 44 ++- .../mocks/expt_insight_analysis_record.go | 52 +-- .../mysql/mocks/expt_item_result.go | 142 +++---- .../mysql/mocks/expt_result_export_record.go | 44 ++- .../experiment/mysql/mocks/expt_run_log.go | 46 ++- .../repo/experiment/mysql/mocks/expt_stats.go | 54 +-- .../mocks/expt_turn_annotate_record_ref.go | 68 ++-- .../mocks/expt_turn_evaluator_result_ref.go | 36 +- .../mysql/mocks/expt_turn_result.go | 179 ++++----- .../expt_turn_result_filter_key_mapping.go | 36 +- .../repo/experiment/redis/dao/mocks/quota.go | 33 +- .../domain/authn/repo/mocks/authn_repo.go | 62 +-- .../domain/user/service/mocks/user_service.go | 80 ++-- .../domain/component/metrics/mocks/metrics.go | 4 +- .../component/rpc/mocks/auth_provider.go | 8 +- .../rpc/mocks/dataset_provider_mock.go | 57 +-- .../domain/component/rpc/mocks/evaluator.go | 25 +- .../domain/component/rpc/mocks/file.go | 1 + .../domain/component/rpc/mocks/tag.go | 1 + .../domain/component/rpc/mocks/user.go | 9 +- .../domain/task/repo/mocks/Task.go | 277 ++++++------- .../domain/task/service/mocks/task_service.go | 41 +- .../tracehub/mocks/trace_hub_service.go | 33 +- .../collector/confmap/mocks/provider.go | 85 ++++ .../entity/collector/mocks/conf_provider.go | 57 +++ .../domain/trace/repo/mocks/view.go | 41 +- .../domain/trace/service/mocks/ingestion.go | 17 +- .../trace/service/mocks/span_processor.go | 1 + .../mocks/trace_export_service_mock.go | 17 +- .../trace/service/mocks/trace_service.go | 137 +++---- .../infra/repo/mysql/mocks/task.go | 148 +++++++ .../infra/repo/mysql/mocks/task_run.go | 132 +++++++ .../infra/repo/redis/dao/mocks/Task_dao.go | 205 ++++++++++ .../repo/redis/dao/mocks/task_run_dao.go | 114 ++++++ .../infra/rpc/dataset/mocks/mock_dataset.go | 10 + .../mocks/mock_datasetservice_client.go | 263 ++++++------- .../mocks/mock_evaluation_set.go | 10 + .../mocks/mock_evaluationsetservice_client.go | 363 ++++++++++++++++++ 63 files changed, 2840 insertions(+), 1402 deletions(-) create mode 100644 backend/infra/middleware/session/mocks/session_service.go create mode 100644 backend/modules/evaluation/domain/component/metrics/mocks/openapi_evaluation_set.go create mode 100644 backend/modules/observability/domain/trace/entity/collector/confmap/mocks/provider.go create mode 100644 backend/modules/observability/domain/trace/entity/collector/mocks/conf_provider.go create mode 100644 backend/modules/observability/infra/repo/mysql/mocks/task.go create mode 100644 backend/modules/observability/infra/repo/mysql/mocks/task_run.go create mode 100644 backend/modules/observability/infra/repo/redis/dao/mocks/Task_dao.go create mode 100644 backend/modules/observability/infra/repo/redis/dao/mocks/task_run_dao.go create mode 100644 backend/modules/observability/infra/rpc/dataset/mocks/mock_dataset.go create mode 100644 backend/modules/observability/infra/rpc/evaluationset/mocks/mock_evaluation_set.go create mode 100644 backend/modules/observability/infra/rpc/evaluationset/mocks/mock_evaluationsetservice_client.go diff --git a/backend/infra/limiter/mocks/rate_limiter.go b/backend/infra/limiter/mocks/rate_limiter.go index a36427311..81e2fb50c 100644 --- a/backend/infra/limiter/mocks/rate_limiter.go +++ b/backend/infra/limiter/mocks/rate_limiter.go @@ -21,6 +21,7 @@ import ( type MockIRateLimiter struct { ctrl *gomock.Controller recorder *MockIRateLimiterMockRecorder + isgomock struct{} } // MockIRateLimiterMockRecorder is the mock recorder for MockIRateLimiter. @@ -41,10 +42,10 @@ func (m *MockIRateLimiter) EXPECT() *MockIRateLimiterMockRecorder { } // AllowN mocks base method. -func (m *MockIRateLimiter) AllowN(arg0 context.Context, arg1 string, arg2 int, arg3 ...limiter.LimitOptionFn) (*limiter.Result, error) { +func (m *MockIRateLimiter) AllowN(ctx context.Context, key string, n int, opts ...limiter.LimitOptionFn) (*limiter.Result, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, key, n} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "AllowN", varargs...) @@ -54,8 +55,8 @@ func (m *MockIRateLimiter) AllowN(arg0 context.Context, arg1 string, arg2 int, a } // AllowN indicates an expected call of AllowN. -func (mr *MockIRateLimiterMockRecorder) AllowN(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call { +func (mr *MockIRateLimiterMockRecorder) AllowN(ctx, key, n any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, key, n}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AllowN", reflect.TypeOf((*MockIRateLimiter)(nil).AllowN), varargs...) } diff --git a/backend/infra/limiter/mocks/rate_limiter_factory.go b/backend/infra/limiter/mocks/rate_limiter_factory.go index 3a6c0154a..57612b48f 100644 --- a/backend/infra/limiter/mocks/rate_limiter_factory.go +++ b/backend/infra/limiter/mocks/rate_limiter_factory.go @@ -20,6 +20,7 @@ import ( type MockIRateLimiterFactory struct { ctrl *gomock.Controller recorder *MockIRateLimiterFactoryMockRecorder + isgomock struct{} } // MockIRateLimiterFactoryMockRecorder is the mock recorder for MockIRateLimiterFactory. @@ -40,10 +41,10 @@ func (m *MockIRateLimiterFactory) EXPECT() *MockIRateLimiterFactoryMockRecorder } // NewRateLimiter mocks base method. -func (m *MockIRateLimiterFactory) NewRateLimiter(arg0 ...limiter.FactoryOptionFn) limiter.IRateLimiter { +func (m *MockIRateLimiterFactory) NewRateLimiter(opts ...limiter.FactoryOptionFn) limiter.IRateLimiter { m.ctrl.T.Helper() varargs := []any{} - for _, a := range arg0 { + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "NewRateLimiter", varargs...) @@ -52,7 +53,7 @@ func (m *MockIRateLimiterFactory) NewRateLimiter(arg0 ...limiter.FactoryOptionFn } // NewRateLimiter indicates an expected call of NewRateLimiter. -func (mr *MockIRateLimiterFactoryMockRecorder) NewRateLimiter(arg0 ...any) *gomock.Call { +func (mr *MockIRateLimiterFactoryMockRecorder) NewRateLimiter(opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewRateLimiter", reflect.TypeOf((*MockIRateLimiterFactory)(nil).NewRateLimiter), arg0...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewRateLimiter", reflect.TypeOf((*MockIRateLimiterFactory)(nil).NewRateLimiter), opts...) } diff --git a/backend/infra/middleware/session/mocks/session_service.go b/backend/infra/middleware/session/mocks/session_service.go new file mode 100644 index 000000000..beede4f94 --- /dev/null +++ b/backend/infra/middleware/session/mocks/session_service.go @@ -0,0 +1,72 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/infra/middleware/session (interfaces: ISessionService) +// +// Generated by this command: +// +// mockgen -destination=mocks/session_service.go -package=mock_session . ISessionService +// + +// Package mock_session is a generated GoMock package. +package mock_session + +import ( + context "context" + reflect "reflect" + + session "github.com/coze-dev/coze-loop/backend/infra/middleware/session" + gomock "go.uber.org/mock/gomock" +) + +// MockISessionService is a mock of ISessionService interface. +type MockISessionService struct { + ctrl *gomock.Controller + recorder *MockISessionServiceMockRecorder + isgomock struct{} +} + +// MockISessionServiceMockRecorder is the mock recorder for MockISessionService. +type MockISessionServiceMockRecorder struct { + mock *MockISessionService +} + +// NewMockISessionService creates a new mock instance. +func NewMockISessionService(ctrl *gomock.Controller) *MockISessionService { + mock := &MockISessionService{ctrl: ctrl} + mock.recorder = &MockISessionServiceMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockISessionService) EXPECT() *MockISessionServiceMockRecorder { + return m.recorder +} + +// GenerateSessionKey mocks base method. +func (m *MockISessionService) GenerateSessionKey(ctx context.Context, arg1 *session.Session) (string, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GenerateSessionKey", ctx, arg1) + ret0, _ := ret[0].(string) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GenerateSessionKey indicates an expected call of GenerateSessionKey. +func (mr *MockISessionServiceMockRecorder) GenerateSessionKey(ctx, arg1 any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GenerateSessionKey", reflect.TypeOf((*MockISessionService)(nil).GenerateSessionKey), ctx, arg1) +} + +// ValidateSession mocks base method. +func (m *MockISessionService) ValidateSession(ctx context.Context, sessionID string) (*session.Session, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ValidateSession", ctx, sessionID) + ret0, _ := ret[0].(*session.Session) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ValidateSession indicates an expected call of ValidateSession. +func (mr *MockISessionServiceMockRecorder) ValidateSession(ctx, sessionID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateSession", reflect.TypeOf((*MockISessionService)(nil).ValidateSession), ctx, sessionID) +} diff --git a/backend/infra/platestwrite/mocks/latest_write_tracker.go b/backend/infra/platestwrite/mocks/latest_write_tracker.go index 382a0f98d..adbdaf1b5 100644 --- a/backend/infra/platestwrite/mocks/latest_write_tracker.go +++ b/backend/infra/platestwrite/mocks/latest_write_tracker.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/infra/platestwrite (interfaces: ILatestWriteTracker) +// +// Generated by this command: +// +// mockgen -destination ./mocks/latest_write_tracker.go --package mocks . ILatestWriteTracker +// // Package mocks is a generated GoMock package. package mocks @@ -8,15 +13,15 @@ import ( context "context" reflect "reflect" - gomock "go.uber.org/mock/gomock" - platestwrite "github.com/coze-dev/coze-loop/backend/infra/platestwrite" + gomock "go.uber.org/mock/gomock" ) // MockILatestWriteTracker is a mock of ILatestWriteTracker interface. type MockILatestWriteTracker struct { ctrl *gomock.Controller recorder *MockILatestWriteTrackerMockRecorder + isgomock struct{} } // MockILatestWriteTrackerMockRecorder is the mock recorder for MockILatestWriteTracker. @@ -37,46 +42,46 @@ func (m *MockILatestWriteTracker) EXPECT() *MockILatestWriteTrackerMockRecorder } // CheckWriteFlagByID mocks base method. -func (m *MockILatestWriteTracker) CheckWriteFlagByID(arg0 context.Context, arg1 platestwrite.ResourceType, arg2 int64) bool { +func (m *MockILatestWriteTracker) CheckWriteFlagByID(ctx context.Context, resourceType platestwrite.ResourceType, id int64) bool { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CheckWriteFlagByID", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "CheckWriteFlagByID", ctx, resourceType, id) ret0, _ := ret[0].(bool) return ret0 } // CheckWriteFlagByID indicates an expected call of CheckWriteFlagByID. -func (mr *MockILatestWriteTrackerMockRecorder) CheckWriteFlagByID(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockILatestWriteTrackerMockRecorder) CheckWriteFlagByID(ctx, resourceType, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckWriteFlagByID", reflect.TypeOf((*MockILatestWriteTracker)(nil).CheckWriteFlagByID), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckWriteFlagByID", reflect.TypeOf((*MockILatestWriteTracker)(nil).CheckWriteFlagByID), ctx, resourceType, id) } // CheckWriteFlagBySearchParam mocks base method. -func (m *MockILatestWriteTracker) CheckWriteFlagBySearchParam(arg0 context.Context, arg1 platestwrite.ResourceType, arg2 string) bool { +func (m *MockILatestWriteTracker) CheckWriteFlagBySearchParam(ctx context.Context, resourceType platestwrite.ResourceType, searchParam string) bool { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CheckWriteFlagBySearchParam", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "CheckWriteFlagBySearchParam", ctx, resourceType, searchParam) ret0, _ := ret[0].(bool) return ret0 } // CheckWriteFlagBySearchParam indicates an expected call of CheckWriteFlagBySearchParam. -func (mr *MockILatestWriteTrackerMockRecorder) CheckWriteFlagBySearchParam(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockILatestWriteTrackerMockRecorder) CheckWriteFlagBySearchParam(ctx, resourceType, searchParam any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckWriteFlagBySearchParam", reflect.TypeOf((*MockILatestWriteTracker)(nil).CheckWriteFlagBySearchParam), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckWriteFlagBySearchParam", reflect.TypeOf((*MockILatestWriteTracker)(nil).CheckWriteFlagBySearchParam), ctx, resourceType, searchParam) } // SetWriteFlag mocks base method. -func (m *MockILatestWriteTracker) SetWriteFlag(arg0 context.Context, arg1 platestwrite.ResourceType, arg2 int64, arg3 ...platestwrite.SetWriteFlagOpt) { +func (m *MockILatestWriteTracker) SetWriteFlag(ctx context.Context, resourceType platestwrite.ResourceType, resourceID int64, opts ...platestwrite.SetWriteFlagOpt) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, resourceType, resourceID} + for _, a := range opts { varargs = append(varargs, a) } m.ctrl.Call(m, "SetWriteFlag", varargs...) } // SetWriteFlag indicates an expected call of SetWriteFlag. -func (mr *MockILatestWriteTrackerMockRecorder) SetWriteFlag(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockILatestWriteTrackerMockRecorder) SetWriteFlag(ctx, resourceType, resourceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, resourceType, resourceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetWriteFlag", reflect.TypeOf((*MockILatestWriteTracker)(nil).SetWriteFlag), varargs...) } diff --git a/backend/modules/data/domain/component/conf/mocks/conf.go b/backend/modules/data/domain/component/conf/mocks/conf.go index f450af452..dbbd91059 100644 --- a/backend/modules/data/domain/component/conf/mocks/conf.go +++ b/backend/modules/data/domain/component/conf/mocks/conf.go @@ -20,6 +20,7 @@ import ( type MockIConfig struct { ctrl *gomock.Controller recorder *MockIConfigMockRecorder + isgomock struct{} } // MockIConfigMockRecorder is the mock recorder for MockIConfig. diff --git a/backend/modules/data/domain/component/rpc/mocks/user_provider.go b/backend/modules/data/domain/component/rpc/mocks/user_provider.go index 699798a92..34e5f3ebd 100644 --- a/backend/modules/data/domain/component/rpc/mocks/user_provider.go +++ b/backend/modules/data/domain/component/rpc/mocks/user_provider.go @@ -21,6 +21,7 @@ import ( type MockIUserProvider struct { ctrl *gomock.Controller recorder *MockIUserProviderMockRecorder + isgomock struct{} } // MockIUserProviderMockRecorder is the mock recorder for MockIUserProvider. @@ -41,16 +42,16 @@ func (m *MockIUserProvider) EXPECT() *MockIUserProviderMockRecorder { } // MGetUserInfo mocks base method. -func (m *MockIUserProvider) MGetUserInfo(arg0 context.Context, arg1 []string) ([]*entity.UserInfo, error) { +func (m *MockIUserProvider) MGetUserInfo(ctx context.Context, userIDs []string) ([]*entity.UserInfo, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "MGetUserInfo", arg0, arg1) + ret := m.ctrl.Call(m, "MGetUserInfo", ctx, userIDs) ret0, _ := ret[0].([]*entity.UserInfo) ret1, _ := ret[1].(error) return ret0, ret1 } // MGetUserInfo indicates an expected call of MGetUserInfo. -func (mr *MockIUserProviderMockRecorder) MGetUserInfo(arg0, arg1 any) *gomock.Call { +func (mr *MockIUserProviderMockRecorder) MGetUserInfo(ctx, userIDs any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetUserInfo", reflect.TypeOf((*MockIUserProvider)(nil).MGetUserInfo), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetUserInfo", reflect.TypeOf((*MockIUserProvider)(nil).MGetUserInfo), ctx, userIDs) } diff --git a/backend/modules/data/domain/component/userinfo/mocks/userinfo.go b/backend/modules/data/domain/component/userinfo/mocks/userinfo.go index 0b41e6b54..cb7ed53c8 100644 --- a/backend/modules/data/domain/component/userinfo/mocks/userinfo.go +++ b/backend/modules/data/domain/component/userinfo/mocks/userinfo.go @@ -20,6 +20,7 @@ import ( type MockUserInfoService struct { ctrl *gomock.Controller recorder *MockUserInfoServiceMockRecorder + isgomock struct{} } // MockUserInfoServiceMockRecorder is the mock recorder for MockUserInfoService. @@ -40,13 +41,13 @@ func (m *MockUserInfoService) EXPECT() *MockUserInfoServiceMockRecorder { } // PackUserInfo mocks base method. -func (m *MockUserInfoService) PackUserInfo(arg0 context.Context, arg1 any) { +func (m *MockUserInfoService) PackUserInfo(ctx context.Context, userInfoCarrier any) { m.ctrl.T.Helper() - m.ctrl.Call(m, "PackUserInfo", arg0, arg1) + m.ctrl.Call(m, "PackUserInfo", ctx, userInfoCarrier) } // PackUserInfo indicates an expected call of PackUserInfo. -func (mr *MockUserInfoServiceMockRecorder) PackUserInfo(arg0, arg1 any) *gomock.Call { +func (mr *MockUserInfoServiceMockRecorder) PackUserInfo(ctx, userInfoCarrier any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PackUserInfo", reflect.TypeOf((*MockUserInfoService)(nil).PackUserInfo), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PackUserInfo", reflect.TypeOf((*MockUserInfoService)(nil).PackUserInfo), ctx, userInfoCarrier) } diff --git a/backend/modules/data/domain/tag/repo/mocks/tag_mock.go b/backend/modules/data/domain/tag/repo/mocks/tag_mock.go index 813cbccdf..c4460d262 100644 --- a/backend/modules/data/domain/tag/repo/mocks/tag_mock.go +++ b/backend/modules/data/domain/tag/repo/mocks/tag_mock.go @@ -23,6 +23,7 @@ import ( type MockITagAPI struct { ctrl *gomock.Controller recorder *MockITagAPIMockRecorder + isgomock struct{} } // MockITagAPIMockRecorder is the mock recorder for MockITagAPI. @@ -43,10 +44,10 @@ func (m *MockITagAPI) EXPECT() *MockITagAPIMockRecorder { } // CountTagKeys mocks base method. -func (m *MockITagAPI) CountTagKeys(arg0 context.Context, arg1 *entity.MGetTagKeyParam, arg2 ...db.Option) (int64, error) { +func (m *MockITagAPI) CountTagKeys(ctx context.Context, param *entity.MGetTagKeyParam, opts ...db.Option) (int64, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, param} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CountTagKeys", varargs...) @@ -56,17 +57,17 @@ func (m *MockITagAPI) CountTagKeys(arg0 context.Context, arg1 *entity.MGetTagKey } // CountTagKeys indicates an expected call of CountTagKeys. -func (mr *MockITagAPIMockRecorder) CountTagKeys(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) CountTagKeys(ctx, param any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, param}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CountTagKeys", reflect.TypeOf((*MockITagAPI)(nil).CountTagKeys), varargs...) } // DeleteTagKey mocks base method. -func (m *MockITagAPI) DeleteTagKey(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) error { +func (m *MockITagAPI) DeleteTagKey(ctx context.Context, spaceID, id int64, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, id} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteTagKey", varargs...) @@ -75,17 +76,17 @@ func (m *MockITagAPI) DeleteTagKey(arg0 context.Context, arg1, arg2 int64, arg3 } // DeleteTagKey indicates an expected call of DeleteTagKey. -func (mr *MockITagAPIMockRecorder) DeleteTagKey(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) DeleteTagKey(ctx, spaceID, id any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, id}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTagKey", reflect.TypeOf((*MockITagAPI)(nil).DeleteTagKey), varargs...) } // DeleteTagValue mocks base method. -func (m *MockITagAPI) DeleteTagValue(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) error { +func (m *MockITagAPI) DeleteTagValue(ctx context.Context, spaceID, id int64, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, id} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteTagValue", varargs...) @@ -94,17 +95,17 @@ func (m *MockITagAPI) DeleteTagValue(arg0 context.Context, arg1, arg2 int64, arg } // DeleteTagValue indicates an expected call of DeleteTagValue. -func (mr *MockITagAPIMockRecorder) DeleteTagValue(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) DeleteTagValue(ctx, spaceID, id any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, id}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTagValue", reflect.TypeOf((*MockITagAPI)(nil).DeleteTagValue), varargs...) } // GetTagKey mocks base method. -func (m *MockITagAPI) GetTagKey(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) (*entity.TagKey, error) { +func (m *MockITagAPI) GetTagKey(ctx context.Context, spaceID, id int64, opts ...db.Option) (*entity.TagKey, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, id} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetTagKey", varargs...) @@ -114,17 +115,17 @@ func (m *MockITagAPI) GetTagKey(arg0 context.Context, arg1, arg2 int64, arg3 ... } // GetTagKey indicates an expected call of GetTagKey. -func (mr *MockITagAPIMockRecorder) GetTagKey(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) GetTagKey(ctx, spaceID, id any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, id}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTagKey", reflect.TypeOf((*MockITagAPI)(nil).GetTagKey), varargs...) } // GetTagValue mocks base method. -func (m *MockITagAPI) GetTagValue(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) (*entity.TagValue, error) { +func (m *MockITagAPI) GetTagValue(ctx context.Context, spaceID, id int64, opts ...db.Option) (*entity.TagValue, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, id} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetTagValue", varargs...) @@ -134,17 +135,17 @@ func (m *MockITagAPI) GetTagValue(arg0 context.Context, arg1, arg2 int64, arg3 . } // GetTagValue indicates an expected call of GetTagValue. -func (mr *MockITagAPIMockRecorder) GetTagValue(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) GetTagValue(ctx, spaceID, id any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, id}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTagValue", reflect.TypeOf((*MockITagAPI)(nil).GetTagValue), varargs...) } // MCreateTagKeys mocks base method. -func (m *MockITagAPI) MCreateTagKeys(arg0 context.Context, arg1 []*entity.TagKey, arg2 ...db.Option) error { +func (m *MockITagAPI) MCreateTagKeys(ctx context.Context, val []*entity.TagKey, opt ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, val} + for _, a := range opt { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "MCreateTagKeys", varargs...) @@ -153,17 +154,17 @@ func (m *MockITagAPI) MCreateTagKeys(arg0 context.Context, arg1 []*entity.TagKey } // MCreateTagKeys indicates an expected call of MCreateTagKeys. -func (mr *MockITagAPIMockRecorder) MCreateTagKeys(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) MCreateTagKeys(ctx, val any, opt ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, val}, opt...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MCreateTagKeys", reflect.TypeOf((*MockITagAPI)(nil).MCreateTagKeys), varargs...) } // MCreateTagValues mocks base method. -func (m *MockITagAPI) MCreateTagValues(arg0 context.Context, arg1 []*entity.TagValue, arg2 ...db.Option) error { +func (m *MockITagAPI) MCreateTagValues(ctx context.Context, val []*entity.TagValue, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, val} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "MCreateTagValues", varargs...) @@ -172,17 +173,17 @@ func (m *MockITagAPI) MCreateTagValues(arg0 context.Context, arg1 []*entity.TagV } // MCreateTagValues indicates an expected call of MCreateTagValues. -func (mr *MockITagAPIMockRecorder) MCreateTagValues(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) MCreateTagValues(ctx, val any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, val}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MCreateTagValues", reflect.TypeOf((*MockITagAPI)(nil).MCreateTagValues), varargs...) } // MGetTagKeys mocks base method. -func (m *MockITagAPI) MGetTagKeys(arg0 context.Context, arg1 *entity.MGetTagKeyParam, arg2 ...db.Option) ([]*entity.TagKey, *pagination.PageResult, error) { +func (m *MockITagAPI) MGetTagKeys(ctx context.Context, param *entity.MGetTagKeyParam, opts ...db.Option) ([]*entity.TagKey, *pagination.PageResult, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, param} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "MGetTagKeys", varargs...) @@ -193,17 +194,17 @@ func (m *MockITagAPI) MGetTagKeys(arg0 context.Context, arg1 *entity.MGetTagKeyP } // MGetTagKeys indicates an expected call of MGetTagKeys. -func (mr *MockITagAPIMockRecorder) MGetTagKeys(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) MGetTagKeys(ctx, param any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, param}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetTagKeys", reflect.TypeOf((*MockITagAPI)(nil).MGetTagKeys), varargs...) } // MGetTagValue mocks base method. -func (m *MockITagAPI) MGetTagValue(arg0 context.Context, arg1 *entity.MGetTagValueParam, arg2 ...db.Option) ([]*entity.TagValue, *pagination.PageResult, error) { +func (m *MockITagAPI) MGetTagValue(ctx context.Context, param *entity.MGetTagValueParam, opts ...db.Option) ([]*entity.TagValue, *pagination.PageResult, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, param} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "MGetTagValue", varargs...) @@ -214,17 +215,17 @@ func (m *MockITagAPI) MGetTagValue(arg0 context.Context, arg1 *entity.MGetTagVal } // MGetTagValue indicates an expected call of MGetTagValue. -func (mr *MockITagAPIMockRecorder) MGetTagValue(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) MGetTagValue(ctx, param any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, param}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetTagValue", reflect.TypeOf((*MockITagAPI)(nil).MGetTagValue), varargs...) } // PatchTagKey mocks base method. -func (m *MockITagAPI) PatchTagKey(arg0 context.Context, arg1, arg2 int64, arg3 *entity.TagKey, arg4 ...db.Option) error { +func (m *MockITagAPI) PatchTagKey(ctx context.Context, spaceID, id int64, patch *entity.TagKey, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, id, patch} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "PatchTagKey", varargs...) @@ -233,17 +234,17 @@ func (m *MockITagAPI) PatchTagKey(arg0 context.Context, arg1, arg2 int64, arg3 * } // PatchTagKey indicates an expected call of PatchTagKey. -func (mr *MockITagAPIMockRecorder) PatchTagKey(arg0, arg1, arg2, arg3 any, arg4 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) PatchTagKey(ctx, spaceID, id, patch any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, id, patch}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PatchTagKey", reflect.TypeOf((*MockITagAPI)(nil).PatchTagKey), varargs...) } // PatchTagValue mocks base method. -func (m *MockITagAPI) PatchTagValue(arg0 context.Context, arg1, arg2 int64, arg3 *entity.TagValue, arg4 ...db.Option) error { +func (m *MockITagAPI) PatchTagValue(ctx context.Context, spaceID, id int64, patch *entity.TagValue, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, id, patch} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "PatchTagValue", varargs...) @@ -252,17 +253,17 @@ func (m *MockITagAPI) PatchTagValue(arg0 context.Context, arg1, arg2 int64, arg3 } // PatchTagValue indicates an expected call of PatchTagValue. -func (mr *MockITagAPIMockRecorder) PatchTagValue(arg0, arg1, arg2, arg3 any, arg4 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) PatchTagValue(ctx, spaceID, id, patch any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, id, patch}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PatchTagValue", reflect.TypeOf((*MockITagAPI)(nil).PatchTagValue), varargs...) } // UpdateTagKeysStatus mocks base method. -func (m *MockITagAPI) UpdateTagKeysStatus(arg0 context.Context, arg1, arg2 int64, arg3 int32, arg4 entity.TagStatus, arg5 bool, arg6 ...db.Option) error { +func (m *MockITagAPI) UpdateTagKeysStatus(ctx context.Context, spaceID, tagKeyID int64, versionNum int32, toStatus entity.TagStatus, updateInfo bool, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3, arg4, arg5} - for _, a := range arg6 { + varargs := []any{ctx, spaceID, tagKeyID, versionNum, toStatus, updateInfo} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateTagKeysStatus", varargs...) @@ -271,17 +272,17 @@ func (m *MockITagAPI) UpdateTagKeysStatus(arg0 context.Context, arg1, arg2 int64 } // UpdateTagKeysStatus indicates an expected call of UpdateTagKeysStatus. -func (mr *MockITagAPIMockRecorder) UpdateTagKeysStatus(arg0, arg1, arg2, arg3, arg4, arg5 any, arg6 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) UpdateTagKeysStatus(ctx, spaceID, tagKeyID, versionNum, toStatus, updateInfo any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) + varargs := append([]any{ctx, spaceID, tagKeyID, versionNum, toStatus, updateInfo}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTagKeysStatus", reflect.TypeOf((*MockITagAPI)(nil).UpdateTagKeysStatus), varargs...) } // UpdateTagValuesStatus mocks base method. -func (m *MockITagAPI) UpdateTagValuesStatus(arg0 context.Context, arg1, arg2 int64, arg3 int32, arg4 entity.TagStatus, arg5 bool, arg6 ...db.Option) error { +func (m *MockITagAPI) UpdateTagValuesStatus(ctx context.Context, spaceID, tagKeyID int64, versionNum int32, toStatus entity.TagStatus, updateInfo bool, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3, arg4, arg5} - for _, a := range arg6 { + varargs := []any{ctx, spaceID, tagKeyID, versionNum, toStatus, updateInfo} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateTagValuesStatus", varargs...) @@ -290,8 +291,8 @@ func (m *MockITagAPI) UpdateTagValuesStatus(arg0 context.Context, arg1, arg2 int } // UpdateTagValuesStatus indicates an expected call of UpdateTagValuesStatus. -func (mr *MockITagAPIMockRecorder) UpdateTagValuesStatus(arg0, arg1, arg2, arg3, arg4, arg5 any, arg6 ...any) *gomock.Call { +func (mr *MockITagAPIMockRecorder) UpdateTagValuesStatus(ctx, spaceID, tagKeyID, versionNum, toStatus, updateInfo any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) + varargs := append([]any{ctx, spaceID, tagKeyID, versionNum, toStatus, updateInfo}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTagValuesStatus", reflect.TypeOf((*MockITagAPI)(nil).UpdateTagValuesStatus), varargs...) } diff --git a/backend/modules/data/domain/tag/service/mocks/tag_service_mock.go b/backend/modules/data/domain/tag/service/mocks/tag_service_mock.go index dcb71ea55..d7128a8cf 100644 --- a/backend/modules/data/domain/tag/service/mocks/tag_service_mock.go +++ b/backend/modules/data/domain/tag/service/mocks/tag_service_mock.go @@ -23,6 +23,7 @@ import ( type MockITagService struct { ctrl *gomock.Controller recorder *MockITagServiceMockRecorder + isgomock struct{} } // MockITagServiceMockRecorder is the mock recorder for MockITagService. @@ -43,10 +44,10 @@ func (m *MockITagService) EXPECT() *MockITagServiceMockRecorder { } // ArchiveOptionTag mocks base method. -func (m *MockITagService) ArchiveOptionTag(arg0 context.Context, arg1, arg2 int64, arg3 *entity.TagKey, arg4 ...db.Option) error { +func (m *MockITagService) ArchiveOptionTag(ctx context.Context, spaceID, tagKeyID int64, val *entity.TagKey, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, tagKeyID, val} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ArchiveOptionTag", varargs...) @@ -55,47 +56,47 @@ func (m *MockITagService) ArchiveOptionTag(arg0 context.Context, arg1, arg2 int6 } // ArchiveOptionTag indicates an expected call of ArchiveOptionTag. -func (mr *MockITagServiceMockRecorder) ArchiveOptionTag(arg0, arg1, arg2, arg3 any, arg4 ...any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) ArchiveOptionTag(ctx, spaceID, tagKeyID, val any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, tagKeyID, val}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ArchiveOptionTag", reflect.TypeOf((*MockITagService)(nil).ArchiveOptionTag), varargs...) } // BatchGetTagsByTagKeyIDs mocks base method. -func (m *MockITagService) BatchGetTagsByTagKeyIDs(arg0 context.Context, arg1 int64, arg2 []int64) ([]*entity.TagKey, error) { +func (m *MockITagService) BatchGetTagsByTagKeyIDs(ctx context.Context, spaceID int64, tagKeyIDs []int64) ([]*entity.TagKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchGetTagsByTagKeyIDs", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "BatchGetTagsByTagKeyIDs", ctx, spaceID, tagKeyIDs) ret0, _ := ret[0].([]*entity.TagKey) ret1, _ := ret[1].(error) return ret0, ret1 } // BatchGetTagsByTagKeyIDs indicates an expected call of BatchGetTagsByTagKeyIDs. -func (mr *MockITagServiceMockRecorder) BatchGetTagsByTagKeyIDs(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) BatchGetTagsByTagKeyIDs(ctx, spaceID, tagKeyIDs any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetTagsByTagKeyIDs", reflect.TypeOf((*MockITagService)(nil).BatchGetTagsByTagKeyIDs), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetTagsByTagKeyIDs", reflect.TypeOf((*MockITagService)(nil).BatchGetTagsByTagKeyIDs), ctx, spaceID, tagKeyIDs) } // BatchUpdateTagStatus mocks base method. -func (m *MockITagService) BatchUpdateTagStatus(arg0 context.Context, arg1 int64, arg2 []int64, arg3 entity.TagStatus) (map[int64]string, error) { +func (m *MockITagService) BatchUpdateTagStatus(ctx context.Context, spaceID int64, tagKeyIDs []int64, toStatus entity.TagStatus) (map[int64]string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchUpdateTagStatus", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "BatchUpdateTagStatus", ctx, spaceID, tagKeyIDs, toStatus) ret0, _ := ret[0].(map[int64]string) ret1, _ := ret[1].(error) return ret0, ret1 } // BatchUpdateTagStatus indicates an expected call of BatchUpdateTagStatus. -func (mr *MockITagServiceMockRecorder) BatchUpdateTagStatus(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) BatchUpdateTagStatus(ctx, spaceID, tagKeyIDs, toStatus any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchUpdateTagStatus", reflect.TypeOf((*MockITagService)(nil).BatchUpdateTagStatus), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchUpdateTagStatus", reflect.TypeOf((*MockITagService)(nil).BatchUpdateTagStatus), ctx, spaceID, tagKeyIDs, toStatus) } // CreateTag mocks base method. -func (m *MockITagService) CreateTag(arg0 context.Context, arg1 int64, arg2 *entity.TagKey, arg3 ...db.Option) (int64, error) { +func (m *MockITagService) CreateTag(ctx context.Context, spaceID int64, val *entity.TagKey, opts ...db.Option) (int64, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, val} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateTag", varargs...) @@ -105,17 +106,17 @@ func (m *MockITagService) CreateTag(arg0 context.Context, arg1 int64, arg2 *enti } // CreateTag indicates an expected call of CreateTag. -func (mr *MockITagServiceMockRecorder) CreateTag(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) CreateTag(ctx, spaceID, val any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, val}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTag", reflect.TypeOf((*MockITagService)(nil).CreateTag), varargs...) } // GetAllTagKeyVersionsByKeyID mocks base method. -func (m *MockITagService) GetAllTagKeyVersionsByKeyID(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) ([]*entity.TagKey, error) { +func (m *MockITagService) GetAllTagKeyVersionsByKeyID(ctx context.Context, spaceID, tagKeyID int64, opts ...db.Option) ([]*entity.TagKey, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, tagKeyID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetAllTagKeyVersionsByKeyID", varargs...) @@ -125,17 +126,17 @@ func (m *MockITagService) GetAllTagKeyVersionsByKeyID(arg0 context.Context, arg1 } // GetAllTagKeyVersionsByKeyID indicates an expected call of GetAllTagKeyVersionsByKeyID. -func (mr *MockITagServiceMockRecorder) GetAllTagKeyVersionsByKeyID(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) GetAllTagKeyVersionsByKeyID(ctx, spaceID, tagKeyID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, tagKeyID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTagKeyVersionsByKeyID", reflect.TypeOf((*MockITagService)(nil).GetAllTagKeyVersionsByKeyID), varargs...) } // GetAndBuildTagValues mocks base method. -func (m *MockITagService) GetAndBuildTagValues(arg0 context.Context, arg1, arg2 int64, arg3 int32, arg4 ...db.Option) ([]*entity.TagValue, error) { +func (m *MockITagService) GetAndBuildTagValues(ctx context.Context, spaceID, tagKeyID int64, versionNum int32, opts ...db.Option) ([]*entity.TagValue, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, tagKeyID, versionNum} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetAndBuildTagValues", varargs...) @@ -145,17 +146,17 @@ func (m *MockITagService) GetAndBuildTagValues(arg0 context.Context, arg1, arg2 } // GetAndBuildTagValues indicates an expected call of GetAndBuildTagValues. -func (mr *MockITagServiceMockRecorder) GetAndBuildTagValues(arg0, arg1, arg2, arg3 any, arg4 ...any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) GetAndBuildTagValues(ctx, spaceID, tagKeyID, versionNum any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, tagKeyID, versionNum}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAndBuildTagValues", reflect.TypeOf((*MockITagService)(nil).GetAndBuildTagValues), varargs...) } // GetLatestTag mocks base method. -func (m *MockITagService) GetLatestTag(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) (*entity.TagKey, error) { +func (m *MockITagService) GetLatestTag(ctx context.Context, spaceID, tagKeyID int64, opts ...db.Option) (*entity.TagKey, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, tagKeyID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetLatestTag", varargs...) @@ -165,31 +166,31 @@ func (m *MockITagService) GetLatestTag(arg0 context.Context, arg1, arg2 int64, a } // GetLatestTag indicates an expected call of GetLatestTag. -func (mr *MockITagServiceMockRecorder) GetLatestTag(arg0, arg1, arg2 any, arg3 ...any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) GetLatestTag(ctx, spaceID, tagKeyID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, tagKeyID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestTag", reflect.TypeOf((*MockITagService)(nil).GetLatestTag), varargs...) } // GetTagDetail mocks base method. -func (m *MockITagService) GetTagDetail(arg0 context.Context, arg1 int64, arg2 *entity.GetTagDetailReq) (*entity.GetTagDetailResp, error) { +func (m *MockITagService) GetTagDetail(ctx context.Context, spaceID int64, param *entity.GetTagDetailReq) (*entity.GetTagDetailResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTagDetail", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetTagDetail", ctx, spaceID, param) ret0, _ := ret[0].(*entity.GetTagDetailResp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTagDetail indicates an expected call of GetTagDetail. -func (mr *MockITagServiceMockRecorder) GetTagDetail(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) GetTagDetail(ctx, spaceID, param any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTagDetail", reflect.TypeOf((*MockITagService)(nil).GetTagDetail), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTagDetail", reflect.TypeOf((*MockITagService)(nil).GetTagDetail), ctx, spaceID, param) } // GetTagSpec mocks base method. -func (m *MockITagService) GetTagSpec(arg0 context.Context, arg1 int64) (int64, int64, int64, error) { +func (m *MockITagService) GetTagSpec(ctx context.Context, spaceID int64) (int64, int64, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTagSpec", arg0, arg1) + ret := m.ctrl.Call(m, "GetTagSpec", ctx, spaceID) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(int64) @@ -198,15 +199,15 @@ func (m *MockITagService) GetTagSpec(arg0 context.Context, arg1 int64) (int64, i } // GetTagSpec indicates an expected call of GetTagSpec. -func (mr *MockITagServiceMockRecorder) GetTagSpec(arg0, arg1 any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) GetTagSpec(ctx, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTagSpec", reflect.TypeOf((*MockITagService)(nil).GetTagSpec), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTagSpec", reflect.TypeOf((*MockITagService)(nil).GetTagSpec), ctx, spaceID) } // SearchTags mocks base method. -func (m *MockITagService) SearchTags(arg0 context.Context, arg1 int64, arg2 *entity.MGetTagKeyParam) ([]*entity.TagKey, *pagination.PageResult, error) { +func (m *MockITagService) SearchTags(ctx context.Context, spaceID int64, param *entity.MGetTagKeyParam) ([]*entity.TagKey, *pagination.PageResult, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SearchTags", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "SearchTags", ctx, spaceID, param) ret0, _ := ret[0].([]*entity.TagKey) ret1, _ := ret[1].(*pagination.PageResult) ret2, _ := ret[2].(error) @@ -214,16 +215,16 @@ func (m *MockITagService) SearchTags(arg0 context.Context, arg1 int64, arg2 *ent } // SearchTags indicates an expected call of SearchTags. -func (mr *MockITagServiceMockRecorder) SearchTags(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) SearchTags(ctx, spaceID, param any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchTags", reflect.TypeOf((*MockITagService)(nil).SearchTags), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchTags", reflect.TypeOf((*MockITagService)(nil).SearchTags), ctx, spaceID, param) } // UpdateOptionTag mocks base method. -func (m *MockITagService) UpdateOptionTag(arg0 context.Context, arg1, arg2 int64, arg3 *entity.TagKey, arg4 ...db.Option) error { +func (m *MockITagService) UpdateOptionTag(ctx context.Context, spaceID, tagKeyID int64, val *entity.TagKey, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, tagKeyID, val} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateOptionTag", varargs...) @@ -232,17 +233,17 @@ func (m *MockITagService) UpdateOptionTag(arg0 context.Context, arg1, arg2 int64 } // UpdateOptionTag indicates an expected call of UpdateOptionTag. -func (mr *MockITagServiceMockRecorder) UpdateOptionTag(arg0, arg1, arg2, arg3 any, arg4 ...any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) UpdateOptionTag(ctx, spaceID, tagKeyID, val any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, tagKeyID, val}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOptionTag", reflect.TypeOf((*MockITagService)(nil).UpdateOptionTag), varargs...) } // UpdateTag mocks base method. -func (m *MockITagService) UpdateTag(arg0 context.Context, arg1, arg2 int64, arg3 *entity.TagKey, arg4 ...db.Option) error { +func (m *MockITagService) UpdateTag(ctx context.Context, spaceID, tagKeyID int64, val *entity.TagKey, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, tagKeyID, val} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateTag", varargs...) @@ -251,17 +252,17 @@ func (m *MockITagService) UpdateTag(arg0 context.Context, arg1, arg2 int64, arg3 } // UpdateTag indicates an expected call of UpdateTag. -func (mr *MockITagServiceMockRecorder) UpdateTag(arg0, arg1, arg2, arg3 any, arg4 ...any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) UpdateTag(ctx, spaceID, tagKeyID, val any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, tagKeyID, val}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTag", reflect.TypeOf((*MockITagService)(nil).UpdateTag), varargs...) } // UpdateTagStatus mocks base method. -func (m *MockITagService) UpdateTagStatus(arg0 context.Context, arg1, arg2 int64, arg3 int32, arg4 entity.TagStatus, arg5, arg6 bool, arg7 ...db.Option) error { +func (m *MockITagService) UpdateTagStatus(ctx context.Context, spaceID, tagKeyID int64, versionNum int32, status entity.TagStatus, needLock, updatedInfo bool, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []any{arg0, arg1, arg2, arg3, arg4, arg5, arg6} - for _, a := range arg7 { + varargs := []any{ctx, spaceID, tagKeyID, versionNum, status, needLock, updatedInfo} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateTagStatus", varargs...) @@ -270,22 +271,22 @@ func (m *MockITagService) UpdateTagStatus(arg0 context.Context, arg1, arg2 int64 } // UpdateTagStatus indicates an expected call of UpdateTagStatus. -func (mr *MockITagServiceMockRecorder) UpdateTagStatus(arg0, arg1, arg2, arg3, arg4, arg5, arg6 any, arg7 ...any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) UpdateTagStatus(ctx, spaceID, tagKeyID, versionNum, status, needLock, updatedInfo any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1, arg2, arg3, arg4, arg5, arg6}, arg7...) + varargs := append([]any{ctx, spaceID, tagKeyID, versionNum, status, needLock, updatedInfo}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTagStatus", reflect.TypeOf((*MockITagService)(nil).UpdateTagStatus), varargs...) } // UpdateTagStatusWithNewVersion mocks base method. -func (m *MockITagService) UpdateTagStatusWithNewVersion(arg0 context.Context, arg1, arg2 int64, arg3 entity.TagStatus) error { +func (m *MockITagService) UpdateTagStatusWithNewVersion(ctx context.Context, spaceID, tagKeyID int64, status entity.TagStatus) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTagStatusWithNewVersion", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "UpdateTagStatusWithNewVersion", ctx, spaceID, tagKeyID, status) ret0, _ := ret[0].(error) return ret0 } // UpdateTagStatusWithNewVersion indicates an expected call of UpdateTagStatusWithNewVersion. -func (mr *MockITagServiceMockRecorder) UpdateTagStatusWithNewVersion(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockITagServiceMockRecorder) UpdateTagStatusWithNewVersion(ctx, spaceID, tagKeyID, status any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTagStatusWithNewVersion", reflect.TypeOf((*MockITagService)(nil).UpdateTagStatusWithNewVersion), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTagStatusWithNewVersion", reflect.TypeOf((*MockITagService)(nil).UpdateTagStatusWithNewVersion), ctx, spaceID, tagKeyID, status) } diff --git a/backend/modules/evaluation/domain/component/metrics/mocks/openapi_evaluation_set.go b/backend/modules/evaluation/domain/component/metrics/mocks/openapi_evaluation_set.go new file mode 100644 index 000000000..7fd06bab5 --- /dev/null +++ b/backend/modules/evaluation/domain/component/metrics/mocks/openapi_evaluation_set.go @@ -0,0 +1,53 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/component/metrics (interfaces: OpenAPIEvaluationMetrics) +// +// Generated by this command: +// +// mockgen -destination=mocks/openapi_evaluation_set.go -package=mocks . OpenAPIEvaluationMetrics +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + gomock "go.uber.org/mock/gomock" +) + +// MockOpenAPIEvaluationMetrics is a mock of OpenAPIEvaluationMetrics interface. +type MockOpenAPIEvaluationMetrics struct { + ctrl *gomock.Controller + recorder *MockOpenAPIEvaluationMetricsMockRecorder + isgomock struct{} +} + +// MockOpenAPIEvaluationMetricsMockRecorder is the mock recorder for MockOpenAPIEvaluationMetrics. +type MockOpenAPIEvaluationMetricsMockRecorder struct { + mock *MockOpenAPIEvaluationMetrics +} + +// NewMockOpenAPIEvaluationMetrics creates a new mock instance. +func NewMockOpenAPIEvaluationMetrics(ctrl *gomock.Controller) *MockOpenAPIEvaluationMetrics { + mock := &MockOpenAPIEvaluationMetrics{ctrl: ctrl} + mock.recorder = &MockOpenAPIEvaluationMetricsMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockOpenAPIEvaluationMetrics) EXPECT() *MockOpenAPIEvaluationMetricsMockRecorder { + return m.recorder +} + +// EmitOpenAPIMetric mocks base method. +func (m *MockOpenAPIEvaluationMetrics) EmitOpenAPIMetric(ctx context.Context, spaceID, evaluationSetID int64, method string, startTime int64, err error) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "EmitOpenAPIMetric", ctx, spaceID, evaluationSetID, method, startTime, err) +} + +// EmitOpenAPIMetric indicates an expected call of EmitOpenAPIMetric. +func (mr *MockOpenAPIEvaluationMetricsMockRecorder) EmitOpenAPIMetric(ctx, spaceID, evaluationSetID, method, startTime, err any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EmitOpenAPIMetric", reflect.TypeOf((*MockOpenAPIEvaluationMetrics)(nil).EmitOpenAPIMetric), ctx, spaceID, evaluationSetID, method, startTime, err) +} diff --git a/backend/modules/evaluation/domain/entity/mocks/expt_scheduler_mock.go b/backend/modules/evaluation/domain/entity/mocks/expt_scheduler_mock.go index 9e2eb4a87..9d4b6e419 100644 --- a/backend/modules/evaluation/domain/entity/mocks/expt_scheduler_mock.go +++ b/backend/modules/evaluation/domain/entity/mocks/expt_scheduler_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/entity (interfaces: ExptSchedulerMode) +// +// Generated by this command: +// +// mockgen -destination ./mocks/expt_scheduler_mock.go --package mocks . ExptSchedulerMode +// // Package mocks is a generated GoMock package. package mocks @@ -16,6 +21,7 @@ import ( type MockExptSchedulerMode struct { ctrl *gomock.Controller recorder *MockExptSchedulerModeMockRecorder + isgomock struct{} } // MockExptSchedulerModeMockRecorder is the mock recorder for MockExptSchedulerMode. @@ -36,32 +42,32 @@ func (m *MockExptSchedulerMode) EXPECT() *MockExptSchedulerModeMockRecorder { } // ExptEnd mocks base method. -func (m *MockExptSchedulerMode) ExptEnd(arg0 context.Context, arg1 *entity.ExptScheduleEvent, arg2 *entity.Experiment, arg3, arg4 int) (bool, error) { +func (m *MockExptSchedulerMode) ExptEnd(ctx context.Context, event *entity.ExptScheduleEvent, expt *entity.Experiment, toSubmit, incomplete int) (bool, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ExptEnd", arg0, arg1, arg2, arg3, arg4) + ret := m.ctrl.Call(m, "ExptEnd", ctx, event, expt, toSubmit, incomplete) ret0, _ := ret[0].(bool) ret1, _ := ret[1].(error) return ret0, ret1 } // ExptEnd indicates an expected call of ExptEnd. -func (mr *MockExptSchedulerModeMockRecorder) ExptEnd(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { +func (mr *MockExptSchedulerModeMockRecorder) ExptEnd(ctx, event, expt, toSubmit, incomplete any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExptEnd", reflect.TypeOf((*MockExptSchedulerMode)(nil).ExptEnd), arg0, arg1, arg2, arg3, arg4) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExptEnd", reflect.TypeOf((*MockExptSchedulerMode)(nil).ExptEnd), ctx, event, expt, toSubmit, incomplete) } // ExptStart mocks base method. -func (m *MockExptSchedulerMode) ExptStart(arg0 context.Context, arg1 *entity.ExptScheduleEvent, arg2 *entity.Experiment) error { +func (m *MockExptSchedulerMode) ExptStart(ctx context.Context, event *entity.ExptScheduleEvent, expt *entity.Experiment) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ExptStart", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "ExptStart", ctx, event, expt) ret0, _ := ret[0].(error) return ret0 } // ExptStart indicates an expected call of ExptStart. -func (mr *MockExptSchedulerModeMockRecorder) ExptStart(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptSchedulerModeMockRecorder) ExptStart(ctx, event, expt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExptStart", reflect.TypeOf((*MockExptSchedulerMode)(nil).ExptStart), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExptStart", reflect.TypeOf((*MockExptSchedulerMode)(nil).ExptStart), ctx, event, expt) } // Mode mocks base method. @@ -79,37 +85,37 @@ func (mr *MockExptSchedulerModeMockRecorder) Mode() *gomock.Call { } // NextTick mocks base method. -func (m *MockExptSchedulerMode) NextTick(arg0 context.Context, arg1 *entity.ExptScheduleEvent, arg2 bool) error { +func (m *MockExptSchedulerMode) NextTick(ctx context.Context, event *entity.ExptScheduleEvent, nextTick bool) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "NextTick", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "NextTick", ctx, event, nextTick) ret0, _ := ret[0].(error) return ret0 } // NextTick indicates an expected call of NextTick. -func (mr *MockExptSchedulerModeMockRecorder) NextTick(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptSchedulerModeMockRecorder) NextTick(ctx, event, nextTick any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NextTick", reflect.TypeOf((*MockExptSchedulerMode)(nil).NextTick), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NextTick", reflect.TypeOf((*MockExptSchedulerMode)(nil).NextTick), ctx, event, nextTick) } // PublishResult mocks base method. -func (m *MockExptSchedulerMode) PublishResult(arg0 context.Context, arg1 []*entity.ExptTurnEvaluatorResultRef, arg2 *entity.ExptScheduleEvent) error { +func (m *MockExptSchedulerMode) PublishResult(ctx context.Context, turnEvaluatorRefs []*entity.ExptTurnEvaluatorResultRef, event *entity.ExptScheduleEvent) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PublishResult", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "PublishResult", ctx, turnEvaluatorRefs, event) ret0, _ := ret[0].(error) return ret0 } // PublishResult indicates an expected call of PublishResult. -func (mr *MockExptSchedulerModeMockRecorder) PublishResult(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptSchedulerModeMockRecorder) PublishResult(ctx, turnEvaluatorRefs, event any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishResult", reflect.TypeOf((*MockExptSchedulerMode)(nil).PublishResult), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishResult", reflect.TypeOf((*MockExptSchedulerMode)(nil).PublishResult), ctx, turnEvaluatorRefs, event) } // ScanEvalItems mocks base method. -func (m *MockExptSchedulerMode) ScanEvalItems(arg0 context.Context, arg1 *entity.ExptScheduleEvent, arg2 *entity.Experiment) ([]*entity.ExptEvalItem, []*entity.ExptEvalItem, []*entity.ExptEvalItem, error) { +func (m *MockExptSchedulerMode) ScanEvalItems(ctx context.Context, event *entity.ExptScheduleEvent, expt *entity.Experiment) ([]*entity.ExptEvalItem, []*entity.ExptEvalItem, []*entity.ExptEvalItem, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ScanEvalItems", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "ScanEvalItems", ctx, event, expt) ret0, _ := ret[0].([]*entity.ExptEvalItem) ret1, _ := ret[1].([]*entity.ExptEvalItem) ret2, _ := ret[2].([]*entity.ExptEvalItem) @@ -118,35 +124,35 @@ func (m *MockExptSchedulerMode) ScanEvalItems(arg0 context.Context, arg1 *entity } // ScanEvalItems indicates an expected call of ScanEvalItems. -func (mr *MockExptSchedulerModeMockRecorder) ScanEvalItems(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptSchedulerModeMockRecorder) ScanEvalItems(ctx, event, expt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScanEvalItems", reflect.TypeOf((*MockExptSchedulerMode)(nil).ScanEvalItems), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScanEvalItems", reflect.TypeOf((*MockExptSchedulerMode)(nil).ScanEvalItems), ctx, event, expt) } // ScheduleEnd mocks base method. -func (m *MockExptSchedulerMode) ScheduleEnd(arg0 context.Context, arg1 *entity.ExptScheduleEvent, arg2 *entity.Experiment, arg3, arg4 int) error { +func (m *MockExptSchedulerMode) ScheduleEnd(ctx context.Context, event *entity.ExptScheduleEvent, expt *entity.Experiment, toSubmit, incomplete int) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ScheduleEnd", arg0, arg1, arg2, arg3, arg4) + ret := m.ctrl.Call(m, "ScheduleEnd", ctx, event, expt, toSubmit, incomplete) ret0, _ := ret[0].(error) return ret0 } // ScheduleEnd indicates an expected call of ScheduleEnd. -func (mr *MockExptSchedulerModeMockRecorder) ScheduleEnd(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { +func (mr *MockExptSchedulerModeMockRecorder) ScheduleEnd(ctx, event, expt, toSubmit, incomplete any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScheduleEnd", reflect.TypeOf((*MockExptSchedulerMode)(nil).ScheduleEnd), arg0, arg1, arg2, arg3, arg4) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScheduleEnd", reflect.TypeOf((*MockExptSchedulerMode)(nil).ScheduleEnd), ctx, event, expt, toSubmit, incomplete) } // ScheduleStart mocks base method. -func (m *MockExptSchedulerMode) ScheduleStart(arg0 context.Context, arg1 *entity.ExptScheduleEvent, arg2 *entity.Experiment) error { +func (m *MockExptSchedulerMode) ScheduleStart(ctx context.Context, event *entity.ExptScheduleEvent, expt *entity.Experiment) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ScheduleStart", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "ScheduleStart", ctx, event, expt) ret0, _ := ret[0].(error) return ret0 } // ScheduleStart indicates an expected call of ScheduleStart. -func (mr *MockExptSchedulerModeMockRecorder) ScheduleStart(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptSchedulerModeMockRecorder) ScheduleStart(ctx, event, expt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScheduleStart", reflect.TypeOf((*MockExptSchedulerMode)(nil).ScheduleStart), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScheduleStart", reflect.TypeOf((*MockExptSchedulerMode)(nil).ScheduleStart), ctx, event, expt) } diff --git a/backend/modules/evaluation/domain/events/mocks/evaluator_event_publisher_mock.go b/backend/modules/evaluation/domain/events/mocks/evaluator_event_publisher_mock.go index 6d15b6e2b..e97b66442 100644 --- a/backend/modules/evaluation/domain/events/mocks/evaluator_event_publisher_mock.go +++ b/backend/modules/evaluation/domain/events/mocks/evaluator_event_publisher_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/events (interfaces: EvaluatorEventPublisher) +// +// Generated by this command: +// +// mockgen -destination mocks/evaluator_event_publisher_mock.go -package mocks . EvaluatorEventPublisher +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockEvaluatorEventPublisher struct { ctrl *gomock.Controller recorder *MockEvaluatorEventPublisherMockRecorder + isgomock struct{} } // MockEvaluatorEventPublisherMockRecorder is the mock recorder for MockEvaluatorEventPublisher. @@ -37,15 +43,15 @@ func (m *MockEvaluatorEventPublisher) EXPECT() *MockEvaluatorEventPublisherMockR } // PublishEvaluatorRecordCorrection mocks base method. -func (m *MockEvaluatorEventPublisher) PublishEvaluatorRecordCorrection(arg0 context.Context, arg1 *entity.EvaluatorRecordCorrectionEvent, arg2 *time.Duration) error { +func (m *MockEvaluatorEventPublisher) PublishEvaluatorRecordCorrection(ctx context.Context, events *entity.EvaluatorRecordCorrectionEvent, duration *time.Duration) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PublishEvaluatorRecordCorrection", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "PublishEvaluatorRecordCorrection", ctx, events, duration) ret0, _ := ret[0].(error) return ret0 } // PublishEvaluatorRecordCorrection indicates an expected call of PublishEvaluatorRecordCorrection. -func (mr *MockEvaluatorEventPublisherMockRecorder) PublishEvaluatorRecordCorrection(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockEvaluatorEventPublisherMockRecorder) PublishEvaluatorRecordCorrection(ctx, events, duration any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishEvaluatorRecordCorrection", reflect.TypeOf((*MockEvaluatorEventPublisher)(nil).PublishEvaluatorRecordCorrection), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishEvaluatorRecordCorrection", reflect.TypeOf((*MockEvaluatorEventPublisher)(nil).PublishEvaluatorRecordCorrection), ctx, events, duration) } diff --git a/backend/modules/evaluation/domain/events/mocks/expt_event_publisher_mock.go b/backend/modules/evaluation/domain/events/mocks/expt_event_publisher_mock.go index d14d4577c..86cc51fde 100644 --- a/backend/modules/evaluation/domain/events/mocks/expt_event_publisher_mock.go +++ b/backend/modules/evaluation/domain/events/mocks/expt_event_publisher_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/events (interfaces: ExptEventPublisher) +// +// Generated by this command: +// +// mockgen -destination mocks/expt_event_publisher_mock.go -package mocks . ExptEventPublisher +// // Package mocks is a generated GoMock package. package mocks @@ -9,15 +14,15 @@ import ( reflect "reflect" time "time" - "go.uber.org/mock/gomock" - entity "github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/entity" + gomock "go.uber.org/mock/gomock" ) // MockExptEventPublisher is a mock of ExptEventPublisher interface. type MockExptEventPublisher struct { ctrl *gomock.Controller recorder *MockExptEventPublisherMockRecorder + isgomock struct{} } // MockExptEventPublisherMockRecorder is the mock recorder for MockExptEventPublisher. @@ -38,99 +43,99 @@ func (m *MockExptEventPublisher) EXPECT() *MockExptEventPublisherMockRecorder { } // BatchPublishExptRecordEvalEvent mocks base method. -func (m *MockExptEventPublisher) BatchPublishExptRecordEvalEvent(arg0 context.Context, arg1 []*entity.ExptItemEvalEvent, arg2 *time.Duration) error { +func (m *MockExptEventPublisher) BatchPublishExptRecordEvalEvent(ctx context.Context, events []*entity.ExptItemEvalEvent, duration *time.Duration) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchPublishExptRecordEvalEvent", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "BatchPublishExptRecordEvalEvent", ctx, events, duration) ret0, _ := ret[0].(error) return ret0 } // BatchPublishExptRecordEvalEvent indicates an expected call of BatchPublishExptRecordEvalEvent. -func (mr *MockExptEventPublisherMockRecorder) BatchPublishExptRecordEvalEvent(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptEventPublisherMockRecorder) BatchPublishExptRecordEvalEvent(ctx, events, duration any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchPublishExptRecordEvalEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).BatchPublishExptRecordEvalEvent), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchPublishExptRecordEvalEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).BatchPublishExptRecordEvalEvent), ctx, events, duration) } // PublishExptAggrCalculateEvent mocks base method. -func (m *MockExptEventPublisher) PublishExptAggrCalculateEvent(arg0 context.Context, arg1 []*entity.AggrCalculateEvent, arg2 *time.Duration) error { +func (m *MockExptEventPublisher) PublishExptAggrCalculateEvent(ctx context.Context, events []*entity.AggrCalculateEvent, duration *time.Duration) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PublishExptAggrCalculateEvent", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "PublishExptAggrCalculateEvent", ctx, events, duration) ret0, _ := ret[0].(error) return ret0 } // PublishExptAggrCalculateEvent indicates an expected call of PublishExptAggrCalculateEvent. -func (mr *MockExptEventPublisherMockRecorder) PublishExptAggrCalculateEvent(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptEventPublisherMockRecorder) PublishExptAggrCalculateEvent(ctx, events, duration any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptAggrCalculateEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptAggrCalculateEvent), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptAggrCalculateEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptAggrCalculateEvent), ctx, events, duration) } // PublishExptExportCSVEvent mocks base method. -func (m *MockExptEventPublisher) PublishExptExportCSVEvent(arg0 context.Context, arg1 *entity.ExportCSVEvent, arg2 *time.Duration) error { +func (m *MockExptEventPublisher) PublishExptExportCSVEvent(ctx context.Context, events *entity.ExportCSVEvent, duration *time.Duration) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PublishExptExportCSVEvent", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "PublishExptExportCSVEvent", ctx, events, duration) ret0, _ := ret[0].(error) return ret0 } // PublishExptExportCSVEvent indicates an expected call of PublishExptExportCSVEvent. -func (mr *MockExptEventPublisherMockRecorder) PublishExptExportCSVEvent(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptEventPublisherMockRecorder) PublishExptExportCSVEvent(ctx, events, duration any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptExportCSVEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptExportCSVEvent), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptExportCSVEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptExportCSVEvent), ctx, events, duration) } // PublishExptOnlineEvalResult mocks base method. -func (m *MockExptEventPublisher) PublishExptOnlineEvalResult(arg0 context.Context, arg1 *entity.OnlineExptEvalResultEvent, arg2 *time.Duration) error { +func (m *MockExptEventPublisher) PublishExptOnlineEvalResult(ctx context.Context, events *entity.OnlineExptEvalResultEvent, duration *time.Duration) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PublishExptOnlineEvalResult", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "PublishExptOnlineEvalResult", ctx, events, duration) ret0, _ := ret[0].(error) return ret0 } // PublishExptOnlineEvalResult indicates an expected call of PublishExptOnlineEvalResult. -func (mr *MockExptEventPublisherMockRecorder) PublishExptOnlineEvalResult(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptEventPublisherMockRecorder) PublishExptOnlineEvalResult(ctx, events, duration any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptOnlineEvalResult", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptOnlineEvalResult), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptOnlineEvalResult", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptOnlineEvalResult), ctx, events, duration) } // PublishExptRecordEvalEvent mocks base method. -func (m *MockExptEventPublisher) PublishExptRecordEvalEvent(arg0 context.Context, arg1 *entity.ExptItemEvalEvent, arg2 *time.Duration) error { +func (m *MockExptEventPublisher) PublishExptRecordEvalEvent(ctx context.Context, event *entity.ExptItemEvalEvent, duration *time.Duration) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PublishExptRecordEvalEvent", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "PublishExptRecordEvalEvent", ctx, event, duration) ret0, _ := ret[0].(error) return ret0 } // PublishExptRecordEvalEvent indicates an expected call of PublishExptRecordEvalEvent. -func (mr *MockExptEventPublisherMockRecorder) PublishExptRecordEvalEvent(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptEventPublisherMockRecorder) PublishExptRecordEvalEvent(ctx, event, duration any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptRecordEvalEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptRecordEvalEvent), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptRecordEvalEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptRecordEvalEvent), ctx, event, duration) } // PublishExptScheduleEvent mocks base method. -func (m *MockExptEventPublisher) PublishExptScheduleEvent(arg0 context.Context, arg1 *entity.ExptScheduleEvent, arg2 *time.Duration) error { +func (m *MockExptEventPublisher) PublishExptScheduleEvent(ctx context.Context, event *entity.ExptScheduleEvent, duration *time.Duration) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PublishExptScheduleEvent", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "PublishExptScheduleEvent", ctx, event, duration) ret0, _ := ret[0].(error) return ret0 } // PublishExptScheduleEvent indicates an expected call of PublishExptScheduleEvent. -func (mr *MockExptEventPublisherMockRecorder) PublishExptScheduleEvent(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptEventPublisherMockRecorder) PublishExptScheduleEvent(ctx, event, duration any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptScheduleEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptScheduleEvent), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptScheduleEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptScheduleEvent), ctx, event, duration) } // PublishExptTurnResultFilterEvent mocks base method. -func (m *MockExptEventPublisher) PublishExptTurnResultFilterEvent(arg0 context.Context, arg1 *entity.ExptTurnResultFilterEvent, arg2 *time.Duration) error { +func (m *MockExptEventPublisher) PublishExptTurnResultFilterEvent(ctx context.Context, event *entity.ExptTurnResultFilterEvent, duration *time.Duration) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PublishExptTurnResultFilterEvent", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "PublishExptTurnResultFilterEvent", ctx, event, duration) ret0, _ := ret[0].(error) return ret0 } // PublishExptTurnResultFilterEvent indicates an expected call of PublishExptTurnResultFilterEvent. -func (mr *MockExptEventPublisherMockRecorder) PublishExptTurnResultFilterEvent(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockExptEventPublisherMockRecorder) PublishExptTurnResultFilterEvent(ctx, event, duration any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptTurnResultFilterEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptTurnResultFilterEvent), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishExptTurnResultFilterEvent", reflect.TypeOf((*MockExptEventPublisher)(nil).PublishExptTurnResultFilterEvent), ctx, event, duration) } diff --git a/backend/modules/evaluation/domain/repo/mocks/evaluator_mock.go b/backend/modules/evaluation/domain/repo/mocks/evaluator_mock.go index 29a9ac313..5177a5010 100644 --- a/backend/modules/evaluation/domain/repo/mocks/evaluator_mock.go +++ b/backend/modules/evaluation/domain/repo/mocks/evaluator_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/repo (interfaces: IEvaluatorRepo) +// +// Generated by this command: +// +// mockgen -destination mocks/evaluator_mock.go -package mocks . IEvaluatorRepo +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockIEvaluatorRepo struct { ctrl *gomock.Controller recorder *MockIEvaluatorRepoMockRecorder + isgomock struct{} } // MockIEvaluatorRepoMockRecorder is the mock recorder for MockIEvaluatorRepo. @@ -37,192 +43,192 @@ func (m *MockIEvaluatorRepo) EXPECT() *MockIEvaluatorRepoMockRecorder { } // BatchDeleteEvaluator mocks base method. -func (m *MockIEvaluatorRepo) BatchDeleteEvaluator(arg0 context.Context, arg1 []int64, arg2 string) error { +func (m *MockIEvaluatorRepo) BatchDeleteEvaluator(ctx context.Context, ids []int64, userID string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchDeleteEvaluator", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "BatchDeleteEvaluator", ctx, ids, userID) ret0, _ := ret[0].(error) return ret0 } // BatchDeleteEvaluator indicates an expected call of BatchDeleteEvaluator. -func (mr *MockIEvaluatorRepoMockRecorder) BatchDeleteEvaluator(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) BatchDeleteEvaluator(ctx, ids, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchDeleteEvaluator", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchDeleteEvaluator), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchDeleteEvaluator", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchDeleteEvaluator), ctx, ids, userID) } // BatchGetEvaluatorByVersionID mocks base method. -func (m *MockIEvaluatorRepo) BatchGetEvaluatorByVersionID(arg0 context.Context, arg1 *int64, arg2 []int64, arg3 bool) ([]*entity.Evaluator, error) { +func (m *MockIEvaluatorRepo) BatchGetEvaluatorByVersionID(ctx context.Context, spaceID *int64, ids []int64, includeDeleted bool) ([]*entity.Evaluator, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchGetEvaluatorByVersionID", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "BatchGetEvaluatorByVersionID", ctx, spaceID, ids, includeDeleted) ret0, _ := ret[0].([]*entity.Evaluator) ret1, _ := ret[1].(error) return ret0, ret1 } // BatchGetEvaluatorByVersionID indicates an expected call of BatchGetEvaluatorByVersionID. -func (mr *MockIEvaluatorRepoMockRecorder) BatchGetEvaluatorByVersionID(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) BatchGetEvaluatorByVersionID(ctx, spaceID, ids, includeDeleted any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorByVersionID", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchGetEvaluatorByVersionID), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorByVersionID", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchGetEvaluatorByVersionID), ctx, spaceID, ids, includeDeleted) } // BatchGetEvaluatorDraftByEvaluatorID mocks base method. -func (m *MockIEvaluatorRepo) BatchGetEvaluatorDraftByEvaluatorID(arg0 context.Context, arg1 int64, arg2 []int64, arg3 bool) ([]*entity.Evaluator, error) { +func (m *MockIEvaluatorRepo) BatchGetEvaluatorDraftByEvaluatorID(ctx context.Context, spaceID int64, ids []int64, includeDeleted bool) ([]*entity.Evaluator, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchGetEvaluatorDraftByEvaluatorID", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "BatchGetEvaluatorDraftByEvaluatorID", ctx, spaceID, ids, includeDeleted) ret0, _ := ret[0].([]*entity.Evaluator) ret1, _ := ret[1].(error) return ret0, ret1 } // BatchGetEvaluatorDraftByEvaluatorID indicates an expected call of BatchGetEvaluatorDraftByEvaluatorID. -func (mr *MockIEvaluatorRepoMockRecorder) BatchGetEvaluatorDraftByEvaluatorID(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) BatchGetEvaluatorDraftByEvaluatorID(ctx, spaceID, ids, includeDeleted any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorDraftByEvaluatorID", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchGetEvaluatorDraftByEvaluatorID), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorDraftByEvaluatorID", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchGetEvaluatorDraftByEvaluatorID), ctx, spaceID, ids, includeDeleted) } // BatchGetEvaluatorMetaByID mocks base method. -func (m *MockIEvaluatorRepo) BatchGetEvaluatorMetaByID(arg0 context.Context, arg1 []int64, arg2 bool) ([]*entity.Evaluator, error) { +func (m *MockIEvaluatorRepo) BatchGetEvaluatorMetaByID(ctx context.Context, ids []int64, includeDeleted bool) ([]*entity.Evaluator, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchGetEvaluatorMetaByID", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "BatchGetEvaluatorMetaByID", ctx, ids, includeDeleted) ret0, _ := ret[0].([]*entity.Evaluator) ret1, _ := ret[1].(error) return ret0, ret1 } // BatchGetEvaluatorMetaByID indicates an expected call of BatchGetEvaluatorMetaByID. -func (mr *MockIEvaluatorRepoMockRecorder) BatchGetEvaluatorMetaByID(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) BatchGetEvaluatorMetaByID(ctx, ids, includeDeleted any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorMetaByID", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchGetEvaluatorMetaByID), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorMetaByID", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchGetEvaluatorMetaByID), ctx, ids, includeDeleted) } // BatchGetEvaluatorVersionsByEvaluatorIDs mocks base method. -func (m *MockIEvaluatorRepo) BatchGetEvaluatorVersionsByEvaluatorIDs(arg0 context.Context, arg1 []int64, arg2 bool) ([]*entity.Evaluator, error) { +func (m *MockIEvaluatorRepo) BatchGetEvaluatorVersionsByEvaluatorIDs(ctx context.Context, evaluatorIDs []int64, includeDeleted bool) ([]*entity.Evaluator, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchGetEvaluatorVersionsByEvaluatorIDs", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "BatchGetEvaluatorVersionsByEvaluatorIDs", ctx, evaluatorIDs, includeDeleted) ret0, _ := ret[0].([]*entity.Evaluator) ret1, _ := ret[1].(error) return ret0, ret1 } // BatchGetEvaluatorVersionsByEvaluatorIDs indicates an expected call of BatchGetEvaluatorVersionsByEvaluatorIDs. -func (mr *MockIEvaluatorRepoMockRecorder) BatchGetEvaluatorVersionsByEvaluatorIDs(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) BatchGetEvaluatorVersionsByEvaluatorIDs(ctx, evaluatorIDs, includeDeleted any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorVersionsByEvaluatorIDs", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchGetEvaluatorVersionsByEvaluatorIDs), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorVersionsByEvaluatorIDs", reflect.TypeOf((*MockIEvaluatorRepo)(nil).BatchGetEvaluatorVersionsByEvaluatorIDs), ctx, evaluatorIDs, includeDeleted) } // CheckNameExist mocks base method. -func (m *MockIEvaluatorRepo) CheckNameExist(arg0 context.Context, arg1, arg2 int64, arg3 string) (bool, error) { +func (m *MockIEvaluatorRepo) CheckNameExist(ctx context.Context, spaceID, evaluatorID int64, name string) (bool, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CheckNameExist", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "CheckNameExist", ctx, spaceID, evaluatorID, name) ret0, _ := ret[0].(bool) ret1, _ := ret[1].(error) return ret0, ret1 } // CheckNameExist indicates an expected call of CheckNameExist. -func (mr *MockIEvaluatorRepoMockRecorder) CheckNameExist(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) CheckNameExist(ctx, spaceID, evaluatorID, name any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckNameExist", reflect.TypeOf((*MockIEvaluatorRepo)(nil).CheckNameExist), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckNameExist", reflect.TypeOf((*MockIEvaluatorRepo)(nil).CheckNameExist), ctx, spaceID, evaluatorID, name) } // CheckVersionExist mocks base method. -func (m *MockIEvaluatorRepo) CheckVersionExist(arg0 context.Context, arg1 int64, arg2 string) (bool, error) { +func (m *MockIEvaluatorRepo) CheckVersionExist(ctx context.Context, evaluatorID int64, version string) (bool, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CheckVersionExist", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "CheckVersionExist", ctx, evaluatorID, version) ret0, _ := ret[0].(bool) ret1, _ := ret[1].(error) return ret0, ret1 } // CheckVersionExist indicates an expected call of CheckVersionExist. -func (mr *MockIEvaluatorRepoMockRecorder) CheckVersionExist(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) CheckVersionExist(ctx, evaluatorID, version any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckVersionExist", reflect.TypeOf((*MockIEvaluatorRepo)(nil).CheckVersionExist), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckVersionExist", reflect.TypeOf((*MockIEvaluatorRepo)(nil).CheckVersionExist), ctx, evaluatorID, version) } // CreateEvaluator mocks base method. -func (m *MockIEvaluatorRepo) CreateEvaluator(arg0 context.Context, arg1 *entity.Evaluator) (int64, error) { +func (m *MockIEvaluatorRepo) CreateEvaluator(ctx context.Context, evaluator *entity.Evaluator) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateEvaluator", arg0, arg1) + ret := m.ctrl.Call(m, "CreateEvaluator", ctx, evaluator) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateEvaluator indicates an expected call of CreateEvaluator. -func (mr *MockIEvaluatorRepoMockRecorder) CreateEvaluator(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) CreateEvaluator(ctx, evaluator any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluator", reflect.TypeOf((*MockIEvaluatorRepo)(nil).CreateEvaluator), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluator", reflect.TypeOf((*MockIEvaluatorRepo)(nil).CreateEvaluator), ctx, evaluator) } // ListEvaluator mocks base method. -func (m *MockIEvaluatorRepo) ListEvaluator(arg0 context.Context, arg1 *repo.ListEvaluatorRequest) (*repo.ListEvaluatorResponse, error) { +func (m *MockIEvaluatorRepo) ListEvaluator(ctx context.Context, req *repo.ListEvaluatorRequest) (*repo.ListEvaluatorResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListEvaluator", arg0, arg1) + ret := m.ctrl.Call(m, "ListEvaluator", ctx, req) ret0, _ := ret[0].(*repo.ListEvaluatorResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ListEvaluator indicates an expected call of ListEvaluator. -func (mr *MockIEvaluatorRepoMockRecorder) ListEvaluator(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) ListEvaluator(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluator", reflect.TypeOf((*MockIEvaluatorRepo)(nil).ListEvaluator), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluator", reflect.TypeOf((*MockIEvaluatorRepo)(nil).ListEvaluator), ctx, req) } // ListEvaluatorVersion mocks base method. -func (m *MockIEvaluatorRepo) ListEvaluatorVersion(arg0 context.Context, arg1 *repo.ListEvaluatorVersionRequest) (*repo.ListEvaluatorVersionResponse, error) { +func (m *MockIEvaluatorRepo) ListEvaluatorVersion(ctx context.Context, req *repo.ListEvaluatorVersionRequest) (*repo.ListEvaluatorVersionResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListEvaluatorVersion", arg0, arg1) + ret := m.ctrl.Call(m, "ListEvaluatorVersion", ctx, req) ret0, _ := ret[0].(*repo.ListEvaluatorVersionResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ListEvaluatorVersion indicates an expected call of ListEvaluatorVersion. -func (mr *MockIEvaluatorRepoMockRecorder) ListEvaluatorVersion(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) ListEvaluatorVersion(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluatorVersion", reflect.TypeOf((*MockIEvaluatorRepo)(nil).ListEvaluatorVersion), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluatorVersion", reflect.TypeOf((*MockIEvaluatorRepo)(nil).ListEvaluatorVersion), ctx, req) } // SubmitEvaluatorVersion mocks base method. -func (m *MockIEvaluatorRepo) SubmitEvaluatorVersion(arg0 context.Context, arg1 *entity.Evaluator) error { +func (m *MockIEvaluatorRepo) SubmitEvaluatorVersion(ctx context.Context, evaluatorVersionDO *entity.Evaluator) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SubmitEvaluatorVersion", arg0, arg1) + ret := m.ctrl.Call(m, "SubmitEvaluatorVersion", ctx, evaluatorVersionDO) ret0, _ := ret[0].(error) return ret0 } // SubmitEvaluatorVersion indicates an expected call of SubmitEvaluatorVersion. -func (mr *MockIEvaluatorRepoMockRecorder) SubmitEvaluatorVersion(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) SubmitEvaluatorVersion(ctx, evaluatorVersionDO any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SubmitEvaluatorVersion", reflect.TypeOf((*MockIEvaluatorRepo)(nil).SubmitEvaluatorVersion), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SubmitEvaluatorVersion", reflect.TypeOf((*MockIEvaluatorRepo)(nil).SubmitEvaluatorVersion), ctx, evaluatorVersionDO) } // UpdateEvaluatorDraft mocks base method. -func (m *MockIEvaluatorRepo) UpdateEvaluatorDraft(arg0 context.Context, arg1 *entity.Evaluator) error { +func (m *MockIEvaluatorRepo) UpdateEvaluatorDraft(ctx context.Context, version *entity.Evaluator) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateEvaluatorDraft", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateEvaluatorDraft", ctx, version) ret0, _ := ret[0].(error) return ret0 } // UpdateEvaluatorDraft indicates an expected call of UpdateEvaluatorDraft. -func (mr *MockIEvaluatorRepoMockRecorder) UpdateEvaluatorDraft(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) UpdateEvaluatorDraft(ctx, version any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorDraft", reflect.TypeOf((*MockIEvaluatorRepo)(nil).UpdateEvaluatorDraft), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorDraft", reflect.TypeOf((*MockIEvaluatorRepo)(nil).UpdateEvaluatorDraft), ctx, version) } // UpdateEvaluatorMeta mocks base method. -func (m *MockIEvaluatorRepo) UpdateEvaluatorMeta(arg0 context.Context, arg1 int64, arg2, arg3, arg4 string) error { +func (m *MockIEvaluatorRepo) UpdateEvaluatorMeta(ctx context.Context, id int64, name, description, userID string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateEvaluatorMeta", arg0, arg1, arg2, arg3, arg4) + ret := m.ctrl.Call(m, "UpdateEvaluatorMeta", ctx, id, name, description, userID) ret0, _ := ret[0].(error) return ret0 } // UpdateEvaluatorMeta indicates an expected call of UpdateEvaluatorMeta. -func (mr *MockIEvaluatorRepoMockRecorder) UpdateEvaluatorMeta(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRepoMockRecorder) UpdateEvaluatorMeta(ctx, id, name, description, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorMeta", reflect.TypeOf((*MockIEvaluatorRepo)(nil).UpdateEvaluatorMeta), arg0, arg1, arg2, arg3, arg4) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorMeta", reflect.TypeOf((*MockIEvaluatorRepo)(nil).UpdateEvaluatorMeta), ctx, id, name, description, userID) } diff --git a/backend/modules/evaluation/domain/repo/mocks/evaluator_record_mock.go b/backend/modules/evaluation/domain/repo/mocks/evaluator_record_mock.go index 30bd9b3a2..e0ca5d9e1 100644 --- a/backend/modules/evaluation/domain/repo/mocks/evaluator_record_mock.go +++ b/backend/modules/evaluation/domain/repo/mocks/evaluator_record_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/repo (interfaces: IEvaluatorRecordRepo) +// +// Generated by this command: +// +// mockgen -destination mocks/evaluator_record_mock.go -package mocks . IEvaluatorRecordRepo +// // Package mocks is a generated GoMock package. package mocks @@ -16,6 +21,7 @@ import ( type MockIEvaluatorRecordRepo struct { ctrl *gomock.Controller recorder *MockIEvaluatorRecordRepoMockRecorder + isgomock struct{} } // MockIEvaluatorRecordRepoMockRecorder is the mock recorder for MockIEvaluatorRecordRepo. @@ -36,59 +42,59 @@ func (m *MockIEvaluatorRecordRepo) EXPECT() *MockIEvaluatorRecordRepoMockRecorde } // BatchGetEvaluatorRecord mocks base method. -func (m *MockIEvaluatorRecordRepo) BatchGetEvaluatorRecord(arg0 context.Context, arg1 []int64, arg2 bool) ([]*entity.EvaluatorRecord, error) { +func (m *MockIEvaluatorRecordRepo) BatchGetEvaluatorRecord(ctx context.Context, evaluatorRecordIDs []int64, includeDeleted bool) ([]*entity.EvaluatorRecord, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchGetEvaluatorRecord", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "BatchGetEvaluatorRecord", ctx, evaluatorRecordIDs, includeDeleted) ret0, _ := ret[0].([]*entity.EvaluatorRecord) ret1, _ := ret[1].(error) return ret0, ret1 } // BatchGetEvaluatorRecord indicates an expected call of BatchGetEvaluatorRecord. -func (mr *MockIEvaluatorRecordRepoMockRecorder) BatchGetEvaluatorRecord(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRecordRepoMockRecorder) BatchGetEvaluatorRecord(ctx, evaluatorRecordIDs, includeDeleted any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRecordRepo)(nil).BatchGetEvaluatorRecord), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRecordRepo)(nil).BatchGetEvaluatorRecord), ctx, evaluatorRecordIDs, includeDeleted) } // CorrectEvaluatorRecord mocks base method. -func (m *MockIEvaluatorRecordRepo) CorrectEvaluatorRecord(arg0 context.Context, arg1 *entity.EvaluatorRecord) error { +func (m *MockIEvaluatorRecordRepo) CorrectEvaluatorRecord(ctx context.Context, evaluatorRecordDO *entity.EvaluatorRecord) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CorrectEvaluatorRecord", arg0, arg1) + ret := m.ctrl.Call(m, "CorrectEvaluatorRecord", ctx, evaluatorRecordDO) ret0, _ := ret[0].(error) return ret0 } // CorrectEvaluatorRecord indicates an expected call of CorrectEvaluatorRecord. -func (mr *MockIEvaluatorRecordRepoMockRecorder) CorrectEvaluatorRecord(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRecordRepoMockRecorder) CorrectEvaluatorRecord(ctx, evaluatorRecordDO any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CorrectEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRecordRepo)(nil).CorrectEvaluatorRecord), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CorrectEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRecordRepo)(nil).CorrectEvaluatorRecord), ctx, evaluatorRecordDO) } // CreateEvaluatorRecord mocks base method. -func (m *MockIEvaluatorRecordRepo) CreateEvaluatorRecord(arg0 context.Context, arg1 *entity.EvaluatorRecord) error { +func (m *MockIEvaluatorRecordRepo) CreateEvaluatorRecord(ctx context.Context, evaluatorRecord *entity.EvaluatorRecord) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateEvaluatorRecord", arg0, arg1) + ret := m.ctrl.Call(m, "CreateEvaluatorRecord", ctx, evaluatorRecord) ret0, _ := ret[0].(error) return ret0 } // CreateEvaluatorRecord indicates an expected call of CreateEvaluatorRecord. -func (mr *MockIEvaluatorRecordRepoMockRecorder) CreateEvaluatorRecord(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRecordRepoMockRecorder) CreateEvaluatorRecord(ctx, evaluatorRecord any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRecordRepo)(nil).CreateEvaluatorRecord), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRecordRepo)(nil).CreateEvaluatorRecord), ctx, evaluatorRecord) } // GetEvaluatorRecord mocks base method. -func (m *MockIEvaluatorRecordRepo) GetEvaluatorRecord(arg0 context.Context, arg1 int64, arg2 bool) (*entity.EvaluatorRecord, error) { +func (m *MockIEvaluatorRecordRepo) GetEvaluatorRecord(ctx context.Context, evaluatorRecordID int64, includeDeleted bool) (*entity.EvaluatorRecord, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetEvaluatorRecord", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetEvaluatorRecord", ctx, evaluatorRecordID, includeDeleted) ret0, _ := ret[0].(*entity.EvaluatorRecord) ret1, _ := ret[1].(error) return ret0, ret1 } // GetEvaluatorRecord indicates an expected call of GetEvaluatorRecord. -func (mr *MockIEvaluatorRecordRepoMockRecorder) GetEvaluatorRecord(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIEvaluatorRecordRepoMockRecorder) GetEvaluatorRecord(ctx, evaluatorRecordID, includeDeleted any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRecordRepo)(nil).GetEvaluatorRecord), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRecordRepo)(nil).GetEvaluatorRecord), ctx, evaluatorRecordID, includeDeleted) } diff --git a/backend/modules/evaluation/domain/repo/mocks/ratelimiter_mock.go b/backend/modules/evaluation/domain/repo/mocks/ratelimiter_mock.go index 4303f9745..55d4b8839 100644 --- a/backend/modules/evaluation/domain/repo/mocks/ratelimiter_mock.go +++ b/backend/modules/evaluation/domain/repo/mocks/ratelimiter_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/repo (interfaces: RateLimiter) +// +// Generated by this command: +// +// mockgen -destination mocks/ratelimiter_mock.go -package mocks . RateLimiter +// // Package mocks is a generated GoMock package. package mocks @@ -15,6 +20,7 @@ import ( type MockRateLimiter struct { ctrl *gomock.Controller recorder *MockRateLimiterMockRecorder + isgomock struct{} } // MockRateLimiterMockRecorder is the mock recorder for MockRateLimiter. @@ -35,15 +41,15 @@ func (m *MockRateLimiter) EXPECT() *MockRateLimiterMockRecorder { } // AllowInvoke mocks base method. -func (m *MockRateLimiter) AllowInvoke(arg0 context.Context, arg1 int64) bool { +func (m *MockRateLimiter) AllowInvoke(ctx context.Context, spaceID int64) bool { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "AllowInvoke", arg0, arg1) + ret := m.ctrl.Call(m, "AllowInvoke", ctx, spaceID) ret0, _ := ret[0].(bool) return ret0 } // AllowInvoke indicates an expected call of AllowInvoke. -func (mr *MockRateLimiterMockRecorder) AllowInvoke(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockRateLimiterMockRecorder) AllowInvoke(ctx, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AllowInvoke", reflect.TypeOf((*MockRateLimiter)(nil).AllowInvoke), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AllowInvoke", reflect.TypeOf((*MockRateLimiter)(nil).AllowInvoke), ctx, spaceID) } diff --git a/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_mock.go b/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_mock.go index 7214747f3..50db0136d 100644 --- a/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_mock.go +++ b/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/evaluator/mysql (interfaces: EvaluatorDAO) +// +// Generated by this command: +// +// mockgen -destination mocks/evaluator_mock.go -package=mocks . EvaluatorDAO +// // Package mocks is a generated GoMock package. package mocks @@ -18,6 +23,7 @@ import ( type MockEvaluatorDAO struct { ctrl *gomock.Controller recorder *MockEvaluatorDAOMockRecorder + isgomock struct{} } // MockEvaluatorDAOMockRecorder is the mock recorder for MockEvaluatorDAO. @@ -38,10 +44,10 @@ func (m *MockEvaluatorDAO) EXPECT() *MockEvaluatorDAOMockRecorder { } // BatchDeleteEvaluator mocks base method. -func (m *MockEvaluatorDAO) BatchDeleteEvaluator(arg0 context.Context, arg1 []int64, arg2 string, arg3 ...db.Option) error { +func (m *MockEvaluatorDAO) BatchDeleteEvaluator(ctx context.Context, ids []int64, userID string, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, ids, userID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchDeleteEvaluator", varargs...) @@ -50,17 +56,17 @@ func (m *MockEvaluatorDAO) BatchDeleteEvaluator(arg0 context.Context, arg1 []int } // BatchDeleteEvaluator indicates an expected call of BatchDeleteEvaluator. -func (mr *MockEvaluatorDAOMockRecorder) BatchDeleteEvaluator(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) BatchDeleteEvaluator(ctx, ids, userID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, ids, userID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchDeleteEvaluator", reflect.TypeOf((*MockEvaluatorDAO)(nil).BatchDeleteEvaluator), varargs...) } // BatchGetEvaluatorByID mocks base method. -func (m *MockEvaluatorDAO) BatchGetEvaluatorByID(arg0 context.Context, arg1 []int64, arg2 bool, arg3 ...db.Option) ([]*model.Evaluator, error) { +func (m *MockEvaluatorDAO) BatchGetEvaluatorByID(ctx context.Context, ids []int64, includeDeleted bool, opts ...db.Option) ([]*model.Evaluator, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, ids, includeDeleted} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetEvaluatorByID", varargs...) @@ -70,17 +76,17 @@ func (m *MockEvaluatorDAO) BatchGetEvaluatorByID(arg0 context.Context, arg1 []in } // BatchGetEvaluatorByID indicates an expected call of BatchGetEvaluatorByID. -func (mr *MockEvaluatorDAOMockRecorder) BatchGetEvaluatorByID(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) BatchGetEvaluatorByID(ctx, ids, includeDeleted any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, ids, includeDeleted}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorByID", reflect.TypeOf((*MockEvaluatorDAO)(nil).BatchGetEvaluatorByID), varargs...) } // CheckNameExist mocks base method. -func (m *MockEvaluatorDAO) CheckNameExist(arg0 context.Context, arg1, arg2 int64, arg3 string, arg4 ...db.Option) (bool, error) { +func (m *MockEvaluatorDAO) CheckNameExist(ctx context.Context, spaceID, evaluatorID int64, name string, opts ...db.Option) (bool, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, evaluatorID, name} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CheckNameExist", varargs...) @@ -90,17 +96,17 @@ func (m *MockEvaluatorDAO) CheckNameExist(arg0 context.Context, arg1, arg2 int64 } // CheckNameExist indicates an expected call of CheckNameExist. -func (mr *MockEvaluatorDAOMockRecorder) CheckNameExist(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) CheckNameExist(ctx, spaceID, evaluatorID, name any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, evaluatorID, name}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckNameExist", reflect.TypeOf((*MockEvaluatorDAO)(nil).CheckNameExist), varargs...) } // CreateEvaluator mocks base method. -func (m *MockEvaluatorDAO) CreateEvaluator(arg0 context.Context, arg1 *model.Evaluator, arg2 ...db.Option) error { +func (m *MockEvaluatorDAO) CreateEvaluator(ctx context.Context, evaluator *model.Evaluator, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, evaluator} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateEvaluator", varargs...) @@ -109,17 +115,17 @@ func (m *MockEvaluatorDAO) CreateEvaluator(arg0 context.Context, arg1 *model.Eva } // CreateEvaluator indicates an expected call of CreateEvaluator. -func (mr *MockEvaluatorDAOMockRecorder) CreateEvaluator(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) CreateEvaluator(ctx, evaluator any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, evaluator}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluator", reflect.TypeOf((*MockEvaluatorDAO)(nil).CreateEvaluator), varargs...) } // GetEvaluatorByID mocks base method. -func (m *MockEvaluatorDAO) GetEvaluatorByID(arg0 context.Context, arg1 int64, arg2 bool, arg3 ...db.Option) (*model.Evaluator, error) { +func (m *MockEvaluatorDAO) GetEvaluatorByID(ctx context.Context, id int64, includeDeleted bool, opts ...db.Option) (*model.Evaluator, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, id, includeDeleted} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetEvaluatorByID", varargs...) @@ -129,17 +135,17 @@ func (m *MockEvaluatorDAO) GetEvaluatorByID(arg0 context.Context, arg1 int64, ar } // GetEvaluatorByID indicates an expected call of GetEvaluatorByID. -func (mr *MockEvaluatorDAOMockRecorder) GetEvaluatorByID(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) GetEvaluatorByID(ctx, id, includeDeleted any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, id, includeDeleted}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEvaluatorByID", reflect.TypeOf((*MockEvaluatorDAO)(nil).GetEvaluatorByID), varargs...) } // ListEvaluator mocks base method. -func (m *MockEvaluatorDAO) ListEvaluator(arg0 context.Context, arg1 *mysql.ListEvaluatorRequest, arg2 ...db.Option) (*mysql.ListEvaluatorResponse, error) { +func (m *MockEvaluatorDAO) ListEvaluator(ctx context.Context, req *mysql.ListEvaluatorRequest, opts ...db.Option) (*mysql.ListEvaluatorResponse, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListEvaluator", varargs...) @@ -149,17 +155,17 @@ func (m *MockEvaluatorDAO) ListEvaluator(arg0 context.Context, arg1 *mysql.ListE } // ListEvaluator indicates an expected call of ListEvaluator. -func (mr *MockEvaluatorDAOMockRecorder) ListEvaluator(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) ListEvaluator(ctx, req any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluator", reflect.TypeOf((*MockEvaluatorDAO)(nil).ListEvaluator), varargs...) } // UpdateEvaluatorDraftSubmitted mocks base method. -func (m *MockEvaluatorDAO) UpdateEvaluatorDraftSubmitted(arg0 context.Context, arg1 int64, arg2 bool, arg3 string, arg4 ...db.Option) error { +func (m *MockEvaluatorDAO) UpdateEvaluatorDraftSubmitted(ctx context.Context, evaluatorID int64, draftSubmitted bool, userID string, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, evaluatorID, draftSubmitted, userID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateEvaluatorDraftSubmitted", varargs...) @@ -168,17 +174,17 @@ func (m *MockEvaluatorDAO) UpdateEvaluatorDraftSubmitted(arg0 context.Context, a } // UpdateEvaluatorDraftSubmitted indicates an expected call of UpdateEvaluatorDraftSubmitted. -func (mr *MockEvaluatorDAOMockRecorder) UpdateEvaluatorDraftSubmitted(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) UpdateEvaluatorDraftSubmitted(ctx, evaluatorID, draftSubmitted, userID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, evaluatorID, draftSubmitted, userID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorDraftSubmitted", reflect.TypeOf((*MockEvaluatorDAO)(nil).UpdateEvaluatorDraftSubmitted), varargs...) } // UpdateEvaluatorLatestVersion mocks base method. -func (m *MockEvaluatorDAO) UpdateEvaluatorLatestVersion(arg0 context.Context, arg1 int64, arg2, arg3 string, arg4 ...db.Option) error { +func (m *MockEvaluatorDAO) UpdateEvaluatorLatestVersion(ctx context.Context, evaluatorID int64, version, userID string, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, evaluatorID, version, userID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateEvaluatorLatestVersion", varargs...) @@ -187,17 +193,17 @@ func (m *MockEvaluatorDAO) UpdateEvaluatorLatestVersion(arg0 context.Context, ar } // UpdateEvaluatorLatestVersion indicates an expected call of UpdateEvaluatorLatestVersion. -func (mr *MockEvaluatorDAOMockRecorder) UpdateEvaluatorLatestVersion(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) UpdateEvaluatorLatestVersion(ctx, evaluatorID, version, userID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, evaluatorID, version, userID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorLatestVersion", reflect.TypeOf((*MockEvaluatorDAO)(nil).UpdateEvaluatorLatestVersion), varargs...) } // UpdateEvaluatorMeta mocks base method. -func (m *MockEvaluatorDAO) UpdateEvaluatorMeta(arg0 context.Context, arg1 *model.Evaluator, arg2 ...db.Option) error { +func (m *MockEvaluatorDAO) UpdateEvaluatorMeta(ctx context.Context, do *model.Evaluator, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, do} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateEvaluatorMeta", varargs...) @@ -206,8 +212,8 @@ func (m *MockEvaluatorDAO) UpdateEvaluatorMeta(arg0 context.Context, arg1 *model } // UpdateEvaluatorMeta indicates an expected call of UpdateEvaluatorMeta. -func (mr *MockEvaluatorDAOMockRecorder) UpdateEvaluatorMeta(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorDAOMockRecorder) UpdateEvaluatorMeta(ctx, do any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, do}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorMeta", reflect.TypeOf((*MockEvaluatorDAO)(nil).UpdateEvaluatorMeta), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_record_mock.go b/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_record_mock.go index bc48d275e..90fda9a1b 100644 --- a/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_record_mock.go +++ b/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_record_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/evaluator/mysql (interfaces: EvaluatorRecordDAO) +// +// Generated by this command: +// +// mockgen -destination mocks/evaluator_record_mock.go -package=mocks . EvaluatorRecordDAO +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockEvaluatorRecordDAO struct { ctrl *gomock.Controller recorder *MockEvaluatorRecordDAOMockRecorder + isgomock struct{} } // MockEvaluatorRecordDAOMockRecorder is the mock recorder for MockEvaluatorRecordDAO. @@ -37,10 +43,10 @@ func (m *MockEvaluatorRecordDAO) EXPECT() *MockEvaluatorRecordDAOMockRecorder { } // BatchGetEvaluatorRecord mocks base method. -func (m *MockEvaluatorRecordDAO) BatchGetEvaluatorRecord(arg0 context.Context, arg1 []int64, arg2 bool, arg3 ...db.Option) ([]*model.EvaluatorRecord, error) { +func (m *MockEvaluatorRecordDAO) BatchGetEvaluatorRecord(ctx context.Context, evaluatorRecordIDs []int64, includeDeleted bool, opts ...db.Option) ([]*model.EvaluatorRecord, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, evaluatorRecordIDs, includeDeleted} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetEvaluatorRecord", varargs...) @@ -50,17 +56,17 @@ func (m *MockEvaluatorRecordDAO) BatchGetEvaluatorRecord(arg0 context.Context, a } // BatchGetEvaluatorRecord indicates an expected call of BatchGetEvaluatorRecord. -func (mr *MockEvaluatorRecordDAOMockRecorder) BatchGetEvaluatorRecord(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorRecordDAOMockRecorder) BatchGetEvaluatorRecord(ctx, evaluatorRecordIDs, includeDeleted any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, evaluatorRecordIDs, includeDeleted}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorRecord", reflect.TypeOf((*MockEvaluatorRecordDAO)(nil).BatchGetEvaluatorRecord), varargs...) } // CreateEvaluatorRecord mocks base method. -func (m *MockEvaluatorRecordDAO) CreateEvaluatorRecord(arg0 context.Context, arg1 *model.EvaluatorRecord, arg2 ...db.Option) error { +func (m *MockEvaluatorRecordDAO) CreateEvaluatorRecord(ctx context.Context, evaluatorRecord *model.EvaluatorRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, evaluatorRecord} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateEvaluatorRecord", varargs...) @@ -69,17 +75,17 @@ func (m *MockEvaluatorRecordDAO) CreateEvaluatorRecord(arg0 context.Context, arg } // CreateEvaluatorRecord indicates an expected call of CreateEvaluatorRecord. -func (mr *MockEvaluatorRecordDAOMockRecorder) CreateEvaluatorRecord(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorRecordDAOMockRecorder) CreateEvaluatorRecord(ctx, evaluatorRecord any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, evaluatorRecord}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluatorRecord", reflect.TypeOf((*MockEvaluatorRecordDAO)(nil).CreateEvaluatorRecord), varargs...) } // GetEvaluatorRecord mocks base method. -func (m *MockEvaluatorRecordDAO) GetEvaluatorRecord(arg0 context.Context, arg1 int64, arg2 bool, arg3 ...db.Option) (*model.EvaluatorRecord, error) { +func (m *MockEvaluatorRecordDAO) GetEvaluatorRecord(ctx context.Context, evaluatorRecordID int64, includeDeleted bool, opts ...db.Option) (*model.EvaluatorRecord, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, evaluatorRecordID, includeDeleted} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetEvaluatorRecord", varargs...) @@ -89,17 +95,17 @@ func (m *MockEvaluatorRecordDAO) GetEvaluatorRecord(arg0 context.Context, arg1 i } // GetEvaluatorRecord indicates an expected call of GetEvaluatorRecord. -func (mr *MockEvaluatorRecordDAOMockRecorder) GetEvaluatorRecord(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorRecordDAOMockRecorder) GetEvaluatorRecord(ctx, evaluatorRecordID, includeDeleted any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, evaluatorRecordID, includeDeleted}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEvaluatorRecord", reflect.TypeOf((*MockEvaluatorRecordDAO)(nil).GetEvaluatorRecord), varargs...) } // UpdateEvaluatorRecord mocks base method. -func (m *MockEvaluatorRecordDAO) UpdateEvaluatorRecord(arg0 context.Context, arg1 *model.EvaluatorRecord, arg2 ...db.Option) error { +func (m *MockEvaluatorRecordDAO) UpdateEvaluatorRecord(ctx context.Context, evaluatorRecord *model.EvaluatorRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, evaluatorRecord} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateEvaluatorRecord", varargs...) @@ -108,8 +114,8 @@ func (m *MockEvaluatorRecordDAO) UpdateEvaluatorRecord(arg0 context.Context, arg } // UpdateEvaluatorRecord indicates an expected call of UpdateEvaluatorRecord. -func (mr *MockEvaluatorRecordDAOMockRecorder) UpdateEvaluatorRecord(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorRecordDAOMockRecorder) UpdateEvaluatorRecord(ctx, evaluatorRecord any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, evaluatorRecord}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorRecord", reflect.TypeOf((*MockEvaluatorRecordDAO)(nil).UpdateEvaluatorRecord), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_version_mock.go b/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_version_mock.go index 2922b3b2f..a0d7965eb 100644 --- a/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_version_mock.go +++ b/backend/modules/evaluation/infra/repo/evaluator/mysql/mocks/evaluator_version_mock.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/evaluator/mysql (interfaces: EvaluatorVersionDAO) +// +// Generated by this command: +// +// mockgen -destination mocks/evaluator_version_mock.go -package=mocks . EvaluatorVersionDAO +// // Package mocks is a generated GoMock package. package mocks @@ -18,6 +23,7 @@ import ( type MockEvaluatorVersionDAO struct { ctrl *gomock.Controller recorder *MockEvaluatorVersionDAOMockRecorder + isgomock struct{} } // MockEvaluatorVersionDAOMockRecorder is the mock recorder for MockEvaluatorVersionDAO. @@ -38,10 +44,10 @@ func (m *MockEvaluatorVersionDAO) EXPECT() *MockEvaluatorVersionDAOMockRecorder } // BatchDeleteEvaluatorVersionByEvaluatorIDs mocks base method. -func (m *MockEvaluatorVersionDAO) BatchDeleteEvaluatorVersionByEvaluatorIDs(arg0 context.Context, arg1 []int64, arg2 string, arg3 ...db.Option) error { +func (m *MockEvaluatorVersionDAO) BatchDeleteEvaluatorVersionByEvaluatorIDs(ctx context.Context, evaluatorIDs []int64, userID string, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, evaluatorIDs, userID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchDeleteEvaluatorVersionByEvaluatorIDs", varargs...) @@ -50,17 +56,17 @@ func (m *MockEvaluatorVersionDAO) BatchDeleteEvaluatorVersionByEvaluatorIDs(arg0 } // BatchDeleteEvaluatorVersionByEvaluatorIDs indicates an expected call of BatchDeleteEvaluatorVersionByEvaluatorIDs. -func (mr *MockEvaluatorVersionDAOMockRecorder) BatchDeleteEvaluatorVersionByEvaluatorIDs(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) BatchDeleteEvaluatorVersionByEvaluatorIDs(ctx, evaluatorIDs, userID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, evaluatorIDs, userID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchDeleteEvaluatorVersionByEvaluatorIDs", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).BatchDeleteEvaluatorVersionByEvaluatorIDs), varargs...) } // BatchGetEvaluatorDraftByEvaluatorID mocks base method. -func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorDraftByEvaluatorID(arg0 context.Context, arg1 []int64, arg2 bool, arg3 ...db.Option) ([]*model.EvaluatorVersion, error) { +func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorDraftByEvaluatorID(ctx context.Context, evaluatorIDs []int64, includeDeleted bool, opts ...db.Option) ([]*model.EvaluatorVersion, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, evaluatorIDs, includeDeleted} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetEvaluatorDraftByEvaluatorID", varargs...) @@ -70,17 +76,17 @@ func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorDraftByEvaluatorID(arg0 conte } // BatchGetEvaluatorDraftByEvaluatorID indicates an expected call of BatchGetEvaluatorDraftByEvaluatorID. -func (mr *MockEvaluatorVersionDAOMockRecorder) BatchGetEvaluatorDraftByEvaluatorID(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) BatchGetEvaluatorDraftByEvaluatorID(ctx, evaluatorIDs, includeDeleted any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, evaluatorIDs, includeDeleted}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorDraftByEvaluatorID", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).BatchGetEvaluatorDraftByEvaluatorID), varargs...) } // BatchGetEvaluatorVersionByID mocks base method. -func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorVersionByID(arg0 context.Context, arg1 *int64, arg2 []int64, arg3 bool, arg4 ...db.Option) ([]*model.EvaluatorVersion, error) { +func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorVersionByID(ctx context.Context, spaceID *int64, ids []int64, includeDeleted bool, opts ...db.Option) ([]*model.EvaluatorVersion, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, ids, includeDeleted} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetEvaluatorVersionByID", varargs...) @@ -90,17 +96,17 @@ func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorVersionByID(arg0 context.Cont } // BatchGetEvaluatorVersionByID indicates an expected call of BatchGetEvaluatorVersionByID. -func (mr *MockEvaluatorVersionDAOMockRecorder) BatchGetEvaluatorVersionByID(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) BatchGetEvaluatorVersionByID(ctx, spaceID, ids, includeDeleted any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, ids, includeDeleted}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorVersionByID", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).BatchGetEvaluatorVersionByID), varargs...) } // BatchGetEvaluatorVersionsByEvaluatorIDs mocks base method. -func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorVersionsByEvaluatorIDs(arg0 context.Context, arg1 []int64, arg2 bool, arg3 ...db.Option) ([]*model.EvaluatorVersion, error) { +func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorVersionsByEvaluatorIDs(ctx context.Context, evaluatorIDs []int64, includeDeleted bool, opts ...db.Option) ([]*model.EvaluatorVersion, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, evaluatorIDs, includeDeleted} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetEvaluatorVersionsByEvaluatorIDs", varargs...) @@ -110,17 +116,17 @@ func (m *MockEvaluatorVersionDAO) BatchGetEvaluatorVersionsByEvaluatorIDs(arg0 c } // BatchGetEvaluatorVersionsByEvaluatorIDs indicates an expected call of BatchGetEvaluatorVersionsByEvaluatorIDs. -func (mr *MockEvaluatorVersionDAOMockRecorder) BatchGetEvaluatorVersionsByEvaluatorIDs(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) BatchGetEvaluatorVersionsByEvaluatorIDs(ctx, evaluatorIDs, includeDeleted any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, evaluatorIDs, includeDeleted}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorVersionsByEvaluatorIDs", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).BatchGetEvaluatorVersionsByEvaluatorIDs), varargs...) } // CheckVersionExist mocks base method. -func (m *MockEvaluatorVersionDAO) CheckVersionExist(arg0 context.Context, arg1 int64, arg2 string, arg3 ...db.Option) (bool, error) { +func (m *MockEvaluatorVersionDAO) CheckVersionExist(ctx context.Context, evaluatorID int64, version string, opts ...db.Option) (bool, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, evaluatorID, version} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CheckVersionExist", varargs...) @@ -130,17 +136,17 @@ func (m *MockEvaluatorVersionDAO) CheckVersionExist(arg0 context.Context, arg1 i } // CheckVersionExist indicates an expected call of CheckVersionExist. -func (mr *MockEvaluatorVersionDAOMockRecorder) CheckVersionExist(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) CheckVersionExist(ctx, evaluatorID, version any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, evaluatorID, version}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckVersionExist", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).CheckVersionExist), varargs...) } // CreateEvaluatorVersion mocks base method. -func (m *MockEvaluatorVersionDAO) CreateEvaluatorVersion(arg0 context.Context, arg1 *model.EvaluatorVersion, arg2 ...db.Option) error { +func (m *MockEvaluatorVersionDAO) CreateEvaluatorVersion(ctx context.Context, version *model.EvaluatorVersion, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, version} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateEvaluatorVersion", varargs...) @@ -149,17 +155,17 @@ func (m *MockEvaluatorVersionDAO) CreateEvaluatorVersion(arg0 context.Context, a } // CreateEvaluatorVersion indicates an expected call of CreateEvaluatorVersion. -func (mr *MockEvaluatorVersionDAOMockRecorder) CreateEvaluatorVersion(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) CreateEvaluatorVersion(ctx, version any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, version}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluatorVersion", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).CreateEvaluatorVersion), varargs...) } // DeleteEvaluatorVersion mocks base method. -func (m *MockEvaluatorVersionDAO) DeleteEvaluatorVersion(arg0 context.Context, arg1 int64, arg2 string, arg3 ...db.Option) error { +func (m *MockEvaluatorVersionDAO) DeleteEvaluatorVersion(ctx context.Context, id int64, userID string, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, id, userID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteEvaluatorVersion", varargs...) @@ -168,17 +174,17 @@ func (m *MockEvaluatorVersionDAO) DeleteEvaluatorVersion(arg0 context.Context, a } // DeleteEvaluatorVersion indicates an expected call of DeleteEvaluatorVersion. -func (mr *MockEvaluatorVersionDAOMockRecorder) DeleteEvaluatorVersion(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) DeleteEvaluatorVersion(ctx, id, userID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, id, userID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteEvaluatorVersion", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).DeleteEvaluatorVersion), varargs...) } // ListEvaluatorVersion mocks base method. -func (m *MockEvaluatorVersionDAO) ListEvaluatorVersion(arg0 context.Context, arg1 *mysql.ListEvaluatorVersionRequest, arg2 ...db.Option) (*mysql.ListEvaluatorVersionResponse, error) { +func (m *MockEvaluatorVersionDAO) ListEvaluatorVersion(ctx context.Context, req *mysql.ListEvaluatorVersionRequest, opts ...db.Option) (*mysql.ListEvaluatorVersionResponse, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListEvaluatorVersion", varargs...) @@ -188,17 +194,17 @@ func (m *MockEvaluatorVersionDAO) ListEvaluatorVersion(arg0 context.Context, arg } // ListEvaluatorVersion indicates an expected call of ListEvaluatorVersion. -func (mr *MockEvaluatorVersionDAOMockRecorder) ListEvaluatorVersion(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) ListEvaluatorVersion(ctx, req any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluatorVersion", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).ListEvaluatorVersion), varargs...) } // UpdateEvaluatorDraft mocks base method. -func (m *MockEvaluatorVersionDAO) UpdateEvaluatorDraft(arg0 context.Context, arg1 *model.EvaluatorVersion, arg2 ...db.Option) error { +func (m *MockEvaluatorVersionDAO) UpdateEvaluatorDraft(ctx context.Context, version *model.EvaluatorVersion, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, version} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateEvaluatorDraft", varargs...) @@ -207,8 +213,8 @@ func (m *MockEvaluatorVersionDAO) UpdateEvaluatorDraft(arg0 context.Context, arg } // UpdateEvaluatorDraft indicates an expected call of UpdateEvaluatorDraft. -func (mr *MockEvaluatorVersionDAOMockRecorder) UpdateEvaluatorDraft(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockEvaluatorVersionDAOMockRecorder) UpdateEvaluatorDraft(ctx, version any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, version}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorDraft", reflect.TypeOf((*MockEvaluatorVersionDAO)(nil).UpdateEvaluatorDraft), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/ck/mocks/expt.go b/backend/modules/evaluation/infra/repo/experiment/ck/mocks/expt.go index e11be126e..242e6ce26 100644 --- a/backend/modules/evaluation/infra/repo/experiment/ck/mocks/expt.go +++ b/backend/modules/evaluation/infra/repo/experiment/ck/mocks/expt.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/ck (interfaces: IExptTurnResultFilterDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt.go -package=mocks . IExptTurnResultFilterDAO +// // Package mocks is a generated GoMock package. package mocks @@ -18,6 +23,7 @@ import ( type MockIExptTurnResultFilterDAO struct { ctrl *gomock.Controller recorder *MockIExptTurnResultFilterDAOMockRecorder + isgomock struct{} } // MockIExptTurnResultFilterDAOMockRecorder is the mock recorder for MockIExptTurnResultFilterDAO. @@ -38,24 +44,24 @@ func (m *MockIExptTurnResultFilterDAO) EXPECT() *MockIExptTurnResultFilterDAOMoc } // GetByExptIDItemIDs mocks base method. -func (m *MockIExptTurnResultFilterDAO) GetByExptIDItemIDs(arg0 context.Context, arg1, arg2, arg3 string, arg4 []string) ([]*model0.ExptTurnResultFilter, error) { +func (m *MockIExptTurnResultFilterDAO) GetByExptIDItemIDs(ctx context.Context, spaceID, exptID, createdDate string, itemIDs []string) ([]*model0.ExptTurnResultFilter, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetByExptIDItemIDs", arg0, arg1, arg2, arg3, arg4) + ret := m.ctrl.Call(m, "GetByExptIDItemIDs", ctx, spaceID, exptID, createdDate, itemIDs) ret0, _ := ret[0].([]*model0.ExptTurnResultFilter) ret1, _ := ret[1].(error) return ret0, ret1 } // GetByExptIDItemIDs indicates an expected call of GetByExptIDItemIDs. -func (mr *MockIExptTurnResultFilterDAOMockRecorder) GetByExptIDItemIDs(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { +func (mr *MockIExptTurnResultFilterDAOMockRecorder) GetByExptIDItemIDs(ctx, spaceID, exptID, createdDate, itemIDs any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByExptIDItemIDs", reflect.TypeOf((*MockIExptTurnResultFilterDAO)(nil).GetByExptIDItemIDs), arg0, arg1, arg2, arg3, arg4) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByExptIDItemIDs", reflect.TypeOf((*MockIExptTurnResultFilterDAO)(nil).GetByExptIDItemIDs), ctx, spaceID, exptID, createdDate, itemIDs) } // QueryItemIDStates mocks base method. -func (m *MockIExptTurnResultFilterDAO) QueryItemIDStates(arg0 context.Context, arg1 *ck.ExptTurnResultFilterQueryCond) (map[string]int32, int64, error) { +func (m *MockIExptTurnResultFilterDAO) QueryItemIDStates(ctx context.Context, cond *ck.ExptTurnResultFilterQueryCond) (map[string]int32, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "QueryItemIDStates", arg0, arg1) + ret := m.ctrl.Call(m, "QueryItemIDStates", ctx, cond) ret0, _ := ret[0].(map[string]int32) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(error) @@ -63,21 +69,21 @@ func (m *MockIExptTurnResultFilterDAO) QueryItemIDStates(arg0 context.Context, a } // QueryItemIDStates indicates an expected call of QueryItemIDStates. -func (mr *MockIExptTurnResultFilterDAOMockRecorder) QueryItemIDStates(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptTurnResultFilterDAOMockRecorder) QueryItemIDStates(ctx, cond any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryItemIDStates", reflect.TypeOf((*MockIExptTurnResultFilterDAO)(nil).QueryItemIDStates), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryItemIDStates", reflect.TypeOf((*MockIExptTurnResultFilterDAO)(nil).QueryItemIDStates), ctx, cond) } // Save mocks base method. -func (m *MockIExptTurnResultFilterDAO) Save(arg0 context.Context, arg1 []*model.ExptTurnResultFilter) error { +func (m *MockIExptTurnResultFilterDAO) Save(ctx context.Context, filter []*model.ExptTurnResultFilter) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Save", arg0, arg1) + ret := m.ctrl.Call(m, "Save", ctx, filter) ret0, _ := ret[0].(error) return ret0 } // Save indicates an expected call of Save. -func (mr *MockIExptTurnResultFilterDAOMockRecorder) Save(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptTurnResultFilterDAOMockRecorder) Save(ctx, filter any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Save", reflect.TypeOf((*MockIExptTurnResultFilterDAO)(nil).Save), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Save", reflect.TypeOf((*MockIExptTurnResultFilterDAO)(nil).Save), ctx, filter) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/annotate_record.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/annotate_record.go index c1059477a..60c0fe23c 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/annotate_record.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/annotate_record.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IAnnotateRecordDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/annotate_record.go -package=mocks . IAnnotateRecordDAO +// // Package mocks is a generated GoMock package. package mocks @@ -8,16 +13,16 @@ import ( context "context" reflect "reflect" - "go.uber.org/mock/gomock" - db "github.com/coze-dev/coze-loop/backend/infra/db" model "github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql/gorm_gen/model" + gomock "go.uber.org/mock/gomock" ) // MockIAnnotateRecordDAO is a mock of IAnnotateRecordDAO interface. type MockIAnnotateRecordDAO struct { ctrl *gomock.Controller recorder *MockIAnnotateRecordDAOMockRecorder + isgomock struct{} } // MockIAnnotateRecordDAOMockRecorder is the mock recorder for MockIAnnotateRecordDAO. @@ -38,10 +43,10 @@ func (m *MockIAnnotateRecordDAO) EXPECT() *MockIAnnotateRecordDAOMockRecorder { } // BatchSave mocks base method. -func (m *MockIAnnotateRecordDAO) BatchSave(arg0 context.Context, arg1 []*model.AnnotateRecord, arg2 ...db.Option) error { +func (m *MockIAnnotateRecordDAO) BatchSave(ctx context.Context, annotateRecord []*model.AnnotateRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, annotateRecord} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchSave", varargs...) @@ -50,32 +55,32 @@ func (m *MockIAnnotateRecordDAO) BatchSave(arg0 context.Context, arg1 []*model.A } // BatchSave indicates an expected call of BatchSave. -func (mr *MockIAnnotateRecordDAOMockRecorder) BatchSave(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIAnnotateRecordDAOMockRecorder) BatchSave(ctx, annotateRecord any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, annotateRecord}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchSave", reflect.TypeOf((*MockIAnnotateRecordDAO)(nil).BatchSave), varargs...) } // MGetByID mocks base method. -func (m *MockIAnnotateRecordDAO) MGetByID(arg0 context.Context, arg1 []int64) ([]*model.AnnotateRecord, error) { +func (m *MockIAnnotateRecordDAO) MGetByID(ctx context.Context, ids []int64) ([]*model.AnnotateRecord, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "MGetByID", arg0, arg1) + ret := m.ctrl.Call(m, "MGetByID", ctx, ids) ret0, _ := ret[0].([]*model.AnnotateRecord) ret1, _ := ret[1].(error) return ret0, ret1 } // MGetByID indicates an expected call of MGetByID. -func (mr *MockIAnnotateRecordDAOMockRecorder) MGetByID(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIAnnotateRecordDAOMockRecorder) MGetByID(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByID", reflect.TypeOf((*MockIAnnotateRecordDAO)(nil).MGetByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByID", reflect.TypeOf((*MockIAnnotateRecordDAO)(nil).MGetByID), ctx, ids) } // Save mocks base method. -func (m *MockIAnnotateRecordDAO) Save(arg0 context.Context, arg1 *model.AnnotateRecord, arg2 ...db.Option) error { +func (m *MockIAnnotateRecordDAO) Save(ctx context.Context, annotateRecord *model.AnnotateRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, annotateRecord} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Save", varargs...) @@ -84,17 +89,17 @@ func (m *MockIAnnotateRecordDAO) Save(arg0 context.Context, arg1 *model.Annotate } // Save indicates an expected call of Save. -func (mr *MockIAnnotateRecordDAOMockRecorder) Save(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIAnnotateRecordDAOMockRecorder) Save(ctx, annotateRecord any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, annotateRecord}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Save", reflect.TypeOf((*MockIAnnotateRecordDAO)(nil).Save), varargs...) } // Update mocks base method. -func (m *MockIAnnotateRecordDAO) Update(arg0 context.Context, arg1 *model.AnnotateRecord, arg2 ...db.Option) error { +func (m *MockIAnnotateRecordDAO) Update(ctx context.Context, annotateRecord *model.AnnotateRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, annotateRecord} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Update", varargs...) @@ -103,8 +108,8 @@ func (m *MockIAnnotateRecordDAO) Update(arg0 context.Context, arg1 *model.Annota } // Update indicates an expected call of Update. -func (mr *MockIAnnotateRecordDAOMockRecorder) Update(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIAnnotateRecordDAOMockRecorder) Update(ctx, annotateRecord any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, annotateRecord}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockIAnnotateRecordDAO)(nil).Update), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt.go index d37347328..a17e1faa4 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt.go -package=mocks . IExptDAO +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockIExptDAO struct { ctrl *gomock.Controller recorder *MockIExptDAOMockRecorder + isgomock struct{} } // MockIExptDAOMockRecorder is the mock recorder for MockIExptDAO. @@ -37,67 +43,67 @@ func (m *MockIExptDAO) EXPECT() *MockIExptDAOMockRecorder { } // Create mocks base method. -func (m *MockIExptDAO) Create(arg0 context.Context, arg1 *model.Experiment) error { +func (m *MockIExptDAO) Create(ctx context.Context, expt *model.Experiment) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Create", arg0, arg1) + ret := m.ctrl.Call(m, "Create", ctx, expt) ret0, _ := ret[0].(error) return ret0 } // Create indicates an expected call of Create. -func (mr *MockIExptDAOMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptDAOMockRecorder) Create(ctx, expt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptDAO)(nil).Create), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptDAO)(nil).Create), ctx, expt) } // Delete mocks base method. -func (m *MockIExptDAO) Delete(arg0 context.Context, arg1 int64) error { +func (m *MockIExptDAO) Delete(ctx context.Context, id int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Delete", arg0, arg1) + ret := m.ctrl.Call(m, "Delete", ctx, id) ret0, _ := ret[0].(error) return ret0 } // Delete indicates an expected call of Delete. -func (mr *MockIExptDAOMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptDAOMockRecorder) Delete(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockIExptDAO)(nil).Delete), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockIExptDAO)(nil).Delete), ctx, id) } // GetByID mocks base method. -func (m *MockIExptDAO) GetByID(arg0 context.Context, arg1 int64) (*model.Experiment, error) { +func (m *MockIExptDAO) GetByID(ctx context.Context, id int64) (*model.Experiment, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetByID", ctx, id) ret0, _ := ret[0].(*model.Experiment) ret1, _ := ret[1].(error) return ret0, ret1 } // GetByID indicates an expected call of GetByID. -func (mr *MockIExptDAOMockRecorder) GetByID(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptDAOMockRecorder) GetByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByID", reflect.TypeOf((*MockIExptDAO)(nil).GetByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByID", reflect.TypeOf((*MockIExptDAO)(nil).GetByID), ctx, id) } // GetByName mocks base method. -func (m *MockIExptDAO) GetByName(arg0 context.Context, arg1 string, arg2 int64) (*model.Experiment, error) { +func (m *MockIExptDAO) GetByName(ctx context.Context, name string, spaceID int64) (*model.Experiment, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetByName", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetByName", ctx, name, spaceID) ret0, _ := ret[0].(*model.Experiment) ret1, _ := ret[1].(error) return ret0, ret1 } // GetByName indicates an expected call of GetByName. -func (mr *MockIExptDAOMockRecorder) GetByName(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIExptDAOMockRecorder) GetByName(ctx, name, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByName", reflect.TypeOf((*MockIExptDAO)(nil).GetByName), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByName", reflect.TypeOf((*MockIExptDAO)(nil).GetByName), ctx, name, spaceID) } // List mocks base method. -func (m *MockIExptDAO) List(arg0 context.Context, arg1, arg2 int32, arg3 *entity.ExptListFilter, arg4 []*entity.OrderBy, arg5 int64) ([]*model.Experiment, int64, error) { +func (m *MockIExptDAO) List(ctx context.Context, page, size int32, filter *entity.ExptListFilter, orders []*entity.OrderBy, spaceID int64) ([]*model.Experiment, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "List", arg0, arg1, arg2, arg3, arg4, arg5) + ret := m.ctrl.Call(m, "List", ctx, page, size, filter, orders, spaceID) ret0, _ := ret[0].([]*model.Experiment) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(error) @@ -105,50 +111,50 @@ func (m *MockIExptDAO) List(arg0 context.Context, arg1, arg2 int32, arg3 *entity } // List indicates an expected call of List. -func (mr *MockIExptDAOMockRecorder) List(arg0, arg1, arg2, arg3, arg4, arg5 interface{}) *gomock.Call { +func (mr *MockIExptDAOMockRecorder) List(ctx, page, size, filter, orders, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockIExptDAO)(nil).List), arg0, arg1, arg2, arg3, arg4, arg5) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockIExptDAO)(nil).List), ctx, page, size, filter, orders, spaceID) } // MDelete mocks base method. -func (m *MockIExptDAO) MDelete(arg0 context.Context, arg1 []int64) error { +func (m *MockIExptDAO) MDelete(ctx context.Context, ids []int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "MDelete", arg0, arg1) + ret := m.ctrl.Call(m, "MDelete", ctx, ids) ret0, _ := ret[0].(error) return ret0 } // MDelete indicates an expected call of MDelete. -func (mr *MockIExptDAOMockRecorder) MDelete(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptDAOMockRecorder) MDelete(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MDelete", reflect.TypeOf((*MockIExptDAO)(nil).MDelete), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MDelete", reflect.TypeOf((*MockIExptDAO)(nil).MDelete), ctx, ids) } // MGetByID mocks base method. -func (m *MockIExptDAO) MGetByID(arg0 context.Context, arg1 []int64) ([]*model.Experiment, error) { +func (m *MockIExptDAO) MGetByID(ctx context.Context, ids []int64) ([]*model.Experiment, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "MGetByID", arg0, arg1) + ret := m.ctrl.Call(m, "MGetByID", ctx, ids) ret0, _ := ret[0].([]*model.Experiment) ret1, _ := ret[1].(error) return ret0, ret1 } // MGetByID indicates an expected call of MGetByID. -func (mr *MockIExptDAOMockRecorder) MGetByID(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptDAOMockRecorder) MGetByID(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByID", reflect.TypeOf((*MockIExptDAO)(nil).MGetByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByID", reflect.TypeOf((*MockIExptDAO)(nil).MGetByID), ctx, ids) } // Update mocks base method. -func (m *MockIExptDAO) Update(arg0 context.Context, arg1 *model.Experiment) error { +func (m *MockIExptDAO) Update(ctx context.Context, expt *model.Experiment) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Update", arg0, arg1) + ret := m.ctrl.Call(m, "Update", ctx, expt) ret0, _ := ret[0].(error) return ret0 } // Update indicates an expected call of Update. -func (mr *MockIExptDAOMockRecorder) Update(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptDAOMockRecorder) Update(ctx, expt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockIExptDAO)(nil).Update), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockIExptDAO)(nil).Update), ctx, expt) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_aggr_result.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_aggr_result.go index bff63e345..e4d87aaca 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_aggr_result.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_aggr_result.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: ExptAggrResultDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_aggr_result.go -package mocks . ExptAggrResultDAO +// // Package mocks is a generated GoMock package. package mocks @@ -8,16 +13,16 @@ import ( context "context" reflect "reflect" - "go.uber.org/mock/gomock" - db "github.com/coze-dev/coze-loop/backend/infra/db" model "github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql/gorm_gen/model" + gomock "go.uber.org/mock/gomock" ) // MockExptAggrResultDAO is a mock of ExptAggrResultDAO interface. type MockExptAggrResultDAO struct { ctrl *gomock.Controller recorder *MockExptAggrResultDAOMockRecorder + isgomock struct{} } // MockExptAggrResultDAOMockRecorder is the mock recorder for MockExptAggrResultDAO. @@ -38,10 +43,10 @@ func (m *MockExptAggrResultDAO) EXPECT() *MockExptAggrResultDAOMockRecorder { } // BatchCreateExptAggrResult mocks base method. -func (m *MockExptAggrResultDAO) BatchCreateExptAggrResult(arg0 context.Context, arg1 []*model.ExptAggrResult, arg2 ...db.Option) error { +func (m *MockExptAggrResultDAO) BatchCreateExptAggrResult(ctx context.Context, exptAggrResults []*model.ExptAggrResult, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, exptAggrResults} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchCreateExptAggrResult", varargs...) @@ -50,17 +55,17 @@ func (m *MockExptAggrResultDAO) BatchCreateExptAggrResult(arg0 context.Context, } // BatchCreateExptAggrResult indicates an expected call of BatchCreateExptAggrResult. -func (mr *MockExptAggrResultDAOMockRecorder) BatchCreateExptAggrResult(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptAggrResultDAOMockRecorder) BatchCreateExptAggrResult(ctx, exptAggrResults any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, exptAggrResults}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateExptAggrResult", reflect.TypeOf((*MockExptAggrResultDAO)(nil).BatchCreateExptAggrResult), varargs...) } // BatchGetExptAggrResultByExperimentIDs mocks base method. -func (m *MockExptAggrResultDAO) BatchGetExptAggrResultByExperimentIDs(arg0 context.Context, arg1 []int64, arg2 ...db.Option) ([]*model.ExptAggrResult, error) { +func (m *MockExptAggrResultDAO) BatchGetExptAggrResultByExperimentIDs(ctx context.Context, experimentIDs []int64, opts ...db.Option) ([]*model.ExptAggrResult, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, experimentIDs} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetExptAggrResultByExperimentIDs", varargs...) @@ -70,17 +75,17 @@ func (m *MockExptAggrResultDAO) BatchGetExptAggrResultByExperimentIDs(arg0 conte } // BatchGetExptAggrResultByExperimentIDs indicates an expected call of BatchGetExptAggrResultByExperimentIDs. -func (mr *MockExptAggrResultDAOMockRecorder) BatchGetExptAggrResultByExperimentIDs(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptAggrResultDAOMockRecorder) BatchGetExptAggrResultByExperimentIDs(ctx, experimentIDs any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, experimentIDs}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetExptAggrResultByExperimentIDs", reflect.TypeOf((*MockExptAggrResultDAO)(nil).BatchGetExptAggrResultByExperimentIDs), varargs...) } // CreateExptAggrResult mocks base method. -func (m *MockExptAggrResultDAO) CreateExptAggrResult(arg0 context.Context, arg1 *model.ExptAggrResult, arg2 ...db.Option) error { +func (m *MockExptAggrResultDAO) CreateExptAggrResult(ctx context.Context, exptAggrResult *model.ExptAggrResult, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, exptAggrResult} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateExptAggrResult", varargs...) @@ -89,17 +94,17 @@ func (m *MockExptAggrResultDAO) CreateExptAggrResult(arg0 context.Context, arg1 } // CreateExptAggrResult indicates an expected call of CreateExptAggrResult. -func (mr *MockExptAggrResultDAOMockRecorder) CreateExptAggrResult(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptAggrResultDAOMockRecorder) CreateExptAggrResult(ctx, exptAggrResult any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, exptAggrResult}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateExptAggrResult", reflect.TypeOf((*MockExptAggrResultDAO)(nil).CreateExptAggrResult), varargs...) } // DeleteExptAggrResult mocks base method. -func (m *MockExptAggrResultDAO) DeleteExptAggrResult(arg0 context.Context, arg1 *model.ExptAggrResult, arg2 ...db.Option) error { +func (m *MockExptAggrResultDAO) DeleteExptAggrResult(ctx context.Context, exptAggrResult *model.ExptAggrResult, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, exptAggrResult} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteExptAggrResult", varargs...) @@ -108,17 +113,17 @@ func (m *MockExptAggrResultDAO) DeleteExptAggrResult(arg0 context.Context, arg1 } // DeleteExptAggrResult indicates an expected call of DeleteExptAggrResult. -func (mr *MockExptAggrResultDAOMockRecorder) DeleteExptAggrResult(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptAggrResultDAOMockRecorder) DeleteExptAggrResult(ctx, exptAggrResult any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, exptAggrResult}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteExptAggrResult", reflect.TypeOf((*MockExptAggrResultDAO)(nil).DeleteExptAggrResult), varargs...) } // GetExptAggrResult mocks base method. -func (m *MockExptAggrResultDAO) GetExptAggrResult(arg0 context.Context, arg1 int64, arg2 int32, arg3 string, arg4 ...db.Option) (*model.ExptAggrResult, error) { +func (m *MockExptAggrResultDAO) GetExptAggrResult(ctx context.Context, experimentID int64, fieldType int32, fieldKey string, opts ...db.Option) (*model.ExptAggrResult, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, experimentID, fieldType, fieldKey} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetExptAggrResult", varargs...) @@ -128,17 +133,17 @@ func (m *MockExptAggrResultDAO) GetExptAggrResult(arg0 context.Context, arg1 int } // GetExptAggrResult indicates an expected call of GetExptAggrResult. -func (mr *MockExptAggrResultDAOMockRecorder) GetExptAggrResult(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockExptAggrResultDAOMockRecorder) GetExptAggrResult(ctx, experimentID, fieldType, fieldKey any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, experimentID, fieldType, fieldKey}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExptAggrResult", reflect.TypeOf((*MockExptAggrResultDAO)(nil).GetExptAggrResult), varargs...) } // GetExptAggrResultByExperimentID mocks base method. -func (m *MockExptAggrResultDAO) GetExptAggrResultByExperimentID(arg0 context.Context, arg1 int64, arg2 ...db.Option) ([]*model.ExptAggrResult, error) { +func (m *MockExptAggrResultDAO) GetExptAggrResultByExperimentID(ctx context.Context, experimentID int64, opts ...db.Option) ([]*model.ExptAggrResult, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, experimentID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetExptAggrResultByExperimentID", varargs...) @@ -148,17 +153,17 @@ func (m *MockExptAggrResultDAO) GetExptAggrResultByExperimentID(arg0 context.Con } // GetExptAggrResultByExperimentID indicates an expected call of GetExptAggrResultByExperimentID. -func (mr *MockExptAggrResultDAOMockRecorder) GetExptAggrResultByExperimentID(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptAggrResultDAOMockRecorder) GetExptAggrResultByExperimentID(ctx, experimentID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, experimentID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExptAggrResultByExperimentID", reflect.TypeOf((*MockExptAggrResultDAO)(nil).GetExptAggrResultByExperimentID), varargs...) } // UpdateAndGetLatestVersion mocks base method. -func (m *MockExptAggrResultDAO) UpdateAndGetLatestVersion(arg0 context.Context, arg1 int64, arg2 int32, arg3 string, arg4 ...db.Option) (int64, error) { +func (m *MockExptAggrResultDAO) UpdateAndGetLatestVersion(ctx context.Context, experimentID int64, fieldType int32, fieldKey string, opts ...db.Option) (int64, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, experimentID, fieldType, fieldKey} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateAndGetLatestVersion", varargs...) @@ -168,17 +173,17 @@ func (m *MockExptAggrResultDAO) UpdateAndGetLatestVersion(arg0 context.Context, } // UpdateAndGetLatestVersion indicates an expected call of UpdateAndGetLatestVersion. -func (mr *MockExptAggrResultDAOMockRecorder) UpdateAndGetLatestVersion(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockExptAggrResultDAOMockRecorder) UpdateAndGetLatestVersion(ctx, experimentID, fieldType, fieldKey any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, experimentID, fieldType, fieldKey}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAndGetLatestVersion", reflect.TypeOf((*MockExptAggrResultDAO)(nil).UpdateAndGetLatestVersion), varargs...) } // UpdateExptAggrResultByVersion mocks base method. -func (m *MockExptAggrResultDAO) UpdateExptAggrResultByVersion(arg0 context.Context, arg1 *model.ExptAggrResult, arg2 int64, arg3 ...db.Option) error { +func (m *MockExptAggrResultDAO) UpdateExptAggrResultByVersion(ctx context.Context, exptAggrResult *model.ExptAggrResult, taskVersion int64, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, exptAggrResult, taskVersion} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateExptAggrResultByVersion", varargs...) @@ -187,8 +192,8 @@ func (m *MockExptAggrResultDAO) UpdateExptAggrResultByVersion(arg0 context.Conte } // UpdateExptAggrResultByVersion indicates an expected call of UpdateExptAggrResultByVersion. -func (mr *MockExptAggrResultDAOMockRecorder) UpdateExptAggrResultByVersion(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockExptAggrResultDAOMockRecorder) UpdateExptAggrResultByVersion(ctx, exptAggrResult, taskVersion any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, exptAggrResult, taskVersion}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateExptAggrResultByVersion", reflect.TypeOf((*MockExptAggrResultDAO)(nil).UpdateExptAggrResultByVersion), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_evaluator_ref.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_evaluator_ref.go index a7a8b3e4d..ca7ae9367 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_evaluator_ref.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_evaluator_ref.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptEvaluatorRefDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_evaluator_ref.go -package mocks . IExptEvaluatorRefDAO +// // Package mocks is a generated GoMock package. package mocks @@ -16,6 +21,7 @@ import ( type MockIExptEvaluatorRefDAO struct { ctrl *gomock.Controller recorder *MockIExptEvaluatorRefDAOMockRecorder + isgomock struct{} } // MockIExptEvaluatorRefDAOMockRecorder is the mock recorder for MockIExptEvaluatorRefDAO. @@ -36,30 +42,30 @@ func (m *MockIExptEvaluatorRefDAO) EXPECT() *MockIExptEvaluatorRefDAOMockRecorde } // Create mocks base method. -func (m *MockIExptEvaluatorRefDAO) Create(arg0 context.Context, arg1 []*model.ExptEvaluatorRef) error { +func (m *MockIExptEvaluatorRefDAO) Create(ctx context.Context, exptEvaluatorRef []*model.ExptEvaluatorRef) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Create", arg0, arg1) + ret := m.ctrl.Call(m, "Create", ctx, exptEvaluatorRef) ret0, _ := ret[0].(error) return ret0 } // Create indicates an expected call of Create. -func (mr *MockIExptEvaluatorRefDAOMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptEvaluatorRefDAOMockRecorder) Create(ctx, exptEvaluatorRef any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptEvaluatorRefDAO)(nil).Create), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptEvaluatorRefDAO)(nil).Create), ctx, exptEvaluatorRef) } // MGetByExptID mocks base method. -func (m *MockIExptEvaluatorRefDAO) MGetByExptID(arg0 context.Context, arg1 []int64, arg2 int64) ([]*model.ExptEvaluatorRef, error) { +func (m *MockIExptEvaluatorRefDAO) MGetByExptID(ctx context.Context, exptIDs []int64, spaceID int64) ([]*model.ExptEvaluatorRef, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "MGetByExptID", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "MGetByExptID", ctx, exptIDs, spaceID) ret0, _ := ret[0].([]*model.ExptEvaluatorRef) ret1, _ := ret[1].(error) return ret0, ret1 } // MGetByExptID indicates an expected call of MGetByExptID. -func (mr *MockIExptEvaluatorRefDAOMockRecorder) MGetByExptID(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIExptEvaluatorRefDAOMockRecorder) MGetByExptID(ctx, exptIDs, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByExptID", reflect.TypeOf((*MockIExptEvaluatorRefDAO)(nil).MGetByExptID), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByExptID", reflect.TypeOf((*MockIExptEvaluatorRefDAO)(nil).MGetByExptID), ctx, exptIDs, spaceID) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_feedback_comment.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_feedback_comment.go index faee0a332..691ec4388 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_feedback_comment.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_feedback_comment.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptInsightAnalysisFeedbackCommentDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_insight_analysis_feedback_comment.go -package mocks . IExptInsightAnalysisFeedbackCommentDAO +// // Package mocks is a generated GoMock package. package mocks @@ -18,6 +23,7 @@ import ( type MockIExptInsightAnalysisFeedbackCommentDAO struct { ctrl *gomock.Controller recorder *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder + isgomock struct{} } // MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder is the mock recorder for MockIExptInsightAnalysisFeedbackCommentDAO. @@ -38,10 +44,10 @@ func (m *MockIExptInsightAnalysisFeedbackCommentDAO) EXPECT() *MockIExptInsightA } // Create mocks base method. -func (m *MockIExptInsightAnalysisFeedbackCommentDAO) Create(arg0 context.Context, arg1 *model.ExptInsightAnalysisFeedbackComment, arg2 ...db.Option) error { +func (m *MockIExptInsightAnalysisFeedbackCommentDAO) Create(ctx context.Context, feedbackComment *model.ExptInsightAnalysisFeedbackComment, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, feedbackComment} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Create", varargs...) @@ -50,31 +56,31 @@ func (m *MockIExptInsightAnalysisFeedbackCommentDAO) Create(arg0 context.Context } // Create indicates an expected call of Create. -func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) Create(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) Create(ctx, feedbackComment any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, feedbackComment}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackCommentDAO)(nil).Create), varargs...) } // Delete mocks base method. -func (m *MockIExptInsightAnalysisFeedbackCommentDAO) Delete(arg0 context.Context, arg1, arg2, arg3 int64) error { +func (m *MockIExptInsightAnalysisFeedbackCommentDAO) Delete(ctx context.Context, spaceID, exptID, commentID int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Delete", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "Delete", ctx, spaceID, exptID, commentID) ret0, _ := ret[0].(error) return ret0 } // Delete indicates an expected call of Delete. -func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) Delete(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) Delete(ctx, spaceID, exptID, commentID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackCommentDAO)(nil).Delete), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackCommentDAO)(nil).Delete), ctx, spaceID, exptID, commentID) } // GetByRecordID mocks base method. -func (m *MockIExptInsightAnalysisFeedbackCommentDAO) GetByRecordID(arg0 context.Context, arg1, arg2, arg3 int64, arg4 ...db.Option) (*model.ExptInsightAnalysisFeedbackComment, error) { +func (m *MockIExptInsightAnalysisFeedbackCommentDAO) GetByRecordID(ctx context.Context, spaceID, exptID, recordID int64, opts ...db.Option) (*model.ExptInsightAnalysisFeedbackComment, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, exptID, recordID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetByRecordID", varargs...) @@ -84,16 +90,16 @@ func (m *MockIExptInsightAnalysisFeedbackCommentDAO) GetByRecordID(arg0 context. } // GetByRecordID indicates an expected call of GetByRecordID. -func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) GetByRecordID(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) GetByRecordID(ctx, spaceID, exptID, recordID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, exptID, recordID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByRecordID", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackCommentDAO)(nil).GetByRecordID), varargs...) } // List mocks base method. -func (m *MockIExptInsightAnalysisFeedbackCommentDAO) List(arg0 context.Context, arg1, arg2, arg3 int64, arg4 entity.Page) ([]*model.ExptInsightAnalysisFeedbackComment, int64, error) { +func (m *MockIExptInsightAnalysisFeedbackCommentDAO) List(ctx context.Context, spaceID, exptID, recordID int64, page entity.Page) ([]*model.ExptInsightAnalysisFeedbackComment, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "List", arg0, arg1, arg2, arg3, arg4) + ret := m.ctrl.Call(m, "List", ctx, spaceID, exptID, recordID, page) ret0, _ := ret[0].([]*model.ExptInsightAnalysisFeedbackComment) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(error) @@ -101,16 +107,16 @@ func (m *MockIExptInsightAnalysisFeedbackCommentDAO) List(arg0 context.Context, } // List indicates an expected call of List. -func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) List(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) List(ctx, spaceID, exptID, recordID, page any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackCommentDAO)(nil).List), arg0, arg1, arg2, arg3, arg4) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackCommentDAO)(nil).List), ctx, spaceID, exptID, recordID, page) } // Update mocks base method. -func (m *MockIExptInsightAnalysisFeedbackCommentDAO) Update(arg0 context.Context, arg1 *model.ExptInsightAnalysisFeedbackComment, arg2 ...db.Option) error { +func (m *MockIExptInsightAnalysisFeedbackCommentDAO) Update(ctx context.Context, feedbackComment *model.ExptInsightAnalysisFeedbackComment, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, feedbackComment} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Update", varargs...) @@ -119,8 +125,8 @@ func (m *MockIExptInsightAnalysisFeedbackCommentDAO) Update(arg0 context.Context } // Update indicates an expected call of Update. -func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) Update(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackCommentDAOMockRecorder) Update(ctx, feedbackComment any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, feedbackComment}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackCommentDAO)(nil).Update), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_feedback_vote.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_feedback_vote.go index addd54589..13964dd1b 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_feedback_vote.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_feedback_vote.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptInsightAnalysisFeedbackVoteDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_insight_analysis_feedback_vote.go -package mocks . IExptInsightAnalysisFeedbackVoteDAO +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockIExptInsightAnalysisFeedbackVoteDAO struct { ctrl *gomock.Controller recorder *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder + isgomock struct{} } // MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder is the mock recorder for MockIExptInsightAnalysisFeedbackVoteDAO. @@ -37,9 +43,9 @@ func (m *MockIExptInsightAnalysisFeedbackVoteDAO) EXPECT() *MockIExptInsightAnal } // Count mocks base method. -func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Count(arg0 context.Context, arg1, arg2, arg3 int64) (int64, int64, error) { +func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Count(ctx context.Context, spaceID, exptID, recordID int64) (int64, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Count", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "Count", ctx, spaceID, exptID, recordID) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(error) @@ -47,16 +53,16 @@ func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Count(arg0 context.Context, ar } // Count indicates an expected call of Count. -func (mr *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder) Count(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder) Count(ctx, spaceID, exptID, recordID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Count", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackVoteDAO)(nil).Count), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Count", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackVoteDAO)(nil).Count), ctx, spaceID, exptID, recordID) } // Create mocks base method. -func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Create(arg0 context.Context, arg1 *model.ExptInsightAnalysisFeedbackVote, arg2 ...db.Option) error { +func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Create(ctx context.Context, feedbackVote *model.ExptInsightAnalysisFeedbackVote, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, feedbackVote} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Create", varargs...) @@ -65,17 +71,17 @@ func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Create(arg0 context.Context, a } // Create indicates an expected call of Create. -func (mr *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder) Create(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder) Create(ctx, feedbackVote any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, feedbackVote}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackVoteDAO)(nil).Create), varargs...) } // GetByUser mocks base method. -func (m *MockIExptInsightAnalysisFeedbackVoteDAO) GetByUser(arg0 context.Context, arg1, arg2, arg3 int64, arg4 string, arg5 ...db.Option) (*model.ExptInsightAnalysisFeedbackVote, error) { +func (m *MockIExptInsightAnalysisFeedbackVoteDAO) GetByUser(ctx context.Context, spaceID, exptID, recordID int64, userID string, opts ...db.Option) (*model.ExptInsightAnalysisFeedbackVote, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, spaceID, exptID, recordID, userID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetByUser", varargs...) @@ -85,17 +91,17 @@ func (m *MockIExptInsightAnalysisFeedbackVoteDAO) GetByUser(arg0 context.Context } // GetByUser indicates an expected call of GetByUser. -func (mr *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder) GetByUser(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder) GetByUser(ctx, spaceID, exptID, recordID, userID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, spaceID, exptID, recordID, userID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByUser", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackVoteDAO)(nil).GetByUser), varargs...) } // Update mocks base method. -func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Update(arg0 context.Context, arg1 *model.ExptInsightAnalysisFeedbackVote, arg2 ...db.Option) error { +func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Update(ctx context.Context, feedbackVote *model.ExptInsightAnalysisFeedbackVote, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, feedbackVote} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Update", varargs...) @@ -104,8 +110,8 @@ func (m *MockIExptInsightAnalysisFeedbackVoteDAO) Update(arg0 context.Context, a } // Update indicates an expected call of Update. -func (mr *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder) Update(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisFeedbackVoteDAOMockRecorder) Update(ctx, feedbackVote any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, feedbackVote}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockIExptInsightAnalysisFeedbackVoteDAO)(nil).Update), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_record.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_record.go index ccffbe7fc..63def3504 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_record.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_insight_analysis_record.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptInsightAnalysisRecordDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_insight_analysis_record.go -package mocks . IExptInsightAnalysisRecordDAO +// // Package mocks is a generated GoMock package. package mocks @@ -18,6 +23,7 @@ import ( type MockIExptInsightAnalysisRecordDAO struct { ctrl *gomock.Controller recorder *MockIExptInsightAnalysisRecordDAOMockRecorder + isgomock struct{} } // MockIExptInsightAnalysisRecordDAOMockRecorder is the mock recorder for MockIExptInsightAnalysisRecordDAO. @@ -38,10 +44,10 @@ func (m *MockIExptInsightAnalysisRecordDAO) EXPECT() *MockIExptInsightAnalysisRe } // Create mocks base method. -func (m *MockIExptInsightAnalysisRecordDAO) Create(arg0 context.Context, arg1 *model.ExptInsightAnalysisRecord, arg2 ...db.Option) error { +func (m *MockIExptInsightAnalysisRecordDAO) Create(ctx context.Context, record *model.ExptInsightAnalysisRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, record} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Create", varargs...) @@ -50,31 +56,31 @@ func (m *MockIExptInsightAnalysisRecordDAO) Create(arg0 context.Context, arg1 *m } // Create indicates an expected call of Create. -func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) Create(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) Create(ctx, record any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, record}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptInsightAnalysisRecordDAO)(nil).Create), varargs...) } // Delete mocks base method. -func (m *MockIExptInsightAnalysisRecordDAO) Delete(arg0 context.Context, arg1, arg2, arg3 int64) error { +func (m *MockIExptInsightAnalysisRecordDAO) Delete(ctx context.Context, spaceID, exptID, recordID int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Delete", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "Delete", ctx, spaceID, exptID, recordID) ret0, _ := ret[0].(error) return ret0 } // Delete indicates an expected call of Delete. -func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) Delete(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) Delete(ctx, spaceID, exptID, recordID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockIExptInsightAnalysisRecordDAO)(nil).Delete), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockIExptInsightAnalysisRecordDAO)(nil).Delete), ctx, spaceID, exptID, recordID) } // GetByID mocks base method. -func (m *MockIExptInsightAnalysisRecordDAO) GetByID(arg0 context.Context, arg1, arg2, arg3 int64, arg4 ...db.Option) (*model.ExptInsightAnalysisRecord, error) { +func (m *MockIExptInsightAnalysisRecordDAO) GetByID(ctx context.Context, spaceID, exptID, recordID int64, opts ...db.Option) (*model.ExptInsightAnalysisRecord, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, exptID, recordID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetByID", varargs...) @@ -84,16 +90,16 @@ func (m *MockIExptInsightAnalysisRecordDAO) GetByID(arg0 context.Context, arg1, } // GetByID indicates an expected call of GetByID. -func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) GetByID(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) GetByID(ctx, spaceID, exptID, recordID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, exptID, recordID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByID", reflect.TypeOf((*MockIExptInsightAnalysisRecordDAO)(nil).GetByID), varargs...) } // List mocks base method. -func (m *MockIExptInsightAnalysisRecordDAO) List(arg0 context.Context, arg1, arg2 int64, arg3 entity.Page) ([]*model.ExptInsightAnalysisRecord, int64, error) { +func (m *MockIExptInsightAnalysisRecordDAO) List(ctx context.Context, spaceID, exptID int64, page entity.Page) ([]*model.ExptInsightAnalysisRecord, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "List", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "List", ctx, spaceID, exptID, page) ret0, _ := ret[0].([]*model.ExptInsightAnalysisRecord) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(error) @@ -101,16 +107,16 @@ func (m *MockIExptInsightAnalysisRecordDAO) List(arg0 context.Context, arg1, arg } // List indicates an expected call of List. -func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) List(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) List(ctx, spaceID, exptID, page any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockIExptInsightAnalysisRecordDAO)(nil).List), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockIExptInsightAnalysisRecordDAO)(nil).List), ctx, spaceID, exptID, page) } // Update mocks base method. -func (m *MockIExptInsightAnalysisRecordDAO) Update(arg0 context.Context, arg1 *model.ExptInsightAnalysisRecord, arg2 ...db.Option) error { +func (m *MockIExptInsightAnalysisRecordDAO) Update(ctx context.Context, record *model.ExptInsightAnalysisRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, record} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Update", varargs...) @@ -119,8 +125,8 @@ func (m *MockIExptInsightAnalysisRecordDAO) Update(arg0 context.Context, arg1 *m } // Update indicates an expected call of Update. -func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) Update(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptInsightAnalysisRecordDAOMockRecorder) Update(ctx, record any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, record}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockIExptInsightAnalysisRecordDAO)(nil).Update), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_item_result.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_item_result.go index 03d486f73..70685b05c 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_item_result.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_item_result.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptItemResultDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_item_result.go -package mocks . IExptItemResultDAO +// // Package mocks is a generated GoMock package. package mocks @@ -18,6 +23,7 @@ import ( type MockIExptItemResultDAO struct { ctrl *gomock.Controller recorder *MockIExptItemResultDAOMockRecorder + isgomock struct{} } // MockIExptItemResultDAOMockRecorder is the mock recorder for MockIExptItemResultDAO. @@ -38,10 +44,10 @@ func (m *MockIExptItemResultDAO) EXPECT() *MockIExptItemResultDAOMockRecorder { } // BatchCreateNX mocks base method. -func (m *MockIExptItemResultDAO) BatchCreateNX(arg0 context.Context, arg1 []*model.ExptItemResult, arg2 ...db.Option) error { +func (m *MockIExptItemResultDAO) BatchCreateNX(ctx context.Context, itemResults []*model.ExptItemResult, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, itemResults} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchCreateNX", varargs...) @@ -50,17 +56,17 @@ func (m *MockIExptItemResultDAO) BatchCreateNX(arg0 context.Context, arg1 []*mod } // BatchCreateNX indicates an expected call of BatchCreateNX. -func (mr *MockIExptItemResultDAOMockRecorder) BatchCreateNX(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) BatchCreateNX(ctx, itemResults any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, itemResults}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateNX", reflect.TypeOf((*MockIExptItemResultDAO)(nil).BatchCreateNX), varargs...) } // BatchCreateNXRunLogs mocks base method. -func (m *MockIExptItemResultDAO) BatchCreateNXRunLogs(arg0 context.Context, arg1 []*model.ExptItemResultRunLog, arg2 ...db.Option) error { +func (m *MockIExptItemResultDAO) BatchCreateNXRunLogs(ctx context.Context, itemRunLogs []*model.ExptItemResultRunLog, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, itemRunLogs} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchCreateNXRunLogs", varargs...) @@ -69,17 +75,17 @@ func (m *MockIExptItemResultDAO) BatchCreateNXRunLogs(arg0 context.Context, arg1 } // BatchCreateNXRunLogs indicates an expected call of BatchCreateNXRunLogs. -func (mr *MockIExptItemResultDAOMockRecorder) BatchCreateNXRunLogs(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) BatchCreateNXRunLogs(ctx, itemRunLogs any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, itemRunLogs}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateNXRunLogs", reflect.TypeOf((*MockIExptItemResultDAO)(nil).BatchCreateNXRunLogs), varargs...) } // BatchGet mocks base method. -func (m *MockIExptItemResultDAO) BatchGet(arg0 context.Context, arg1, arg2 int64, arg3 []int64, arg4 ...db.Option) ([]*model.ExptItemResult, error) { +func (m *MockIExptItemResultDAO) BatchGet(ctx context.Context, spaceID, exptID int64, itemIDs []int64, opts ...db.Option) ([]*model.ExptItemResult, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, exptID, itemIDs} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGet", varargs...) @@ -89,32 +95,32 @@ func (m *MockIExptItemResultDAO) BatchGet(arg0 context.Context, arg1, arg2 int64 } // BatchGet indicates an expected call of BatchGet. -func (mr *MockIExptItemResultDAOMockRecorder) BatchGet(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) BatchGet(ctx, spaceID, exptID, itemIDs any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, exptID, itemIDs}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGet", reflect.TypeOf((*MockIExptItemResultDAO)(nil).BatchGet), varargs...) } // GetItemIDListByExptID mocks base method. -func (m *MockIExptItemResultDAO) GetItemIDListByExptID(arg0 context.Context, arg1, arg2 int64) ([]int64, error) { +func (m *MockIExptItemResultDAO) GetItemIDListByExptID(ctx context.Context, exptID, spaceID int64) ([]int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetItemIDListByExptID", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetItemIDListByExptID", ctx, exptID, spaceID) ret0, _ := ret[0].([]int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetItemIDListByExptID indicates an expected call of GetItemIDListByExptID. -func (mr *MockIExptItemResultDAOMockRecorder) GetItemIDListByExptID(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) GetItemIDListByExptID(ctx, exptID, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetItemIDListByExptID", reflect.TypeOf((*MockIExptItemResultDAO)(nil).GetItemIDListByExptID), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetItemIDListByExptID", reflect.TypeOf((*MockIExptItemResultDAO)(nil).GetItemIDListByExptID), ctx, exptID, spaceID) } // GetItemRunLog mocks base method. -func (m *MockIExptItemResultDAO) GetItemRunLog(arg0 context.Context, arg1, arg2, arg3, arg4 int64, arg5 ...db.Option) (*model.ExptItemResultRunLog, error) { +func (m *MockIExptItemResultDAO) GetItemRunLog(ctx context.Context, exptID, exptRunID, itemID, spaceID int64, opts ...db.Option) (*model.ExptItemResultRunLog, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, exptID, exptRunID, itemID, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetItemRunLog", varargs...) @@ -124,17 +130,17 @@ func (m *MockIExptItemResultDAO) GetItemRunLog(arg0 context.Context, arg1, arg2, } // GetItemRunLog indicates an expected call of GetItemRunLog. -func (mr *MockIExptItemResultDAOMockRecorder) GetItemRunLog(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) GetItemRunLog(ctx, exptID, exptRunID, itemID, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, exptID, exptRunID, itemID, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetItemRunLog", reflect.TypeOf((*MockIExptItemResultDAO)(nil).GetItemRunLog), varargs...) } // GetItemTurnResults mocks base method. -func (m *MockIExptItemResultDAO) GetItemTurnResults(arg0 context.Context, arg1, arg2, arg3 int64, arg4 ...db.Option) ([]*model.ExptTurnResult, error) { +func (m *MockIExptItemResultDAO) GetItemTurnResults(ctx context.Context, spaceID, exptID, itemID int64, opts ...db.Option) ([]*model.ExptTurnResult, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, exptID, itemID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetItemTurnResults", varargs...) @@ -144,17 +150,17 @@ func (m *MockIExptItemResultDAO) GetItemTurnResults(arg0 context.Context, arg1, } // GetItemTurnResults indicates an expected call of GetItemTurnResults. -func (mr *MockIExptItemResultDAOMockRecorder) GetItemTurnResults(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) GetItemTurnResults(ctx, spaceID, exptID, itemID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, exptID, itemID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetItemTurnResults", reflect.TypeOf((*MockIExptItemResultDAO)(nil).GetItemTurnResults), varargs...) } // GetMaxItemIdxByExptID mocks base method. -func (m *MockIExptItemResultDAO) GetMaxItemIdxByExptID(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) (int32, error) { +func (m *MockIExptItemResultDAO) GetMaxItemIdxByExptID(ctx context.Context, exptID, spaceID int64, opts ...db.Option) (int32, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, exptID, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetMaxItemIdxByExptID", varargs...) @@ -164,16 +170,16 @@ func (m *MockIExptItemResultDAO) GetMaxItemIdxByExptID(arg0 context.Context, arg } // GetMaxItemIdxByExptID indicates an expected call of GetMaxItemIdxByExptID. -func (mr *MockIExptItemResultDAOMockRecorder) GetMaxItemIdxByExptID(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) GetMaxItemIdxByExptID(ctx, exptID, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, exptID, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetMaxItemIdxByExptID", reflect.TypeOf((*MockIExptItemResultDAO)(nil).GetMaxItemIdxByExptID), varargs...) } // ListItemResultsByExptID mocks base method. -func (m *MockIExptItemResultDAO) ListItemResultsByExptID(arg0 context.Context, arg1, arg2 int64, arg3 entity.Page, arg4 bool) ([]*model.ExptItemResult, int64, error) { +func (m *MockIExptItemResultDAO) ListItemResultsByExptID(ctx context.Context, exptID, spaceID int64, page entity.Page, desc bool) ([]*model.ExptItemResult, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListItemResultsByExptID", arg0, arg1, arg2, arg3, arg4) + ret := m.ctrl.Call(m, "ListItemResultsByExptID", ctx, exptID, spaceID, page, desc) ret0, _ := ret[0].([]*model.ExptItemResult) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(error) @@ -181,16 +187,16 @@ func (m *MockIExptItemResultDAO) ListItemResultsByExptID(arg0 context.Context, a } // ListItemResultsByExptID indicates an expected call of ListItemResultsByExptID. -func (mr *MockIExptItemResultDAOMockRecorder) ListItemResultsByExptID(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) ListItemResultsByExptID(ctx, exptID, spaceID, page, desc any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListItemResultsByExptID", reflect.TypeOf((*MockIExptItemResultDAO)(nil).ListItemResultsByExptID), arg0, arg1, arg2, arg3, arg4) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListItemResultsByExptID", reflect.TypeOf((*MockIExptItemResultDAO)(nil).ListItemResultsByExptID), ctx, exptID, spaceID, page, desc) } // MGetItemRunLog mocks base method. -func (m *MockIExptItemResultDAO) MGetItemRunLog(arg0 context.Context, arg1, arg2 int64, arg3 []int64, arg4 int64, arg5 ...db.Option) ([]*model.ExptItemResultRunLog, error) { +func (m *MockIExptItemResultDAO) MGetItemRunLog(ctx context.Context, exptID, exptRunID int64, itemIDs []int64, spaceID int64, opts ...db.Option) ([]*model.ExptItemResultRunLog, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, exptID, exptRunID, itemIDs, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "MGetItemRunLog", varargs...) @@ -200,17 +206,17 @@ func (m *MockIExptItemResultDAO) MGetItemRunLog(arg0 context.Context, arg1, arg2 } // MGetItemRunLog indicates an expected call of MGetItemRunLog. -func (mr *MockIExptItemResultDAOMockRecorder) MGetItemRunLog(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) MGetItemRunLog(ctx, exptID, exptRunID, itemIDs, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, exptID, exptRunID, itemIDs, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetItemRunLog", reflect.TypeOf((*MockIExptItemResultDAO)(nil).MGetItemRunLog), varargs...) } // SaveItemResults mocks base method. -func (m *MockIExptItemResultDAO) SaveItemResults(arg0 context.Context, arg1 []*model.ExptItemResult, arg2 ...db.Option) error { +func (m *MockIExptItemResultDAO) SaveItemResults(ctx context.Context, itemResults []*model.ExptItemResult, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, itemResults} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "SaveItemResults", varargs...) @@ -219,17 +225,17 @@ func (m *MockIExptItemResultDAO) SaveItemResults(arg0 context.Context, arg1 []*m } // SaveItemResults indicates an expected call of SaveItemResults. -func (mr *MockIExptItemResultDAOMockRecorder) SaveItemResults(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) SaveItemResults(ctx, itemResults any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, itemResults}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveItemResults", reflect.TypeOf((*MockIExptItemResultDAO)(nil).SaveItemResults), varargs...) } // ScanItemResults mocks base method. -func (m *MockIExptItemResultDAO) ScanItemResults(arg0 context.Context, arg1, arg2, arg3 int64, arg4 []int32, arg5 int64, arg6 ...db.Option) ([]*model.ExptItemResult, int64, error) { +func (m *MockIExptItemResultDAO) ScanItemResults(ctx context.Context, exptID, cursor, limit int64, status []int32, spaceID int64, opts ...db.Option) ([]*model.ExptItemResult, int64, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4, arg5} - for _, a := range arg6 { + varargs := []any{ctx, exptID, cursor, limit, status, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ScanItemResults", varargs...) @@ -240,17 +246,17 @@ func (m *MockIExptItemResultDAO) ScanItemResults(arg0 context.Context, arg1, arg } // ScanItemResults indicates an expected call of ScanItemResults. -func (mr *MockIExptItemResultDAOMockRecorder) ScanItemResults(arg0, arg1, arg2, arg3, arg4, arg5 interface{}, arg6 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) ScanItemResults(ctx, exptID, cursor, limit, status, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) + varargs := append([]any{ctx, exptID, cursor, limit, status, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScanItemResults", reflect.TypeOf((*MockIExptItemResultDAO)(nil).ScanItemResults), varargs...) } // ScanItemRunLogs mocks base method. -func (m *MockIExptItemResultDAO) ScanItemRunLogs(arg0 context.Context, arg1, arg2 int64, arg3 *entity.ExptItemRunLogFilter, arg4, arg5, arg6 int64, arg7 ...db.Option) ([]*model.ExptItemResultRunLog, int64, error) { +func (m *MockIExptItemResultDAO) ScanItemRunLogs(ctx context.Context, exptID, exptRunID int64, filter *entity.ExptItemRunLogFilter, cursor, limit, spaceID int64, opts ...db.Option) ([]*model.ExptItemResultRunLog, int64, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4, arg5, arg6} - for _, a := range arg7 { + varargs := []any{ctx, exptID, exptRunID, filter, cursor, limit, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ScanItemRunLogs", varargs...) @@ -261,17 +267,17 @@ func (m *MockIExptItemResultDAO) ScanItemRunLogs(arg0 context.Context, arg1, arg } // ScanItemRunLogs indicates an expected call of ScanItemRunLogs. -func (mr *MockIExptItemResultDAOMockRecorder) ScanItemRunLogs(arg0, arg1, arg2, arg3, arg4, arg5, arg6 interface{}, arg7 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) ScanItemRunLogs(ctx, exptID, exptRunID, filter, cursor, limit, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4, arg5, arg6}, arg7...) + varargs := append([]any{ctx, exptID, exptRunID, filter, cursor, limit, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScanItemRunLogs", reflect.TypeOf((*MockIExptItemResultDAO)(nil).ScanItemRunLogs), varargs...) } // UpdateItemRunLog mocks base method. -func (m *MockIExptItemResultDAO) UpdateItemRunLog(arg0 context.Context, arg1, arg2 int64, arg3 []int64, arg4 map[string]interface{}, arg5 int64, arg6 ...db.Option) error { +func (m *MockIExptItemResultDAO) UpdateItemRunLog(ctx context.Context, exptID, exptRunID int64, itemID []int64, ufields map[string]any, spaceID int64, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4, arg5} - for _, a := range arg6 { + varargs := []any{ctx, exptID, exptRunID, itemID, ufields, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateItemRunLog", varargs...) @@ -280,17 +286,17 @@ func (m *MockIExptItemResultDAO) UpdateItemRunLog(arg0 context.Context, arg1, ar } // UpdateItemRunLog indicates an expected call of UpdateItemRunLog. -func (mr *MockIExptItemResultDAOMockRecorder) UpdateItemRunLog(arg0, arg1, arg2, arg3, arg4, arg5 interface{}, arg6 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) UpdateItemRunLog(ctx, exptID, exptRunID, itemID, ufields, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) + varargs := append([]any{ctx, exptID, exptRunID, itemID, ufields, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateItemRunLog", reflect.TypeOf((*MockIExptItemResultDAO)(nil).UpdateItemRunLog), varargs...) } // UpdateItemsResult mocks base method. -func (m *MockIExptItemResultDAO) UpdateItemsResult(arg0 context.Context, arg1, arg2 int64, arg3 []int64, arg4 map[string]interface{}, arg5 ...db.Option) error { +func (m *MockIExptItemResultDAO) UpdateItemsResult(ctx context.Context, spaceID, exptID int64, itemID []int64, ufields map[string]any, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, spaceID, exptID, itemID, ufields} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateItemsResult", varargs...) @@ -299,8 +305,8 @@ func (m *MockIExptItemResultDAO) UpdateItemsResult(arg0 context.Context, arg1, a } // UpdateItemsResult indicates an expected call of UpdateItemsResult. -func (mr *MockIExptItemResultDAOMockRecorder) UpdateItemsResult(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockIExptItemResultDAOMockRecorder) UpdateItemsResult(ctx, spaceID, exptID, itemID, ufields any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, spaceID, exptID, itemID, ufields}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateItemsResult", reflect.TypeOf((*MockIExptItemResultDAO)(nil).UpdateItemsResult), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_result_export_record.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_result_export_record.go index 6385624a6..663371ba0 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_result_export_record.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_result_export_record.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: ExptResultExportRecordDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_result_export_record.go -package=mocks . ExptResultExportRecordDAO +// // Package mocks is a generated GoMock package. package mocks @@ -18,6 +23,7 @@ import ( type MockExptResultExportRecordDAO struct { ctrl *gomock.Controller recorder *MockExptResultExportRecordDAOMockRecorder + isgomock struct{} } // MockExptResultExportRecordDAOMockRecorder is the mock recorder for MockExptResultExportRecordDAO. @@ -38,10 +44,10 @@ func (m *MockExptResultExportRecordDAO) EXPECT() *MockExptResultExportRecordDAOM } // Create mocks base method. -func (m *MockExptResultExportRecordDAO) Create(arg0 context.Context, arg1 *model.ExptResultExportRecord, arg2 ...db.Option) error { +func (m *MockExptResultExportRecordDAO) Create(ctx context.Context, exptResultExportRecord *model.ExptResultExportRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, exptResultExportRecord} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Create", varargs...) @@ -50,17 +56,17 @@ func (m *MockExptResultExportRecordDAO) Create(arg0 context.Context, arg1 *model } // Create indicates an expected call of Create. -func (mr *MockExptResultExportRecordDAOMockRecorder) Create(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptResultExportRecordDAOMockRecorder) Create(ctx, exptResultExportRecord any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, exptResultExportRecord}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockExptResultExportRecordDAO)(nil).Create), varargs...) } // Get mocks base method. -func (m *MockExptResultExportRecordDAO) Get(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) (*model.ExptResultExportRecord, error) { +func (m *MockExptResultExportRecordDAO) Get(ctx context.Context, spaceID, exportID int64, opts ...db.Option) (*model.ExptResultExportRecord, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, exportID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Get", varargs...) @@ -70,16 +76,16 @@ func (m *MockExptResultExportRecordDAO) Get(arg0 context.Context, arg1, arg2 int } // Get indicates an expected call of Get. -func (mr *MockExptResultExportRecordDAOMockRecorder) Get(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockExptResultExportRecordDAOMockRecorder) Get(ctx, spaceID, exportID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, exportID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockExptResultExportRecordDAO)(nil).Get), varargs...) } // List mocks base method. -func (m *MockExptResultExportRecordDAO) List(arg0 context.Context, arg1, arg2 int64, arg3 entity.Page, arg4 *int32) ([]*model.ExptResultExportRecord, int64, error) { +func (m *MockExptResultExportRecordDAO) List(ctx context.Context, spaceID, exptID int64, page entity.Page, csvExportStatus *int32) ([]*model.ExptResultExportRecord, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "List", arg0, arg1, arg2, arg3, arg4) + ret := m.ctrl.Call(m, "List", ctx, spaceID, exptID, page, csvExportStatus) ret0, _ := ret[0].([]*model.ExptResultExportRecord) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(error) @@ -87,16 +93,16 @@ func (m *MockExptResultExportRecordDAO) List(arg0 context.Context, arg1, arg2 in } // List indicates an expected call of List. -func (mr *MockExptResultExportRecordDAOMockRecorder) List(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { +func (mr *MockExptResultExportRecordDAOMockRecorder) List(ctx, spaceID, exptID, page, csvExportStatus any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockExptResultExportRecordDAO)(nil).List), arg0, arg1, arg2, arg3, arg4) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockExptResultExportRecordDAO)(nil).List), ctx, spaceID, exptID, page, csvExportStatus) } // Update mocks base method. -func (m *MockExptResultExportRecordDAO) Update(arg0 context.Context, arg1 *model.ExptResultExportRecord, arg2 ...db.Option) error { +func (m *MockExptResultExportRecordDAO) Update(ctx context.Context, exptResultExportRecord *model.ExptResultExportRecord, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, exptResultExportRecord} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Update", varargs...) @@ -105,8 +111,8 @@ func (m *MockExptResultExportRecordDAO) Update(arg0 context.Context, arg1 *model } // Update indicates an expected call of Update. -func (mr *MockExptResultExportRecordDAOMockRecorder) Update(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptResultExportRecordDAOMockRecorder) Update(ctx, exptResultExportRecord any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, exptResultExportRecord}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockExptResultExportRecordDAO)(nil).Update), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_run_log.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_run_log.go index 156e67cc6..ef1c1d21b 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_run_log.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_run_log.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptRunLogDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_run_log.go -package mocks . IExptRunLogDAO +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockIExptRunLogDAO struct { ctrl *gomock.Controller recorder *MockIExptRunLogDAOMockRecorder + isgomock struct{} } // MockIExptRunLogDAOMockRecorder is the mock recorder for MockIExptRunLogDAO. @@ -37,10 +43,10 @@ func (m *MockIExptRunLogDAO) EXPECT() *MockIExptRunLogDAOMockRecorder { } // Create mocks base method. -func (m *MockIExptRunLogDAO) Create(arg0 context.Context, arg1 *model.ExptRunLog, arg2 ...db.Option) error { +func (m *MockIExptRunLogDAO) Create(ctx context.Context, exptRunLog *model.ExptRunLog, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, exptRunLog} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Create", varargs...) @@ -49,17 +55,17 @@ func (m *MockIExptRunLogDAO) Create(arg0 context.Context, arg1 *model.ExptRunLog } // Create indicates an expected call of Create. -func (mr *MockIExptRunLogDAOMockRecorder) Create(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptRunLogDAOMockRecorder) Create(ctx, exptRunLog any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, exptRunLog}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptRunLogDAO)(nil).Create), varargs...) } // Get mocks base method. -func (m *MockIExptRunLogDAO) Get(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) (*model.ExptRunLog, error) { +func (m *MockIExptRunLogDAO) Get(ctx context.Context, exptID, exptRunID int64, opts ...db.Option) (*model.ExptRunLog, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, exptID, exptRunID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Get", varargs...) @@ -69,17 +75,17 @@ func (m *MockIExptRunLogDAO) Get(arg0 context.Context, arg1, arg2 int64, arg3 .. } // Get indicates an expected call of Get. -func (mr *MockIExptRunLogDAOMockRecorder) Get(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockIExptRunLogDAOMockRecorder) Get(ctx, exptID, exptRunID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, exptID, exptRunID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockIExptRunLogDAO)(nil).Get), varargs...) } // Save mocks base method. -func (m *MockIExptRunLogDAO) Save(arg0 context.Context, arg1 *model.ExptRunLog, arg2 ...db.Option) error { +func (m *MockIExptRunLogDAO) Save(ctx context.Context, exptRunLog *model.ExptRunLog, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, exptRunLog} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Save", varargs...) @@ -88,17 +94,17 @@ func (m *MockIExptRunLogDAO) Save(arg0 context.Context, arg1 *model.ExptRunLog, } // Save indicates an expected call of Save. -func (mr *MockIExptRunLogDAOMockRecorder) Save(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptRunLogDAOMockRecorder) Save(ctx, exptRunLog any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, exptRunLog}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Save", reflect.TypeOf((*MockIExptRunLogDAO)(nil).Save), varargs...) } // Update mocks base method. -func (m *MockIExptRunLogDAO) Update(arg0 context.Context, arg1, arg2 int64, arg3 map[string]interface{}, arg4 ...db.Option) error { +func (m *MockIExptRunLogDAO) Update(ctx context.Context, exptID, exptRunID int64, ufields map[string]any, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, exptID, exptRunID, ufields} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Update", varargs...) @@ -107,8 +113,8 @@ func (m *MockIExptRunLogDAO) Update(arg0 context.Context, arg1, arg2 int64, arg3 } // Update indicates an expected call of Update. -func (mr *MockIExptRunLogDAOMockRecorder) Update(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockIExptRunLogDAOMockRecorder) Update(ctx, exptID, exptRunID, ufields any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, exptID, exptRunID, ufields}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockIExptRunLogDAO)(nil).Update), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_stats.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_stats.go index cb02bfedb..b2c61129c 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_stats.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_stats.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptStatsDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_stats.go -package mocks . IExptStatsDAO +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockIExptStatsDAO struct { ctrl *gomock.Controller recorder *MockIExptStatsDAOMockRecorder + isgomock struct{} } // MockIExptStatsDAOMockRecorder is the mock recorder for MockIExptStatsDAO. @@ -37,87 +43,87 @@ func (m *MockIExptStatsDAO) EXPECT() *MockIExptStatsDAOMockRecorder { } // ArithOperateCount mocks base method. -func (m *MockIExptStatsDAO) ArithOperateCount(arg0 context.Context, arg1, arg2 int64, arg3 *entity.StatsCntArithOp) error { +func (m *MockIExptStatsDAO) ArithOperateCount(ctx context.Context, exptID, spaceID int64, cntArithOp *entity.StatsCntArithOp) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ArithOperateCount", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "ArithOperateCount", ctx, exptID, spaceID, cntArithOp) ret0, _ := ret[0].(error) return ret0 } // ArithOperateCount indicates an expected call of ArithOperateCount. -func (mr *MockIExptStatsDAOMockRecorder) ArithOperateCount(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIExptStatsDAOMockRecorder) ArithOperateCount(ctx, exptID, spaceID, cntArithOp any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ArithOperateCount", reflect.TypeOf((*MockIExptStatsDAO)(nil).ArithOperateCount), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ArithOperateCount", reflect.TypeOf((*MockIExptStatsDAO)(nil).ArithOperateCount), ctx, exptID, spaceID, cntArithOp) } // Create mocks base method. -func (m *MockIExptStatsDAO) Create(arg0 context.Context, arg1 *model.ExptStats) error { +func (m *MockIExptStatsDAO) Create(ctx context.Context, stats *model.ExptStats) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Create", arg0, arg1) + ret := m.ctrl.Call(m, "Create", ctx, stats) ret0, _ := ret[0].(error) return ret0 } // Create indicates an expected call of Create. -func (mr *MockIExptStatsDAOMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptStatsDAOMockRecorder) Create(ctx, stats any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptStatsDAO)(nil).Create), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIExptStatsDAO)(nil).Create), ctx, stats) } // Get mocks base method. -func (m *MockIExptStatsDAO) Get(arg0 context.Context, arg1, arg2 int64) (*model.ExptStats, error) { +func (m *MockIExptStatsDAO) Get(ctx context.Context, exptID, spaceID int64) (*model.ExptStats, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Get", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "Get", ctx, exptID, spaceID) ret0, _ := ret[0].(*model.ExptStats) ret1, _ := ret[1].(error) return ret0, ret1 } // Get indicates an expected call of Get. -func (mr *MockIExptStatsDAOMockRecorder) Get(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIExptStatsDAOMockRecorder) Get(ctx, exptID, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockIExptStatsDAO)(nil).Get), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockIExptStatsDAO)(nil).Get), ctx, exptID, spaceID) } // MGet mocks base method. -func (m *MockIExptStatsDAO) MGet(arg0 context.Context, arg1 []int64, arg2 int64) ([]*model.ExptStats, error) { +func (m *MockIExptStatsDAO) MGet(ctx context.Context, exptIDs []int64, spaceID int64) ([]*model.ExptStats, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "MGet", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "MGet", ctx, exptIDs, spaceID) ret0, _ := ret[0].([]*model.ExptStats) ret1, _ := ret[1].(error) return ret0, ret1 } // MGet indicates an expected call of MGet. -func (mr *MockIExptStatsDAOMockRecorder) MGet(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIExptStatsDAOMockRecorder) MGet(ctx, exptIDs, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGet", reflect.TypeOf((*MockIExptStatsDAO)(nil).MGet), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGet", reflect.TypeOf((*MockIExptStatsDAO)(nil).MGet), ctx, exptIDs, spaceID) } // Save mocks base method. -func (m *MockIExptStatsDAO) Save(arg0 context.Context, arg1 *model.ExptStats) error { +func (m *MockIExptStatsDAO) Save(ctx context.Context, stats *model.ExptStats) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Save", arg0, arg1) + ret := m.ctrl.Call(m, "Save", ctx, stats) ret0, _ := ret[0].(error) return ret0 } // Save indicates an expected call of Save. -func (mr *MockIExptStatsDAOMockRecorder) Save(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptStatsDAOMockRecorder) Save(ctx, stats any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Save", reflect.TypeOf((*MockIExptStatsDAO)(nil).Save), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Save", reflect.TypeOf((*MockIExptStatsDAO)(nil).Save), ctx, stats) } // UpdateByExptID mocks base method. -func (m *MockIExptStatsDAO) UpdateByExptID(arg0 context.Context, arg1, arg2 int64, arg3 *model.ExptStats) error { +func (m *MockIExptStatsDAO) UpdateByExptID(ctx context.Context, exptID, spaceID int64, stats *model.ExptStats) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateByExptID", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "UpdateByExptID", ctx, exptID, spaceID, stats) ret0, _ := ret[0].(error) return ret0 } // UpdateByExptID indicates an expected call of UpdateByExptID. -func (mr *MockIExptStatsDAOMockRecorder) UpdateByExptID(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIExptStatsDAOMockRecorder) UpdateByExptID(ctx, exptID, spaceID, stats any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateByExptID", reflect.TypeOf((*MockIExptStatsDAO)(nil).UpdateByExptID), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateByExptID", reflect.TypeOf((*MockIExptStatsDAO)(nil).UpdateByExptID), ctx, exptID, spaceID, stats) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_annotate_record_ref.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_annotate_record_ref.go index 178987666..f291d781e 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_annotate_record_ref.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_annotate_record_ref.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptTurnAnnotateRecordRefDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_turn_annotate_record_ref.go -package=mocks . IExptTurnAnnotateRecordRefDAO +// // Package mocks is a generated GoMock package. package mocks @@ -10,13 +15,14 @@ import ( db "github.com/coze-dev/coze-loop/backend/infra/db" model "github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql/gorm_gen/model" - "go.uber.org/mock/gomock" + gomock "go.uber.org/mock/gomock" ) // MockIExptTurnAnnotateRecordRefDAO is a mock of IExptTurnAnnotateRecordRefDAO interface. type MockIExptTurnAnnotateRecordRefDAO struct { ctrl *gomock.Controller recorder *MockIExptTurnAnnotateRecordRefDAOMockRecorder + isgomock struct{} } // MockIExptTurnAnnotateRecordRefDAOMockRecorder is the mock recorder for MockIExptTurnAnnotateRecordRefDAO. @@ -37,10 +43,10 @@ func (m *MockIExptTurnAnnotateRecordRefDAO) EXPECT() *MockIExptTurnAnnotateRecor } // BatchGet mocks base method. -func (m *MockIExptTurnAnnotateRecordRefDAO) BatchGet(arg0 context.Context, arg1 int64, arg2 []int64, arg3 ...db.Option) ([]*model.ExptTurnAnnotateRecordRef, error) { +func (m *MockIExptTurnAnnotateRecordRefDAO) BatchGet(ctx context.Context, spaceID int64, exptTurnResultIDs []int64, opts ...db.Option) ([]*model.ExptTurnAnnotateRecordRef, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, exptTurnResultIDs} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGet", varargs...) @@ -50,17 +56,17 @@ func (m *MockIExptTurnAnnotateRecordRefDAO) BatchGet(arg0 context.Context, arg1 } // BatchGet indicates an expected call of BatchGet. -func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) BatchGet(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) BatchGet(ctx, spaceID, exptTurnResultIDs any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, exptTurnResultIDs}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGet", reflect.TypeOf((*MockIExptTurnAnnotateRecordRefDAO)(nil).BatchGet), varargs...) } // BatchGetByExptIDs mocks base method. -func (m *MockIExptTurnAnnotateRecordRefDAO) BatchGetByExptIDs(arg0 context.Context, arg1 int64, arg2 []int64, arg3 ...db.Option) ([]*model.ExptTurnAnnotateRecordRef, error) { +func (m *MockIExptTurnAnnotateRecordRefDAO) BatchGetByExptIDs(ctx context.Context, spaceID int64, exptIDs []int64, opts ...db.Option) ([]*model.ExptTurnAnnotateRecordRef, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, exptIDs} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetByExptIDs", varargs...) @@ -70,17 +76,17 @@ func (m *MockIExptTurnAnnotateRecordRefDAO) BatchGetByExptIDs(arg0 context.Conte } // BatchGetByExptIDs indicates an expected call of BatchGetByExptIDs. -func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) BatchGetByExptIDs(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) BatchGetByExptIDs(ctx, spaceID, exptIDs any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, exptIDs}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetByExptIDs", reflect.TypeOf((*MockIExptTurnAnnotateRecordRefDAO)(nil).BatchGetByExptIDs), varargs...) } // DeleteByTagKeyID mocks base method. -func (m *MockIExptTurnAnnotateRecordRefDAO) DeleteByTagKeyID(arg0 context.Context, arg1, arg2, arg3 int64, arg4 ...db.Option) error { +func (m *MockIExptTurnAnnotateRecordRefDAO) DeleteByTagKeyID(ctx context.Context, spaceID, exptID, tagKeyID int64, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, exptID, tagKeyID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteByTagKeyID", varargs...) @@ -89,17 +95,17 @@ func (m *MockIExptTurnAnnotateRecordRefDAO) DeleteByTagKeyID(arg0 context.Contex } // DeleteByTagKeyID indicates an expected call of DeleteByTagKeyID. -func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) DeleteByTagKeyID(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) DeleteByTagKeyID(ctx, spaceID, exptID, tagKeyID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, exptID, tagKeyID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteByTagKeyID", reflect.TypeOf((*MockIExptTurnAnnotateRecordRefDAO)(nil).DeleteByTagKeyID), varargs...) } // GetByExptID mocks base method. -func (m *MockIExptTurnAnnotateRecordRefDAO) GetByExptID(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) ([]*model.ExptTurnAnnotateRecordRef, error) { +func (m *MockIExptTurnAnnotateRecordRefDAO) GetByExptID(ctx context.Context, spaceID, exptID int64, opts ...db.Option) ([]*model.ExptTurnAnnotateRecordRef, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, exptID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetByExptID", varargs...) @@ -109,17 +115,17 @@ func (m *MockIExptTurnAnnotateRecordRefDAO) GetByExptID(arg0 context.Context, ar } // GetByExptID indicates an expected call of GetByExptID. -func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) GetByExptID(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) GetByExptID(ctx, spaceID, exptID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, exptID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByExptID", reflect.TypeOf((*MockIExptTurnAnnotateRecordRefDAO)(nil).GetByExptID), varargs...) } // GetByTagKeyID mocks base method. -func (m *MockIExptTurnAnnotateRecordRefDAO) GetByTagKeyID(arg0 context.Context, arg1, arg2, arg3 int64, arg4 ...db.Option) ([]*model.ExptTurnAnnotateRecordRef, error) { +func (m *MockIExptTurnAnnotateRecordRefDAO) GetByTagKeyID(ctx context.Context, spaceID, exptID, tagKeyID int64, opts ...db.Option) ([]*model.ExptTurnAnnotateRecordRef, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, exptID, tagKeyID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetByTagKeyID", varargs...) @@ -129,17 +135,17 @@ func (m *MockIExptTurnAnnotateRecordRefDAO) GetByTagKeyID(arg0 context.Context, } // GetByTagKeyID indicates an expected call of GetByTagKeyID. -func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) GetByTagKeyID(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) GetByTagKeyID(ctx, spaceID, exptID, tagKeyID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, exptID, tagKeyID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByTagKeyID", reflect.TypeOf((*MockIExptTurnAnnotateRecordRefDAO)(nil).GetByTagKeyID), varargs...) } // Save mocks base method. -func (m *MockIExptTurnAnnotateRecordRefDAO) Save(arg0 context.Context, arg1 *model.ExptTurnAnnotateRecordRef, arg2 ...db.Option) error { +func (m *MockIExptTurnAnnotateRecordRefDAO) Save(ctx context.Context, refs *model.ExptTurnAnnotateRecordRef, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, refs} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Save", varargs...) @@ -148,8 +154,8 @@ func (m *MockIExptTurnAnnotateRecordRefDAO) Save(arg0 context.Context, arg1 *mod } // Save indicates an expected call of Save. -func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) Save(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnAnnotateRecordRefDAOMockRecorder) Save(ctx, refs any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, refs}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Save", reflect.TypeOf((*MockIExptTurnAnnotateRecordRefDAO)(nil).Save), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_evaluator_result_ref.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_evaluator_result_ref.go index 4341e56db..ff61bfd9f 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_evaluator_result_ref.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_evaluator_result_ref.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptTurnEvaluatorResultRefDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_turn_evaluator_result_ref.go -package mocks . IExptTurnEvaluatorResultRefDAO +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockIExptTurnEvaluatorResultRefDAO struct { ctrl *gomock.Controller recorder *MockIExptTurnEvaluatorResultRefDAOMockRecorder + isgomock struct{} } // MockIExptTurnEvaluatorResultRefDAOMockRecorder is the mock recorder for MockIExptTurnEvaluatorResultRefDAO. @@ -37,10 +43,10 @@ func (m *MockIExptTurnEvaluatorResultRefDAO) EXPECT() *MockIExptTurnEvaluatorRes } // BatchGet mocks base method. -func (m *MockIExptTurnEvaluatorResultRefDAO) BatchGet(arg0 context.Context, arg1 int64, arg2 []int64, arg3 ...db.Option) ([]*model.ExptTurnEvaluatorResultRef, error) { +func (m *MockIExptTurnEvaluatorResultRefDAO) BatchGet(ctx context.Context, spaceID int64, exptTurnResultIDs []int64, opts ...db.Option) ([]*model.ExptTurnEvaluatorResultRef, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, exptTurnResultIDs} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGet", varargs...) @@ -50,17 +56,17 @@ func (m *MockIExptTurnEvaluatorResultRefDAO) BatchGet(arg0 context.Context, arg1 } // BatchGet indicates an expected call of BatchGet. -func (mr *MockIExptTurnEvaluatorResultRefDAOMockRecorder) BatchGet(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnEvaluatorResultRefDAOMockRecorder) BatchGet(ctx, spaceID, exptTurnResultIDs any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, exptTurnResultIDs}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGet", reflect.TypeOf((*MockIExptTurnEvaluatorResultRefDAO)(nil).BatchGet), varargs...) } // GetByExptEvaluatorVersionID mocks base method. -func (m *MockIExptTurnEvaluatorResultRefDAO) GetByExptEvaluatorVersionID(arg0 context.Context, arg1, arg2, arg3 int64, arg4 ...db.Option) ([]*model.ExptTurnEvaluatorResultRef, error) { +func (m *MockIExptTurnEvaluatorResultRefDAO) GetByExptEvaluatorVersionID(ctx context.Context, spaceID, exptID, evaluatorVersionID int64, opts ...db.Option) ([]*model.ExptTurnEvaluatorResultRef, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, exptID, evaluatorVersionID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetByExptEvaluatorVersionID", varargs...) @@ -70,17 +76,17 @@ func (m *MockIExptTurnEvaluatorResultRefDAO) GetByExptEvaluatorVersionID(arg0 co } // GetByExptEvaluatorVersionID indicates an expected call of GetByExptEvaluatorVersionID. -func (mr *MockIExptTurnEvaluatorResultRefDAOMockRecorder) GetByExptEvaluatorVersionID(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnEvaluatorResultRefDAOMockRecorder) GetByExptEvaluatorVersionID(ctx, spaceID, exptID, evaluatorVersionID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, exptID, evaluatorVersionID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByExptEvaluatorVersionID", reflect.TypeOf((*MockIExptTurnEvaluatorResultRefDAO)(nil).GetByExptEvaluatorVersionID), varargs...) } // GetByExptID mocks base method. -func (m *MockIExptTurnEvaluatorResultRefDAO) GetByExptID(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) ([]*model.ExptTurnEvaluatorResultRef, error) { +func (m *MockIExptTurnEvaluatorResultRefDAO) GetByExptID(ctx context.Context, spaceID, exptID int64, opts ...db.Option) ([]*model.ExptTurnEvaluatorResultRef, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, exptID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetByExptID", varargs...) @@ -90,8 +96,8 @@ func (m *MockIExptTurnEvaluatorResultRefDAO) GetByExptID(arg0 context.Context, a } // GetByExptID indicates an expected call of GetByExptID. -func (mr *MockIExptTurnEvaluatorResultRefDAOMockRecorder) GetByExptID(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnEvaluatorResultRefDAOMockRecorder) GetByExptID(ctx, spaceID, exptID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, exptID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByExptID", reflect.TypeOf((*MockIExptTurnEvaluatorResultRefDAO)(nil).GetByExptID), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_result.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_result.go index f3d2e1e6b..ef34df51e 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_result.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_result.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: ExptTurnResultDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_turn_result.go -package mocks . ExptTurnResultDAO +// // Package mocks is a generated GoMock package. package mocks @@ -8,17 +13,17 @@ import ( context "context" reflect "reflect" - "go.uber.org/mock/gomock" - db "github.com/coze-dev/coze-loop/backend/infra/db" entity "github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/entity" model "github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql/gorm_gen/model" + gomock "go.uber.org/mock/gomock" ) // MockExptTurnResultDAO is a mock of ExptTurnResultDAO interface. type MockExptTurnResultDAO struct { ctrl *gomock.Controller recorder *MockExptTurnResultDAOMockRecorder + isgomock struct{} } // MockExptTurnResultDAOMockRecorder is the mock recorder for MockExptTurnResultDAO. @@ -39,10 +44,10 @@ func (m *MockExptTurnResultDAO) EXPECT() *MockExptTurnResultDAOMockRecorder { } // BatchCreateNX mocks base method. -func (m *MockExptTurnResultDAO) BatchCreateNX(arg0 context.Context, arg1 []*model.ExptTurnResult, arg2 ...db.Option) error { +func (m *MockExptTurnResultDAO) BatchCreateNX(ctx context.Context, turnResults []*model.ExptTurnResult, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, turnResults} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchCreateNX", varargs...) @@ -51,17 +56,17 @@ func (m *MockExptTurnResultDAO) BatchCreateNX(arg0 context.Context, arg1 []*mode } // BatchCreateNX indicates an expected call of BatchCreateNX. -func (mr *MockExptTurnResultDAOMockRecorder) BatchCreateNX(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) BatchCreateNX(ctx, turnResults any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, turnResults}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateNX", reflect.TypeOf((*MockExptTurnResultDAO)(nil).BatchCreateNX), varargs...) } // BatchCreateNXRunLog mocks base method. -func (m *MockExptTurnResultDAO) BatchCreateNXRunLog(arg0 context.Context, arg1 []*model.ExptTurnResultRunLog, arg2 ...db.Option) error { +func (m *MockExptTurnResultDAO) BatchCreateNXRunLog(ctx context.Context, turnResults []*model.ExptTurnResultRunLog, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, turnResults} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchCreateNXRunLog", varargs...) @@ -70,17 +75,17 @@ func (m *MockExptTurnResultDAO) BatchCreateNXRunLog(arg0 context.Context, arg1 [ } // BatchCreateNXRunLog indicates an expected call of BatchCreateNXRunLog. -func (mr *MockExptTurnResultDAOMockRecorder) BatchCreateNXRunLog(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) BatchCreateNXRunLog(ctx, turnResults any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, turnResults}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateNXRunLog", reflect.TypeOf((*MockExptTurnResultDAO)(nil).BatchCreateNXRunLog), varargs...) } // BatchGet mocks base method. -func (m *MockExptTurnResultDAO) BatchGet(arg0 context.Context, arg1, arg2 int64, arg3 []int64, arg4 ...db.Option) ([]*model.ExptTurnResult, error) { +func (m *MockExptTurnResultDAO) BatchGet(ctx context.Context, spaceID, exptID int64, itemIDs []int64, opts ...db.Option) ([]*model.ExptTurnResult, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, spaceID, exptID, itemIDs} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGet", varargs...) @@ -90,17 +95,17 @@ func (m *MockExptTurnResultDAO) BatchGet(arg0 context.Context, arg1, arg2 int64, } // BatchGet indicates an expected call of BatchGet. -func (mr *MockExptTurnResultDAOMockRecorder) BatchGet(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) BatchGet(ctx, spaceID, exptID, itemIDs any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, spaceID, exptID, itemIDs}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGet", reflect.TypeOf((*MockExptTurnResultDAO)(nil).BatchGet), varargs...) } // CreateTurnEvaluatorRefs mocks base method. -func (m *MockExptTurnResultDAO) CreateTurnEvaluatorRefs(arg0 context.Context, arg1 []*model.ExptTurnEvaluatorResultRef, arg2 ...db.Option) error { +func (m *MockExptTurnResultDAO) CreateTurnEvaluatorRefs(ctx context.Context, turnResults []*model.ExptTurnEvaluatorResultRef, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, turnResults} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateTurnEvaluatorRefs", varargs...) @@ -109,17 +114,17 @@ func (m *MockExptTurnResultDAO) CreateTurnEvaluatorRefs(arg0 context.Context, ar } // CreateTurnEvaluatorRefs indicates an expected call of CreateTurnEvaluatorRefs. -func (mr *MockExptTurnResultDAOMockRecorder) CreateTurnEvaluatorRefs(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) CreateTurnEvaluatorRefs(ctx, turnResults any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, turnResults}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTurnEvaluatorRefs", reflect.TypeOf((*MockExptTurnResultDAO)(nil).CreateTurnEvaluatorRefs), varargs...) } // Get mocks base method. -func (m *MockExptTurnResultDAO) Get(arg0 context.Context, arg1, arg2, arg3, arg4 int64, arg5 ...db.Option) (*model.ExptTurnResult, error) { +func (m *MockExptTurnResultDAO) Get(ctx context.Context, spaceID, exptID, itemID, turnID int64, opts ...db.Option) (*model.ExptTurnResult, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, spaceID, exptID, itemID, turnID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Get", varargs...) @@ -129,17 +134,17 @@ func (m *MockExptTurnResultDAO) Get(arg0 context.Context, arg1, arg2, arg3, arg4 } // Get indicates an expected call of Get. -func (mr *MockExptTurnResultDAOMockRecorder) Get(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) Get(ctx, spaceID, exptID, itemID, turnID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, spaceID, exptID, itemID, turnID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockExptTurnResultDAO)(nil).Get), varargs...) } // GetItemTurnResults mocks base method. -func (m *MockExptTurnResultDAO) GetItemTurnResults(arg0 context.Context, arg1, arg2, arg3 int64, arg4 ...db.Option) ([]*model.ExptTurnResult, error) { +func (m *MockExptTurnResultDAO) GetItemTurnResults(ctx context.Context, exptID, itemID, spaceID int64, opts ...db.Option) ([]*model.ExptTurnResult, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3} - for _, a := range arg4 { + varargs := []any{ctx, exptID, itemID, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetItemTurnResults", varargs...) @@ -149,17 +154,17 @@ func (m *MockExptTurnResultDAO) GetItemTurnResults(arg0 context.Context, arg1, a } // GetItemTurnResults indicates an expected call of GetItemTurnResults. -func (mr *MockExptTurnResultDAOMockRecorder) GetItemTurnResults(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) GetItemTurnResults(ctx, exptID, itemID, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) + varargs := append([]any{ctx, exptID, itemID, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetItemTurnResults", reflect.TypeOf((*MockExptTurnResultDAO)(nil).GetItemTurnResults), varargs...) } // GetItemTurnRunLogs mocks base method. -func (m *MockExptTurnResultDAO) GetItemTurnRunLogs(arg0 context.Context, arg1, arg2, arg3, arg4 int64, arg5 ...db.Option) ([]*model.ExptTurnResultRunLog, error) { +func (m *MockExptTurnResultDAO) GetItemTurnRunLogs(ctx context.Context, exptID, exptRunID, itemID, spaceID int64, opts ...db.Option) ([]*model.ExptTurnResultRunLog, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, exptID, exptRunID, itemID, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetItemTurnRunLogs", varargs...) @@ -169,17 +174,17 @@ func (m *MockExptTurnResultDAO) GetItemTurnRunLogs(arg0 context.Context, arg1, a } // GetItemTurnRunLogs indicates an expected call of GetItemTurnRunLogs. -func (mr *MockExptTurnResultDAOMockRecorder) GetItemTurnRunLogs(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) GetItemTurnRunLogs(ctx, exptID, exptRunID, itemID, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, exptID, exptRunID, itemID, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetItemTurnRunLogs", reflect.TypeOf((*MockExptTurnResultDAO)(nil).GetItemTurnRunLogs), varargs...) } // ListTurnResult mocks base method. -func (m *MockExptTurnResultDAO) ListTurnResult(arg0 context.Context, arg1, arg2 int64, arg3 *entity.ExptTurnResultFilter, arg4 entity.Page, arg5 bool, arg6 ...db.Option) ([]*model.ExptTurnResult, int64, error) { +func (m *MockExptTurnResultDAO) ListTurnResult(ctx context.Context, spaceID, exptID int64, filter *entity.ExptTurnResultFilter, page entity.Page, desc bool, opts ...db.Option) ([]*model.ExptTurnResult, int64, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4, arg5} - for _, a := range arg6 { + varargs := []any{ctx, spaceID, exptID, filter, page, desc} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListTurnResult", varargs...) @@ -190,17 +195,17 @@ func (m *MockExptTurnResultDAO) ListTurnResult(arg0 context.Context, arg1, arg2 } // ListTurnResult indicates an expected call of ListTurnResult. -func (mr *MockExptTurnResultDAOMockRecorder) ListTurnResult(arg0, arg1, arg2, arg3, arg4, arg5 interface{}, arg6 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) ListTurnResult(ctx, spaceID, exptID, filter, page, desc any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) + varargs := append([]any{ctx, spaceID, exptID, filter, page, desc}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTurnResult", reflect.TypeOf((*MockExptTurnResultDAO)(nil).ListTurnResult), varargs...) } // ListTurnResultByItemIDs mocks base method. -func (m *MockExptTurnResultDAO) ListTurnResultByItemIDs(arg0 context.Context, arg1, arg2 int64, arg3 []int64, arg4 entity.Page, arg5 bool, arg6 ...db.Option) ([]*model.ExptTurnResult, int64, error) { +func (m *MockExptTurnResultDAO) ListTurnResultByItemIDs(ctx context.Context, spaceID, exptID int64, itemIDs []int64, page entity.Page, desc bool, opts ...db.Option) ([]*model.ExptTurnResult, int64, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4, arg5} - for _, a := range arg6 { + varargs := []any{ctx, spaceID, exptID, itemIDs, page, desc} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListTurnResultByItemIDs", varargs...) @@ -211,17 +216,17 @@ func (m *MockExptTurnResultDAO) ListTurnResultByItemIDs(arg0 context.Context, ar } // ListTurnResultByItemIDs indicates an expected call of ListTurnResultByItemIDs. -func (mr *MockExptTurnResultDAOMockRecorder) ListTurnResultByItemIDs(arg0, arg1, arg2, arg3, arg4, arg5 interface{}, arg6 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) ListTurnResultByItemIDs(ctx, spaceID, exptID, itemIDs, page, desc any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) + varargs := append([]any{ctx, spaceID, exptID, itemIDs, page, desc}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTurnResultByItemIDs", reflect.TypeOf((*MockExptTurnResultDAO)(nil).ListTurnResultByItemIDs), varargs...) } // MGetItemTurnRunLogs mocks base method. -func (m *MockExptTurnResultDAO) MGetItemTurnRunLogs(arg0 context.Context, arg1, arg2 int64, arg3 []int64, arg4 int64, arg5 ...db.Option) ([]*model.ExptTurnResultRunLog, error) { +func (m *MockExptTurnResultDAO) MGetItemTurnRunLogs(ctx context.Context, exptID, exptRunID int64, itemIDs []int64, spaceID int64, opts ...db.Option) ([]*model.ExptTurnResultRunLog, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, exptID, exptRunID, itemIDs, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "MGetItemTurnRunLogs", varargs...) @@ -231,17 +236,17 @@ func (m *MockExptTurnResultDAO) MGetItemTurnRunLogs(arg0 context.Context, arg1, } // MGetItemTurnRunLogs indicates an expected call of MGetItemTurnRunLogs. -func (mr *MockExptTurnResultDAOMockRecorder) MGetItemTurnRunLogs(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) MGetItemTurnRunLogs(ctx, exptID, exptRunID, itemIDs, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, exptID, exptRunID, itemIDs, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetItemTurnRunLogs", reflect.TypeOf((*MockExptTurnResultDAO)(nil).MGetItemTurnRunLogs), varargs...) } // SaveTurnResults mocks base method. -func (m *MockExptTurnResultDAO) SaveTurnResults(arg0 context.Context, arg1 []*model.ExptTurnResult, arg2 ...db.Option) error { +func (m *MockExptTurnResultDAO) SaveTurnResults(ctx context.Context, turnResults []*model.ExptTurnResult, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, turnResults} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "SaveTurnResults", varargs...) @@ -250,17 +255,17 @@ func (m *MockExptTurnResultDAO) SaveTurnResults(arg0 context.Context, arg1 []*mo } // SaveTurnResults indicates an expected call of SaveTurnResults. -func (mr *MockExptTurnResultDAOMockRecorder) SaveTurnResults(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) SaveTurnResults(ctx, turnResults any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, turnResults}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveTurnResults", reflect.TypeOf((*MockExptTurnResultDAO)(nil).SaveTurnResults), varargs...) } // SaveTurnRunLogs mocks base method. -func (m *MockExptTurnResultDAO) SaveTurnRunLogs(arg0 context.Context, arg1 []*model.ExptTurnResultRunLog, arg2 ...db.Option) error { +func (m *MockExptTurnResultDAO) SaveTurnRunLogs(ctx context.Context, turnResults []*model.ExptTurnResultRunLog, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, turnResults} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "SaveTurnRunLogs", varargs...) @@ -269,17 +274,17 @@ func (m *MockExptTurnResultDAO) SaveTurnRunLogs(arg0 context.Context, arg1 []*mo } // SaveTurnRunLogs indicates an expected call of SaveTurnRunLogs. -func (mr *MockExptTurnResultDAOMockRecorder) SaveTurnRunLogs(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) SaveTurnRunLogs(ctx, turnResults any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, turnResults}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveTurnRunLogs", reflect.TypeOf((*MockExptTurnResultDAO)(nil).SaveTurnRunLogs), varargs...) } // ScanTurnResults mocks base method. -func (m *MockExptTurnResultDAO) ScanTurnResults(arg0 context.Context, arg1 int64, arg2 []int32, arg3, arg4, arg5 int64, arg6 ...db.Option) ([]*model.ExptTurnResult, int64, error) { +func (m *MockExptTurnResultDAO) ScanTurnResults(ctx context.Context, exptID int64, status []int32, cursor, limit, spaceID int64, opts ...db.Option) ([]*model.ExptTurnResult, int64, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4, arg5} - for _, a := range arg6 { + varargs := []any{ctx, exptID, status, cursor, limit, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ScanTurnResults", varargs...) @@ -290,17 +295,17 @@ func (m *MockExptTurnResultDAO) ScanTurnResults(arg0 context.Context, arg1 int64 } // ScanTurnResults indicates an expected call of ScanTurnResults. -func (mr *MockExptTurnResultDAOMockRecorder) ScanTurnResults(arg0, arg1, arg2, arg3, arg4, arg5 interface{}, arg6 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) ScanTurnResults(ctx, exptID, status, cursor, limit, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) + varargs := append([]any{ctx, exptID, status, cursor, limit, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScanTurnResults", reflect.TypeOf((*MockExptTurnResultDAO)(nil).ScanTurnResults), varargs...) } // ScanTurnRunLogs mocks base method. -func (m *MockExptTurnResultDAO) ScanTurnRunLogs(arg0 context.Context, arg1, arg2, arg3, arg4 int64, arg5 ...db.Option) ([]*model.ExptTurnResultRunLog, int64, error) { +func (m *MockExptTurnResultDAO) ScanTurnRunLogs(ctx context.Context, exptID, cursor, limit, spaceID int64, opts ...db.Option) ([]*model.ExptTurnResultRunLog, int64, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, exptID, cursor, limit, spaceID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ScanTurnRunLogs", varargs...) @@ -311,17 +316,17 @@ func (m *MockExptTurnResultDAO) ScanTurnRunLogs(arg0 context.Context, arg1, arg2 } // ScanTurnRunLogs indicates an expected call of ScanTurnRunLogs. -func (mr *MockExptTurnResultDAOMockRecorder) ScanTurnRunLogs(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) ScanTurnRunLogs(ctx, exptID, cursor, limit, spaceID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, exptID, cursor, limit, spaceID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScanTurnRunLogs", reflect.TypeOf((*MockExptTurnResultDAO)(nil).ScanTurnRunLogs), varargs...) } // UpdateTurnResults mocks base method. -func (m *MockExptTurnResultDAO) UpdateTurnResults(arg0 context.Context, arg1 int64, arg2 []*entity.ItemTurnID, arg3 int64, arg4 map[string]interface{}, arg5 ...db.Option) error { +func (m *MockExptTurnResultDAO) UpdateTurnResults(ctx context.Context, exptID int64, itemTurnIDs []*entity.ItemTurnID, spaceID int64, ufields map[string]any, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, exptID, itemTurnIDs, spaceID, ufields} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateTurnResults", varargs...) @@ -330,17 +335,17 @@ func (m *MockExptTurnResultDAO) UpdateTurnResults(arg0 context.Context, arg1 int } // UpdateTurnResults indicates an expected call of UpdateTurnResults. -func (mr *MockExptTurnResultDAOMockRecorder) UpdateTurnResults(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) UpdateTurnResults(ctx, exptID, itemTurnIDs, spaceID, ufields any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, exptID, itemTurnIDs, spaceID, ufields}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTurnResults", reflect.TypeOf((*MockExptTurnResultDAO)(nil).UpdateTurnResults), varargs...) } // UpdateTurnResultsWithItemIDs mocks base method. -func (m *MockExptTurnResultDAO) UpdateTurnResultsWithItemIDs(arg0 context.Context, arg1 int64, arg2 []int64, arg3 int64, arg4 map[string]interface{}, arg5 ...db.Option) error { +func (m *MockExptTurnResultDAO) UpdateTurnResultsWithItemIDs(ctx context.Context, exptID int64, itemIDs []int64, spaceID int64, ufields map[string]any, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} - for _, a := range arg5 { + varargs := []any{ctx, exptID, itemIDs, spaceID, ufields} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateTurnResultsWithItemIDs", varargs...) @@ -349,17 +354,17 @@ func (m *MockExptTurnResultDAO) UpdateTurnResultsWithItemIDs(arg0 context.Contex } // UpdateTurnResultsWithItemIDs indicates an expected call of UpdateTurnResultsWithItemIDs. -func (mr *MockExptTurnResultDAOMockRecorder) UpdateTurnResultsWithItemIDs(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) UpdateTurnResultsWithItemIDs(ctx, exptID, itemIDs, spaceID, ufields any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) + varargs := append([]any{ctx, exptID, itemIDs, spaceID, ufields}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTurnResultsWithItemIDs", reflect.TypeOf((*MockExptTurnResultDAO)(nil).UpdateTurnResultsWithItemIDs), varargs...) } // UpdateTurnRunLogWithItemIDs mocks base method. -func (m *MockExptTurnResultDAO) UpdateTurnRunLogWithItemIDs(arg0 context.Context, arg1, arg2, arg3 int64, arg4 []int64, arg5 map[string]interface{}, arg6 ...db.Option) error { +func (m *MockExptTurnResultDAO) UpdateTurnRunLogWithItemIDs(ctx context.Context, spaceID, exptID, exptRunID int64, itemIDs []int64, ufields map[string]any, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2, arg3, arg4, arg5} - for _, a := range arg6 { + varargs := []any{ctx, spaceID, exptID, exptRunID, itemIDs, ufields} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateTurnRunLogWithItemIDs", varargs...) @@ -368,8 +373,8 @@ func (m *MockExptTurnResultDAO) UpdateTurnRunLogWithItemIDs(arg0 context.Context } // UpdateTurnRunLogWithItemIDs indicates an expected call of UpdateTurnRunLogWithItemIDs. -func (mr *MockExptTurnResultDAOMockRecorder) UpdateTurnRunLogWithItemIDs(arg0, arg1, arg2, arg3, arg4, arg5 interface{}, arg6 ...interface{}) *gomock.Call { +func (mr *MockExptTurnResultDAOMockRecorder) UpdateTurnRunLogWithItemIDs(ctx, spaceID, exptID, exptRunID, itemIDs, ufields any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) + varargs := append([]any{ctx, spaceID, exptID, exptRunID, itemIDs, ufields}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTurnRunLogWithItemIDs", reflect.TypeOf((*MockExptTurnResultDAO)(nil).UpdateTurnRunLogWithItemIDs), varargs...) } diff --git a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_result_filter_key_mapping.go b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_result_filter_key_mapping.go index 7207c8597..541a91ea4 100644 --- a/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_result_filter_key_mapping.go +++ b/backend/modules/evaluation/infra/repo/experiment/mysql/mocks/expt_turn_result_filter_key_mapping.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql (interfaces: IExptTurnResultFilterKeyMappingDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/expt_turn_result_filter_key_mapping.go -package mocks . IExptTurnResultFilterKeyMappingDAO +// // Package mocks is a generated GoMock package. package mocks @@ -10,13 +15,14 @@ import ( db "github.com/coze-dev/coze-loop/backend/infra/db" model "github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/mysql/gorm_gen/model" - "go.uber.org/mock/gomock" + gomock "go.uber.org/mock/gomock" ) // MockIExptTurnResultFilterKeyMappingDAO is a mock of IExptTurnResultFilterKeyMappingDAO interface. type MockIExptTurnResultFilterKeyMappingDAO struct { ctrl *gomock.Controller recorder *MockIExptTurnResultFilterKeyMappingDAOMockRecorder + isgomock struct{} } // MockIExptTurnResultFilterKeyMappingDAOMockRecorder is the mock recorder for MockIExptTurnResultFilterKeyMappingDAO. @@ -37,10 +43,10 @@ func (m *MockIExptTurnResultFilterKeyMappingDAO) EXPECT() *MockIExptTurnResultFi } // Delete mocks base method. -func (m *MockIExptTurnResultFilterKeyMappingDAO) Delete(arg0 context.Context, arg1 *model.ExptTurnResultFilterKeyMapping, arg2 ...db.Option) error { +func (m *MockIExptTurnResultFilterKeyMappingDAO) Delete(ctx context.Context, mapping *model.ExptTurnResultFilterKeyMapping, opts ...db.Option) error { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, mapping} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Delete", varargs...) @@ -49,17 +55,17 @@ func (m *MockIExptTurnResultFilterKeyMappingDAO) Delete(arg0 context.Context, ar } // Delete indicates an expected call of Delete. -func (mr *MockIExptTurnResultFilterKeyMappingDAOMockRecorder) Delete(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnResultFilterKeyMappingDAOMockRecorder) Delete(ctx, mapping any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1}, arg2...) + varargs := append([]any{ctx, mapping}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockIExptTurnResultFilterKeyMappingDAO)(nil).Delete), varargs...) } // GetByExptID mocks base method. -func (m *MockIExptTurnResultFilterKeyMappingDAO) GetByExptID(arg0 context.Context, arg1, arg2 int64, arg3 ...db.Option) ([]*model.ExptTurnResultFilterKeyMapping, error) { +func (m *MockIExptTurnResultFilterKeyMappingDAO) GetByExptID(ctx context.Context, spaceID, exptID int64, opts ...db.Option) ([]*model.ExptTurnResultFilterKeyMapping, error) { m.ctrl.T.Helper() - varargs := []interface{}{arg0, arg1, arg2} - for _, a := range arg3 { + varargs := []any{ctx, spaceID, exptID} + for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetByExptID", varargs...) @@ -69,22 +75,22 @@ func (m *MockIExptTurnResultFilterKeyMappingDAO) GetByExptID(arg0 context.Contex } // GetByExptID indicates an expected call of GetByExptID. -func (mr *MockIExptTurnResultFilterKeyMappingDAOMockRecorder) GetByExptID(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { +func (mr *MockIExptTurnResultFilterKeyMappingDAOMockRecorder) GetByExptID(ctx, spaceID, exptID any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) + varargs := append([]any{ctx, spaceID, exptID}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByExptID", reflect.TypeOf((*MockIExptTurnResultFilterKeyMappingDAO)(nil).GetByExptID), varargs...) } // Insert mocks base method. -func (m *MockIExptTurnResultFilterKeyMappingDAO) Insert(arg0 context.Context, arg1 []*model.ExptTurnResultFilterKeyMapping) error { +func (m *MockIExptTurnResultFilterKeyMappingDAO) Insert(ctx context.Context, exptTurnResultFilterKeyMappings []*model.ExptTurnResultFilterKeyMapping) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Insert", arg0, arg1) + ret := m.ctrl.Call(m, "Insert", ctx, exptTurnResultFilterKeyMappings) ret0, _ := ret[0].(error) return ret0 } // Insert indicates an expected call of Insert. -func (mr *MockIExptTurnResultFilterKeyMappingDAOMockRecorder) Insert(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIExptTurnResultFilterKeyMappingDAOMockRecorder) Insert(ctx, exptTurnResultFilterKeyMappings any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Insert", reflect.TypeOf((*MockIExptTurnResultFilterKeyMappingDAO)(nil).Insert), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Insert", reflect.TypeOf((*MockIExptTurnResultFilterKeyMappingDAO)(nil).Insert), ctx, exptTurnResultFilterKeyMappings) } diff --git a/backend/modules/evaluation/infra/repo/experiment/redis/dao/mocks/quota.go b/backend/modules/evaluation/infra/repo/experiment/redis/dao/mocks/quota.go index a7e3a9faf..dd4916691 100644 --- a/backend/modules/evaluation/infra/repo/experiment/redis/dao/mocks/quota.go +++ b/backend/modules/evaluation/infra/repo/experiment/redis/dao/mocks/quota.go @@ -1,22 +1,27 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: github.com/coze-dev/coze-loop/backend/modules/evaluation/infra/repo/experiment/redis/dao (interfaces: IQuotaDAO) +// Source: quota.go +// +// Generated by this command: +// +// mockgen -source=quota.go -destination=mocks/quota.go -package=mocks -mock_names=IQuotaDAO=MockIQuotaDAO +// // Package mocks is a generated GoMock package. package mocks import ( - "context" - "reflect" + context "context" + reflect "reflect" - "go.uber.org/mock/gomock" - - "github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/entity" + entity "github.com/coze-dev/coze-loop/backend/modules/evaluation/domain/entity" + gomock "go.uber.org/mock/gomock" ) // MockIQuotaDAO is a mock of IQuotaDAO interface. type MockIQuotaDAO struct { ctrl *gomock.Controller recorder *MockIQuotaDAOMockRecorder + isgomock struct{} } // MockIQuotaDAOMockRecorder is the mock recorder for MockIQuotaDAO. @@ -37,30 +42,30 @@ func (m *MockIQuotaDAO) EXPECT() *MockIQuotaDAOMockRecorder { } // GetQuotaSpaceExpt mocks base method. -func (m *MockIQuotaDAO) GetQuotaSpaceExpt(arg0 context.Context, arg1 int64) (*entity.QuotaSpaceExpt, error) { +func (m *MockIQuotaDAO) GetQuotaSpaceExpt(ctx context.Context, spaceID int64) (*entity.QuotaSpaceExpt, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetQuotaSpaceExpt", arg0, arg1) + ret := m.ctrl.Call(m, "GetQuotaSpaceExpt", ctx, spaceID) ret0, _ := ret[0].(*entity.QuotaSpaceExpt) ret1, _ := ret[1].(error) return ret0, ret1 } // GetQuotaSpaceExpt indicates an expected call of GetQuotaSpaceExpt. -func (mr *MockIQuotaDAOMockRecorder) GetQuotaSpaceExpt(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIQuotaDAOMockRecorder) GetQuotaSpaceExpt(ctx, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetQuotaSpaceExpt", reflect.TypeOf((*MockIQuotaDAO)(nil).GetQuotaSpaceExpt), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetQuotaSpaceExpt", reflect.TypeOf((*MockIQuotaDAO)(nil).GetQuotaSpaceExpt), ctx, spaceID) } // SetQuotaSpaceExpt mocks base method. -func (m *MockIQuotaDAO) SetQuotaSpaceExpt(arg0 context.Context, arg1 int64, arg2 *entity.QuotaSpaceExpt) error { +func (m *MockIQuotaDAO) SetQuotaSpaceExpt(ctx context.Context, spaceID int64, qse *entity.QuotaSpaceExpt) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SetQuotaSpaceExpt", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "SetQuotaSpaceExpt", ctx, spaceID, qse) ret0, _ := ret[0].(error) return ret0 } // SetQuotaSpaceExpt indicates an expected call of SetQuotaSpaceExpt. -func (mr *MockIQuotaDAOMockRecorder) SetQuotaSpaceExpt(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIQuotaDAOMockRecorder) SetQuotaSpaceExpt(ctx, spaceID, qse any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetQuotaSpaceExpt", reflect.TypeOf((*MockIQuotaDAO)(nil).SetQuotaSpaceExpt), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetQuotaSpaceExpt", reflect.TypeOf((*MockIQuotaDAO)(nil).SetQuotaSpaceExpt), ctx, spaceID, qse) } diff --git a/backend/modules/foundation/domain/authn/repo/mocks/authn_repo.go b/backend/modules/foundation/domain/authn/repo/mocks/authn_repo.go index 1c8fe1cf5..83dd9ecc2 100644 --- a/backend/modules/foundation/domain/authn/repo/mocks/authn_repo.go +++ b/backend/modules/foundation/domain/authn/repo/mocks/authn_repo.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. // Source: github.com/coze-dev/coze-loop/backend/modules/foundation/domain/authn/repo (interfaces: IAuthNRepo) +// +// Generated by this command: +// +// mockgen -destination=mocks/authn_repo.go -package=mocks . IAuthNRepo +// // Package mocks is a generated GoMock package. package mocks @@ -16,6 +21,7 @@ import ( type MockIAuthNRepo struct { ctrl *gomock.Controller recorder *MockIAuthNRepoMockRecorder + isgomock struct{} } // MockIAuthNRepoMockRecorder is the mock recorder for MockIAuthNRepo. @@ -36,9 +42,9 @@ func (m *MockIAuthNRepo) EXPECT() *MockIAuthNRepoMockRecorder { } // CreateAPIKey mocks base method. -func (m *MockIAuthNRepo) CreateAPIKey(arg0 context.Context, arg1 *entity.APIKey) (int64, string, error) { +func (m *MockIAuthNRepo) CreateAPIKey(ctx context.Context, apiKeyEntity *entity.APIKey) (int64, string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateAPIKey", arg0, arg1) + ret := m.ctrl.Call(m, "CreateAPIKey", ctx, apiKeyEntity) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(string) ret2, _ := ret[2].(error) @@ -46,94 +52,94 @@ func (m *MockIAuthNRepo) CreateAPIKey(arg0 context.Context, arg1 *entity.APIKey) } // CreateAPIKey indicates an expected call of CreateAPIKey. -func (mr *MockIAuthNRepoMockRecorder) CreateAPIKey(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIAuthNRepoMockRecorder) CreateAPIKey(ctx, apiKeyEntity any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateAPIKey", reflect.TypeOf((*MockIAuthNRepo)(nil).CreateAPIKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateAPIKey", reflect.TypeOf((*MockIAuthNRepo)(nil).CreateAPIKey), ctx, apiKeyEntity) } // DeleteAPIKey mocks base method. -func (m *MockIAuthNRepo) DeleteAPIKey(arg0 context.Context, arg1 int64) error { +func (m *MockIAuthNRepo) DeleteAPIKey(ctx context.Context, apiKeyID int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteAPIKey", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteAPIKey", ctx, apiKeyID) ret0, _ := ret[0].(error) return ret0 } // DeleteAPIKey indicates an expected call of DeleteAPIKey. -func (mr *MockIAuthNRepoMockRecorder) DeleteAPIKey(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIAuthNRepoMockRecorder) DeleteAPIKey(ctx, apiKeyID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAPIKey", reflect.TypeOf((*MockIAuthNRepo)(nil).DeleteAPIKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAPIKey", reflect.TypeOf((*MockIAuthNRepo)(nil).DeleteAPIKey), ctx, apiKeyID) } // FlushAPIKeyUsedTime mocks base method. -func (m *MockIAuthNRepo) FlushAPIKeyUsedTime(arg0 context.Context, arg1 int64) error { +func (m *MockIAuthNRepo) FlushAPIKeyUsedTime(ctx context.Context, apiKeyID int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "FlushAPIKeyUsedTime", arg0, arg1) + ret := m.ctrl.Call(m, "FlushAPIKeyUsedTime", ctx, apiKeyID) ret0, _ := ret[0].(error) return ret0 } // FlushAPIKeyUsedTime indicates an expected call of FlushAPIKeyUsedTime. -func (mr *MockIAuthNRepoMockRecorder) FlushAPIKeyUsedTime(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIAuthNRepoMockRecorder) FlushAPIKeyUsedTime(ctx, apiKeyID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FlushAPIKeyUsedTime", reflect.TypeOf((*MockIAuthNRepo)(nil).FlushAPIKeyUsedTime), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FlushAPIKeyUsedTime", reflect.TypeOf((*MockIAuthNRepo)(nil).FlushAPIKeyUsedTime), ctx, apiKeyID) } // GetAPIKeyByIDs mocks base method. -func (m *MockIAuthNRepo) GetAPIKeyByIDs(arg0 context.Context, arg1 []int64) ([]*entity.APIKey, error) { +func (m *MockIAuthNRepo) GetAPIKeyByIDs(ctx context.Context, apiKeyIDs []int64) ([]*entity.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAPIKeyByIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetAPIKeyByIDs", ctx, apiKeyIDs) ret0, _ := ret[0].([]*entity.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAPIKeyByIDs indicates an expected call of GetAPIKeyByIDs. -func (mr *MockIAuthNRepoMockRecorder) GetAPIKeyByIDs(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIAuthNRepoMockRecorder) GetAPIKeyByIDs(ctx, apiKeyIDs any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByIDs", reflect.TypeOf((*MockIAuthNRepo)(nil).GetAPIKeyByIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByIDs", reflect.TypeOf((*MockIAuthNRepo)(nil).GetAPIKeyByIDs), ctx, apiKeyIDs) } // GetAPIKeyByKey mocks base method. -func (m *MockIAuthNRepo) GetAPIKeyByKey(arg0 context.Context, arg1 string) (*entity.APIKey, error) { +func (m *MockIAuthNRepo) GetAPIKeyByKey(ctx context.Context, key string) (*entity.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAPIKeyByKey", arg0, arg1) + ret := m.ctrl.Call(m, "GetAPIKeyByKey", ctx, key) ret0, _ := ret[0].(*entity.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAPIKeyByKey indicates an expected call of GetAPIKeyByKey. -func (mr *MockIAuthNRepoMockRecorder) GetAPIKeyByKey(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIAuthNRepoMockRecorder) GetAPIKeyByKey(ctx, key any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByKey", reflect.TypeOf((*MockIAuthNRepo)(nil).GetAPIKeyByKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByKey", reflect.TypeOf((*MockIAuthNRepo)(nil).GetAPIKeyByKey), ctx, key) } // GetAPIKeyByUser mocks base method. -func (m *MockIAuthNRepo) GetAPIKeyByUser(arg0 context.Context, arg1 int64, arg2, arg3 int) ([]*entity.APIKey, error) { +func (m *MockIAuthNRepo) GetAPIKeyByUser(ctx context.Context, userID int64, pageNumber, pageSize int) ([]*entity.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAPIKeyByUser", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "GetAPIKeyByUser", ctx, userID, pageNumber, pageSize) ret0, _ := ret[0].([]*entity.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAPIKeyByUser indicates an expected call of GetAPIKeyByUser. -func (mr *MockIAuthNRepoMockRecorder) GetAPIKeyByUser(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { +func (mr *MockIAuthNRepoMockRecorder) GetAPIKeyByUser(ctx, userID, pageNumber, pageSize any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByUser", reflect.TypeOf((*MockIAuthNRepo)(nil).GetAPIKeyByUser), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByUser", reflect.TypeOf((*MockIAuthNRepo)(nil).GetAPIKeyByUser), ctx, userID, pageNumber, pageSize) } // UpdateAPIKeyName mocks base method. -func (m *MockIAuthNRepo) UpdateAPIKeyName(arg0 context.Context, arg1 int64, arg2 string) error { +func (m *MockIAuthNRepo) UpdateAPIKeyName(ctx context.Context, apiKeyID int64, name string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateAPIKeyName", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "UpdateAPIKeyName", ctx, apiKeyID, name) ret0, _ := ret[0].(error) return ret0 } // UpdateAPIKeyName indicates an expected call of UpdateAPIKeyName. -func (mr *MockIAuthNRepoMockRecorder) UpdateAPIKeyName(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIAuthNRepoMockRecorder) UpdateAPIKeyName(ctx, apiKeyID, name any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAPIKeyName", reflect.TypeOf((*MockIAuthNRepo)(nil).UpdateAPIKeyName), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAPIKeyName", reflect.TypeOf((*MockIAuthNRepo)(nil).UpdateAPIKeyName), ctx, apiKeyID, name) } diff --git a/backend/modules/foundation/domain/user/service/mocks/user_service.go b/backend/modules/foundation/domain/user/service/mocks/user_service.go index df25c9f34..e9eb2d800 100644 --- a/backend/modules/foundation/domain/user/service/mocks/user_service.go +++ b/backend/modules/foundation/domain/user/service/mocks/user_service.go @@ -1,5 +1,10 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: github.com/coze-dev/coze-loop/backend/modules/foundation/domain/user/service (interfaces: IUserService) +// Source: interface.go +// +// Generated by this command: +// +// mockgen -source=interface.go -destination=mocks/user_service.go -package=mocks -mock_names=IUserService=MockIUserService +// // Package mocks is a generated GoMock package. package mocks @@ -17,6 +22,7 @@ import ( type MockIUserService struct { ctrl *gomock.Controller recorder *MockIUserServiceMockRecorder + isgomock struct{} } // MockIUserServiceMockRecorder is the mock recorder for MockIUserService. @@ -37,54 +43,54 @@ func (m *MockIUserService) EXPECT() *MockIUserServiceMockRecorder { } // Create mocks base method. -func (m *MockIUserService) Create(arg0 context.Context, arg1 *service.CreateUserRequest) (*entity.User, error) { +func (m *MockIUserService) Create(ctx context.Context, req *service.CreateUserRequest) (*entity.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Create", arg0, arg1) + ret := m.ctrl.Call(m, "Create", ctx, req) ret0, _ := ret[0].(*entity.User) ret1, _ := ret[1].(error) return ret0, ret1 } // Create indicates an expected call of Create. -func (mr *MockIUserServiceMockRecorder) Create(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) Create(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIUserService)(nil).Create), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockIUserService)(nil).Create), ctx, req) } // CreateSession mocks base method. -func (m *MockIUserService) CreateSession(arg0 context.Context, arg1 int64) (string, error) { +func (m *MockIUserService) CreateSession(ctx context.Context, userID int64) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateSession", arg0, arg1) + ret := m.ctrl.Call(m, "CreateSession", ctx, userID) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateSession indicates an expected call of CreateSession. -func (mr *MockIUserServiceMockRecorder) CreateSession(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) CreateSession(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSession", reflect.TypeOf((*MockIUserService)(nil).CreateSession), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSession", reflect.TypeOf((*MockIUserService)(nil).CreateSession), ctx, userID) } // GetUserProfile mocks base method. -func (m *MockIUserService) GetUserProfile(arg0 context.Context, arg1 int64) (*entity.User, error) { +func (m *MockIUserService) GetUserProfile(ctx context.Context, userID int64) (*entity.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserProfile", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserProfile", ctx, userID) ret0, _ := ret[0].(*entity.User) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserProfile indicates an expected call of GetUserProfile. -func (mr *MockIUserServiceMockRecorder) GetUserProfile(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) GetUserProfile(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserProfile", reflect.TypeOf((*MockIUserService)(nil).GetUserProfile), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserProfile", reflect.TypeOf((*MockIUserService)(nil).GetUserProfile), ctx, userID) } // GetUserSpaceList mocks base method. -func (m *MockIUserService) GetUserSpaceList(arg0 context.Context, arg1 *service.ListUserSpaceRequest) ([]*entity.Space, int32, error) { +func (m *MockIUserService) GetUserSpaceList(ctx context.Context, req *service.ListUserSpaceRequest) ([]*entity.Space, int32, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserSpaceList", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserSpaceList", ctx, req) ret0, _ := ret[0].([]*entity.Space) ret1, _ := ret[1].(int32) ret2, _ := ret[2].(error) @@ -92,80 +98,80 @@ func (m *MockIUserService) GetUserSpaceList(arg0 context.Context, arg1 *service. } // GetUserSpaceList indicates an expected call of GetUserSpaceList. -func (mr *MockIUserServiceMockRecorder) GetUserSpaceList(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) GetUserSpaceList(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserSpaceList", reflect.TypeOf((*MockIUserService)(nil).GetUserSpaceList), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserSpaceList", reflect.TypeOf((*MockIUserService)(nil).GetUserSpaceList), ctx, req) } // Login mocks base method. -func (m *MockIUserService) Login(arg0 context.Context, arg1, arg2 string) (*entity.User, error) { +func (m *MockIUserService) Login(ctx context.Context, email, password string) (*entity.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Login", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "Login", ctx, email, password) ret0, _ := ret[0].(*entity.User) ret1, _ := ret[1].(error) return ret0, ret1 } // Login indicates an expected call of Login. -func (mr *MockIUserServiceMockRecorder) Login(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) Login(ctx, email, password any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Login", reflect.TypeOf((*MockIUserService)(nil).Login), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Login", reflect.TypeOf((*MockIUserService)(nil).Login), ctx, email, password) } // Logout mocks base method. -func (m *MockIUserService) Logout(arg0 context.Context, arg1 int64) error { +func (m *MockIUserService) Logout(ctx context.Context, userID int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Logout", arg0, arg1) + ret := m.ctrl.Call(m, "Logout", ctx, userID) ret0, _ := ret[0].(error) return ret0 } // Logout indicates an expected call of Logout. -func (mr *MockIUserServiceMockRecorder) Logout(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) Logout(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Logout", reflect.TypeOf((*MockIUserService)(nil).Logout), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Logout", reflect.TypeOf((*MockIUserService)(nil).Logout), ctx, userID) } // MGetUserProfiles mocks base method. -func (m *MockIUserService) MGetUserProfiles(arg0 context.Context, arg1 []int64) ([]*entity.User, error) { +func (m *MockIUserService) MGetUserProfiles(ctx context.Context, userIDs []int64) ([]*entity.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "MGetUserProfiles", arg0, arg1) + ret := m.ctrl.Call(m, "MGetUserProfiles", ctx, userIDs) ret0, _ := ret[0].([]*entity.User) ret1, _ := ret[1].(error) return ret0, ret1 } // MGetUserProfiles indicates an expected call of MGetUserProfiles. -func (mr *MockIUserServiceMockRecorder) MGetUserProfiles(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) MGetUserProfiles(ctx, userIDs any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetUserProfiles", reflect.TypeOf((*MockIUserService)(nil).MGetUserProfiles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetUserProfiles", reflect.TypeOf((*MockIUserService)(nil).MGetUserProfiles), ctx, userIDs) } // ResetPassword mocks base method. -func (m *MockIUserService) ResetPassword(arg0 context.Context, arg1, arg2 string) error { +func (m *MockIUserService) ResetPassword(ctx context.Context, email, password string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ResetPassword", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "ResetPassword", ctx, email, password) ret0, _ := ret[0].(error) return ret0 } // ResetPassword indicates an expected call of ResetPassword. -func (mr *MockIUserServiceMockRecorder) ResetPassword(arg0, arg1, arg2 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) ResetPassword(ctx, email, password any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetPassword", reflect.TypeOf((*MockIUserService)(nil).ResetPassword), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetPassword", reflect.TypeOf((*MockIUserService)(nil).ResetPassword), ctx, email, password) } // UpdateProfile mocks base method. -func (m *MockIUserService) UpdateProfile(arg0 context.Context, arg1 *service.UpdateProfileRequest) (*entity.User, error) { +func (m *MockIUserService) UpdateProfile(ctx context.Context, req *service.UpdateProfileRequest) (*entity.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateProfile", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateProfile", ctx, req) ret0, _ := ret[0].(*entity.User) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateProfile indicates an expected call of UpdateProfile. -func (mr *MockIUserServiceMockRecorder) UpdateProfile(arg0, arg1 interface{}) *gomock.Call { +func (mr *MockIUserServiceMockRecorder) UpdateProfile(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProfile", reflect.TypeOf((*MockIUserService)(nil).UpdateProfile), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProfile", reflect.TypeOf((*MockIUserService)(nil).UpdateProfile), ctx, req) } diff --git a/backend/modules/observability/domain/component/metrics/mocks/metrics.go b/backend/modules/observability/domain/component/metrics/mocks/metrics.go index 3bf3770b8..b6ed0a1ce 100644 --- a/backend/modules/observability/domain/component/metrics/mocks/metrics.go +++ b/backend/modules/observability/domain/component/metrics/mocks/metrics.go @@ -71,7 +71,7 @@ func (m *MockITraceMetrics) EmitTraceOapi(method string, workspaceId int64, plat } // EmitTraceOapi indicates an expected call of EmitTraceOapi. -func (mr *MockITraceMetricsMockRecorder) EmitTraceOapi(method, workspaceId, platformType, spanType, spanSize, errorCode, start, isError any) *gomock.Call { +func (mr *MockITraceMetricsMockRecorder) EmitTraceOapi(method, workspaceId, platformType, spanListType, spanSize, errorCode, start, isError any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EmitTraceOapi", reflect.TypeOf((*MockITraceMetrics)(nil).EmitTraceOapi), method, workspaceId, platformType, spanType, spanSize, errorCode, start, isError) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EmitTraceOapi", reflect.TypeOf((*MockITraceMetrics)(nil).EmitTraceOapi), method, workspaceId, platformType, spanListType, spanSize, errorCode, start, isError) } diff --git a/backend/modules/observability/domain/component/rpc/mocks/auth_provider.go b/backend/modules/observability/domain/component/rpc/mocks/auth_provider.go index 78ec78448..881d90c1f 100644 --- a/backend/modules/observability/domain/component/rpc/mocks/auth_provider.go +++ b/backend/modules/observability/domain/component/rpc/mocks/auth_provider.go @@ -69,17 +69,17 @@ func (mr *MockIAuthProviderMockRecorder) CheckQueryPermission(ctx, workspaceId, } // CheckTaskPermission mocks base method. -func (m *MockIAuthProvider) CheckTaskPermission(arg0 context.Context, arg1, arg2, arg3 string) error { +func (m *MockIAuthProvider) CheckTaskPermission(ctx context.Context, action, workspaceId, taskId string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CheckTaskPermission", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "CheckTaskPermission", ctx, action, workspaceId, taskId) ret0, _ := ret[0].(error) return ret0 } // CheckTaskPermission indicates an expected call of CheckTaskPermission. -func (mr *MockIAuthProviderMockRecorder) CheckTaskPermission(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockIAuthProviderMockRecorder) CheckTaskPermission(ctx, action, workspaceId, taskId any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckTaskPermission", reflect.TypeOf((*MockIAuthProvider)(nil).CheckTaskPermission), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckTaskPermission", reflect.TypeOf((*MockIAuthProvider)(nil).CheckTaskPermission), ctx, action, workspaceId, taskId) } // CheckViewPermission mocks base method. diff --git a/backend/modules/observability/domain/component/rpc/mocks/dataset_provider_mock.go b/backend/modules/observability/domain/component/rpc/mocks/dataset_provider_mock.go index efebc6c00..bc346ae06 100644 --- a/backend/modules/observability/domain/component/rpc/mocks/dataset_provider_mock.go +++ b/backend/modules/observability/domain/component/rpc/mocks/dataset_provider_mock.go @@ -21,6 +21,7 @@ import ( type MockIDatasetProvider struct { ctrl *gomock.Controller recorder *MockIDatasetProviderMockRecorder + isgomock struct{} } // MockIDatasetProviderMockRecorder is the mock recorder for MockIDatasetProvider. @@ -41,9 +42,9 @@ func (m *MockIDatasetProvider) EXPECT() *MockIDatasetProviderMockRecorder { } // AddDatasetItems mocks base method. -func (m *MockIDatasetProvider) AddDatasetItems(arg0 context.Context, arg1 int64, arg2 entity.DatasetCategory, arg3 []*entity.DatasetItem) ([]*entity.DatasetItem, []entity.ItemErrorGroup, error) { +func (m *MockIDatasetProvider) AddDatasetItems(ctx context.Context, datasetID int64, category entity.DatasetCategory, items []*entity.DatasetItem) ([]*entity.DatasetItem, []entity.ItemErrorGroup, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "AddDatasetItems", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "AddDatasetItems", ctx, datasetID, category, items) ret0, _ := ret[0].([]*entity.DatasetItem) ret1, _ := ret[1].([]entity.ItemErrorGroup) ret2, _ := ret[2].(error) @@ -51,88 +52,88 @@ func (m *MockIDatasetProvider) AddDatasetItems(arg0 context.Context, arg1 int64, } // AddDatasetItems indicates an expected call of AddDatasetItems. -func (mr *MockIDatasetProviderMockRecorder) AddDatasetItems(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockIDatasetProviderMockRecorder) AddDatasetItems(ctx, datasetID, category, items any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddDatasetItems", reflect.TypeOf((*MockIDatasetProvider)(nil).AddDatasetItems), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddDatasetItems", reflect.TypeOf((*MockIDatasetProvider)(nil).AddDatasetItems), ctx, datasetID, category, items) } // ClearDatasetItems mocks base method. -func (m *MockIDatasetProvider) ClearDatasetItems(arg0 context.Context, arg1, arg2 int64, arg3 entity.DatasetCategory) error { +func (m *MockIDatasetProvider) ClearDatasetItems(ctx context.Context, workspaceID, datasetID int64, category entity.DatasetCategory) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ClearDatasetItems", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "ClearDatasetItems", ctx, workspaceID, datasetID, category) ret0, _ := ret[0].(error) return ret0 } // ClearDatasetItems indicates an expected call of ClearDatasetItems. -func (mr *MockIDatasetProviderMockRecorder) ClearDatasetItems(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockIDatasetProviderMockRecorder) ClearDatasetItems(ctx, workspaceID, datasetID, category any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearDatasetItems", reflect.TypeOf((*MockIDatasetProvider)(nil).ClearDatasetItems), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearDatasetItems", reflect.TypeOf((*MockIDatasetProvider)(nil).ClearDatasetItems), ctx, workspaceID, datasetID, category) } // CreateDataset mocks base method. -func (m *MockIDatasetProvider) CreateDataset(arg0 context.Context, arg1 *entity.Dataset) (int64, error) { +func (m *MockIDatasetProvider) CreateDataset(ctx context.Context, dataset *entity.Dataset) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateDataset", arg0, arg1) + ret := m.ctrl.Call(m, "CreateDataset", ctx, dataset) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateDataset indicates an expected call of CreateDataset. -func (mr *MockIDatasetProviderMockRecorder) CreateDataset(arg0, arg1 any) *gomock.Call { +func (mr *MockIDatasetProviderMockRecorder) CreateDataset(ctx, dataset any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataset", reflect.TypeOf((*MockIDatasetProvider)(nil).CreateDataset), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataset", reflect.TypeOf((*MockIDatasetProvider)(nil).CreateDataset), ctx, dataset) } // GetDataset mocks base method. -func (m *MockIDatasetProvider) GetDataset(arg0 context.Context, arg1, arg2 int64, arg3 entity.DatasetCategory) (*entity.Dataset, error) { +func (m *MockIDatasetProvider) GetDataset(ctx context.Context, workspaceID, datasetID int64, category entity.DatasetCategory) (*entity.Dataset, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDataset", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "GetDataset", ctx, workspaceID, datasetID, category) ret0, _ := ret[0].(*entity.Dataset) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDataset indicates an expected call of GetDataset. -func (mr *MockIDatasetProviderMockRecorder) GetDataset(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockIDatasetProviderMockRecorder) GetDataset(ctx, workspaceID, datasetID, category any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataset", reflect.TypeOf((*MockIDatasetProvider)(nil).GetDataset), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataset", reflect.TypeOf((*MockIDatasetProvider)(nil).GetDataset), ctx, workspaceID, datasetID, category) } // SearchDatasets mocks base method. -func (m *MockIDatasetProvider) SearchDatasets(arg0 context.Context, arg1, arg2 int64, arg3 entity.DatasetCategory, arg4 string) ([]*entity.Dataset, error) { +func (m *MockIDatasetProvider) SearchDatasets(ctx context.Context, workspaceID, datasetID int64, category entity.DatasetCategory, name string) ([]*entity.Dataset, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SearchDatasets", arg0, arg1, arg2, arg3, arg4) + ret := m.ctrl.Call(m, "SearchDatasets", ctx, workspaceID, datasetID, category, name) ret0, _ := ret[0].([]*entity.Dataset) ret1, _ := ret[1].(error) return ret0, ret1 } // SearchDatasets indicates an expected call of SearchDatasets. -func (mr *MockIDatasetProviderMockRecorder) SearchDatasets(arg0, arg1, arg2, arg3, arg4 any) *gomock.Call { +func (mr *MockIDatasetProviderMockRecorder) SearchDatasets(ctx, workspaceID, datasetID, category, name any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchDatasets", reflect.TypeOf((*MockIDatasetProvider)(nil).SearchDatasets), arg0, arg1, arg2, arg3, arg4) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchDatasets", reflect.TypeOf((*MockIDatasetProvider)(nil).SearchDatasets), ctx, workspaceID, datasetID, category, name) } // UpdateDatasetSchema mocks base method. -func (m *MockIDatasetProvider) UpdateDatasetSchema(arg0 context.Context, arg1 *entity.Dataset) error { +func (m *MockIDatasetProvider) UpdateDatasetSchema(ctx context.Context, dataset *entity.Dataset) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateDatasetSchema", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateDatasetSchema", ctx, dataset) ret0, _ := ret[0].(error) return ret0 } // UpdateDatasetSchema indicates an expected call of UpdateDatasetSchema. -func (mr *MockIDatasetProviderMockRecorder) UpdateDatasetSchema(arg0, arg1 any) *gomock.Call { +func (mr *MockIDatasetProviderMockRecorder) UpdateDatasetSchema(ctx, dataset any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDatasetSchema", reflect.TypeOf((*MockIDatasetProvider)(nil).UpdateDatasetSchema), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDatasetSchema", reflect.TypeOf((*MockIDatasetProvider)(nil).UpdateDatasetSchema), ctx, dataset) } // ValidateDatasetItems mocks base method. -func (m *MockIDatasetProvider) ValidateDatasetItems(arg0 context.Context, arg1 *entity.Dataset, arg2 []*entity.DatasetItem, arg3 *bool) ([]*entity.DatasetItem, []entity.ItemErrorGroup, error) { +func (m *MockIDatasetProvider) ValidateDatasetItems(ctx context.Context, dataset *entity.Dataset, items []*entity.DatasetItem, ignoreCurrentCount *bool) ([]*entity.DatasetItem, []entity.ItemErrorGroup, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ValidateDatasetItems", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "ValidateDatasetItems", ctx, dataset, items, ignoreCurrentCount) ret0, _ := ret[0].([]*entity.DatasetItem) ret1, _ := ret[1].([]entity.ItemErrorGroup) ret2, _ := ret[2].(error) @@ -140,7 +141,7 @@ func (m *MockIDatasetProvider) ValidateDatasetItems(arg0 context.Context, arg1 * } // ValidateDatasetItems indicates an expected call of ValidateDatasetItems. -func (mr *MockIDatasetProviderMockRecorder) ValidateDatasetItems(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockIDatasetProviderMockRecorder) ValidateDatasetItems(ctx, dataset, items, ignoreCurrentCount any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateDatasetItems", reflect.TypeOf((*MockIDatasetProvider)(nil).ValidateDatasetItems), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateDatasetItems", reflect.TypeOf((*MockIDatasetProvider)(nil).ValidateDatasetItems), ctx, dataset, items, ignoreCurrentCount) } diff --git a/backend/modules/observability/domain/component/rpc/mocks/evaluator.go b/backend/modules/observability/domain/component/rpc/mocks/evaluator.go index 57bd0a127..9a4fcfec0 100644 --- a/backend/modules/observability/domain/component/rpc/mocks/evaluator.go +++ b/backend/modules/observability/domain/component/rpc/mocks/evaluator.go @@ -21,6 +21,7 @@ import ( type MockIEvaluatorRPCAdapter struct { ctrl *gomock.Controller recorder *MockIEvaluatorRPCAdapterMockRecorder + isgomock struct{} } // MockIEvaluatorRPCAdapterMockRecorder is the mock recorder for MockIEvaluatorRPCAdapter. @@ -41,9 +42,9 @@ func (m *MockIEvaluatorRPCAdapter) EXPECT() *MockIEvaluatorRPCAdapterMockRecorde } // BatchGetEvaluatorVersions mocks base method. -func (m *MockIEvaluatorRPCAdapter) BatchGetEvaluatorVersions(arg0 context.Context, arg1 *rpc.BatchGetEvaluatorVersionsParam) ([]*rpc.Evaluator, map[int64]*rpc.Evaluator, error) { +func (m *MockIEvaluatorRPCAdapter) BatchGetEvaluatorVersions(ctx context.Context, param *rpc.BatchGetEvaluatorVersionsParam) ([]*rpc.Evaluator, map[int64]*rpc.Evaluator, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchGetEvaluatorVersions", arg0, arg1) + ret := m.ctrl.Call(m, "BatchGetEvaluatorVersions", ctx, param) ret0, _ := ret[0].([]*rpc.Evaluator) ret1, _ := ret[1].(map[int64]*rpc.Evaluator) ret2, _ := ret[2].(error) @@ -51,36 +52,36 @@ func (m *MockIEvaluatorRPCAdapter) BatchGetEvaluatorVersions(arg0 context.Contex } // BatchGetEvaluatorVersions indicates an expected call of BatchGetEvaluatorVersions. -func (mr *MockIEvaluatorRPCAdapterMockRecorder) BatchGetEvaluatorVersions(arg0, arg1 any) *gomock.Call { +func (mr *MockIEvaluatorRPCAdapterMockRecorder) BatchGetEvaluatorVersions(ctx, param any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorVersions", reflect.TypeOf((*MockIEvaluatorRPCAdapter)(nil).BatchGetEvaluatorVersions), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluatorVersions", reflect.TypeOf((*MockIEvaluatorRPCAdapter)(nil).BatchGetEvaluatorVersions), ctx, param) } // ListEvaluators mocks base method. -func (m *MockIEvaluatorRPCAdapter) ListEvaluators(arg0 context.Context, arg1 *rpc.ListEvaluatorsParam) ([]*rpc.Evaluator, error) { +func (m *MockIEvaluatorRPCAdapter) ListEvaluators(ctx context.Context, param *rpc.ListEvaluatorsParam) ([]*rpc.Evaluator, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListEvaluators", arg0, arg1) + ret := m.ctrl.Call(m, "ListEvaluators", ctx, param) ret0, _ := ret[0].([]*rpc.Evaluator) ret1, _ := ret[1].(error) return ret0, ret1 } // ListEvaluators indicates an expected call of ListEvaluators. -func (mr *MockIEvaluatorRPCAdapterMockRecorder) ListEvaluators(arg0, arg1 any) *gomock.Call { +func (mr *MockIEvaluatorRPCAdapterMockRecorder) ListEvaluators(ctx, param any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluators", reflect.TypeOf((*MockIEvaluatorRPCAdapter)(nil).ListEvaluators), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluators", reflect.TypeOf((*MockIEvaluatorRPCAdapter)(nil).ListEvaluators), ctx, param) } // UpdateEvaluatorRecord mocks base method. -func (m *MockIEvaluatorRPCAdapter) UpdateEvaluatorRecord(arg0 context.Context, arg1 *rpc.UpdateEvaluatorRecordParam) error { +func (m *MockIEvaluatorRPCAdapter) UpdateEvaluatorRecord(ctx context.Context, param *rpc.UpdateEvaluatorRecordParam) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateEvaluatorRecord", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateEvaluatorRecord", ctx, param) ret0, _ := ret[0].(error) return ret0 } // UpdateEvaluatorRecord indicates an expected call of UpdateEvaluatorRecord. -func (mr *MockIEvaluatorRPCAdapterMockRecorder) UpdateEvaluatorRecord(arg0, arg1 any) *gomock.Call { +func (mr *MockIEvaluatorRPCAdapterMockRecorder) UpdateEvaluatorRecord(ctx, param any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRPCAdapter)(nil).UpdateEvaluatorRecord), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluatorRecord", reflect.TypeOf((*MockIEvaluatorRPCAdapter)(nil).UpdateEvaluatorRecord), ctx, param) } diff --git a/backend/modules/observability/domain/component/rpc/mocks/file.go b/backend/modules/observability/domain/component/rpc/mocks/file.go index ed18922bc..2ea9cda65 100644 --- a/backend/modules/observability/domain/component/rpc/mocks/file.go +++ b/backend/modules/observability/domain/component/rpc/mocks/file.go @@ -20,6 +20,7 @@ import ( type MockIFileProvider struct { ctrl *gomock.Controller recorder *MockIFileProviderMockRecorder + isgomock struct{} } // MockIFileProviderMockRecorder is the mock recorder for MockIFileProvider. diff --git a/backend/modules/observability/domain/component/rpc/mocks/tag.go b/backend/modules/observability/domain/component/rpc/mocks/tag.go index 67b72557d..db5d94694 100644 --- a/backend/modules/observability/domain/component/rpc/mocks/tag.go +++ b/backend/modules/observability/domain/component/rpc/mocks/tag.go @@ -21,6 +21,7 @@ import ( type MockITagRPCAdapter struct { ctrl *gomock.Controller recorder *MockITagRPCAdapterMockRecorder + isgomock struct{} } // MockITagRPCAdapterMockRecorder is the mock recorder for MockITagRPCAdapter. diff --git a/backend/modules/observability/domain/component/rpc/mocks/user.go b/backend/modules/observability/domain/component/rpc/mocks/user.go index eb5db57ab..bb55cfe63 100644 --- a/backend/modules/observability/domain/component/rpc/mocks/user.go +++ b/backend/modules/observability/domain/component/rpc/mocks/user.go @@ -21,6 +21,7 @@ import ( type MockIUserProvider struct { ctrl *gomock.Controller recorder *MockIUserProviderMockRecorder + isgomock struct{} } // MockIUserProviderMockRecorder is the mock recorder for MockIUserProvider. @@ -41,9 +42,9 @@ func (m *MockIUserProvider) EXPECT() *MockIUserProviderMockRecorder { } // GetUserInfo mocks base method. -func (m *MockIUserProvider) GetUserInfo(arg0 context.Context, arg1 []string) ([]*common.UserInfo, map[string]*common.UserInfo, error) { +func (m *MockIUserProvider) GetUserInfo(ctx context.Context, userIDs []string) ([]*common.UserInfo, map[string]*common.UserInfo, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserInfo", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserInfo", ctx, userIDs) ret0, _ := ret[0].([]*common.UserInfo) ret1, _ := ret[1].(map[string]*common.UserInfo) ret2, _ := ret[2].(error) @@ -51,7 +52,7 @@ func (m *MockIUserProvider) GetUserInfo(arg0 context.Context, arg1 []string) ([] } // GetUserInfo indicates an expected call of GetUserInfo. -func (mr *MockIUserProviderMockRecorder) GetUserInfo(arg0, arg1 any) *gomock.Call { +func (mr *MockIUserProviderMockRecorder) GetUserInfo(ctx, userIDs any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserInfo", reflect.TypeOf((*MockIUserProvider)(nil).GetUserInfo), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserInfo", reflect.TypeOf((*MockIUserProvider)(nil).GetUserInfo), ctx, userIDs) } diff --git a/backend/modules/observability/domain/task/repo/mocks/Task.go b/backend/modules/observability/domain/task/repo/mocks/Task.go index 6af237884..d9115c1ee 100644 --- a/backend/modules/observability/domain/task/repo/mocks/Task.go +++ b/backend/modules/observability/domain/task/repo/mocks/Task.go @@ -3,7 +3,7 @@ // // Generated by this command: // -// mockgen -destination=modules/observability/domain/task/repo/mocks/Task.go -package=mocks github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/repo ITaskRepo +// mockgen -destination=mocks/Task.go -package=mocks . ITaskRepo // // Package mocks is a generated GoMock package. @@ -22,6 +22,7 @@ import ( type MockITaskRepo struct { ctrl *gomock.Controller recorder *MockITaskRepoMockRecorder + isgomock struct{} } // MockITaskRepoMockRecorder is the mock recorder for MockITaskRepo. @@ -41,126 +42,140 @@ func (m *MockITaskRepo) EXPECT() *MockITaskRepoMockRecorder { return m.recorder } +// AddNonFinalTask mocks base method. +func (m *MockITaskRepo) AddNonFinalTask(ctx context.Context, spaceID string, taskID int64) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddNonFinalTask", ctx, spaceID, taskID) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddNonFinalTask indicates an expected call of AddNonFinalTask. +func (mr *MockITaskRepoMockRecorder) AddNonFinalTask(ctx, spaceID, taskID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddNonFinalTask", reflect.TypeOf((*MockITaskRepo)(nil).AddNonFinalTask), ctx, spaceID, taskID) +} + // CreateTask mocks base method. -func (m *MockITaskRepo) CreateTask(arg0 context.Context, arg1 *entity.ObservabilityTask) (int64, error) { +func (m *MockITaskRepo) CreateTask(ctx context.Context, do *entity.ObservabilityTask) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateTask", arg0, arg1) + ret := m.ctrl.Call(m, "CreateTask", ctx, do) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateTask indicates an expected call of CreateTask. -func (mr *MockITaskRepoMockRecorder) CreateTask(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) CreateTask(ctx, do any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTask", reflect.TypeOf((*MockITaskRepo)(nil).CreateTask), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTask", reflect.TypeOf((*MockITaskRepo)(nil).CreateTask), ctx, do) } // CreateTaskRun mocks base method. -func (m *MockITaskRepo) CreateTaskRun(arg0 context.Context, arg1 *entity.TaskRun) (int64, error) { +func (m *MockITaskRepo) CreateTaskRun(ctx context.Context, do *entity.TaskRun) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateTaskRun", arg0, arg1) + ret := m.ctrl.Call(m, "CreateTaskRun", ctx, do) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateTaskRun indicates an expected call of CreateTaskRun. -func (mr *MockITaskRepoMockRecorder) CreateTaskRun(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) CreateTaskRun(ctx, do any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTaskRun", reflect.TypeOf((*MockITaskRepo)(nil).CreateTaskRun), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTaskRun", reflect.TypeOf((*MockITaskRepo)(nil).CreateTaskRun), ctx, do) } // DecrTaskCount mocks base method. -func (m *MockITaskRepo) DecrTaskCount(arg0 context.Context, arg1, arg2 int64) error { +func (m *MockITaskRepo) DecrTaskCount(ctx context.Context, taskID, ttl int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DecrTaskCount", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "DecrTaskCount", ctx, taskID, ttl) ret0, _ := ret[0].(error) return ret0 } // DecrTaskCount indicates an expected call of DecrTaskCount. -func (mr *MockITaskRepoMockRecorder) DecrTaskCount(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) DecrTaskCount(ctx, taskID, ttl any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskCount", reflect.TypeOf((*MockITaskRepo)(nil).DecrTaskCount), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskCount", reflect.TypeOf((*MockITaskRepo)(nil).DecrTaskCount), ctx, taskID, ttl) } // DecrTaskRunCount mocks base method. -func (m *MockITaskRepo) DecrTaskRunCount(arg0 context.Context, arg1, arg2, arg3 int64) error { +func (m *MockITaskRepo) DecrTaskRunCount(ctx context.Context, taskID, taskRunID, ttl int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DecrTaskRunCount", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "DecrTaskRunCount", ctx, taskID, taskRunID, ttl) ret0, _ := ret[0].(error) return ret0 } // DecrTaskRunCount indicates an expected call of DecrTaskRunCount. -func (mr *MockITaskRepoMockRecorder) DecrTaskRunCount(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) DecrTaskRunCount(ctx, taskID, taskRunID, ttl any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskRunCount", reflect.TypeOf((*MockITaskRepo)(nil).DecrTaskRunCount), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskRunCount", reflect.TypeOf((*MockITaskRepo)(nil).DecrTaskRunCount), ctx, taskID, taskRunID, ttl) } // DecrTaskRunSuccessCount mocks base method. -func (m *MockITaskRepo) DecrTaskRunSuccessCount(arg0 context.Context, arg1, arg2 int64) error { +func (m *MockITaskRepo) DecrTaskRunSuccessCount(ctx context.Context, taskID, taskRunID int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DecrTaskRunSuccessCount", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "DecrTaskRunSuccessCount", ctx, taskID, taskRunID) ret0, _ := ret[0].(error) return ret0 } // DecrTaskRunSuccessCount indicates an expected call of DecrTaskRunSuccessCount. -func (mr *MockITaskRepoMockRecorder) DecrTaskRunSuccessCount(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) DecrTaskRunSuccessCount(ctx, taskID, taskRunID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskRunSuccessCount", reflect.TypeOf((*MockITaskRepo)(nil).DecrTaskRunSuccessCount), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskRunSuccessCount", reflect.TypeOf((*MockITaskRepo)(nil).DecrTaskRunSuccessCount), ctx, taskID, taskRunID) } // DeleteTask mocks base method. -func (m *MockITaskRepo) DeleteTask(arg0 context.Context, arg1 *entity.ObservabilityTask) error { +func (m *MockITaskRepo) DeleteTask(ctx context.Context, do *entity.ObservabilityTask) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteTask", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteTask", ctx, do) ret0, _ := ret[0].(error) return ret0 } // DeleteTask indicates an expected call of DeleteTask. -func (mr *MockITaskRepoMockRecorder) DeleteTask(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) DeleteTask(ctx, do any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTask", reflect.TypeOf((*MockITaskRepo)(nil).DeleteTask), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTask", reflect.TypeOf((*MockITaskRepo)(nil).DeleteTask), ctx, do) } // GetBackfillTaskRun mocks base method. -func (m *MockITaskRepo) GetBackfillTaskRun(arg0 context.Context, arg1 *int64, arg2 int64) (*entity.TaskRun, error) { +func (m *MockITaskRepo) GetBackfillTaskRun(ctx context.Context, workspaceID *int64, taskID int64) (*entity.TaskRun, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetBackfillTaskRun", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetBackfillTaskRun", ctx, workspaceID, taskID) ret0, _ := ret[0].(*entity.TaskRun) ret1, _ := ret[1].(error) return ret0, ret1 } // GetBackfillTaskRun indicates an expected call of GetBackfillTaskRun. -func (mr *MockITaskRepoMockRecorder) GetBackfillTaskRun(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) GetBackfillTaskRun(ctx, workspaceID, taskID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBackfillTaskRun", reflect.TypeOf((*MockITaskRepo)(nil).GetBackfillTaskRun), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBackfillTaskRun", reflect.TypeOf((*MockITaskRepo)(nil).GetBackfillTaskRun), ctx, workspaceID, taskID) } // GetLatestNewDataTaskRun mocks base method. -func (m *MockITaskRepo) GetLatestNewDataTaskRun(arg0 context.Context, arg1 *int64, arg2 int64) (*entity.TaskRun, error) { +func (m *MockITaskRepo) GetLatestNewDataTaskRun(ctx context.Context, workspaceID *int64, taskID int64) (*entity.TaskRun, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLatestNewDataTaskRun", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetLatestNewDataTaskRun", ctx, workspaceID, taskID) ret0, _ := ret[0].(*entity.TaskRun) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLatestNewDataTaskRun indicates an expected call of GetLatestNewDataTaskRun. -func (mr *MockITaskRepoMockRecorder) GetLatestNewDataTaskRun(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) GetLatestNewDataTaskRun(ctx, workspaceID, taskID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestNewDataTaskRun", reflect.TypeOf((*MockITaskRepo)(nil).GetLatestNewDataTaskRun), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestNewDataTaskRun", reflect.TypeOf((*MockITaskRepo)(nil).GetLatestNewDataTaskRun), ctx, workspaceID, taskID) } // GetObjListWithTask mocks base method. -func (m *MockITaskRepo) GetObjListWithTask(arg0 context.Context) ([]string, []string, []*entity.ObservabilityTask) { +func (m *MockITaskRepo) GetObjListWithTask(ctx context.Context) ([]string, []string, []*entity.ObservabilityTask) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetObjListWithTask", arg0) + ret := m.ctrl.Call(m, "GetObjListWithTask", ctx) ret0, _ := ret[0].([]string) ret1, _ := ret[1].([]string) ret2, _ := ret[2].([]*entity.ObservabilityTask) @@ -168,161 +183,176 @@ func (m *MockITaskRepo) GetObjListWithTask(arg0 context.Context) ([]string, []st } // GetObjListWithTask indicates an expected call of GetObjListWithTask. -func (mr *MockITaskRepoMockRecorder) GetObjListWithTask(arg0 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) GetObjListWithTask(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetObjListWithTask", reflect.TypeOf((*MockITaskRepo)(nil).GetObjListWithTask), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetObjListWithTask", reflect.TypeOf((*MockITaskRepo)(nil).GetObjListWithTask), ctx) } // GetTask mocks base method. -func (m *MockITaskRepo) GetTask(arg0 context.Context, arg1 int64, arg2 *int64, arg3 *string) (*entity.ObservabilityTask, error) { +func (m *MockITaskRepo) GetTask(ctx context.Context, id int64, workspaceID *int64, userID *string) (*entity.ObservabilityTask, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTask", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "GetTask", ctx, id, workspaceID, userID) ret0, _ := ret[0].(*entity.ObservabilityTask) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTask indicates an expected call of GetTask. -func (mr *MockITaskRepoMockRecorder) GetTask(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) GetTask(ctx, id, workspaceID, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockITaskRepo)(nil).GetTask), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockITaskRepo)(nil).GetTask), ctx, id, workspaceID, userID) +} + +// GetTaskByRedis mocks base method. +func (m *MockITaskRepo) GetTaskByRedis(ctx context.Context, taskID int64) (*entity.ObservabilityTask, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetTaskByRedis", ctx, taskID) + ret0, _ := ret[0].(*entity.ObservabilityTask) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetTaskByRedis indicates an expected call of GetTaskByRedis. +func (mr *MockITaskRepoMockRecorder) GetTaskByRedis(ctx, taskID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskByRedis", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskByRedis), ctx, taskID) } // GetTaskCount mocks base method. -func (m *MockITaskRepo) GetTaskCount(arg0 context.Context, arg1 int64) (int64, error) { +func (m *MockITaskRepo) GetTaskCount(ctx context.Context, taskID int64) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTaskCount", arg0, arg1) + ret := m.ctrl.Call(m, "GetTaskCount", ctx, taskID) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTaskCount indicates an expected call of GetTaskCount. -func (mr *MockITaskRepoMockRecorder) GetTaskCount(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) GetTaskCount(ctx, taskID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskCount", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskCount), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskCount", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskCount), ctx, taskID) } // GetTaskRunCount mocks base method. -func (m *MockITaskRepo) GetTaskRunCount(arg0 context.Context, arg1, arg2 int64) (int64, error) { +func (m *MockITaskRepo) GetTaskRunCount(ctx context.Context, taskID, taskRunID int64) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTaskRunCount", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetTaskRunCount", ctx, taskID, taskRunID) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTaskRunCount indicates an expected call of GetTaskRunCount. -func (mr *MockITaskRepoMockRecorder) GetTaskRunCount(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) GetTaskRunCount(ctx, taskID, taskRunID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunCount", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskRunCount), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunCount", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskRunCount), ctx, taskID, taskRunID) } // GetTaskRunFailCount mocks base method. -func (m *MockITaskRepo) GetTaskRunFailCount(arg0 context.Context, arg1, arg2 int64) (int64, error) { +func (m *MockITaskRepo) GetTaskRunFailCount(ctx context.Context, taskID, taskRunID int64) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTaskRunFailCount", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetTaskRunFailCount", ctx, taskID, taskRunID) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTaskRunFailCount indicates an expected call of GetTaskRunFailCount. -func (mr *MockITaskRepoMockRecorder) GetTaskRunFailCount(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) GetTaskRunFailCount(ctx, taskID, taskRunID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunFailCount", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskRunFailCount), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunFailCount", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskRunFailCount), ctx, taskID, taskRunID) } // GetTaskRunSuccessCount mocks base method. -func (m *MockITaskRepo) GetTaskRunSuccessCount(arg0 context.Context, arg1, arg2 int64) (int64, error) { +func (m *MockITaskRepo) GetTaskRunSuccessCount(ctx context.Context, taskID, taskRunID int64) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTaskRunSuccessCount", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetTaskRunSuccessCount", ctx, taskID, taskRunID) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTaskRunSuccessCount indicates an expected call of GetTaskRunSuccessCount. -func (mr *MockITaskRepoMockRecorder) GetTaskRunSuccessCount(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) GetTaskRunSuccessCount(ctx, taskID, taskRunID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunSuccessCount", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskRunSuccessCount), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunSuccessCount", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskRunSuccessCount), ctx, taskID, taskRunID) } // IncrTaskCount mocks base method. -func (m *MockITaskRepo) IncrTaskCount(arg0 context.Context, arg1, arg2 int64) error { +func (m *MockITaskRepo) IncrTaskCount(ctx context.Context, taskID, ttl int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "IncrTaskCount", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "IncrTaskCount", ctx, taskID, ttl) ret0, _ := ret[0].(error) return ret0 } // IncrTaskCount indicates an expected call of IncrTaskCount. -func (mr *MockITaskRepoMockRecorder) IncrTaskCount(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) IncrTaskCount(ctx, taskID, ttl any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskCount", reflect.TypeOf((*MockITaskRepo)(nil).IncrTaskCount), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskCount", reflect.TypeOf((*MockITaskRepo)(nil).IncrTaskCount), ctx, taskID, ttl) } // IncrTaskRunCount mocks base method. -func (m *MockITaskRepo) IncrTaskRunCount(arg0 context.Context, arg1, arg2, arg3 int64) error { +func (m *MockITaskRepo) IncrTaskRunCount(ctx context.Context, taskID, taskRunID, ttl int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "IncrTaskRunCount", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "IncrTaskRunCount", ctx, taskID, taskRunID, ttl) ret0, _ := ret[0].(error) return ret0 } // IncrTaskRunCount indicates an expected call of IncrTaskRunCount. -func (mr *MockITaskRepoMockRecorder) IncrTaskRunCount(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) IncrTaskRunCount(ctx, taskID, taskRunID, ttl any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunCount", reflect.TypeOf((*MockITaskRepo)(nil).IncrTaskRunCount), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunCount", reflect.TypeOf((*MockITaskRepo)(nil).IncrTaskRunCount), ctx, taskID, taskRunID, ttl) } // IncrTaskRunFailCount mocks base method. -func (m *MockITaskRepo) IncrTaskRunFailCount(arg0 context.Context, arg1, arg2, arg3 int64) error { +func (m *MockITaskRepo) IncrTaskRunFailCount(ctx context.Context, taskID, taskRunID, ttl int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "IncrTaskRunFailCount", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "IncrTaskRunFailCount", ctx, taskID, taskRunID, ttl) ret0, _ := ret[0].(error) return ret0 } // IncrTaskRunFailCount indicates an expected call of IncrTaskRunFailCount. -func (mr *MockITaskRepoMockRecorder) IncrTaskRunFailCount(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) IncrTaskRunFailCount(ctx, taskID, taskRunID, ttl any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunFailCount", reflect.TypeOf((*MockITaskRepo)(nil).IncrTaskRunFailCount), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunFailCount", reflect.TypeOf((*MockITaskRepo)(nil).IncrTaskRunFailCount), ctx, taskID, taskRunID, ttl) } // IncrTaskRunSuccessCount mocks base method. -func (m *MockITaskRepo) IncrTaskRunSuccessCount(arg0 context.Context, arg1, arg2, arg3 int64) error { +func (m *MockITaskRepo) IncrTaskRunSuccessCount(ctx context.Context, taskID, taskRunID, ttl int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "IncrTaskRunSuccessCount", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "IncrTaskRunSuccessCount", ctx, taskID, taskRunID, ttl) ret0, _ := ret[0].(error) return ret0 } // IncrTaskRunSuccessCount indicates an expected call of IncrTaskRunSuccessCount. -func (mr *MockITaskRepoMockRecorder) IncrTaskRunSuccessCount(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) IncrTaskRunSuccessCount(ctx, taskID, taskRunID, ttl any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunSuccessCount", reflect.TypeOf((*MockITaskRepo)(nil).IncrTaskRunSuccessCount), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunSuccessCount", reflect.TypeOf((*MockITaskRepo)(nil).IncrTaskRunSuccessCount), ctx, taskID, taskRunID, ttl) } // ListNonFinalTask mocks base method. -func (m *MockITaskRepo) ListNonFinalTask(arg0 context.Context, arg1 string) ([]int64, error) { +func (m *MockITaskRepo) ListNonFinalTask(ctx context.Context, spaceID string) ([]int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListNonFinalTask", arg0, arg1) + ret := m.ctrl.Call(m, "ListNonFinalTask", ctx, spaceID) ret0, _ := ret[0].([]int64) ret1, _ := ret[1].(error) return ret0, ret1 } // ListNonFinalTask indicates an expected call of ListNonFinalTask. -func (mr *MockITaskRepoMockRecorder) ListNonFinalTask(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) ListNonFinalTask(ctx, spaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListNonFinalTask", reflect.TypeOf((*MockITaskRepo)(nil).ListNonFinalTask), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListNonFinalTask", reflect.TypeOf((*MockITaskRepo)(nil).ListNonFinalTask), ctx, spaceID) } // ListTasks mocks base method. -func (m *MockITaskRepo) ListTasks(arg0 context.Context, arg1 mysql.ListTaskParam) ([]*entity.ObservabilityTask, int64, error) { +func (m *MockITaskRepo) ListTasks(ctx context.Context, param mysql.ListTaskParam) ([]*entity.ObservabilityTask, int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListTasks", arg0, arg1) + ret := m.ctrl.Call(m, "ListTasks", ctx, param) ret0, _ := ret[0].([]*entity.ObservabilityTask) ret1, _ := ret[1].(int64) ret2, _ := ret[2].(error) @@ -330,120 +360,91 @@ func (m *MockITaskRepo) ListTasks(arg0 context.Context, arg1 mysql.ListTaskParam } // ListTasks indicates an expected call of ListTasks. -func (mr *MockITaskRepoMockRecorder) ListTasks(arg0, arg1 any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTasks", reflect.TypeOf((*MockITaskRepo)(nil).ListTasks), arg0, arg1) -} - -// AddNonFinalTask mocks base method. -func (m *MockITaskRepo) AddNonFinalTask(arg0 context.Context, arg1 string, arg2 int64) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "AddNonFinalTask", arg0, arg1, arg2) - ret0, _ := ret[0].(error) - return ret0 -} - -// AddNonFinalTask indicates an expected call of AddNonFinalTask. -func (mr *MockITaskRepoMockRecorder) AddNonFinalTask(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) ListTasks(ctx, param any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddNonFinalTask", reflect.TypeOf((*MockITaskRepo)(nil).AddNonFinalTask), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTasks", reflect.TypeOf((*MockITaskRepo)(nil).ListTasks), ctx, param) } // RemoveNonFinalTask mocks base method. -func (m *MockITaskRepo) RemoveNonFinalTask(arg0 context.Context, arg1 string, arg2 int64) error { +func (m *MockITaskRepo) RemoveNonFinalTask(ctx context.Context, spaceID string, taskID int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "RemoveNonFinalTask", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "RemoveNonFinalTask", ctx, spaceID, taskID) ret0, _ := ret[0].(error) return ret0 } // RemoveNonFinalTask indicates an expected call of RemoveNonFinalTask. -func (mr *MockITaskRepoMockRecorder) RemoveNonFinalTask(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) RemoveNonFinalTask(ctx, spaceID, taskID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveNonFinalTask", reflect.TypeOf((*MockITaskRepo)(nil).RemoveNonFinalTask), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveNonFinalTask", reflect.TypeOf((*MockITaskRepo)(nil).RemoveNonFinalTask), ctx, spaceID, taskID) } -// GetTaskByRedis mocks base method. -func (m *MockITaskRepo) GetTaskByRedis(arg0 context.Context, arg1 int64) (*entity.ObservabilityTask, error) { +// SetTask mocks base method. +func (m *MockITaskRepo) SetTask(ctx context.Context, task *entity.ObservabilityTask) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTaskByRedis", arg0, arg1) - ret0, _ := ret[0].(*entity.ObservabilityTask) - ret1, _ := ret[1].(error) - return ret0, ret1 + ret := m.ctrl.Call(m, "SetTask", ctx, task) + ret0, _ := ret[0].(error) + return ret0 } -// GetTaskByRedis indicates an expected call of GetTaskByRedis. -func (mr *MockITaskRepoMockRecorder) GetTaskByRedis(arg0, arg1 any) *gomock.Call { +// SetTask indicates an expected call of SetTask. +func (mr *MockITaskRepoMockRecorder) SetTask(ctx, task any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskByRedis", reflect.TypeOf((*MockITaskRepo)(nil).GetTaskByRedis), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTask", reflect.TypeOf((*MockITaskRepo)(nil).SetTask), ctx, task) } // UpdateTask mocks base method. -func (m *MockITaskRepo) UpdateTask(arg0 context.Context, arg1 *entity.ObservabilityTask) error { +func (m *MockITaskRepo) UpdateTask(ctx context.Context, do *entity.ObservabilityTask) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTask", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTask", ctx, do) ret0, _ := ret[0].(error) return ret0 } // UpdateTask indicates an expected call of UpdateTask. -func (mr *MockITaskRepoMockRecorder) UpdateTask(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) UpdateTask(ctx, do any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTask", reflect.TypeOf((*MockITaskRepo)(nil).UpdateTask), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTask", reflect.TypeOf((*MockITaskRepo)(nil).UpdateTask), ctx, do) } // UpdateTaskRun mocks base method. -func (m *MockITaskRepo) UpdateTaskRun(arg0 context.Context, arg1 *entity.TaskRun) error { +func (m *MockITaskRepo) UpdateTaskRun(ctx context.Context, do *entity.TaskRun) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTaskRun", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTaskRun", ctx, do) ret0, _ := ret[0].(error) return ret0 } // UpdateTaskRun indicates an expected call of UpdateTaskRun. -func (mr *MockITaskRepoMockRecorder) UpdateTaskRun(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) UpdateTaskRun(ctx, do any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskRun", reflect.TypeOf((*MockITaskRepo)(nil).UpdateTaskRun), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskRun", reflect.TypeOf((*MockITaskRepo)(nil).UpdateTaskRun), ctx, do) } // UpdateTaskRunWithOCC mocks base method. -func (m *MockITaskRepo) UpdateTaskRunWithOCC(arg0 context.Context, arg1, arg2 int64, arg3 map[string]any) error { +func (m *MockITaskRepo) UpdateTaskRunWithOCC(ctx context.Context, id, workspaceID int64, updateMap map[string]any) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTaskRunWithOCC", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "UpdateTaskRunWithOCC", ctx, id, workspaceID, updateMap) ret0, _ := ret[0].(error) return ret0 } // UpdateTaskRunWithOCC indicates an expected call of UpdateTaskRunWithOCC. -func (mr *MockITaskRepoMockRecorder) UpdateTaskRunWithOCC(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) UpdateTaskRunWithOCC(ctx, id, workspaceID, updateMap any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskRunWithOCC", reflect.TypeOf((*MockITaskRepo)(nil).UpdateTaskRunWithOCC), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskRunWithOCC", reflect.TypeOf((*MockITaskRepo)(nil).UpdateTaskRunWithOCC), ctx, id, workspaceID, updateMap) } // UpdateTaskWithOCC mocks base method. -func (m *MockITaskRepo) UpdateTaskWithOCC(arg0 context.Context, arg1, arg2 int64, arg3 map[string]any) error { +func (m *MockITaskRepo) UpdateTaskWithOCC(ctx context.Context, id, workspaceID int64, updateMap map[string]any) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTaskWithOCC", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "UpdateTaskWithOCC", ctx, id, workspaceID, updateMap) ret0, _ := ret[0].(error) return ret0 } // UpdateTaskWithOCC indicates an expected call of UpdateTaskWithOCC. -func (mr *MockITaskRepoMockRecorder) UpdateTaskWithOCC(arg0, arg1, arg2, arg3 any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskWithOCC", reflect.TypeOf((*MockITaskRepo)(nil).UpdateTaskWithOCC), arg0, arg1, arg2, arg3) -} - -// SetTask mocks base method. -func (m *MockITaskRepo) SetTask(arg0 context.Context, arg1 *entity.ObservabilityTask) error { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SetTask", arg0, arg1) - ret0, _ := ret[0].(error) - return ret0 -} - -// SetTask indicates an expected call of SetTask. -func (mr *MockITaskRepoMockRecorder) SetTask(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskRepoMockRecorder) UpdateTaskWithOCC(ctx, id, workspaceID, updateMap any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTask", reflect.TypeOf((*MockITaskRepo)(nil).SetTask), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskWithOCC", reflect.TypeOf((*MockITaskRepo)(nil).UpdateTaskWithOCC), ctx, id, workspaceID, updateMap) } diff --git a/backend/modules/observability/domain/task/service/mocks/task_service.go b/backend/modules/observability/domain/task/service/mocks/task_service.go index 992300806..ab653fed2 100644 --- a/backend/modules/observability/domain/task/service/mocks/task_service.go +++ b/backend/modules/observability/domain/task/service/mocks/task_service.go @@ -21,6 +21,7 @@ import ( type MockITaskService struct { ctrl *gomock.Controller recorder *MockITaskServiceMockRecorder + isgomock struct{} } // MockITaskServiceMockRecorder is the mock recorder for MockITaskService. @@ -41,75 +42,75 @@ func (m *MockITaskService) EXPECT() *MockITaskServiceMockRecorder { } // CheckTaskName mocks base method. -func (m *MockITaskService) CheckTaskName(arg0 context.Context, arg1 *service.CheckTaskNameReq) (*service.CheckTaskNameResp, error) { +func (m *MockITaskService) CheckTaskName(ctx context.Context, req *service.CheckTaskNameReq) (*service.CheckTaskNameResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CheckTaskName", arg0, arg1) + ret := m.ctrl.Call(m, "CheckTaskName", ctx, req) ret0, _ := ret[0].(*service.CheckTaskNameResp) ret1, _ := ret[1].(error) return ret0, ret1 } // CheckTaskName indicates an expected call of CheckTaskName. -func (mr *MockITaskServiceMockRecorder) CheckTaskName(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskServiceMockRecorder) CheckTaskName(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckTaskName", reflect.TypeOf((*MockITaskService)(nil).CheckTaskName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckTaskName", reflect.TypeOf((*MockITaskService)(nil).CheckTaskName), ctx, req) } // CreateTask mocks base method. -func (m *MockITaskService) CreateTask(arg0 context.Context, arg1 *service.CreateTaskReq) (*service.CreateTaskResp, error) { +func (m *MockITaskService) CreateTask(ctx context.Context, req *service.CreateTaskReq) (*service.CreateTaskResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateTask", arg0, arg1) + ret := m.ctrl.Call(m, "CreateTask", ctx, req) ret0, _ := ret[0].(*service.CreateTaskResp) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateTask indicates an expected call of CreateTask. -func (mr *MockITaskServiceMockRecorder) CreateTask(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskServiceMockRecorder) CreateTask(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTask", reflect.TypeOf((*MockITaskService)(nil).CreateTask), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTask", reflect.TypeOf((*MockITaskService)(nil).CreateTask), ctx, req) } // GetTask mocks base method. -func (m *MockITaskService) GetTask(arg0 context.Context, arg1 *service.GetTaskReq) (*service.GetTaskResp, error) { +func (m *MockITaskService) GetTask(ctx context.Context, req *service.GetTaskReq) (*service.GetTaskResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTask", arg0, arg1) + ret := m.ctrl.Call(m, "GetTask", ctx, req) ret0, _ := ret[0].(*service.GetTaskResp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTask indicates an expected call of GetTask. -func (mr *MockITaskServiceMockRecorder) GetTask(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskServiceMockRecorder) GetTask(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockITaskService)(nil).GetTask), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockITaskService)(nil).GetTask), ctx, req) } // ListTasks mocks base method. -func (m *MockITaskService) ListTasks(arg0 context.Context, arg1 *service.ListTasksReq) (*service.ListTasksResp, error) { +func (m *MockITaskService) ListTasks(ctx context.Context, req *service.ListTasksReq) (*service.ListTasksResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListTasks", arg0, arg1) + ret := m.ctrl.Call(m, "ListTasks", ctx, req) ret0, _ := ret[0].(*service.ListTasksResp) ret1, _ := ret[1].(error) return ret0, ret1 } // ListTasks indicates an expected call of ListTasks. -func (mr *MockITaskServiceMockRecorder) ListTasks(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskServiceMockRecorder) ListTasks(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTasks", reflect.TypeOf((*MockITaskService)(nil).ListTasks), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTasks", reflect.TypeOf((*MockITaskService)(nil).ListTasks), ctx, req) } // UpdateTask mocks base method. -func (m *MockITaskService) UpdateTask(arg0 context.Context, arg1 *service.UpdateTaskReq) error { +func (m *MockITaskService) UpdateTask(ctx context.Context, req *service.UpdateTaskReq) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTask", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTask", ctx, req) ret0, _ := ret[0].(error) return ret0 } // UpdateTask indicates an expected call of UpdateTask. -func (mr *MockITaskServiceMockRecorder) UpdateTask(arg0, arg1 any) *gomock.Call { +func (mr *MockITaskServiceMockRecorder) UpdateTask(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTask", reflect.TypeOf((*MockITaskService)(nil).UpdateTask), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTask", reflect.TypeOf((*MockITaskService)(nil).UpdateTask), ctx, req) } diff --git a/backend/modules/observability/domain/task/service/taskexe/tracehub/mocks/trace_hub_service.go b/backend/modules/observability/domain/task/service/taskexe/tracehub/mocks/trace_hub_service.go index a46b07d6b..c9eb98743 100644 --- a/backend/modules/observability/domain/task/service/taskexe/tracehub/mocks/trace_hub_service.go +++ b/backend/modules/observability/domain/task/service/taskexe/tracehub/mocks/trace_hub_service.go @@ -21,6 +21,7 @@ import ( type MockITraceHubService struct { ctrl *gomock.Controller recorder *MockITraceHubServiceMockRecorder + isgomock struct{} } // MockITraceHubServiceMockRecorder is the mock recorder for MockITraceHubService. @@ -41,57 +42,57 @@ func (m *MockITraceHubService) EXPECT() *MockITraceHubServiceMockRecorder { } // BackFill mocks base method. -func (m *MockITraceHubService) BackFill(arg0 context.Context, arg1 *entity.BackFillEvent) error { +func (m *MockITraceHubService) BackFill(ctx context.Context, event *entity.BackFillEvent) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BackFill", arg0, arg1) + ret := m.ctrl.Call(m, "BackFill", ctx, event) ret0, _ := ret[0].(error) return ret0 } // BackFill indicates an expected call of BackFill. -func (mr *MockITraceHubServiceMockRecorder) BackFill(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceHubServiceMockRecorder) BackFill(ctx, event any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BackFill", reflect.TypeOf((*MockITraceHubService)(nil).BackFill), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BackFill", reflect.TypeOf((*MockITraceHubService)(nil).BackFill), ctx, event) } // CallBack mocks base method. -func (m *MockITraceHubService) CallBack(arg0 context.Context, arg1 *entity.AutoEvalEvent) error { +func (m *MockITraceHubService) CallBack(ctx context.Context, event *entity.AutoEvalEvent) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CallBack", arg0, arg1) + ret := m.ctrl.Call(m, "CallBack", ctx, event) ret0, _ := ret[0].(error) return ret0 } // CallBack indicates an expected call of CallBack. -func (mr *MockITraceHubServiceMockRecorder) CallBack(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceHubServiceMockRecorder) CallBack(ctx, event any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CallBack", reflect.TypeOf((*MockITraceHubService)(nil).CallBack), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CallBack", reflect.TypeOf((*MockITraceHubService)(nil).CallBack), ctx, event) } // Correction mocks base method. -func (m *MockITraceHubService) Correction(arg0 context.Context, arg1 *entity.CorrectionEvent) error { +func (m *MockITraceHubService) Correction(ctx context.Context, event *entity.CorrectionEvent) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Correction", arg0, arg1) + ret := m.ctrl.Call(m, "Correction", ctx, event) ret0, _ := ret[0].(error) return ret0 } // Correction indicates an expected call of Correction. -func (mr *MockITraceHubServiceMockRecorder) Correction(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceHubServiceMockRecorder) Correction(ctx, event any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Correction", reflect.TypeOf((*MockITraceHubService)(nil).Correction), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Correction", reflect.TypeOf((*MockITraceHubService)(nil).Correction), ctx, event) } // SpanTrigger mocks base method. -func (m *MockITraceHubService) SpanTrigger(arg0 context.Context, arg1 *entity.RawSpan) error { +func (m *MockITraceHubService) SpanTrigger(ctx context.Context, event *entity.RawSpan) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SpanTrigger", arg0, arg1) + ret := m.ctrl.Call(m, "SpanTrigger", ctx, event) ret0, _ := ret[0].(error) return ret0 } // SpanTrigger indicates an expected call of SpanTrigger. -func (mr *MockITraceHubServiceMockRecorder) SpanTrigger(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceHubServiceMockRecorder) SpanTrigger(ctx, event any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SpanTrigger", reflect.TypeOf((*MockITraceHubService)(nil).SpanTrigger), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SpanTrigger", reflect.TypeOf((*MockITraceHubService)(nil).SpanTrigger), ctx, event) } diff --git a/backend/modules/observability/domain/trace/entity/collector/confmap/mocks/provider.go b/backend/modules/observability/domain/trace/entity/collector/confmap/mocks/provider.go new file mode 100644 index 000000000..d51b72206 --- /dev/null +++ b/backend/modules/observability/domain/trace/entity/collector/confmap/mocks/provider.go @@ -0,0 +1,85 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/collector/confmap (interfaces: Provider) +// +// Generated by this command: +// +// mockgen -destination=mocks/provider.go -package=mocks . Provider +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + confmap "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/collector/confmap" + gomock "go.uber.org/mock/gomock" +) + +// MockProvider is a mock of Provider interface. +type MockProvider struct { + ctrl *gomock.Controller + recorder *MockProviderMockRecorder + isgomock struct{} +} + +// MockProviderMockRecorder is the mock recorder for MockProvider. +type MockProviderMockRecorder struct { + mock *MockProvider +} + +// NewMockProvider creates a new mock instance. +func NewMockProvider(ctrl *gomock.Controller) *MockProvider { + mock := &MockProvider{ctrl: ctrl} + mock.recorder = &MockProviderMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockProvider) EXPECT() *MockProviderMockRecorder { + return m.recorder +} + +// Retrieve mocks base method. +func (m *MockProvider) Retrieve(ctx context.Context, path string) (*confmap.Retrieved, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Retrieve", ctx, path) + ret0, _ := ret[0].(*confmap.Retrieved) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Retrieve indicates an expected call of Retrieve. +func (mr *MockProviderMockRecorder) Retrieve(ctx, path any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Retrieve", reflect.TypeOf((*MockProvider)(nil).Retrieve), ctx, path) +} + +// Scheme mocks base method. +func (m *MockProvider) Scheme() string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Scheme") + ret0, _ := ret[0].(string) + return ret0 +} + +// Scheme indicates an expected call of Scheme. +func (mr *MockProviderMockRecorder) Scheme() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Scheme", reflect.TypeOf((*MockProvider)(nil).Scheme)) +} + +// Shutdown mocks base method. +func (m *MockProvider) Shutdown(ctx context.Context) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Shutdown", ctx) + ret0, _ := ret[0].(error) + return ret0 +} + +// Shutdown indicates an expected call of Shutdown. +func (mr *MockProviderMockRecorder) Shutdown(ctx any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Shutdown", reflect.TypeOf((*MockProvider)(nil).Shutdown), ctx) +} diff --git a/backend/modules/observability/domain/trace/entity/collector/mocks/conf_provider.go b/backend/modules/observability/domain/trace/entity/collector/mocks/conf_provider.go new file mode 100644 index 000000000..afcfa7256 --- /dev/null +++ b/backend/modules/observability/domain/trace/entity/collector/mocks/conf_provider.go @@ -0,0 +1,57 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/collector (interfaces: ConfigProvider) +// +// Generated by this command: +// +// mockgen -destination=mocks/conf_provider.go -package=mocks . ConfigProvider +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + collector "github.com/coze-dev/coze-loop/backend/modules/observability/domain/trace/entity/collector" + gomock "go.uber.org/mock/gomock" +) + +// MockConfigProvider is a mock of ConfigProvider interface. +type MockConfigProvider struct { + ctrl *gomock.Controller + recorder *MockConfigProviderMockRecorder + isgomock struct{} +} + +// MockConfigProviderMockRecorder is the mock recorder for MockConfigProvider. +type MockConfigProviderMockRecorder struct { + mock *MockConfigProvider +} + +// NewMockConfigProvider creates a new mock instance. +func NewMockConfigProvider(ctrl *gomock.Controller) *MockConfigProvider { + mock := &MockConfigProvider{ctrl: ctrl} + mock.recorder = &MockConfigProviderMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockConfigProvider) EXPECT() *MockConfigProviderMockRecorder { + return m.recorder +} + +// Get mocks base method. +func (m *MockConfigProvider) Get(ctx context.Context, factories collector.Factories) (*collector.Config, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Get", ctx, factories) + ret0, _ := ret[0].(*collector.Config) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Get indicates an expected call of Get. +func (mr *MockConfigProviderMockRecorder) Get(ctx, factories any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockConfigProvider)(nil).Get), ctx, factories) +} diff --git a/backend/modules/observability/domain/trace/repo/mocks/view.go b/backend/modules/observability/domain/trace/repo/mocks/view.go index 419109d6c..12aa19f8c 100644 --- a/backend/modules/observability/domain/trace/repo/mocks/view.go +++ b/backend/modules/observability/domain/trace/repo/mocks/view.go @@ -21,6 +21,7 @@ import ( type MockIViewRepo struct { ctrl *gomock.Controller recorder *MockIViewRepoMockRecorder + isgomock struct{} } // MockIViewRepoMockRecorder is the mock recorder for MockIViewRepo. @@ -41,74 +42,74 @@ func (m *MockIViewRepo) EXPECT() *MockIViewRepoMockRecorder { } // CreateView mocks base method. -func (m *MockIViewRepo) CreateView(arg0 context.Context, arg1 *entity.ObservabilityView) (int64, error) { +func (m *MockIViewRepo) CreateView(ctx context.Context, do *entity.ObservabilityView) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateView", arg0, arg1) + ret := m.ctrl.Call(m, "CreateView", ctx, do) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateView indicates an expected call of CreateView. -func (mr *MockIViewRepoMockRecorder) CreateView(arg0, arg1 any) *gomock.Call { +func (mr *MockIViewRepoMockRecorder) CreateView(ctx, do any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateView", reflect.TypeOf((*MockIViewRepo)(nil).CreateView), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateView", reflect.TypeOf((*MockIViewRepo)(nil).CreateView), ctx, do) } // DeleteView mocks base method. -func (m *MockIViewRepo) DeleteView(arg0 context.Context, arg1, arg2 int64, arg3 string) error { +func (m *MockIViewRepo) DeleteView(ctx context.Context, id, workspaceID int64, userID string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteView", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "DeleteView", ctx, id, workspaceID, userID) ret0, _ := ret[0].(error) return ret0 } // DeleteView indicates an expected call of DeleteView. -func (mr *MockIViewRepoMockRecorder) DeleteView(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockIViewRepoMockRecorder) DeleteView(ctx, id, workspaceID, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteView", reflect.TypeOf((*MockIViewRepo)(nil).DeleteView), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteView", reflect.TypeOf((*MockIViewRepo)(nil).DeleteView), ctx, id, workspaceID, userID) } // GetView mocks base method. -func (m *MockIViewRepo) GetView(arg0 context.Context, arg1 int64, arg2 *int64, arg3 *string) (*entity.ObservabilityView, error) { +func (m *MockIViewRepo) GetView(ctx context.Context, id int64, workspaceID *int64, userID *string) (*entity.ObservabilityView, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetView", arg0, arg1, arg2, arg3) + ret := m.ctrl.Call(m, "GetView", ctx, id, workspaceID, userID) ret0, _ := ret[0].(*entity.ObservabilityView) ret1, _ := ret[1].(error) return ret0, ret1 } // GetView indicates an expected call of GetView. -func (mr *MockIViewRepoMockRecorder) GetView(arg0, arg1, arg2, arg3 any) *gomock.Call { +func (mr *MockIViewRepoMockRecorder) GetView(ctx, id, workspaceID, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetView", reflect.TypeOf((*MockIViewRepo)(nil).GetView), arg0, arg1, arg2, arg3) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetView", reflect.TypeOf((*MockIViewRepo)(nil).GetView), ctx, id, workspaceID, userID) } // ListViews mocks base method. -func (m *MockIViewRepo) ListViews(arg0 context.Context, arg1 int64, arg2 string) ([]*entity.ObservabilityView, error) { +func (m *MockIViewRepo) ListViews(ctx context.Context, workspaceID int64, userID string) ([]*entity.ObservabilityView, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListViews", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "ListViews", ctx, workspaceID, userID) ret0, _ := ret[0].([]*entity.ObservabilityView) ret1, _ := ret[1].(error) return ret0, ret1 } // ListViews indicates an expected call of ListViews. -func (mr *MockIViewRepoMockRecorder) ListViews(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockIViewRepoMockRecorder) ListViews(ctx, workspaceID, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListViews", reflect.TypeOf((*MockIViewRepo)(nil).ListViews), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListViews", reflect.TypeOf((*MockIViewRepo)(nil).ListViews), ctx, workspaceID, userID) } // UpdateView mocks base method. -func (m *MockIViewRepo) UpdateView(arg0 context.Context, arg1 *entity.ObservabilityView) error { +func (m *MockIViewRepo) UpdateView(ctx context.Context, do *entity.ObservabilityView) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateView", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateView", ctx, do) ret0, _ := ret[0].(error) return ret0 } // UpdateView indicates an expected call of UpdateView. -func (mr *MockIViewRepoMockRecorder) UpdateView(arg0, arg1 any) *gomock.Call { +func (mr *MockIViewRepoMockRecorder) UpdateView(ctx, do any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateView", reflect.TypeOf((*MockIViewRepo)(nil).UpdateView), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateView", reflect.TypeOf((*MockIViewRepo)(nil).UpdateView), ctx, do) } diff --git a/backend/modules/observability/domain/trace/service/mocks/ingestion.go b/backend/modules/observability/domain/trace/service/mocks/ingestion.go index 397a9493c..9010a34ad 100644 --- a/backend/modules/observability/domain/trace/service/mocks/ingestion.go +++ b/backend/modules/observability/domain/trace/service/mocks/ingestion.go @@ -20,6 +20,7 @@ import ( type MockIngestionService struct { ctrl *gomock.Controller recorder *MockIngestionServiceMockRecorder + isgomock struct{} } // MockIngestionServiceMockRecorder is the mock recorder for MockIngestionService. @@ -40,27 +41,27 @@ func (m *MockIngestionService) EXPECT() *MockIngestionServiceMockRecorder { } // RunAsync mocks base method. -func (m *MockIngestionService) RunAsync(arg0 context.Context) { +func (m *MockIngestionService) RunAsync(ctx context.Context) { m.ctrl.T.Helper() - m.ctrl.Call(m, "RunAsync", arg0) + m.ctrl.Call(m, "RunAsync", ctx) } // RunAsync indicates an expected call of RunAsync. -func (mr *MockIngestionServiceMockRecorder) RunAsync(arg0 any) *gomock.Call { +func (mr *MockIngestionServiceMockRecorder) RunAsync(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunAsync", reflect.TypeOf((*MockIngestionService)(nil).RunAsync), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunAsync", reflect.TypeOf((*MockIngestionService)(nil).RunAsync), ctx) } // RunSync mocks base method. -func (m *MockIngestionService) RunSync(arg0 context.Context) error { +func (m *MockIngestionService) RunSync(ctx context.Context) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "RunSync", arg0) + ret := m.ctrl.Call(m, "RunSync", ctx) ret0, _ := ret[0].(error) return ret0 } // RunSync indicates an expected call of RunSync. -func (mr *MockIngestionServiceMockRecorder) RunSync(arg0 any) *gomock.Call { +func (mr *MockIngestionServiceMockRecorder) RunSync(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunSync", reflect.TypeOf((*MockIngestionService)(nil).RunSync), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunSync", reflect.TypeOf((*MockIngestionService)(nil).RunSync), ctx) } diff --git a/backend/modules/observability/domain/trace/service/mocks/span_processor.go b/backend/modules/observability/domain/trace/service/mocks/span_processor.go index adcda3d67..1241cd947 100644 --- a/backend/modules/observability/domain/trace/service/mocks/span_processor.go +++ b/backend/modules/observability/domain/trace/service/mocks/span_processor.go @@ -23,6 +23,7 @@ import ( type MockTraceFilterProcessorBuilder struct { ctrl *gomock.Controller recorder *MockTraceFilterProcessorBuilderMockRecorder + isgomock struct{} } // MockTraceFilterProcessorBuilderMockRecorder is the mock recorder for MockTraceFilterProcessorBuilder. diff --git a/backend/modules/observability/domain/trace/service/mocks/trace_export_service_mock.go b/backend/modules/observability/domain/trace/service/mocks/trace_export_service_mock.go index eb5b009e4..79734a876 100644 --- a/backend/modules/observability/domain/trace/service/mocks/trace_export_service_mock.go +++ b/backend/modules/observability/domain/trace/service/mocks/trace_export_service_mock.go @@ -21,6 +21,7 @@ import ( type MockITraceExportService struct { ctrl *gomock.Controller recorder *MockITraceExportServiceMockRecorder + isgomock struct{} } // MockITraceExportServiceMockRecorder is the mock recorder for MockITraceExportService. @@ -41,31 +42,31 @@ func (m *MockITraceExportService) EXPECT() *MockITraceExportServiceMockRecorder } // ExportTracesToDataset mocks base method. -func (m *MockITraceExportService) ExportTracesToDataset(arg0 context.Context, arg1 *service.ExportTracesToDatasetRequest) (*service.ExportTracesToDatasetResponse, error) { +func (m *MockITraceExportService) ExportTracesToDataset(ctx context.Context, req *service.ExportTracesToDatasetRequest) (*service.ExportTracesToDatasetResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ExportTracesToDataset", arg0, arg1) + ret := m.ctrl.Call(m, "ExportTracesToDataset", ctx, req) ret0, _ := ret[0].(*service.ExportTracesToDatasetResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // ExportTracesToDataset indicates an expected call of ExportTracesToDataset. -func (mr *MockITraceExportServiceMockRecorder) ExportTracesToDataset(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceExportServiceMockRecorder) ExportTracesToDataset(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExportTracesToDataset", reflect.TypeOf((*MockITraceExportService)(nil).ExportTracesToDataset), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExportTracesToDataset", reflect.TypeOf((*MockITraceExportService)(nil).ExportTracesToDataset), ctx, req) } // PreviewExportTracesToDataset mocks base method. -func (m *MockITraceExportService) PreviewExportTracesToDataset(arg0 context.Context, arg1 *service.ExportTracesToDatasetRequest) (*service.PreviewExportTracesToDatasetResponse, error) { +func (m *MockITraceExportService) PreviewExportTracesToDataset(ctx context.Context, req *service.ExportTracesToDatasetRequest) (*service.PreviewExportTracesToDatasetResponse, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PreviewExportTracesToDataset", arg0, arg1) + ret := m.ctrl.Call(m, "PreviewExportTracesToDataset", ctx, req) ret0, _ := ret[0].(*service.PreviewExportTracesToDatasetResponse) ret1, _ := ret[1].(error) return ret0, ret1 } // PreviewExportTracesToDataset indicates an expected call of PreviewExportTracesToDataset. -func (mr *MockITraceExportServiceMockRecorder) PreviewExportTracesToDataset(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceExportServiceMockRecorder) PreviewExportTracesToDataset(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PreviewExportTracesToDataset", reflect.TypeOf((*MockITraceExportService)(nil).PreviewExportTracesToDataset), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PreviewExportTracesToDataset", reflect.TypeOf((*MockITraceExportService)(nil).PreviewExportTracesToDataset), ctx, req) } diff --git a/backend/modules/observability/domain/trace/service/mocks/trace_service.go b/backend/modules/observability/domain/trace/service/mocks/trace_service.go index fac7ba3c1..152f819f4 100644 --- a/backend/modules/observability/domain/trace/service/mocks/trace_service.go +++ b/backend/modules/observability/domain/trace/service/mocks/trace_service.go @@ -22,6 +22,7 @@ import ( type MockITraceService struct { ctrl *gomock.Controller recorder *MockITraceServiceMockRecorder + isgomock struct{} } // MockITraceServiceMockRecorder is the mock recorder for MockITraceService. @@ -42,250 +43,250 @@ func (m *MockITraceService) EXPECT() *MockITraceServiceMockRecorder { } // ChangeEvaluatorScore mocks base method. -func (m *MockITraceService) ChangeEvaluatorScore(arg0 context.Context, arg1 *service.ChangeEvaluatorScoreRequest) (*service.ChangeEvaluatorScoreResp, error) { +func (m *MockITraceService) ChangeEvaluatorScore(ctx context.Context, req *service.ChangeEvaluatorScoreRequest) (*service.ChangeEvaluatorScoreResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ChangeEvaluatorScore", arg0, arg1) + ret := m.ctrl.Call(m, "ChangeEvaluatorScore", ctx, req) ret0, _ := ret[0].(*service.ChangeEvaluatorScoreResp) ret1, _ := ret[1].(error) return ret0, ret1 } // ChangeEvaluatorScore indicates an expected call of ChangeEvaluatorScore. -func (mr *MockITraceServiceMockRecorder) ChangeEvaluatorScore(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) ChangeEvaluatorScore(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChangeEvaluatorScore", reflect.TypeOf((*MockITraceService)(nil).ChangeEvaluatorScore), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChangeEvaluatorScore", reflect.TypeOf((*MockITraceService)(nil).ChangeEvaluatorScore), ctx, req) } // CreateAnnotation mocks base method. -func (m *MockITraceService) CreateAnnotation(arg0 context.Context, arg1 *service.CreateAnnotationReq) error { +func (m *MockITraceService) CreateAnnotation(ctx context.Context, req *service.CreateAnnotationReq) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateAnnotation", arg0, arg1) + ret := m.ctrl.Call(m, "CreateAnnotation", ctx, req) ret0, _ := ret[0].(error) return ret0 } // CreateAnnotation indicates an expected call of CreateAnnotation. -func (mr *MockITraceServiceMockRecorder) CreateAnnotation(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) CreateAnnotation(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateAnnotation", reflect.TypeOf((*MockITraceService)(nil).CreateAnnotation), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateAnnotation", reflect.TypeOf((*MockITraceService)(nil).CreateAnnotation), ctx, req) } // CreateManualAnnotation mocks base method. -func (m *MockITraceService) CreateManualAnnotation(arg0 context.Context, arg1 *service.CreateManualAnnotationReq) (*service.CreateManualAnnotationResp, error) { +func (m *MockITraceService) CreateManualAnnotation(ctx context.Context, req *service.CreateManualAnnotationReq) (*service.CreateManualAnnotationResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CreateManualAnnotation", arg0, arg1) + ret := m.ctrl.Call(m, "CreateManualAnnotation", ctx, req) ret0, _ := ret[0].(*service.CreateManualAnnotationResp) ret1, _ := ret[1].(error) return ret0, ret1 } // CreateManualAnnotation indicates an expected call of CreateManualAnnotation. -func (mr *MockITraceServiceMockRecorder) CreateManualAnnotation(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) CreateManualAnnotation(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateManualAnnotation", reflect.TypeOf((*MockITraceService)(nil).CreateManualAnnotation), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateManualAnnotation", reflect.TypeOf((*MockITraceService)(nil).CreateManualAnnotation), ctx, req) } // DeleteAnnotation mocks base method. -func (m *MockITraceService) DeleteAnnotation(arg0 context.Context, arg1 *service.DeleteAnnotationReq) error { +func (m *MockITraceService) DeleteAnnotation(ctx context.Context, req *service.DeleteAnnotationReq) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteAnnotation", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteAnnotation", ctx, req) ret0, _ := ret[0].(error) return ret0 } // DeleteAnnotation indicates an expected call of DeleteAnnotation. -func (mr *MockITraceServiceMockRecorder) DeleteAnnotation(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) DeleteAnnotation(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAnnotation", reflect.TypeOf((*MockITraceService)(nil).DeleteAnnotation), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAnnotation", reflect.TypeOf((*MockITraceService)(nil).DeleteAnnotation), ctx, req) } // DeleteManualAnnotation mocks base method. -func (m *MockITraceService) DeleteManualAnnotation(arg0 context.Context, arg1 *service.DeleteManualAnnotationReq) error { +func (m *MockITraceService) DeleteManualAnnotation(ctx context.Context, req *service.DeleteManualAnnotationReq) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteManualAnnotation", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteManualAnnotation", ctx, req) ret0, _ := ret[0].(error) return ret0 } // DeleteManualAnnotation indicates an expected call of DeleteManualAnnotation. -func (mr *MockITraceServiceMockRecorder) DeleteManualAnnotation(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) DeleteManualAnnotation(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteManualAnnotation", reflect.TypeOf((*MockITraceService)(nil).DeleteManualAnnotation), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteManualAnnotation", reflect.TypeOf((*MockITraceService)(nil).DeleteManualAnnotation), ctx, req) } // ExtractSpanInfo mocks base method. -func (m *MockITraceService) ExtractSpanInfo(arg0 context.Context, arg1 *service.ExtractSpanInfoRequest) (*service.ExtractSpanInfoResp, error) { +func (m *MockITraceService) ExtractSpanInfo(ctx context.Context, req *service.ExtractSpanInfoRequest) (*service.ExtractSpanInfoResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ExtractSpanInfo", arg0, arg1) + ret := m.ctrl.Call(m, "ExtractSpanInfo", ctx, req) ret0, _ := ret[0].(*service.ExtractSpanInfoResp) ret1, _ := ret[1].(error) return ret0, ret1 } // ExtractSpanInfo indicates an expected call of ExtractSpanInfo. -func (mr *MockITraceServiceMockRecorder) ExtractSpanInfo(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) ExtractSpanInfo(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExtractSpanInfo", reflect.TypeOf((*MockITraceService)(nil).ExtractSpanInfo), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExtractSpanInfo", reflect.TypeOf((*MockITraceService)(nil).ExtractSpanInfo), ctx, req) } // GetTrace mocks base method. -func (m *MockITraceService) GetTrace(arg0 context.Context, arg1 *service.GetTraceReq) (*service.GetTraceResp, error) { +func (m *MockITraceService) GetTrace(ctx context.Context, req *service.GetTraceReq) (*service.GetTraceResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTrace", arg0, arg1) + ret := m.ctrl.Call(m, "GetTrace", ctx, req) ret0, _ := ret[0].(*service.GetTraceResp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTrace indicates an expected call of GetTrace. -func (mr *MockITraceServiceMockRecorder) GetTrace(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) GetTrace(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTrace", reflect.TypeOf((*MockITraceService)(nil).GetTrace), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTrace", reflect.TypeOf((*MockITraceService)(nil).GetTrace), ctx, req) } // GetTracesAdvanceInfo mocks base method. -func (m *MockITraceService) GetTracesAdvanceInfo(arg0 context.Context, arg1 *service.GetTracesAdvanceInfoReq) (*service.GetTracesAdvanceInfoResp, error) { +func (m *MockITraceService) GetTracesAdvanceInfo(ctx context.Context, req *service.GetTracesAdvanceInfoReq) (*service.GetTracesAdvanceInfoResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTracesAdvanceInfo", arg0, arg1) + ret := m.ctrl.Call(m, "GetTracesAdvanceInfo", ctx, req) ret0, _ := ret[0].(*service.GetTracesAdvanceInfoResp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTracesAdvanceInfo indicates an expected call of GetTracesAdvanceInfo. -func (mr *MockITraceServiceMockRecorder) GetTracesAdvanceInfo(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) GetTracesAdvanceInfo(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTracesAdvanceInfo", reflect.TypeOf((*MockITraceService)(nil).GetTracesAdvanceInfo), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTracesAdvanceInfo", reflect.TypeOf((*MockITraceService)(nil).GetTracesAdvanceInfo), ctx, req) } // GetTracesMetaInfo mocks base method. -func (m *MockITraceService) GetTracesMetaInfo(arg0 context.Context, arg1 *service.GetTracesMetaInfoReq) (*service.GetTracesMetaInfoResp, error) { +func (m *MockITraceService) GetTracesMetaInfo(ctx context.Context, req *service.GetTracesMetaInfoReq) (*service.GetTracesMetaInfoResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTracesMetaInfo", arg0, arg1) + ret := m.ctrl.Call(m, "GetTracesMetaInfo", ctx, req) ret0, _ := ret[0].(*service.GetTracesMetaInfoResp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTracesMetaInfo indicates an expected call of GetTracesMetaInfo. -func (mr *MockITraceServiceMockRecorder) GetTracesMetaInfo(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) GetTracesMetaInfo(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTracesMetaInfo", reflect.TypeOf((*MockITraceService)(nil).GetTracesMetaInfo), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTracesMetaInfo", reflect.TypeOf((*MockITraceService)(nil).GetTracesMetaInfo), ctx, req) } // IngestTraces mocks base method. -func (m *MockITraceService) IngestTraces(arg0 context.Context, arg1 *service.IngestTracesReq) error { +func (m *MockITraceService) IngestTraces(ctx context.Context, req *service.IngestTracesReq) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "IngestTraces", arg0, arg1) + ret := m.ctrl.Call(m, "IngestTraces", ctx, req) ret0, _ := ret[0].(error) return ret0 } // IngestTraces indicates an expected call of IngestTraces. -func (mr *MockITraceServiceMockRecorder) IngestTraces(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) IngestTraces(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IngestTraces", reflect.TypeOf((*MockITraceService)(nil).IngestTraces), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IngestTraces", reflect.TypeOf((*MockITraceService)(nil).IngestTraces), ctx, req) } // ListAnnotationEvaluators mocks base method. -func (m *MockITraceService) ListAnnotationEvaluators(arg0 context.Context, arg1 *service.ListAnnotationEvaluatorsRequest) (*service.ListAnnotationEvaluatorsResp, error) { +func (m *MockITraceService) ListAnnotationEvaluators(ctx context.Context, req *service.ListAnnotationEvaluatorsRequest) (*service.ListAnnotationEvaluatorsResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListAnnotationEvaluators", arg0, arg1) + ret := m.ctrl.Call(m, "ListAnnotationEvaluators", ctx, req) ret0, _ := ret[0].(*service.ListAnnotationEvaluatorsResp) ret1, _ := ret[1].(error) return ret0, ret1 } // ListAnnotationEvaluators indicates an expected call of ListAnnotationEvaluators. -func (mr *MockITraceServiceMockRecorder) ListAnnotationEvaluators(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) ListAnnotationEvaluators(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAnnotationEvaluators", reflect.TypeOf((*MockITraceService)(nil).ListAnnotationEvaluators), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAnnotationEvaluators", reflect.TypeOf((*MockITraceService)(nil).ListAnnotationEvaluators), ctx, req) } // ListAnnotations mocks base method. -func (m *MockITraceService) ListAnnotations(arg0 context.Context, arg1 *service.ListAnnotationsReq) (*service.ListAnnotationsResp, error) { +func (m *MockITraceService) ListAnnotations(ctx context.Context, req *service.ListAnnotationsReq) (*service.ListAnnotationsResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListAnnotations", arg0, arg1) + ret := m.ctrl.Call(m, "ListAnnotations", ctx, req) ret0, _ := ret[0].(*service.ListAnnotationsResp) ret1, _ := ret[1].(error) return ret0, ret1 } // ListAnnotations indicates an expected call of ListAnnotations. -func (mr *MockITraceServiceMockRecorder) ListAnnotations(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) ListAnnotations(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAnnotations", reflect.TypeOf((*MockITraceService)(nil).ListAnnotations), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAnnotations", reflect.TypeOf((*MockITraceService)(nil).ListAnnotations), ctx, req) } // ListSpans mocks base method. -func (m *MockITraceService) ListSpans(arg0 context.Context, arg1 *service.ListSpansReq) (*service.ListSpansResp, error) { +func (m *MockITraceService) ListSpans(ctx context.Context, req *service.ListSpansReq) (*service.ListSpansResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListSpans", arg0, arg1) + ret := m.ctrl.Call(m, "ListSpans", ctx, req) ret0, _ := ret[0].(*service.ListSpansResp) ret1, _ := ret[1].(error) return ret0, ret1 } // ListSpans indicates an expected call of ListSpans. -func (mr *MockITraceServiceMockRecorder) ListSpans(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) ListSpans(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSpans", reflect.TypeOf((*MockITraceService)(nil).ListSpans), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSpans", reflect.TypeOf((*MockITraceService)(nil).ListSpans), ctx, req) } // ListSpansOApi mocks base method. -func (m *MockITraceService) ListSpansOApi(arg0 context.Context, arg1 *service.ListSpansOApiReq) (*service.ListSpansOApiResp, error) { +func (m *MockITraceService) ListSpansOApi(ctx context.Context, req *service.ListSpansOApiReq) (*service.ListSpansOApiResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListSpansOApi", arg0, arg1) + ret := m.ctrl.Call(m, "ListSpansOApi", ctx, req) ret0, _ := ret[0].(*service.ListSpansOApiResp) ret1, _ := ret[1].(error) return ret0, ret1 } // ListSpansOApi indicates an expected call of ListSpansOApi. -func (mr *MockITraceServiceMockRecorder) ListSpansOApi(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) ListSpansOApi(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSpansOApi", reflect.TypeOf((*MockITraceService)(nil).ListSpansOApi), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListSpansOApi", reflect.TypeOf((*MockITraceService)(nil).ListSpansOApi), ctx, req) } // SearchTraceOApi mocks base method. -func (m *MockITraceService) SearchTraceOApi(arg0 context.Context, arg1 *service.SearchTraceOApiReq) (*service.SearchTraceOApiResp, error) { +func (m *MockITraceService) SearchTraceOApi(ctx context.Context, req *service.SearchTraceOApiReq) (*service.SearchTraceOApiResp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "SearchTraceOApi", arg0, arg1) + ret := m.ctrl.Call(m, "SearchTraceOApi", ctx, req) ret0, _ := ret[0].(*service.SearchTraceOApiResp) ret1, _ := ret[1].(error) return ret0, ret1 } // SearchTraceOApi indicates an expected call of SearchTraceOApi. -func (mr *MockITraceServiceMockRecorder) SearchTraceOApi(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) SearchTraceOApi(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchTraceOApi", reflect.TypeOf((*MockITraceService)(nil).SearchTraceOApi), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SearchTraceOApi", reflect.TypeOf((*MockITraceService)(nil).SearchTraceOApi), ctx, req) } // Send mocks base method. -func (m *MockITraceService) Send(arg0 context.Context, arg1 *entity.AnnotationEvent) error { +func (m *MockITraceService) Send(ctx context.Context, msg *entity.AnnotationEvent) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Send", arg0, arg1) + ret := m.ctrl.Call(m, "Send", ctx, msg) ret0, _ := ret[0].(error) return ret0 } // Send indicates an expected call of Send. -func (mr *MockITraceServiceMockRecorder) Send(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) Send(ctx, msg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockITraceService)(nil).Send), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockITraceService)(nil).Send), ctx, msg) } // UpdateManualAnnotation mocks base method. -func (m *MockITraceService) UpdateManualAnnotation(arg0 context.Context, arg1 *service.UpdateManualAnnotationReq) error { +func (m *MockITraceService) UpdateManualAnnotation(ctx context.Context, req *service.UpdateManualAnnotationReq) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateManualAnnotation", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateManualAnnotation", ctx, req) ret0, _ := ret[0].(error) return ret0 } // UpdateManualAnnotation indicates an expected call of UpdateManualAnnotation. -func (mr *MockITraceServiceMockRecorder) UpdateManualAnnotation(arg0, arg1 any) *gomock.Call { +func (mr *MockITraceServiceMockRecorder) UpdateManualAnnotation(ctx, req any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateManualAnnotation", reflect.TypeOf((*MockITraceService)(nil).UpdateManualAnnotation), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateManualAnnotation", reflect.TypeOf((*MockITraceService)(nil).UpdateManualAnnotation), ctx, req) } diff --git a/backend/modules/observability/infra/repo/mysql/mocks/task.go b/backend/modules/observability/infra/repo/mysql/mocks/task.go new file mode 100644 index 000000000..65c4a961c --- /dev/null +++ b/backend/modules/observability/infra/repo/mysql/mocks/task.go @@ -0,0 +1,148 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/modules/observability/infra/repo/mysql (interfaces: ITaskDao) +// +// Generated by this command: +// +// mockgen -destination=mocks/task.go -package=mocks . ITaskDao +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + mysql "github.com/coze-dev/coze-loop/backend/modules/observability/infra/repo/mysql" + model "github.com/coze-dev/coze-loop/backend/modules/observability/infra/repo/mysql/gorm_gen/model" + gomock "go.uber.org/mock/gomock" +) + +// MockITaskDao is a mock of ITaskDao interface. +type MockITaskDao struct { + ctrl *gomock.Controller + recorder *MockITaskDaoMockRecorder + isgomock struct{} +} + +// MockITaskDaoMockRecorder is the mock recorder for MockITaskDao. +type MockITaskDaoMockRecorder struct { + mock *MockITaskDao +} + +// NewMockITaskDao creates a new mock instance. +func NewMockITaskDao(ctrl *gomock.Controller) *MockITaskDao { + mock := &MockITaskDao{ctrl: ctrl} + mock.recorder = &MockITaskDaoMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockITaskDao) EXPECT() *MockITaskDaoMockRecorder { + return m.recorder +} + +// CreateTask mocks base method. +func (m *MockITaskDao) CreateTask(ctx context.Context, po *model.ObservabilityTask) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateTask", ctx, po) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateTask indicates an expected call of CreateTask. +func (mr *MockITaskDaoMockRecorder) CreateTask(ctx, po any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTask", reflect.TypeOf((*MockITaskDao)(nil).CreateTask), ctx, po) +} + +// DeleteTask mocks base method. +func (m *MockITaskDao) DeleteTask(ctx context.Context, id, workspaceID int64, userID string) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DeleteTask", ctx, id, workspaceID, userID) + ret0, _ := ret[0].(error) + return ret0 +} + +// DeleteTask indicates an expected call of DeleteTask. +func (mr *MockITaskDaoMockRecorder) DeleteTask(ctx, id, workspaceID, userID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTask", reflect.TypeOf((*MockITaskDao)(nil).DeleteTask), ctx, id, workspaceID, userID) +} + +// GetObjListWithTask mocks base method. +func (m *MockITaskDao) GetObjListWithTask(ctx context.Context) ([]string, []string, []*model.ObservabilityTask, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetObjListWithTask", ctx) + ret0, _ := ret[0].([]string) + ret1, _ := ret[1].([]string) + ret2, _ := ret[2].([]*model.ObservabilityTask) + ret3, _ := ret[3].(error) + return ret0, ret1, ret2, ret3 +} + +// GetObjListWithTask indicates an expected call of GetObjListWithTask. +func (mr *MockITaskDaoMockRecorder) GetObjListWithTask(ctx any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetObjListWithTask", reflect.TypeOf((*MockITaskDao)(nil).GetObjListWithTask), ctx) +} + +// GetTask mocks base method. +func (m *MockITaskDao) GetTask(ctx context.Context, id int64, workspaceID *int64, userID *string) (*model.ObservabilityTask, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetTask", ctx, id, workspaceID, userID) + ret0, _ := ret[0].(*model.ObservabilityTask) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetTask indicates an expected call of GetTask. +func (mr *MockITaskDaoMockRecorder) GetTask(ctx, id, workspaceID, userID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockITaskDao)(nil).GetTask), ctx, id, workspaceID, userID) +} + +// ListTasks mocks base method. +func (m *MockITaskDao) ListTasks(ctx context.Context, param mysql.ListTaskParam) ([]*model.ObservabilityTask, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ListTasks", ctx, param) + ret0, _ := ret[0].([]*model.ObservabilityTask) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// ListTasks indicates an expected call of ListTasks. +func (mr *MockITaskDaoMockRecorder) ListTasks(ctx, param any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTasks", reflect.TypeOf((*MockITaskDao)(nil).ListTasks), ctx, param) +} + +// UpdateTask mocks base method. +func (m *MockITaskDao) UpdateTask(ctx context.Context, po *model.ObservabilityTask) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateTask", ctx, po) + ret0, _ := ret[0].(error) + return ret0 +} + +// UpdateTask indicates an expected call of UpdateTask. +func (mr *MockITaskDaoMockRecorder) UpdateTask(ctx, po any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTask", reflect.TypeOf((*MockITaskDao)(nil).UpdateTask), ctx, po) +} + +// UpdateTaskWithOCC mocks base method. +func (m *MockITaskDao) UpdateTaskWithOCC(ctx context.Context, id, workspaceID int64, updateMap map[string]any) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateTaskWithOCC", ctx, id, workspaceID, updateMap) + ret0, _ := ret[0].(error) + return ret0 +} + +// UpdateTaskWithOCC indicates an expected call of UpdateTaskWithOCC. +func (mr *MockITaskDaoMockRecorder) UpdateTaskWithOCC(ctx, id, workspaceID, updateMap any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskWithOCC", reflect.TypeOf((*MockITaskDao)(nil).UpdateTaskWithOCC), ctx, id, workspaceID, updateMap) +} diff --git a/backend/modules/observability/infra/repo/mysql/mocks/task_run.go b/backend/modules/observability/infra/repo/mysql/mocks/task_run.go new file mode 100644 index 000000000..5176f8631 --- /dev/null +++ b/backend/modules/observability/infra/repo/mysql/mocks/task_run.go @@ -0,0 +1,132 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/modules/observability/infra/repo/mysql (interfaces: ITaskRunDao) +// +// Generated by this command: +// +// mockgen -destination=mocks/task_run.go -package=mocks . ITaskRunDao +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + mysql "github.com/coze-dev/coze-loop/backend/modules/observability/infra/repo/mysql" + model "github.com/coze-dev/coze-loop/backend/modules/observability/infra/repo/mysql/gorm_gen/model" + gomock "go.uber.org/mock/gomock" +) + +// MockITaskRunDao is a mock of ITaskRunDao interface. +type MockITaskRunDao struct { + ctrl *gomock.Controller + recorder *MockITaskRunDaoMockRecorder + isgomock struct{} +} + +// MockITaskRunDaoMockRecorder is the mock recorder for MockITaskRunDao. +type MockITaskRunDaoMockRecorder struct { + mock *MockITaskRunDao +} + +// NewMockITaskRunDao creates a new mock instance. +func NewMockITaskRunDao(ctrl *gomock.Controller) *MockITaskRunDao { + mock := &MockITaskRunDao{ctrl: ctrl} + mock.recorder = &MockITaskRunDaoMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockITaskRunDao) EXPECT() *MockITaskRunDaoMockRecorder { + return m.recorder +} + +// CreateTaskRun mocks base method. +func (m *MockITaskRunDao) CreateTaskRun(ctx context.Context, po *model.ObservabilityTaskRun) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CreateTaskRun", ctx, po) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateTaskRun indicates an expected call of CreateTaskRun. +func (mr *MockITaskRunDaoMockRecorder) CreateTaskRun(ctx, po any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTaskRun", reflect.TypeOf((*MockITaskRunDao)(nil).CreateTaskRun), ctx, po) +} + +// GetBackfillTaskRun mocks base method. +func (m *MockITaskRunDao) GetBackfillTaskRun(ctx context.Context, workspaceID *int64, taskID int64) (*model.ObservabilityTaskRun, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetBackfillTaskRun", ctx, workspaceID, taskID) + ret0, _ := ret[0].(*model.ObservabilityTaskRun) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetBackfillTaskRun indicates an expected call of GetBackfillTaskRun. +func (mr *MockITaskRunDaoMockRecorder) GetBackfillTaskRun(ctx, workspaceID, taskID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBackfillTaskRun", reflect.TypeOf((*MockITaskRunDao)(nil).GetBackfillTaskRun), ctx, workspaceID, taskID) +} + +// GetLatestNewDataTaskRun mocks base method. +func (m *MockITaskRunDao) GetLatestNewDataTaskRun(ctx context.Context, workspaceID *int64, taskID int64) (*model.ObservabilityTaskRun, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetLatestNewDataTaskRun", ctx, workspaceID, taskID) + ret0, _ := ret[0].(*model.ObservabilityTaskRun) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetLatestNewDataTaskRun indicates an expected call of GetLatestNewDataTaskRun. +func (mr *MockITaskRunDaoMockRecorder) GetLatestNewDataTaskRun(ctx, workspaceID, taskID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestNewDataTaskRun", reflect.TypeOf((*MockITaskRunDao)(nil).GetLatestNewDataTaskRun), ctx, workspaceID, taskID) +} + +// ListTaskRuns mocks base method. +func (m *MockITaskRunDao) ListTaskRuns(ctx context.Context, param mysql.ListTaskRunParam) ([]*model.ObservabilityTaskRun, int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ListTaskRuns", ctx, param) + ret0, _ := ret[0].([]*model.ObservabilityTaskRun) + ret1, _ := ret[1].(int64) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// ListTaskRuns indicates an expected call of ListTaskRuns. +func (mr *MockITaskRunDaoMockRecorder) ListTaskRuns(ctx, param any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTaskRuns", reflect.TypeOf((*MockITaskRunDao)(nil).ListTaskRuns), ctx, param) +} + +// UpdateTaskRun mocks base method. +func (m *MockITaskRunDao) UpdateTaskRun(ctx context.Context, po *model.ObservabilityTaskRun) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateTaskRun", ctx, po) + ret0, _ := ret[0].(error) + return ret0 +} + +// UpdateTaskRun indicates an expected call of UpdateTaskRun. +func (mr *MockITaskRunDaoMockRecorder) UpdateTaskRun(ctx, po any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskRun", reflect.TypeOf((*MockITaskRunDao)(nil).UpdateTaskRun), ctx, po) +} + +// UpdateTaskRunWithOCC mocks base method. +func (m *MockITaskRunDao) UpdateTaskRunWithOCC(ctx context.Context, id, workspaceID int64, updateMap map[string]any) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateTaskRunWithOCC", ctx, id, workspaceID, updateMap) + ret0, _ := ret[0].(error) + return ret0 +} + +// UpdateTaskRunWithOCC indicates an expected call of UpdateTaskRunWithOCC. +func (mr *MockITaskRunDaoMockRecorder) UpdateTaskRunWithOCC(ctx, id, workspaceID, updateMap any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTaskRunWithOCC", reflect.TypeOf((*MockITaskRunDao)(nil).UpdateTaskRunWithOCC), ctx, id, workspaceID, updateMap) +} diff --git a/backend/modules/observability/infra/repo/redis/dao/mocks/Task_dao.go b/backend/modules/observability/infra/repo/redis/dao/mocks/Task_dao.go new file mode 100644 index 000000000..98375b3e2 --- /dev/null +++ b/backend/modules/observability/infra/repo/redis/dao/mocks/Task_dao.go @@ -0,0 +1,205 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/modules/observability/infra/repo/redis/dao (interfaces: ITaskDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/Task_dao.go -package=mocks . ITaskDAO +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + time "time" + + entity "github.com/coze-dev/coze-loop/backend/modules/observability/domain/task/entity" + gomock "go.uber.org/mock/gomock" +) + +// MockITaskDAO is a mock of ITaskDAO interface. +type MockITaskDAO struct { + ctrl *gomock.Controller + recorder *MockITaskDAOMockRecorder + isgomock struct{} +} + +// MockITaskDAOMockRecorder is the mock recorder for MockITaskDAO. +type MockITaskDAOMockRecorder struct { + mock *MockITaskDAO +} + +// NewMockITaskDAO creates a new mock instance. +func NewMockITaskDAO(ctrl *gomock.Controller) *MockITaskDAO { + mock := &MockITaskDAO{ctrl: ctrl} + mock.recorder = &MockITaskDAOMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockITaskDAO) EXPECT() *MockITaskDAOMockRecorder { + return m.recorder +} + +// AddNonFinalTask mocks base method. +func (m *MockITaskDAO) AddNonFinalTask(ctx context.Context, spaceID string, taskID int64) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddNonFinalTask", ctx, spaceID, taskID) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddNonFinalTask indicates an expected call of AddNonFinalTask. +func (mr *MockITaskDAOMockRecorder) AddNonFinalTask(ctx, spaceID, taskID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddNonFinalTask", reflect.TypeOf((*MockITaskDAO)(nil).AddNonFinalTask), ctx, spaceID, taskID) +} + +// DecrTaskCount mocks base method. +func (m *MockITaskDAO) DecrTaskCount(ctx context.Context, taskID int64, ttl time.Duration) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DecrTaskCount", ctx, taskID, ttl) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DecrTaskCount indicates an expected call of DecrTaskCount. +func (mr *MockITaskDAOMockRecorder) DecrTaskCount(ctx, taskID, ttl any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskCount", reflect.TypeOf((*MockITaskDAO)(nil).DecrTaskCount), ctx, taskID, ttl) +} + +// DecrTaskRunCount mocks base method. +func (m *MockITaskDAO) DecrTaskRunCount(ctx context.Context, taskID, taskRunID int64, ttl time.Duration) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DecrTaskRunCount", ctx, taskID, taskRunID, ttl) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DecrTaskRunCount indicates an expected call of DecrTaskRunCount. +func (mr *MockITaskDAOMockRecorder) DecrTaskRunCount(ctx, taskID, taskRunID, ttl any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskRunCount", reflect.TypeOf((*MockITaskDAO)(nil).DecrTaskRunCount), ctx, taskID, taskRunID, ttl) +} + +// GetTask mocks base method. +func (m *MockITaskDAO) GetTask(ctx context.Context, taskID int64) (*entity.ObservabilityTask, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetTask", ctx, taskID) + ret0, _ := ret[0].(*entity.ObservabilityTask) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetTask indicates an expected call of GetTask. +func (mr *MockITaskDAOMockRecorder) GetTask(ctx, taskID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTask", reflect.TypeOf((*MockITaskDAO)(nil).GetTask), ctx, taskID) +} + +// GetTaskCount mocks base method. +func (m *MockITaskDAO) GetTaskCount(ctx context.Context, taskID int64) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetTaskCount", ctx, taskID) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetTaskCount indicates an expected call of GetTaskCount. +func (mr *MockITaskDAOMockRecorder) GetTaskCount(ctx, taskID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskCount", reflect.TypeOf((*MockITaskDAO)(nil).GetTaskCount), ctx, taskID) +} + +// GetTaskRunCount mocks base method. +func (m *MockITaskDAO) GetTaskRunCount(ctx context.Context, taskID, taskRunID int64) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetTaskRunCount", ctx, taskID, taskRunID) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetTaskRunCount indicates an expected call of GetTaskRunCount. +func (mr *MockITaskDAOMockRecorder) GetTaskRunCount(ctx, taskID, taskRunID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunCount", reflect.TypeOf((*MockITaskDAO)(nil).GetTaskRunCount), ctx, taskID, taskRunID) +} + +// IncrTaskCount mocks base method. +func (m *MockITaskDAO) IncrTaskCount(ctx context.Context, taskID int64, ttl time.Duration) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "IncrTaskCount", ctx, taskID, ttl) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// IncrTaskCount indicates an expected call of IncrTaskCount. +func (mr *MockITaskDAOMockRecorder) IncrTaskCount(ctx, taskID, ttl any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskCount", reflect.TypeOf((*MockITaskDAO)(nil).IncrTaskCount), ctx, taskID, ttl) +} + +// IncrTaskRunCount mocks base method. +func (m *MockITaskDAO) IncrTaskRunCount(ctx context.Context, taskID, taskRunID int64, ttl time.Duration) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "IncrTaskRunCount", ctx, taskID, taskRunID, ttl) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// IncrTaskRunCount indicates an expected call of IncrTaskRunCount. +func (mr *MockITaskDAOMockRecorder) IncrTaskRunCount(ctx, taskID, taskRunID, ttl any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunCount", reflect.TypeOf((*MockITaskDAO)(nil).IncrTaskRunCount), ctx, taskID, taskRunID, ttl) +} + +// ListNonFinalTask mocks base method. +func (m *MockITaskDAO) ListNonFinalTask(ctx context.Context, spaceID string) ([]int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ListNonFinalTask", ctx, spaceID) + ret0, _ := ret[0].([]int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListNonFinalTask indicates an expected call of ListNonFinalTask. +func (mr *MockITaskDAOMockRecorder) ListNonFinalTask(ctx, spaceID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListNonFinalTask", reflect.TypeOf((*MockITaskDAO)(nil).ListNonFinalTask), ctx, spaceID) +} + +// RemoveNonFinalTask mocks base method. +func (m *MockITaskDAO) RemoveNonFinalTask(ctx context.Context, spaceID string, taskID int64) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "RemoveNonFinalTask", ctx, spaceID, taskID) + ret0, _ := ret[0].(error) + return ret0 +} + +// RemoveNonFinalTask indicates an expected call of RemoveNonFinalTask. +func (mr *MockITaskDAOMockRecorder) RemoveNonFinalTask(ctx, spaceID, taskID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveNonFinalTask", reflect.TypeOf((*MockITaskDAO)(nil).RemoveNonFinalTask), ctx, spaceID, taskID) +} + +// SetTask mocks base method. +func (m *MockITaskDAO) SetTask(ctx context.Context, task *entity.ObservabilityTask) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SetTask", ctx, task) + ret0, _ := ret[0].(error) + return ret0 +} + +// SetTask indicates an expected call of SetTask. +func (mr *MockITaskDAOMockRecorder) SetTask(ctx, task any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTask", reflect.TypeOf((*MockITaskDAO)(nil).SetTask), ctx, task) +} diff --git a/backend/modules/observability/infra/repo/redis/dao/mocks/task_run_dao.go b/backend/modules/observability/infra/repo/redis/dao/mocks/task_run_dao.go new file mode 100644 index 000000000..e5a1f3099 --- /dev/null +++ b/backend/modules/observability/infra/repo/redis/dao/mocks/task_run_dao.go @@ -0,0 +1,114 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/modules/observability/infra/repo/redis/dao (interfaces: ITaskRunDAO) +// +// Generated by this command: +// +// mockgen -destination=mocks/task_run_dao.go -package=mocks . ITaskRunDAO +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + time "time" + + gomock "go.uber.org/mock/gomock" +) + +// MockITaskRunDAO is a mock of ITaskRunDAO interface. +type MockITaskRunDAO struct { + ctrl *gomock.Controller + recorder *MockITaskRunDAOMockRecorder + isgomock struct{} +} + +// MockITaskRunDAOMockRecorder is the mock recorder for MockITaskRunDAO. +type MockITaskRunDAOMockRecorder struct { + mock *MockITaskRunDAO +} + +// NewMockITaskRunDAO creates a new mock instance. +func NewMockITaskRunDAO(ctrl *gomock.Controller) *MockITaskRunDAO { + mock := &MockITaskRunDAO{ctrl: ctrl} + mock.recorder = &MockITaskRunDAOMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockITaskRunDAO) EXPECT() *MockITaskRunDAOMockRecorder { + return m.recorder +} + +// DecrTaskRunSuccessCount mocks base method. +func (m *MockITaskRunDAO) DecrTaskRunSuccessCount(ctx context.Context, taskID, taskRunID int64) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DecrTaskRunSuccessCount", ctx, taskID, taskRunID) + ret0, _ := ret[0].(error) + return ret0 +} + +// DecrTaskRunSuccessCount indicates an expected call of DecrTaskRunSuccessCount. +func (mr *MockITaskRunDAOMockRecorder) DecrTaskRunSuccessCount(ctx, taskID, taskRunID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecrTaskRunSuccessCount", reflect.TypeOf((*MockITaskRunDAO)(nil).DecrTaskRunSuccessCount), ctx, taskID, taskRunID) +} + +// GetTaskRunFailCount mocks base method. +func (m *MockITaskRunDAO) GetTaskRunFailCount(ctx context.Context, taskID, taskRunID int64) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetTaskRunFailCount", ctx, taskID, taskRunID) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetTaskRunFailCount indicates an expected call of GetTaskRunFailCount. +func (mr *MockITaskRunDAOMockRecorder) GetTaskRunFailCount(ctx, taskID, taskRunID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunFailCount", reflect.TypeOf((*MockITaskRunDAO)(nil).GetTaskRunFailCount), ctx, taskID, taskRunID) +} + +// GetTaskRunSuccessCount mocks base method. +func (m *MockITaskRunDAO) GetTaskRunSuccessCount(ctx context.Context, taskID, taskRunID int64) (int64, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetTaskRunSuccessCount", ctx, taskID, taskRunID) + ret0, _ := ret[0].(int64) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetTaskRunSuccessCount indicates an expected call of GetTaskRunSuccessCount. +func (mr *MockITaskRunDAOMockRecorder) GetTaskRunSuccessCount(ctx, taskID, taskRunID any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTaskRunSuccessCount", reflect.TypeOf((*MockITaskRunDAO)(nil).GetTaskRunSuccessCount), ctx, taskID, taskRunID) +} + +// IncrTaskRunFailCount mocks base method. +func (m *MockITaskRunDAO) IncrTaskRunFailCount(ctx context.Context, taskID, taskRunID int64, ttl time.Duration) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "IncrTaskRunFailCount", ctx, taskID, taskRunID, ttl) + ret0, _ := ret[0].(error) + return ret0 +} + +// IncrTaskRunFailCount indicates an expected call of IncrTaskRunFailCount. +func (mr *MockITaskRunDAOMockRecorder) IncrTaskRunFailCount(ctx, taskID, taskRunID, ttl any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunFailCount", reflect.TypeOf((*MockITaskRunDAO)(nil).IncrTaskRunFailCount), ctx, taskID, taskRunID, ttl) +} + +// IncrTaskRunSuccessCount mocks base method. +func (m *MockITaskRunDAO) IncrTaskRunSuccessCount(ctx context.Context, taskID, taskRunID int64, ttl time.Duration) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "IncrTaskRunSuccessCount", ctx, taskID, taskRunID, ttl) + ret0, _ := ret[0].(error) + return ret0 +} + +// IncrTaskRunSuccessCount indicates an expected call of IncrTaskRunSuccessCount. +func (mr *MockITaskRunDAOMockRecorder) IncrTaskRunSuccessCount(ctx, taskID, taskRunID, ttl any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrTaskRunSuccessCount", reflect.TypeOf((*MockITaskRunDAO)(nil).IncrTaskRunSuccessCount), ctx, taskID, taskRunID, ttl) +} diff --git a/backend/modules/observability/infra/rpc/dataset/mocks/mock_dataset.go b/backend/modules/observability/infra/rpc/dataset/mocks/mock_dataset.go new file mode 100644 index 000000000..03590841d --- /dev/null +++ b/backend/modules/observability/infra/rpc/dataset/mocks/mock_dataset.go @@ -0,0 +1,10 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: dataset.go +// +// Generated by this command: +// +// mockgen -source=dataset.go -destination=mocks/mock_dataset.go +// + +// Package mock_dataset is a generated GoMock package. +package mock_dataset diff --git a/backend/modules/observability/infra/rpc/dataset/mocks/mock_datasetservice_client.go b/backend/modules/observability/infra/rpc/dataset/mocks/mock_datasetservice_client.go index e36f65e95..59bd1819e 100644 --- a/backend/modules/observability/infra/rpc/dataset/mocks/mock_datasetservice_client.go +++ b/backend/modules/observability/infra/rpc/dataset/mocks/mock_datasetservice_client.go @@ -3,7 +3,7 @@ // // Generated by this command: // -// mockgen -package=mocks -destination=modules/observability/infra/rpc/dataset/mocks/mock_datasetservice_client.go github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/datasetservice Client +// mockgen -package=mocks -destination=mocks/mock_datasetservice_client.go github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/data/datasetservice Client // // Package mocks is a generated GoMock package. @@ -22,6 +22,7 @@ import ( type MockClient struct { ctrl *gomock.Controller recorder *MockClientMockRecorder + isgomock struct{} } // MockClientMockRecorder is the mock recorder for MockClient. @@ -42,10 +43,10 @@ func (m *MockClient) EXPECT() *MockClientMockRecorder { } // BatchCreateDatasetItems mocks base method. -func (m *MockClient) BatchCreateDatasetItems(arg0 context.Context, arg1 *dataset.BatchCreateDatasetItemsRequest, arg2 ...callopt.Option) (*dataset.BatchCreateDatasetItemsResponse, error) { +func (m *MockClient) BatchCreateDatasetItems(ctx context.Context, req *dataset.BatchCreateDatasetItemsRequest, callOptions ...callopt.Option) (*dataset.BatchCreateDatasetItemsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchCreateDatasetItems", varargs...) @@ -55,17 +56,17 @@ func (m *MockClient) BatchCreateDatasetItems(arg0 context.Context, arg1 *dataset } // BatchCreateDatasetItems indicates an expected call of BatchCreateDatasetItems. -func (mr *MockClientMockRecorder) BatchCreateDatasetItems(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) BatchCreateDatasetItems(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateDatasetItems", reflect.TypeOf((*MockClient)(nil).BatchCreateDatasetItems), varargs...) } // BatchDeleteDatasetItems mocks base method. -func (m *MockClient) BatchDeleteDatasetItems(arg0 context.Context, arg1 *dataset.BatchDeleteDatasetItemsRequest, arg2 ...callopt.Option) (*dataset.BatchDeleteDatasetItemsResponse, error) { +func (m *MockClient) BatchDeleteDatasetItems(ctx context.Context, req *dataset.BatchDeleteDatasetItemsRequest, callOptions ...callopt.Option) (*dataset.BatchDeleteDatasetItemsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchDeleteDatasetItems", varargs...) @@ -75,17 +76,17 @@ func (m *MockClient) BatchDeleteDatasetItems(arg0 context.Context, arg1 *dataset } // BatchDeleteDatasetItems indicates an expected call of BatchDeleteDatasetItems. -func (mr *MockClientMockRecorder) BatchDeleteDatasetItems(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) BatchDeleteDatasetItems(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchDeleteDatasetItems", reflect.TypeOf((*MockClient)(nil).BatchDeleteDatasetItems), varargs...) } // BatchGetDatasetItems mocks base method. -func (m *MockClient) BatchGetDatasetItems(arg0 context.Context, arg1 *dataset.BatchGetDatasetItemsRequest, arg2 ...callopt.Option) (*dataset.BatchGetDatasetItemsResponse, error) { +func (m *MockClient) BatchGetDatasetItems(ctx context.Context, req *dataset.BatchGetDatasetItemsRequest, callOptions ...callopt.Option) (*dataset.BatchGetDatasetItemsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetDatasetItems", varargs...) @@ -95,17 +96,17 @@ func (m *MockClient) BatchGetDatasetItems(arg0 context.Context, arg1 *dataset.Ba } // BatchGetDatasetItems indicates an expected call of BatchGetDatasetItems. -func (mr *MockClientMockRecorder) BatchGetDatasetItems(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) BatchGetDatasetItems(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetDatasetItems", reflect.TypeOf((*MockClient)(nil).BatchGetDatasetItems), varargs...) } // BatchGetDatasetItemsByVersion mocks base method. -func (m *MockClient) BatchGetDatasetItemsByVersion(arg0 context.Context, arg1 *dataset.BatchGetDatasetItemsByVersionRequest, arg2 ...callopt.Option) (*dataset.BatchGetDatasetItemsByVersionResponse, error) { +func (m *MockClient) BatchGetDatasetItemsByVersion(ctx context.Context, req *dataset.BatchGetDatasetItemsByVersionRequest, callOptions ...callopt.Option) (*dataset.BatchGetDatasetItemsByVersionResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetDatasetItemsByVersion", varargs...) @@ -115,17 +116,17 @@ func (m *MockClient) BatchGetDatasetItemsByVersion(arg0 context.Context, arg1 *d } // BatchGetDatasetItemsByVersion indicates an expected call of BatchGetDatasetItemsByVersion. -func (mr *MockClientMockRecorder) BatchGetDatasetItemsByVersion(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) BatchGetDatasetItemsByVersion(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetDatasetItemsByVersion", reflect.TypeOf((*MockClient)(nil).BatchGetDatasetItemsByVersion), varargs...) } // BatchGetDatasetVersions mocks base method. -func (m *MockClient) BatchGetDatasetVersions(arg0 context.Context, arg1 *dataset.BatchGetDatasetVersionsRequest, arg2 ...callopt.Option) (*dataset.BatchGetDatasetVersionsResponse, error) { +func (m *MockClient) BatchGetDatasetVersions(ctx context.Context, req *dataset.BatchGetDatasetVersionsRequest, callOptions ...callopt.Option) (*dataset.BatchGetDatasetVersionsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetDatasetVersions", varargs...) @@ -135,17 +136,17 @@ func (m *MockClient) BatchGetDatasetVersions(arg0 context.Context, arg1 *dataset } // BatchGetDatasetVersions indicates an expected call of BatchGetDatasetVersions. -func (mr *MockClientMockRecorder) BatchGetDatasetVersions(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) BatchGetDatasetVersions(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetDatasetVersions", reflect.TypeOf((*MockClient)(nil).BatchGetDatasetVersions), varargs...) } // BatchGetDatasets mocks base method. -func (m *MockClient) BatchGetDatasets(arg0 context.Context, arg1 *dataset.BatchGetDatasetsRequest, arg2 ...callopt.Option) (*dataset.BatchGetDatasetsResponse, error) { +func (m *MockClient) BatchGetDatasets(ctx context.Context, req *dataset.BatchGetDatasetsRequest, callOptions ...callopt.Option) (*dataset.BatchGetDatasetsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchGetDatasets", varargs...) @@ -155,17 +156,17 @@ func (m *MockClient) BatchGetDatasets(arg0 context.Context, arg1 *dataset.BatchG } // BatchGetDatasets indicates an expected call of BatchGetDatasets. -func (mr *MockClientMockRecorder) BatchGetDatasets(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) BatchGetDatasets(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetDatasets", reflect.TypeOf((*MockClient)(nil).BatchGetDatasets), varargs...) } // ClearDatasetItem mocks base method. -func (m *MockClient) ClearDatasetItem(arg0 context.Context, arg1 *dataset.ClearDatasetItemRequest, arg2 ...callopt.Option) (*dataset.ClearDatasetItemResponse, error) { +func (m *MockClient) ClearDatasetItem(ctx context.Context, req *dataset.ClearDatasetItemRequest, callOptions ...callopt.Option) (*dataset.ClearDatasetItemResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ClearDatasetItem", varargs...) @@ -175,17 +176,17 @@ func (m *MockClient) ClearDatasetItem(arg0 context.Context, arg1 *dataset.ClearD } // ClearDatasetItem indicates an expected call of ClearDatasetItem. -func (mr *MockClientMockRecorder) ClearDatasetItem(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) ClearDatasetItem(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearDatasetItem", reflect.TypeOf((*MockClient)(nil).ClearDatasetItem), varargs...) } // CreateDataset mocks base method. -func (m *MockClient) CreateDataset(arg0 context.Context, arg1 *dataset.CreateDatasetRequest, arg2 ...callopt.Option) (*dataset.CreateDatasetResponse, error) { +func (m *MockClient) CreateDataset(ctx context.Context, req *dataset.CreateDatasetRequest, callOptions ...callopt.Option) (*dataset.CreateDatasetResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateDataset", varargs...) @@ -195,17 +196,17 @@ func (m *MockClient) CreateDataset(arg0 context.Context, arg1 *dataset.CreateDat } // CreateDataset indicates an expected call of CreateDataset. -func (mr *MockClientMockRecorder) CreateDataset(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) CreateDataset(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDataset", reflect.TypeOf((*MockClient)(nil).CreateDataset), varargs...) } // CreateDatasetVersion mocks base method. -func (m *MockClient) CreateDatasetVersion(arg0 context.Context, arg1 *dataset.CreateDatasetVersionRequest, arg2 ...callopt.Option) (*dataset.CreateDatasetVersionResponse, error) { +func (m *MockClient) CreateDatasetVersion(ctx context.Context, req *dataset.CreateDatasetVersionRequest, callOptions ...callopt.Option) (*dataset.CreateDatasetVersionResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CreateDatasetVersion", varargs...) @@ -215,17 +216,17 @@ func (m *MockClient) CreateDatasetVersion(arg0 context.Context, arg1 *dataset.Cr } // CreateDatasetVersion indicates an expected call of CreateDatasetVersion. -func (mr *MockClientMockRecorder) CreateDatasetVersion(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) CreateDatasetVersion(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateDatasetVersion", reflect.TypeOf((*MockClient)(nil).CreateDatasetVersion), varargs...) } // DeleteDataset mocks base method. -func (m *MockClient) DeleteDataset(arg0 context.Context, arg1 *dataset.DeleteDatasetRequest, arg2 ...callopt.Option) (*dataset.DeleteDatasetResponse, error) { +func (m *MockClient) DeleteDataset(ctx context.Context, req *dataset.DeleteDatasetRequest, callOptions ...callopt.Option) (*dataset.DeleteDatasetResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteDataset", varargs...) @@ -235,17 +236,17 @@ func (m *MockClient) DeleteDataset(arg0 context.Context, arg1 *dataset.DeleteDat } // DeleteDataset indicates an expected call of DeleteDataset. -func (mr *MockClientMockRecorder) DeleteDataset(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) DeleteDataset(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDataset", reflect.TypeOf((*MockClient)(nil).DeleteDataset), varargs...) } // DeleteDatasetItem mocks base method. -func (m *MockClient) DeleteDatasetItem(arg0 context.Context, arg1 *dataset.DeleteDatasetItemRequest, arg2 ...callopt.Option) (*dataset.DeleteDatasetItemResponse, error) { +func (m *MockClient) DeleteDatasetItem(ctx context.Context, req *dataset.DeleteDatasetItemRequest, callOptions ...callopt.Option) (*dataset.DeleteDatasetItemResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "DeleteDatasetItem", varargs...) @@ -255,17 +256,17 @@ func (m *MockClient) DeleteDatasetItem(arg0 context.Context, arg1 *dataset.Delet } // DeleteDatasetItem indicates an expected call of DeleteDatasetItem. -func (mr *MockClientMockRecorder) DeleteDatasetItem(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) DeleteDatasetItem(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDatasetItem", reflect.TypeOf((*MockClient)(nil).DeleteDatasetItem), varargs...) } // GetDataset mocks base method. -func (m *MockClient) GetDataset(arg0 context.Context, arg1 *dataset.GetDatasetRequest, arg2 ...callopt.Option) (*dataset.GetDatasetResponse, error) { +func (m *MockClient) GetDataset(ctx context.Context, req *dataset.GetDatasetRequest, callOptions ...callopt.Option) (*dataset.GetDatasetResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetDataset", varargs...) @@ -275,17 +276,17 @@ func (m *MockClient) GetDataset(arg0 context.Context, arg1 *dataset.GetDatasetRe } // GetDataset indicates an expected call of GetDataset. -func (mr *MockClientMockRecorder) GetDataset(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) GetDataset(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDataset", reflect.TypeOf((*MockClient)(nil).GetDataset), varargs...) } // GetDatasetIOJob mocks base method. -func (m *MockClient) GetDatasetIOJob(arg0 context.Context, arg1 *dataset.GetDatasetIOJobRequest, arg2 ...callopt.Option) (*dataset.GetDatasetIOJobResponse, error) { +func (m *MockClient) GetDatasetIOJob(ctx context.Context, req *dataset.GetDatasetIOJobRequest, callOptions ...callopt.Option) (*dataset.GetDatasetIOJobResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetDatasetIOJob", varargs...) @@ -295,17 +296,17 @@ func (m *MockClient) GetDatasetIOJob(arg0 context.Context, arg1 *dataset.GetData } // GetDatasetIOJob indicates an expected call of GetDatasetIOJob. -func (mr *MockClientMockRecorder) GetDatasetIOJob(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) GetDatasetIOJob(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDatasetIOJob", reflect.TypeOf((*MockClient)(nil).GetDatasetIOJob), varargs...) } // GetDatasetItem mocks base method. -func (m *MockClient) GetDatasetItem(arg0 context.Context, arg1 *dataset.GetDatasetItemRequest, arg2 ...callopt.Option) (*dataset.GetDatasetItemResponse, error) { +func (m *MockClient) GetDatasetItem(ctx context.Context, req *dataset.GetDatasetItemRequest, callOptions ...callopt.Option) (*dataset.GetDatasetItemResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetDatasetItem", varargs...) @@ -315,17 +316,17 @@ func (m *MockClient) GetDatasetItem(arg0 context.Context, arg1 *dataset.GetDatas } // GetDatasetItem indicates an expected call of GetDatasetItem. -func (mr *MockClientMockRecorder) GetDatasetItem(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) GetDatasetItem(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDatasetItem", reflect.TypeOf((*MockClient)(nil).GetDatasetItem), varargs...) } // GetDatasetSchema mocks base method. -func (m *MockClient) GetDatasetSchema(arg0 context.Context, arg1 *dataset.GetDatasetSchemaRequest, arg2 ...callopt.Option) (*dataset.GetDatasetSchemaResponse, error) { +func (m *MockClient) GetDatasetSchema(ctx context.Context, req *dataset.GetDatasetSchemaRequest, callOptions ...callopt.Option) (*dataset.GetDatasetSchemaResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetDatasetSchema", varargs...) @@ -335,17 +336,17 @@ func (m *MockClient) GetDatasetSchema(arg0 context.Context, arg1 *dataset.GetDat } // GetDatasetSchema indicates an expected call of GetDatasetSchema. -func (mr *MockClientMockRecorder) GetDatasetSchema(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) GetDatasetSchema(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDatasetSchema", reflect.TypeOf((*MockClient)(nil).GetDatasetSchema), varargs...) } // GetDatasetVersion mocks base method. -func (m *MockClient) GetDatasetVersion(arg0 context.Context, arg1 *dataset.GetDatasetVersionRequest, arg2 ...callopt.Option) (*dataset.GetDatasetVersionResponse, error) { +func (m *MockClient) GetDatasetVersion(ctx context.Context, req *dataset.GetDatasetVersionRequest, callOptions ...callopt.Option) (*dataset.GetDatasetVersionResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "GetDatasetVersion", varargs...) @@ -355,17 +356,17 @@ func (m *MockClient) GetDatasetVersion(arg0 context.Context, arg1 *dataset.GetDa } // GetDatasetVersion indicates an expected call of GetDatasetVersion. -func (mr *MockClientMockRecorder) GetDatasetVersion(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) GetDatasetVersion(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDatasetVersion", reflect.TypeOf((*MockClient)(nil).GetDatasetVersion), varargs...) } // ImportDataset mocks base method. -func (m *MockClient) ImportDataset(arg0 context.Context, arg1 *dataset.ImportDatasetRequest, arg2 ...callopt.Option) (*dataset.ImportDatasetResponse, error) { +func (m *MockClient) ImportDataset(ctx context.Context, req *dataset.ImportDatasetRequest, callOptions ...callopt.Option) (*dataset.ImportDatasetResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ImportDataset", varargs...) @@ -375,17 +376,17 @@ func (m *MockClient) ImportDataset(arg0 context.Context, arg1 *dataset.ImportDat } // ImportDataset indicates an expected call of ImportDataset. -func (mr *MockClientMockRecorder) ImportDataset(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) ImportDataset(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ImportDataset", reflect.TypeOf((*MockClient)(nil).ImportDataset), varargs...) } // ListDatasetIOJobs mocks base method. -func (m *MockClient) ListDatasetIOJobs(arg0 context.Context, arg1 *dataset.ListDatasetIOJobsRequest, arg2 ...callopt.Option) (*dataset.ListDatasetIOJobsResponse, error) { +func (m *MockClient) ListDatasetIOJobs(ctx context.Context, req *dataset.ListDatasetIOJobsRequest, callOptions ...callopt.Option) (*dataset.ListDatasetIOJobsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListDatasetIOJobs", varargs...) @@ -395,17 +396,17 @@ func (m *MockClient) ListDatasetIOJobs(arg0 context.Context, arg1 *dataset.ListD } // ListDatasetIOJobs indicates an expected call of ListDatasetIOJobs. -func (mr *MockClientMockRecorder) ListDatasetIOJobs(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) ListDatasetIOJobs(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDatasetIOJobs", reflect.TypeOf((*MockClient)(nil).ListDatasetIOJobs), varargs...) } // ListDatasetItems mocks base method. -func (m *MockClient) ListDatasetItems(arg0 context.Context, arg1 *dataset.ListDatasetItemsRequest, arg2 ...callopt.Option) (*dataset.ListDatasetItemsResponse, error) { +func (m *MockClient) ListDatasetItems(ctx context.Context, req *dataset.ListDatasetItemsRequest, callOptions ...callopt.Option) (*dataset.ListDatasetItemsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListDatasetItems", varargs...) @@ -415,17 +416,17 @@ func (m *MockClient) ListDatasetItems(arg0 context.Context, arg1 *dataset.ListDa } // ListDatasetItems indicates an expected call of ListDatasetItems. -func (mr *MockClientMockRecorder) ListDatasetItems(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) ListDatasetItems(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDatasetItems", reflect.TypeOf((*MockClient)(nil).ListDatasetItems), varargs...) } // ListDatasetItemsByVersion mocks base method. -func (m *MockClient) ListDatasetItemsByVersion(arg0 context.Context, arg1 *dataset.ListDatasetItemsByVersionRequest, arg2 ...callopt.Option) (*dataset.ListDatasetItemsByVersionResponse, error) { +func (m *MockClient) ListDatasetItemsByVersion(ctx context.Context, req *dataset.ListDatasetItemsByVersionRequest, callOptions ...callopt.Option) (*dataset.ListDatasetItemsByVersionResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListDatasetItemsByVersion", varargs...) @@ -435,17 +436,17 @@ func (m *MockClient) ListDatasetItemsByVersion(arg0 context.Context, arg1 *datas } // ListDatasetItemsByVersion indicates an expected call of ListDatasetItemsByVersion. -func (mr *MockClientMockRecorder) ListDatasetItemsByVersion(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) ListDatasetItemsByVersion(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDatasetItemsByVersion", reflect.TypeOf((*MockClient)(nil).ListDatasetItemsByVersion), varargs...) } // ListDatasetVersions mocks base method. -func (m *MockClient) ListDatasetVersions(arg0 context.Context, arg1 *dataset.ListDatasetVersionsRequest, arg2 ...callopt.Option) (*dataset.ListDatasetVersionsResponse, error) { +func (m *MockClient) ListDatasetVersions(ctx context.Context, req *dataset.ListDatasetVersionsRequest, callOptions ...callopt.Option) (*dataset.ListDatasetVersionsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListDatasetVersions", varargs...) @@ -455,17 +456,17 @@ func (m *MockClient) ListDatasetVersions(arg0 context.Context, arg1 *dataset.Lis } // ListDatasetVersions indicates an expected call of ListDatasetVersions. -func (mr *MockClientMockRecorder) ListDatasetVersions(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) ListDatasetVersions(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDatasetVersions", reflect.TypeOf((*MockClient)(nil).ListDatasetVersions), varargs...) } // ListDatasets mocks base method. -func (m *MockClient) ListDatasets(arg0 context.Context, arg1 *dataset.ListDatasetsRequest, arg2 ...callopt.Option) (*dataset.ListDatasetsResponse, error) { +func (m *MockClient) ListDatasets(ctx context.Context, req *dataset.ListDatasetsRequest, callOptions ...callopt.Option) (*dataset.ListDatasetsResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ListDatasets", varargs...) @@ -475,17 +476,17 @@ func (m *MockClient) ListDatasets(arg0 context.Context, arg1 *dataset.ListDatase } // ListDatasets indicates an expected call of ListDatasets. -func (mr *MockClientMockRecorder) ListDatasets(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) ListDatasets(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDatasets", reflect.TypeOf((*MockClient)(nil).ListDatasets), varargs...) } // UpdateDataset mocks base method. -func (m *MockClient) UpdateDataset(arg0 context.Context, arg1 *dataset.UpdateDatasetRequest, arg2 ...callopt.Option) (*dataset.UpdateDatasetResponse, error) { +func (m *MockClient) UpdateDataset(ctx context.Context, req *dataset.UpdateDatasetRequest, callOptions ...callopt.Option) (*dataset.UpdateDatasetResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateDataset", varargs...) @@ -495,17 +496,17 @@ func (m *MockClient) UpdateDataset(arg0 context.Context, arg1 *dataset.UpdateDat } // UpdateDataset indicates an expected call of UpdateDataset. -func (mr *MockClientMockRecorder) UpdateDataset(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) UpdateDataset(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDataset", reflect.TypeOf((*MockClient)(nil).UpdateDataset), varargs...) } // UpdateDatasetItem mocks base method. -func (m *MockClient) UpdateDatasetItem(arg0 context.Context, arg1 *dataset.UpdateDatasetItemRequest, arg2 ...callopt.Option) (*dataset.UpdateDatasetItemResponse, error) { +func (m *MockClient) UpdateDatasetItem(ctx context.Context, req *dataset.UpdateDatasetItemRequest, callOptions ...callopt.Option) (*dataset.UpdateDatasetItemResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateDatasetItem", varargs...) @@ -515,17 +516,17 @@ func (m *MockClient) UpdateDatasetItem(arg0 context.Context, arg1 *dataset.Updat } // UpdateDatasetItem indicates an expected call of UpdateDatasetItem. -func (mr *MockClientMockRecorder) UpdateDatasetItem(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) UpdateDatasetItem(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDatasetItem", reflect.TypeOf((*MockClient)(nil).UpdateDatasetItem), varargs...) } // UpdateDatasetSchema mocks base method. -func (m *MockClient) UpdateDatasetSchema(arg0 context.Context, arg1 *dataset.UpdateDatasetSchemaRequest, arg2 ...callopt.Option) (*dataset.UpdateDatasetSchemaResponse, error) { +func (m *MockClient) UpdateDatasetSchema(ctx context.Context, req *dataset.UpdateDatasetSchemaRequest, callOptions ...callopt.Option) (*dataset.UpdateDatasetSchemaResponse, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "UpdateDatasetSchema", varargs...) @@ -535,17 +536,17 @@ func (m *MockClient) UpdateDatasetSchema(arg0 context.Context, arg1 *dataset.Upd } // UpdateDatasetSchema indicates an expected call of UpdateDatasetSchema. -func (mr *MockClientMockRecorder) UpdateDatasetSchema(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) UpdateDatasetSchema(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDatasetSchema", reflect.TypeOf((*MockClient)(nil).UpdateDatasetSchema), varargs...) } // ValidateDatasetItems mocks base method. -func (m *MockClient) ValidateDatasetItems(arg0 context.Context, arg1 *dataset.ValidateDatasetItemsReq, arg2 ...callopt.Option) (*dataset.ValidateDatasetItemsResp, error) { +func (m *MockClient) ValidateDatasetItems(ctx context.Context, req *dataset.ValidateDatasetItemsReq, callOptions ...callopt.Option) (*dataset.ValidateDatasetItemsResp, error) { m.ctrl.T.Helper() - varargs := []any{arg0, arg1} - for _, a := range arg2 { + varargs := []any{ctx, req} + for _, a := range callOptions { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "ValidateDatasetItems", varargs...) @@ -555,8 +556,8 @@ func (m *MockClient) ValidateDatasetItems(arg0 context.Context, arg1 *dataset.Va } // ValidateDatasetItems indicates an expected call of ValidateDatasetItems. -func (mr *MockClientMockRecorder) ValidateDatasetItems(arg0, arg1 any, arg2 ...any) *gomock.Call { +func (mr *MockClientMockRecorder) ValidateDatasetItems(ctx, req any, callOptions ...any) *gomock.Call { mr.mock.ctrl.T.Helper() - varargs := append([]any{arg0, arg1}, arg2...) + varargs := append([]any{ctx, req}, callOptions...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateDatasetItems", reflect.TypeOf((*MockClient)(nil).ValidateDatasetItems), varargs...) } diff --git a/backend/modules/observability/infra/rpc/evaluationset/mocks/mock_evaluation_set.go b/backend/modules/observability/infra/rpc/evaluationset/mocks/mock_evaluation_set.go new file mode 100644 index 000000000..0a85f9995 --- /dev/null +++ b/backend/modules/observability/infra/rpc/evaluationset/mocks/mock_evaluation_set.go @@ -0,0 +1,10 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: evaluation_set.go +// +// Generated by this command: +// +// mockgen -source=evaluation_set.go -destination=mocks/mock_evaluation_set.go +// + +// Package mock_evaluationset is a generated GoMock package. +package mock_evaluationset diff --git a/backend/modules/observability/infra/rpc/evaluationset/mocks/mock_evaluationsetservice_client.go b/backend/modules/observability/infra/rpc/evaluationset/mocks/mock_evaluationsetservice_client.go new file mode 100644 index 000000000..9dc9e7daa --- /dev/null +++ b/backend/modules/observability/infra/rpc/evaluationset/mocks/mock_evaluationsetservice_client.go @@ -0,0 +1,363 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/evaluationsetservice (interfaces: Client) +// +// Generated by this command: +// +// mockgen -package=mocks -destination=mocks/mock_evaluationsetservice_client.go github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/evaluationsetservice Client +// + +// Package mocks is a generated GoMock package. +package mocks + +import ( + context "context" + reflect "reflect" + + callopt "github.com/cloudwego/kitex/client/callopt" + eval_set "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/evaluation/eval_set" + gomock "go.uber.org/mock/gomock" +) + +// MockClient is a mock of Client interface. +type MockClient struct { + ctrl *gomock.Controller + recorder *MockClientMockRecorder + isgomock struct{} +} + +// MockClientMockRecorder is the mock recorder for MockClient. +type MockClientMockRecorder struct { + mock *MockClient +} + +// NewMockClient creates a new mock instance. +func NewMockClient(ctrl *gomock.Controller) *MockClient { + mock := &MockClient{ctrl: ctrl} + mock.recorder = &MockClientMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockClient) EXPECT() *MockClientMockRecorder { + return m.recorder +} + +// BatchCreateEvaluationSetItems mocks base method. +func (m *MockClient) BatchCreateEvaluationSetItems(ctx context.Context, req *eval_set.BatchCreateEvaluationSetItemsRequest, callOptions ...callopt.Option) (*eval_set.BatchCreateEvaluationSetItemsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchCreateEvaluationSetItems", varargs...) + ret0, _ := ret[0].(*eval_set.BatchCreateEvaluationSetItemsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchCreateEvaluationSetItems indicates an expected call of BatchCreateEvaluationSetItems. +func (mr *MockClientMockRecorder) BatchCreateEvaluationSetItems(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreateEvaluationSetItems", reflect.TypeOf((*MockClient)(nil).BatchCreateEvaluationSetItems), varargs...) +} + +// BatchDeleteEvaluationSetItems mocks base method. +func (m *MockClient) BatchDeleteEvaluationSetItems(ctx context.Context, req *eval_set.BatchDeleteEvaluationSetItemsRequest, callOptions ...callopt.Option) (*eval_set.BatchDeleteEvaluationSetItemsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchDeleteEvaluationSetItems", varargs...) + ret0, _ := ret[0].(*eval_set.BatchDeleteEvaluationSetItemsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchDeleteEvaluationSetItems indicates an expected call of BatchDeleteEvaluationSetItems. +func (mr *MockClientMockRecorder) BatchDeleteEvaluationSetItems(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchDeleteEvaluationSetItems", reflect.TypeOf((*MockClient)(nil).BatchDeleteEvaluationSetItems), varargs...) +} + +// BatchGetEvaluationSetItems mocks base method. +func (m *MockClient) BatchGetEvaluationSetItems(ctx context.Context, req *eval_set.BatchGetEvaluationSetItemsRequest, callOptions ...callopt.Option) (*eval_set.BatchGetEvaluationSetItemsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchGetEvaluationSetItems", varargs...) + ret0, _ := ret[0].(*eval_set.BatchGetEvaluationSetItemsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchGetEvaluationSetItems indicates an expected call of BatchGetEvaluationSetItems. +func (mr *MockClientMockRecorder) BatchGetEvaluationSetItems(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluationSetItems", reflect.TypeOf((*MockClient)(nil).BatchGetEvaluationSetItems), varargs...) +} + +// BatchGetEvaluationSetVersions mocks base method. +func (m *MockClient) BatchGetEvaluationSetVersions(ctx context.Context, req *eval_set.BatchGetEvaluationSetVersionsRequest, callOptions ...callopt.Option) (*eval_set.BatchGetEvaluationSetVersionsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "BatchGetEvaluationSetVersions", varargs...) + ret0, _ := ret[0].(*eval_set.BatchGetEvaluationSetVersionsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BatchGetEvaluationSetVersions indicates an expected call of BatchGetEvaluationSetVersions. +func (mr *MockClientMockRecorder) BatchGetEvaluationSetVersions(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchGetEvaluationSetVersions", reflect.TypeOf((*MockClient)(nil).BatchGetEvaluationSetVersions), varargs...) +} + +// ClearEvaluationSetDraftItem mocks base method. +func (m *MockClient) ClearEvaluationSetDraftItem(ctx context.Context, req *eval_set.ClearEvaluationSetDraftItemRequest, callOptions ...callopt.Option) (*eval_set.ClearEvaluationSetDraftItemResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ClearEvaluationSetDraftItem", varargs...) + ret0, _ := ret[0].(*eval_set.ClearEvaluationSetDraftItemResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ClearEvaluationSetDraftItem indicates an expected call of ClearEvaluationSetDraftItem. +func (mr *MockClientMockRecorder) ClearEvaluationSetDraftItem(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearEvaluationSetDraftItem", reflect.TypeOf((*MockClient)(nil).ClearEvaluationSetDraftItem), varargs...) +} + +// CreateEvaluationSet mocks base method. +func (m *MockClient) CreateEvaluationSet(ctx context.Context, req *eval_set.CreateEvaluationSetRequest, callOptions ...callopt.Option) (*eval_set.CreateEvaluationSetResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CreateEvaluationSet", varargs...) + ret0, _ := ret[0].(*eval_set.CreateEvaluationSetResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateEvaluationSet indicates an expected call of CreateEvaluationSet. +func (mr *MockClientMockRecorder) CreateEvaluationSet(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluationSet", reflect.TypeOf((*MockClient)(nil).CreateEvaluationSet), varargs...) +} + +// CreateEvaluationSetVersion mocks base method. +func (m *MockClient) CreateEvaluationSetVersion(ctx context.Context, req *eval_set.CreateEvaluationSetVersionRequest, callOptions ...callopt.Option) (*eval_set.CreateEvaluationSetVersionResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "CreateEvaluationSetVersion", varargs...) + ret0, _ := ret[0].(*eval_set.CreateEvaluationSetVersionResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CreateEvaluationSetVersion indicates an expected call of CreateEvaluationSetVersion. +func (mr *MockClientMockRecorder) CreateEvaluationSetVersion(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateEvaluationSetVersion", reflect.TypeOf((*MockClient)(nil).CreateEvaluationSetVersion), varargs...) +} + +// DeleteEvaluationSet mocks base method. +func (m *MockClient) DeleteEvaluationSet(ctx context.Context, req *eval_set.DeleteEvaluationSetRequest, callOptions ...callopt.Option) (*eval_set.DeleteEvaluationSetResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "DeleteEvaluationSet", varargs...) + ret0, _ := ret[0].(*eval_set.DeleteEvaluationSetResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DeleteEvaluationSet indicates an expected call of DeleteEvaluationSet. +func (mr *MockClientMockRecorder) DeleteEvaluationSet(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteEvaluationSet", reflect.TypeOf((*MockClient)(nil).DeleteEvaluationSet), varargs...) +} + +// GetEvaluationSet mocks base method. +func (m *MockClient) GetEvaluationSet(ctx context.Context, req *eval_set.GetEvaluationSetRequest, callOptions ...callopt.Option) (*eval_set.GetEvaluationSetResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetEvaluationSet", varargs...) + ret0, _ := ret[0].(*eval_set.GetEvaluationSetResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetEvaluationSet indicates an expected call of GetEvaluationSet. +func (mr *MockClientMockRecorder) GetEvaluationSet(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEvaluationSet", reflect.TypeOf((*MockClient)(nil).GetEvaluationSet), varargs...) +} + +// GetEvaluationSetVersion mocks base method. +func (m *MockClient) GetEvaluationSetVersion(ctx context.Context, req *eval_set.GetEvaluationSetVersionRequest, callOptions ...callopt.Option) (*eval_set.GetEvaluationSetVersionResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "GetEvaluationSetVersion", varargs...) + ret0, _ := ret[0].(*eval_set.GetEvaluationSetVersionResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetEvaluationSetVersion indicates an expected call of GetEvaluationSetVersion. +func (mr *MockClientMockRecorder) GetEvaluationSetVersion(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetEvaluationSetVersion", reflect.TypeOf((*MockClient)(nil).GetEvaluationSetVersion), varargs...) +} + +// ListEvaluationSetItems mocks base method. +func (m *MockClient) ListEvaluationSetItems(ctx context.Context, req *eval_set.ListEvaluationSetItemsRequest, callOptions ...callopt.Option) (*eval_set.ListEvaluationSetItemsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListEvaluationSetItems", varargs...) + ret0, _ := ret[0].(*eval_set.ListEvaluationSetItemsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListEvaluationSetItems indicates an expected call of ListEvaluationSetItems. +func (mr *MockClientMockRecorder) ListEvaluationSetItems(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluationSetItems", reflect.TypeOf((*MockClient)(nil).ListEvaluationSetItems), varargs...) +} + +// ListEvaluationSetVersions mocks base method. +func (m *MockClient) ListEvaluationSetVersions(ctx context.Context, req *eval_set.ListEvaluationSetVersionsRequest, callOptions ...callopt.Option) (*eval_set.ListEvaluationSetVersionsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListEvaluationSetVersions", varargs...) + ret0, _ := ret[0].(*eval_set.ListEvaluationSetVersionsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListEvaluationSetVersions indicates an expected call of ListEvaluationSetVersions. +func (mr *MockClientMockRecorder) ListEvaluationSetVersions(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluationSetVersions", reflect.TypeOf((*MockClient)(nil).ListEvaluationSetVersions), varargs...) +} + +// ListEvaluationSets mocks base method. +func (m *MockClient) ListEvaluationSets(ctx context.Context, req *eval_set.ListEvaluationSetsRequest, callOptions ...callopt.Option) (*eval_set.ListEvaluationSetsResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListEvaluationSets", varargs...) + ret0, _ := ret[0].(*eval_set.ListEvaluationSetsResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListEvaluationSets indicates an expected call of ListEvaluationSets. +func (mr *MockClientMockRecorder) ListEvaluationSets(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListEvaluationSets", reflect.TypeOf((*MockClient)(nil).ListEvaluationSets), varargs...) +} + +// UpdateEvaluationSet mocks base method. +func (m *MockClient) UpdateEvaluationSet(ctx context.Context, req *eval_set.UpdateEvaluationSetRequest, callOptions ...callopt.Option) (*eval_set.UpdateEvaluationSetResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpdateEvaluationSet", varargs...) + ret0, _ := ret[0].(*eval_set.UpdateEvaluationSetResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateEvaluationSet indicates an expected call of UpdateEvaluationSet. +func (mr *MockClientMockRecorder) UpdateEvaluationSet(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluationSet", reflect.TypeOf((*MockClient)(nil).UpdateEvaluationSet), varargs...) +} + +// UpdateEvaluationSetItem mocks base method. +func (m *MockClient) UpdateEvaluationSetItem(ctx context.Context, req *eval_set.UpdateEvaluationSetItemRequest, callOptions ...callopt.Option) (*eval_set.UpdateEvaluationSetItemResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpdateEvaluationSetItem", varargs...) + ret0, _ := ret[0].(*eval_set.UpdateEvaluationSetItemResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateEvaluationSetItem indicates an expected call of UpdateEvaluationSetItem. +func (mr *MockClientMockRecorder) UpdateEvaluationSetItem(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluationSetItem", reflect.TypeOf((*MockClient)(nil).UpdateEvaluationSetItem), varargs...) +} + +// UpdateEvaluationSetSchema mocks base method. +func (m *MockClient) UpdateEvaluationSetSchema(ctx context.Context, req *eval_set.UpdateEvaluationSetSchemaRequest, callOptions ...callopt.Option) (*eval_set.UpdateEvaluationSetSchemaResponse, error) { + m.ctrl.T.Helper() + varargs := []any{ctx, req} + for _, a := range callOptions { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpdateEvaluationSetSchema", varargs...) + ret0, _ := ret[0].(*eval_set.UpdateEvaluationSetSchemaResponse) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateEvaluationSetSchema indicates an expected call of UpdateEvaluationSetSchema. +func (mr *MockClientMockRecorder) UpdateEvaluationSetSchema(ctx, req any, callOptions ...any) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]any{ctx, req}, callOptions...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateEvaluationSetSchema", reflect.TypeOf((*MockClient)(nil).UpdateEvaluationSetSchema), varargs...) +}