325 lines
13 KiB
Go
325 lines
13 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: plugin.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -destination pluginmock/plugin_mock.go --package pluginmock -source plugin.go
|
|
//
|
|
|
|
// Package pluginmock is a generated GoMock package.
|
|
package pluginmock
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
schema "github.com/cloudwego/eino/schema"
|
|
plugin "github.com/coze-dev/coze-studio/backend/api/model/crossdomain/plugin"
|
|
workflow "github.com/coze-dev/coze-studio/backend/api/model/crossdomain/workflow"
|
|
plugin0 "github.com/coze-dev/coze-studio/backend/crossdomain/contract/plugin"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockPluginService is a mock of PluginService interface.
|
|
type MockPluginService struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockPluginServiceMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockPluginServiceMockRecorder is the mock recorder for MockPluginService.
|
|
type MockPluginServiceMockRecorder struct {
|
|
mock *MockPluginService
|
|
}
|
|
|
|
// NewMockPluginService creates a new mock instance.
|
|
func NewMockPluginService(ctrl *gomock.Controller) *MockPluginService {
|
|
mock := &MockPluginService{ctrl: ctrl}
|
|
mock.recorder = &MockPluginServiceMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockPluginService) EXPECT() *MockPluginServiceMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// BindAgentTools mocks base method.
|
|
func (m *MockPluginService) BindAgentTools(ctx context.Context, agentID int64, toolIDs []int64) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "BindAgentTools", ctx, agentID, toolIDs)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// BindAgentTools indicates an expected call of BindAgentTools.
|
|
func (mr *MockPluginServiceMockRecorder) BindAgentTools(ctx, agentID, toolIDs any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BindAgentTools", reflect.TypeOf((*MockPluginService)(nil).BindAgentTools), ctx, agentID, toolIDs)
|
|
}
|
|
|
|
// DeleteDraftPlugin mocks base method.
|
|
func (m *MockPluginService) DeleteDraftPlugin(ctx context.Context, PluginID int64) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "DeleteDraftPlugin", ctx, PluginID)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// DeleteDraftPlugin indicates an expected call of DeleteDraftPlugin.
|
|
func (mr *MockPluginServiceMockRecorder) DeleteDraftPlugin(ctx, PluginID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDraftPlugin", reflect.TypeOf((*MockPluginService)(nil).DeleteDraftPlugin), ctx, PluginID)
|
|
}
|
|
|
|
// DuplicateDraftAgentTools mocks base method.
|
|
func (m *MockPluginService) DuplicateDraftAgentTools(ctx context.Context, fromAgentID, toAgentID int64) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "DuplicateDraftAgentTools", ctx, fromAgentID, toAgentID)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// DuplicateDraftAgentTools indicates an expected call of DuplicateDraftAgentTools.
|
|
func (mr *MockPluginServiceMockRecorder) DuplicateDraftAgentTools(ctx, fromAgentID, toAgentID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DuplicateDraftAgentTools", reflect.TypeOf((*MockPluginService)(nil).DuplicateDraftAgentTools), ctx, fromAgentID, toAgentID)
|
|
}
|
|
|
|
// ExecutePlugin mocks base method.
|
|
func (m *MockPluginService) ExecutePlugin(ctx context.Context, input map[string]any, pe *plugin.PluginEntity, toolID int64, cfg workflow.ExecuteConfig) (map[string]any, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ExecutePlugin", ctx, input, pe, toolID, cfg)
|
|
ret0, _ := ret[0].(map[string]any)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ExecutePlugin indicates an expected call of ExecutePlugin.
|
|
func (mr *MockPluginServiceMockRecorder) ExecutePlugin(ctx, input, pe, toolID, cfg any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExecutePlugin", reflect.TypeOf((*MockPluginService)(nil).ExecutePlugin), ctx, input, pe, toolID, cfg)
|
|
}
|
|
|
|
// ExecuteTool mocks base method.
|
|
func (m *MockPluginService) ExecuteTool(ctx context.Context, req *plugin.ExecuteToolRequest, opts ...plugin.ExecuteToolOpt) (*plugin.ExecuteToolResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
varargs := []any{ctx, req}
|
|
for _, a := range opts {
|
|
varargs = append(varargs, a)
|
|
}
|
|
ret := m.ctrl.Call(m, "ExecuteTool", varargs...)
|
|
ret0, _ := ret[0].(*plugin.ExecuteToolResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// ExecuteTool indicates an expected call of ExecuteTool.
|
|
func (mr *MockPluginServiceMockRecorder) ExecuteTool(ctx, req any, opts ...any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
varargs := append([]any{ctx, req}, opts...)
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExecuteTool", reflect.TypeOf((*MockPluginService)(nil).ExecuteTool), varargs...)
|
|
}
|
|
|
|
// GetAPPAllPlugins mocks base method.
|
|
func (m *MockPluginService) GetAPPAllPlugins(ctx context.Context, appID int64) ([]*plugin.PluginInfo, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetAPPAllPlugins", ctx, appID)
|
|
ret0, _ := ret[0].([]*plugin.PluginInfo)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetAPPAllPlugins indicates an expected call of GetAPPAllPlugins.
|
|
func (mr *MockPluginServiceMockRecorder) GetAPPAllPlugins(ctx, appID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPPAllPlugins", reflect.TypeOf((*MockPluginService)(nil).GetAPPAllPlugins), ctx, appID)
|
|
}
|
|
|
|
// GetPluginInvokableTools mocks base method.
|
|
func (m *MockPluginService) GetPluginInvokableTools(ctx context.Context, req *plugin.ToolsInvokableRequest) (map[int64]plugin0.InvokableTool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetPluginInvokableTools", ctx, req)
|
|
ret0, _ := ret[0].(map[int64]plugin0.InvokableTool)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetPluginInvokableTools indicates an expected call of GetPluginInvokableTools.
|
|
func (mr *MockPluginServiceMockRecorder) GetPluginInvokableTools(ctx, req any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPluginInvokableTools", reflect.TypeOf((*MockPluginService)(nil).GetPluginInvokableTools), ctx, req)
|
|
}
|
|
|
|
// GetPluginToolsInfo mocks base method.
|
|
func (m *MockPluginService) GetPluginToolsInfo(ctx context.Context, req *plugin.ToolsInfoRequest) (*plugin.ToolsInfoResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetPluginToolsInfo", ctx, req)
|
|
ret0, _ := ret[0].(*plugin.ToolsInfoResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetPluginToolsInfo indicates an expected call of GetPluginToolsInfo.
|
|
func (mr *MockPluginServiceMockRecorder) GetPluginToolsInfo(ctx, req any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPluginToolsInfo", reflect.TypeOf((*MockPluginService)(nil).GetPluginToolsInfo), ctx, req)
|
|
}
|
|
|
|
// MGetAgentTools mocks base method.
|
|
func (m *MockPluginService) MGetAgentTools(ctx context.Context, req *plugin.MGetAgentToolsRequest) ([]*plugin.ToolInfo, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "MGetAgentTools", ctx, req)
|
|
ret0, _ := ret[0].([]*plugin.ToolInfo)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// MGetAgentTools indicates an expected call of MGetAgentTools.
|
|
func (mr *MockPluginServiceMockRecorder) MGetAgentTools(ctx, req any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetAgentTools", reflect.TypeOf((*MockPluginService)(nil).MGetAgentTools), ctx, req)
|
|
}
|
|
|
|
// MGetPluginLatestVersion mocks base method.
|
|
func (m *MockPluginService) MGetPluginLatestVersion(ctx context.Context, pluginIDs []int64) (*plugin.MGetPluginLatestVersionResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "MGetPluginLatestVersion", ctx, pluginIDs)
|
|
ret0, _ := ret[0].(*plugin.MGetPluginLatestVersionResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// MGetPluginLatestVersion indicates an expected call of MGetPluginLatestVersion.
|
|
func (mr *MockPluginServiceMockRecorder) MGetPluginLatestVersion(ctx, pluginIDs any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetPluginLatestVersion", reflect.TypeOf((*MockPluginService)(nil).MGetPluginLatestVersion), ctx, pluginIDs)
|
|
}
|
|
|
|
// MGetVersionPlugins mocks base method.
|
|
func (m *MockPluginService) MGetVersionPlugins(ctx context.Context, versionPlugins []plugin.VersionPlugin) ([]*plugin.PluginInfo, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "MGetVersionPlugins", ctx, versionPlugins)
|
|
ret0, _ := ret[0].([]*plugin.PluginInfo)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// MGetVersionPlugins indicates an expected call of MGetVersionPlugins.
|
|
func (mr *MockPluginServiceMockRecorder) MGetVersionPlugins(ctx, versionPlugins any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetVersionPlugins", reflect.TypeOf((*MockPluginService)(nil).MGetVersionPlugins), ctx, versionPlugins)
|
|
}
|
|
|
|
// MGetVersionTools mocks base method.
|
|
func (m *MockPluginService) MGetVersionTools(ctx context.Context, versionTools []plugin.VersionTool) ([]*plugin.ToolInfo, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "MGetVersionTools", ctx, versionTools)
|
|
ret0, _ := ret[0].([]*plugin.ToolInfo)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// MGetVersionTools indicates an expected call of MGetVersionTools.
|
|
func (mr *MockPluginServiceMockRecorder) MGetVersionTools(ctx, versionTools any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetVersionTools", reflect.TypeOf((*MockPluginService)(nil).MGetVersionTools), ctx, versionTools)
|
|
}
|
|
|
|
// PublishAPPPlugins mocks base method.
|
|
func (m *MockPluginService) PublishAPPPlugins(ctx context.Context, req *plugin.PublishAPPPluginsRequest) (*plugin.PublishAPPPluginsResponse, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "PublishAPPPlugins", ctx, req)
|
|
ret0, _ := ret[0].(*plugin.PublishAPPPluginsResponse)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// PublishAPPPlugins indicates an expected call of PublishAPPPlugins.
|
|
func (mr *MockPluginServiceMockRecorder) PublishAPPPlugins(ctx, req any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishAPPPlugins", reflect.TypeOf((*MockPluginService)(nil).PublishAPPPlugins), ctx, req)
|
|
}
|
|
|
|
// PublishAgentTools mocks base method.
|
|
func (m *MockPluginService) PublishAgentTools(ctx context.Context, agentID int64, agentVersion string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "PublishAgentTools", ctx, agentID, agentVersion)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// PublishAgentTools indicates an expected call of PublishAgentTools.
|
|
func (mr *MockPluginServiceMockRecorder) PublishAgentTools(ctx, agentID, agentVersion any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishAgentTools", reflect.TypeOf((*MockPluginService)(nil).PublishAgentTools), ctx, agentID, agentVersion)
|
|
}
|
|
|
|
// PublishPlugin mocks base method.
|
|
func (m *MockPluginService) PublishPlugin(ctx context.Context, req *plugin.PublishPluginRequest) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "PublishPlugin", ctx, req)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// PublishPlugin indicates an expected call of PublishPlugin.
|
|
func (mr *MockPluginServiceMockRecorder) PublishPlugin(ctx, req any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PublishPlugin", reflect.TypeOf((*MockPluginService)(nil).PublishPlugin), ctx, req)
|
|
}
|
|
|
|
// MockInvokableTool is a mock of InvokableTool interface.
|
|
type MockInvokableTool struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockInvokableToolMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockInvokableToolMockRecorder is the mock recorder for MockInvokableTool.
|
|
type MockInvokableToolMockRecorder struct {
|
|
mock *MockInvokableTool
|
|
}
|
|
|
|
// NewMockInvokableTool creates a new mock instance.
|
|
func NewMockInvokableTool(ctrl *gomock.Controller) *MockInvokableTool {
|
|
mock := &MockInvokableTool{ctrl: ctrl}
|
|
mock.recorder = &MockInvokableToolMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockInvokableTool) EXPECT() *MockInvokableToolMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Info mocks base method.
|
|
func (m *MockInvokableTool) Info(ctx context.Context) (*schema.ToolInfo, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Info", ctx)
|
|
ret0, _ := ret[0].(*schema.ToolInfo)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Info indicates an expected call of Info.
|
|
func (mr *MockInvokableToolMockRecorder) Info(ctx any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Info", reflect.TypeOf((*MockInvokableTool)(nil).Info), ctx)
|
|
}
|
|
|
|
// PluginInvoke mocks base method.
|
|
func (m *MockInvokableTool) PluginInvoke(ctx context.Context, argumentsInJSON string, cfg workflow.ExecuteConfig) (string, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "PluginInvoke", ctx, argumentsInJSON, cfg)
|
|
ret0, _ := ret[0].(string)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// PluginInvoke indicates an expected call of PluginInvoke.
|
|
func (mr *MockInvokableToolMockRecorder) PluginInvoke(ctx, argumentsInJSON, cfg any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PluginInvoke", reflect.TypeOf((*MockInvokableTool)(nil).PluginInvoke), ctx, argumentsInJSON, cfg)
|
|
}
|