|
|
|
@ -35,6 +35,7 @@ import ( |
|
|
|
|
workflow "github.com/coze-dev/coze-studio/backend/api/model/crossdomain/workflow" |
|
|
|
|
workflow0 "github.com/coze-dev/coze-studio/backend/api/model/workflow" |
|
|
|
|
workflow1 "github.com/coze-dev/coze-studio/backend/domain/workflow" |
|
|
|
|
config "github.com/coze-dev/coze-studio/backend/domain/workflow/config" |
|
|
|
|
entity "github.com/coze-dev/coze-studio/backend/domain/workflow/entity" |
|
|
|
|
vo "github.com/coze-dev/coze-studio/backend/domain/workflow/entity/vo" |
|
|
|
|
gomock "go.uber.org/mock/gomock" |
|
|
|
@ -65,47 +66,47 @@ func (m *MockService) EXPECT() *MockServiceMockRecorder { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// AsyncExecute mocks base method.
|
|
|
|
|
func (m *MockService) AsyncExecute(ctx context.Context, config workflow.ExecuteConfig, input map[string]any) (int64, error) { |
|
|
|
|
func (m *MockService) AsyncExecute(ctx context.Context, arg1 workflow.ExecuteConfig, input map[string]any) (int64, error) { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|
ret := m.ctrl.Call(m, "AsyncExecute", ctx, config, input) |
|
|
|
|
ret := m.ctrl.Call(m, "AsyncExecute", ctx, arg1, input) |
|
|
|
|
ret0, _ := ret[0].(int64) |
|
|
|
|
ret1, _ := ret[1].(error) |
|
|
|
|
return ret0, ret1 |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// AsyncExecute indicates an expected call of AsyncExecute.
|
|
|
|
|
func (mr *MockServiceMockRecorder) AsyncExecute(ctx, config, input any) *gomock.Call { |
|
|
|
|
func (mr *MockServiceMockRecorder) AsyncExecute(ctx, arg1, input any) *gomock.Call { |
|
|
|
|
mr.mock.ctrl.T.Helper() |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AsyncExecute", reflect.TypeOf((*MockService)(nil).AsyncExecute), ctx, config, input) |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AsyncExecute", reflect.TypeOf((*MockService)(nil).AsyncExecute), ctx, arg1, input) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// AsyncExecuteNode mocks base method.
|
|
|
|
|
func (m *MockService) AsyncExecuteNode(ctx context.Context, nodeID string, config workflow.ExecuteConfig, input map[string]any) (int64, error) { |
|
|
|
|
func (m *MockService) AsyncExecuteNode(ctx context.Context, nodeID string, arg2 workflow.ExecuteConfig, input map[string]any) (int64, error) { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|
ret := m.ctrl.Call(m, "AsyncExecuteNode", ctx, nodeID, config, input) |
|
|
|
|
ret := m.ctrl.Call(m, "AsyncExecuteNode", ctx, nodeID, arg2, input) |
|
|
|
|
ret0, _ := ret[0].(int64) |
|
|
|
|
ret1, _ := ret[1].(error) |
|
|
|
|
return ret0, ret1 |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// AsyncExecuteNode indicates an expected call of AsyncExecuteNode.
|
|
|
|
|
func (mr *MockServiceMockRecorder) AsyncExecuteNode(ctx, nodeID, config, input any) *gomock.Call { |
|
|
|
|
func (mr *MockServiceMockRecorder) AsyncExecuteNode(ctx, nodeID, arg2, input any) *gomock.Call { |
|
|
|
|
mr.mock.ctrl.T.Helper() |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AsyncExecuteNode", reflect.TypeOf((*MockService)(nil).AsyncExecuteNode), ctx, nodeID, config, input) |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AsyncExecuteNode", reflect.TypeOf((*MockService)(nil).AsyncExecuteNode), ctx, nodeID, arg2, input) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// AsyncResume mocks base method.
|
|
|
|
|
func (m *MockService) AsyncResume(ctx context.Context, req *entity.ResumeRequest, config workflow.ExecuteConfig) error { |
|
|
|
|
func (m *MockService) AsyncResume(ctx context.Context, req *entity.ResumeRequest, arg2 workflow.ExecuteConfig) error { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|
ret := m.ctrl.Call(m, "AsyncResume", ctx, req, config) |
|
|
|
|
ret := m.ctrl.Call(m, "AsyncResume", ctx, req, arg2) |
|
|
|
|
ret0, _ := ret[0].(error) |
|
|
|
|
return ret0 |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// AsyncResume indicates an expected call of AsyncResume.
|
|
|
|
|
func (mr *MockServiceMockRecorder) AsyncResume(ctx, req, config any) *gomock.Call { |
|
|
|
|
func (mr *MockServiceMockRecorder) AsyncResume(ctx, req, arg2 any) *gomock.Call { |
|
|
|
|
mr.mock.ctrl.T.Helper() |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AsyncResume", reflect.TypeOf((*MockService)(nil).AsyncResume), ctx, req, config) |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AsyncResume", reflect.TypeOf((*MockService)(nil).AsyncResume), ctx, req, arg2) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Cancel mocks base method.
|
|
|
|
@ -368,18 +369,18 @@ func (mr *MockServiceMockRecorder) QueryNodeProperties(ctx, id any) *gomock.Call |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// ReleaseApplicationWorkflows mocks base method.
|
|
|
|
|
func (m *MockService) ReleaseApplicationWorkflows(ctx context.Context, appID int64, config *vo.ReleaseWorkflowConfig) ([]*vo.ValidateIssue, error) { |
|
|
|
|
func (m *MockService) ReleaseApplicationWorkflows(ctx context.Context, appID int64, arg2 *vo.ReleaseWorkflowConfig) ([]*vo.ValidateIssue, error) { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|
ret := m.ctrl.Call(m, "ReleaseApplicationWorkflows", ctx, appID, config) |
|
|
|
|
ret := m.ctrl.Call(m, "ReleaseApplicationWorkflows", ctx, appID, arg2) |
|
|
|
|
ret0, _ := ret[0].([]*vo.ValidateIssue) |
|
|
|
|
ret1, _ := ret[1].(error) |
|
|
|
|
return ret0, ret1 |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// ReleaseApplicationWorkflows indicates an expected call of ReleaseApplicationWorkflows.
|
|
|
|
|
func (mr *MockServiceMockRecorder) ReleaseApplicationWorkflows(ctx, appID, config any) *gomock.Call { |
|
|
|
|
func (mr *MockServiceMockRecorder) ReleaseApplicationWorkflows(ctx, appID, arg2 any) *gomock.Call { |
|
|
|
|
mr.mock.ctrl.T.Helper() |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReleaseApplicationWorkflows", reflect.TypeOf((*MockService)(nil).ReleaseApplicationWorkflows), ctx, appID, config) |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReleaseApplicationWorkflows", reflect.TypeOf((*MockService)(nil).ReleaseApplicationWorkflows), ctx, appID, arg2) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Save mocks base method.
|
|
|
|
@ -397,39 +398,39 @@ func (mr *MockServiceMockRecorder) Save(ctx, id, arg2 any) *gomock.Call { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// StreamExecute mocks base method.
|
|
|
|
|
func (m *MockService) StreamExecute(ctx context.Context, config workflow.ExecuteConfig, input map[string]any) (*schema.StreamReader[*entity.Message], error) { |
|
|
|
|
func (m *MockService) StreamExecute(ctx context.Context, arg1 workflow.ExecuteConfig, input map[string]any) (*schema.StreamReader[*entity.Message], error) { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|
ret := m.ctrl.Call(m, "StreamExecute", ctx, config, input) |
|
|
|
|
ret := m.ctrl.Call(m, "StreamExecute", ctx, arg1, input) |
|
|
|
|
ret0, _ := ret[0].(*schema.StreamReader[*entity.Message]) |
|
|
|
|
ret1, _ := ret[1].(error) |
|
|
|
|
return ret0, ret1 |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// StreamExecute indicates an expected call of StreamExecute.
|
|
|
|
|
func (mr *MockServiceMockRecorder) StreamExecute(ctx, config, input any) *gomock.Call { |
|
|
|
|
func (mr *MockServiceMockRecorder) StreamExecute(ctx, arg1, input any) *gomock.Call { |
|
|
|
|
mr.mock.ctrl.T.Helper() |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamExecute", reflect.TypeOf((*MockService)(nil).StreamExecute), ctx, config, input) |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamExecute", reflect.TypeOf((*MockService)(nil).StreamExecute), ctx, arg1, input) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// StreamResume mocks base method.
|
|
|
|
|
func (m *MockService) StreamResume(ctx context.Context, req *entity.ResumeRequest, config workflow.ExecuteConfig) (*schema.StreamReader[*entity.Message], error) { |
|
|
|
|
func (m *MockService) StreamResume(ctx context.Context, req *entity.ResumeRequest, arg2 workflow.ExecuteConfig) (*schema.StreamReader[*entity.Message], error) { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|
ret := m.ctrl.Call(m, "StreamResume", ctx, req, config) |
|
|
|
|
ret := m.ctrl.Call(m, "StreamResume", ctx, req, arg2) |
|
|
|
|
ret0, _ := ret[0].(*schema.StreamReader[*entity.Message]) |
|
|
|
|
ret1, _ := ret[1].(error) |
|
|
|
|
return ret0, ret1 |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// StreamResume indicates an expected call of StreamResume.
|
|
|
|
|
func (mr *MockServiceMockRecorder) StreamResume(ctx, req, config any) *gomock.Call { |
|
|
|
|
func (mr *MockServiceMockRecorder) StreamResume(ctx, req, arg2 any) *gomock.Call { |
|
|
|
|
mr.mock.ctrl.T.Helper() |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamResume", reflect.TypeOf((*MockService)(nil).StreamResume), ctx, req, config) |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamResume", reflect.TypeOf((*MockService)(nil).StreamResume), ctx, req, arg2) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// SyncExecute mocks base method.
|
|
|
|
|
func (m *MockService) SyncExecute(ctx context.Context, config workflow.ExecuteConfig, input map[string]any) (*entity.WorkflowExecution, vo.TerminatePlan, error) { |
|
|
|
|
func (m *MockService) SyncExecute(ctx context.Context, arg1 workflow.ExecuteConfig, input map[string]any) (*entity.WorkflowExecution, vo.TerminatePlan, error) { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|
ret := m.ctrl.Call(m, "SyncExecute", ctx, config, input) |
|
|
|
|
ret := m.ctrl.Call(m, "SyncExecute", ctx, arg1, input) |
|
|
|
|
ret0, _ := ret[0].(*entity.WorkflowExecution) |
|
|
|
|
ret1, _ := ret[1].(vo.TerminatePlan) |
|
|
|
|
ret2, _ := ret[2].(error) |
|
|
|
@ -437,9 +438,9 @@ func (m *MockService) SyncExecute(ctx context.Context, config workflow.ExecuteCo |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// SyncExecute indicates an expected call of SyncExecute.
|
|
|
|
|
func (mr *MockServiceMockRecorder) SyncExecute(ctx, config, input any) *gomock.Call { |
|
|
|
|
func (mr *MockServiceMockRecorder) SyncExecute(ctx, arg1, input any) *gomock.Call { |
|
|
|
|
mr.mock.ctrl.T.Helper() |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncExecute", reflect.TypeOf((*MockService)(nil).SyncExecute), ctx, config, input) |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncExecute", reflect.TypeOf((*MockService)(nil).SyncExecute), ctx, arg1, input) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// SyncRelatedWorkflowResources mocks base method.
|
|
|
|
@ -923,6 +924,20 @@ func (mr *MockRepositoryMockRecorder) GetNodeExecutionsByWfExeID(ctx, wfExeID an |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeExecutionsByWfExeID", reflect.TypeOf((*MockRepository)(nil).GetNodeExecutionsByWfExeID), ctx, wfExeID) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// GetNodeOfCodeConfig mocks base method.
|
|
|
|
|
func (m *MockRepository) GetNodeOfCodeConfig() *config.NodeOfCodeConfig { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|
ret := m.ctrl.Call(m, "GetNodeOfCodeConfig") |
|
|
|
|
ret0, _ := ret[0].(*config.NodeOfCodeConfig) |
|
|
|
|
return ret0 |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// GetNodeOfCodeConfig indicates an expected call of GetNodeOfCodeConfig.
|
|
|
|
|
func (mr *MockRepositoryMockRecorder) GetNodeOfCodeConfig() *gomock.Call { |
|
|
|
|
mr.mock.ctrl.T.Helper() |
|
|
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeOfCodeConfig", reflect.TypeOf((*MockRepository)(nil).GetNodeOfCodeConfig)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// GetTestRunLatestExeID mocks base method.
|
|
|
|
|
func (m *MockRepository) GetTestRunLatestExeID(ctx context.Context, wfID, uID int64) (int64, error) { |
|
|
|
|
m.ctrl.T.Helper() |
|
|
|
|