166 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			166 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			Go
		
	
	
	
/*
 | 
						|
 * Copyright 2025 coze-dev Authors
 | 
						|
 *
 | 
						|
 * Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
 * you may not use this file except in compliance with the License.
 | 
						|
 * You may obtain a copy of the License at
 | 
						|
 *
 | 
						|
 *     http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 *
 | 
						|
 * Unless required by applicable law or agreed to in writing, software
 | 
						|
 * distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
 * See the License for the specific language governing permissions and
 | 
						|
 * limitations under the License.
 | 
						|
 */
 | 
						|
 | 
						|
// Code generated by MockGen. DO NOT EDIT.
 | 
						|
// Source: variable.go
 | 
						|
//
 | 
						|
// Generated by this command:
 | 
						|
//
 | 
						|
//	mockgen -destination varmock/var_mock.go --package mockvar -source variable.go
 | 
						|
//
 | 
						|
 | 
						|
// Package mockvar is a generated GoMock package.
 | 
						|
package mockvar
 | 
						|
 | 
						|
import (
 | 
						|
	context "context"
 | 
						|
	reflect "reflect"
 | 
						|
 | 
						|
	variable "github.com/coze-dev/coze-studio/backend/domain/workflow/crossdomain/variable"
 | 
						|
	vo "github.com/coze-dev/coze-studio/backend/domain/workflow/entity/vo"
 | 
						|
	compose "github.com/cloudwego/eino/compose"
 | 
						|
	gomock "go.uber.org/mock/gomock"
 | 
						|
)
 | 
						|
 | 
						|
// MockStore is a mock of Store interface.
 | 
						|
type MockStore struct {
 | 
						|
	ctrl     *gomock.Controller
 | 
						|
	recorder *MockStoreMockRecorder
 | 
						|
	isgomock struct{}
 | 
						|
}
 | 
						|
 | 
						|
// MockStoreMockRecorder is the mock recorder for MockStore.
 | 
						|
type MockStoreMockRecorder struct {
 | 
						|
	mock *MockStore
 | 
						|
}
 | 
						|
 | 
						|
// NewMockStore creates a new mock instance.
 | 
						|
func NewMockStore(ctrl *gomock.Controller) *MockStore {
 | 
						|
	mock := &MockStore{ctrl: ctrl}
 | 
						|
	mock.recorder = &MockStoreMockRecorder{mock}
 | 
						|
	return mock
 | 
						|
}
 | 
						|
 | 
						|
// EXPECT returns an object that allows the caller to indicate expected use.
 | 
						|
func (m *MockStore) EXPECT() *MockStoreMockRecorder {
 | 
						|
	return m.recorder
 | 
						|
}
 | 
						|
 | 
						|
// Get mocks base method.
 | 
						|
func (m *MockStore) Get(ctx context.Context, path compose.FieldPath, opts ...variable.OptionFn) (any, error) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	varargs := []any{ctx, path}
 | 
						|
	for _, a := range opts {
 | 
						|
		varargs = append(varargs, a)
 | 
						|
	}
 | 
						|
	ret := m.ctrl.Call(m, "Get", varargs...)
 | 
						|
	ret0, _ := ret[0].(any)
 | 
						|
	ret1, _ := ret[1].(error)
 | 
						|
	return ret0, ret1
 | 
						|
}
 | 
						|
 | 
						|
// Get indicates an expected call of Get.
 | 
						|
func (mr *MockStoreMockRecorder) Get(ctx, path any, opts ...any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	varargs := append([]any{ctx, path}, opts...)
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockStore)(nil).Get), varargs...)
 | 
						|
}
 | 
						|
 | 
						|
// Init mocks base method.
 | 
						|
func (m *MockStore) Init(ctx context.Context) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	m.ctrl.Call(m, "Init", ctx)
 | 
						|
}
 | 
						|
 | 
						|
// Init indicates an expected call of Init.
 | 
						|
func (mr *MockStoreMockRecorder) Init(ctx any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Init", reflect.TypeOf((*MockStore)(nil).Init), ctx)
 | 
						|
}
 | 
						|
 | 
						|
// Set mocks base method.
 | 
						|
func (m *MockStore) Set(ctx context.Context, path compose.FieldPath, value any, opts ...variable.OptionFn) error {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	varargs := []any{ctx, path, value}
 | 
						|
	for _, a := range opts {
 | 
						|
		varargs = append(varargs, a)
 | 
						|
	}
 | 
						|
	ret := m.ctrl.Call(m, "Set", varargs...)
 | 
						|
	ret0, _ := ret[0].(error)
 | 
						|
	return ret0
 | 
						|
}
 | 
						|
 | 
						|
// Set indicates an expected call of Set.
 | 
						|
func (mr *MockStoreMockRecorder) Set(ctx, path, value any, opts ...any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	varargs := append([]any{ctx, path, value}, opts...)
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Set", reflect.TypeOf((*MockStore)(nil).Set), varargs...)
 | 
						|
}
 | 
						|
 | 
						|
// MockVariablesMetaGetter is a mock of VariablesMetaGetter interface.
 | 
						|
type MockVariablesMetaGetter struct {
 | 
						|
	ctrl     *gomock.Controller
 | 
						|
	recorder *MockVariablesMetaGetterMockRecorder
 | 
						|
	isgomock struct{}
 | 
						|
}
 | 
						|
 | 
						|
// MockVariablesMetaGetterMockRecorder is the mock recorder for MockVariablesMetaGetter.
 | 
						|
type MockVariablesMetaGetterMockRecorder struct {
 | 
						|
	mock *MockVariablesMetaGetter
 | 
						|
}
 | 
						|
 | 
						|
// NewMockVariablesMetaGetter creates a new mock instance.
 | 
						|
func NewMockVariablesMetaGetter(ctrl *gomock.Controller) *MockVariablesMetaGetter {
 | 
						|
	mock := &MockVariablesMetaGetter{ctrl: ctrl}
 | 
						|
	mock.recorder = &MockVariablesMetaGetterMockRecorder{mock}
 | 
						|
	return mock
 | 
						|
}
 | 
						|
 | 
						|
// EXPECT returns an object that allows the caller to indicate expected use.
 | 
						|
func (m *MockVariablesMetaGetter) EXPECT() *MockVariablesMetaGetterMockRecorder {
 | 
						|
	return m.recorder
 | 
						|
}
 | 
						|
 | 
						|
// GetAgentVariablesMeta mocks base method.
 | 
						|
func (m *MockVariablesMetaGetter) GetAgentVariablesMeta(ctx context.Context, id int64, version string) (map[string]*vo.TypeInfo, error) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "GetAgentVariablesMeta", ctx, id, version)
 | 
						|
	ret0, _ := ret[0].(map[string]*vo.TypeInfo)
 | 
						|
	ret1, _ := ret[1].(error)
 | 
						|
	return ret0, ret1
 | 
						|
}
 | 
						|
 | 
						|
// GetAgentVariablesMeta indicates an expected call of GetAgentVariablesMeta.
 | 
						|
func (mr *MockVariablesMetaGetterMockRecorder) GetAgentVariablesMeta(ctx, id, version any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAgentVariablesMeta", reflect.TypeOf((*MockVariablesMetaGetter)(nil).GetAgentVariablesMeta), ctx, id, version)
 | 
						|
}
 | 
						|
 | 
						|
// GetAppVariablesMeta mocks base method.
 | 
						|
func (m *MockVariablesMetaGetter) GetAppVariablesMeta(ctx context.Context, id, version string) (map[string]*vo.TypeInfo, error) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "GetAppVariablesMeta", ctx, id, version)
 | 
						|
	ret0, _ := ret[0].(map[string]*vo.TypeInfo)
 | 
						|
	ret1, _ := ret[1].(error)
 | 
						|
	return ret0, ret1
 | 
						|
}
 | 
						|
 | 
						|
// GetAppVariablesMeta indicates an expected call of GetAppVariablesMeta.
 | 
						|
func (mr *MockVariablesMetaGetterMockRecorder) GetAppVariablesMeta(ctx, id, version any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAppVariablesMeta", reflect.TypeOf((*MockVariablesMetaGetter)(nil).GetAppVariablesMeta), ctx, id, version)
 | 
						|
}
 |