177 lines
		
	
	
		
			6.3 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			177 lines
		
	
	
		
			6.3 KiB
		
	
	
	
		
			Go
		
	
	
	
// Code generated by MockGen. DO NOT EDIT.
 | 
						|
// Source: knowledge.go
 | 
						|
//
 | 
						|
// Generated by this command:
 | 
						|
//
 | 
						|
//	mockgen -destination ../../mock/dal/dao/knowledge.go --package dao -source knowledge.go
 | 
						|
//
 | 
						|
 | 
						|
// Package dao is a generated GoMock package.
 | 
						|
package dao
 | 
						|
 | 
						|
import (
 | 
						|
	context "context"
 | 
						|
	reflect "reflect"
 | 
						|
 | 
						|
	gomock "go.uber.org/mock/gomock"
 | 
						|
	gorm "gorm.io/gorm"
 | 
						|
 | 
						|
	"github.com/coze-dev/coze-studio/backend/domain/knowledge/entity"
 | 
						|
	model "github.com/coze-dev/coze-studio/backend/domain/knowledge/internal/dal/model"
 | 
						|
)
 | 
						|
 | 
						|
// MockKnowledgeRepo is a mock of KnowledgeRepo interface.
 | 
						|
type MockKnowledgeRepo struct {
 | 
						|
	ctrl     *gomock.Controller
 | 
						|
	recorder *MockKnowledgeRepoMockRecorder
 | 
						|
}
 | 
						|
 | 
						|
// MockKnowledgeRepoMockRecorder is the mock recorder for MockKnowledgeRepo.
 | 
						|
type MockKnowledgeRepoMockRecorder struct {
 | 
						|
	mock *MockKnowledgeRepo
 | 
						|
}
 | 
						|
 | 
						|
// NewMockKnowledgeRepo creates a new mock instance.
 | 
						|
func NewMockKnowledgeRepo(ctrl *gomock.Controller) *MockKnowledgeRepo {
 | 
						|
	mock := &MockKnowledgeRepo{ctrl: ctrl}
 | 
						|
	mock.recorder = &MockKnowledgeRepoMockRecorder{mock}
 | 
						|
	return mock
 | 
						|
}
 | 
						|
 | 
						|
// EXPECT returns an object that allows the caller to indicate expected use.
 | 
						|
func (m *MockKnowledgeRepo) EXPECT() *MockKnowledgeRepoMockRecorder {
 | 
						|
	return m.recorder
 | 
						|
}
 | 
						|
 | 
						|
// Create mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) Create(ctx context.Context, knowledge *model.Knowledge) error {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "Create", ctx, knowledge)
 | 
						|
	ret0, _ := ret[0].(error)
 | 
						|
	return ret0
 | 
						|
}
 | 
						|
 | 
						|
// Create indicates an expected call of Create.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) Create(ctx, knowledge any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockKnowledgeRepo)(nil).Create), ctx, knowledge)
 | 
						|
}
 | 
						|
 | 
						|
// Delete mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) Delete(ctx context.Context, id int64) error {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "Delete", ctx, id)
 | 
						|
	ret0, _ := ret[0].(error)
 | 
						|
	return ret0
 | 
						|
}
 | 
						|
 | 
						|
// Delete indicates an expected call of Delete.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) Delete(ctx, id any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockKnowledgeRepo)(nil).Delete), ctx, id)
 | 
						|
}
 | 
						|
 | 
						|
// FilterEnableKnowledge mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) FilterEnableKnowledge(ctx context.Context, ids []int64) ([]*model.Knowledge, error) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "FilterEnableKnowledge", ctx, ids)
 | 
						|
	ret0, _ := ret[0].([]*model.Knowledge)
 | 
						|
	ret1, _ := ret[1].(error)
 | 
						|
	return ret0, ret1
 | 
						|
}
 | 
						|
 | 
						|
// FilterEnableKnowledge indicates an expected call of FilterEnableKnowledge.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) FilterEnableKnowledge(ctx, ids any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FilterEnableKnowledge", reflect.TypeOf((*MockKnowledgeRepo)(nil).FilterEnableKnowledge), ctx, ids)
 | 
						|
}
 | 
						|
 | 
						|
// FindKnowledgeByCondition mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) FindKnowledgeByCondition(ctx context.Context, opts *entity.WhereKnowledgeOption) ([]*model.Knowledge, int64, error) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "FindKnowledgeByCondition", ctx, opts)
 | 
						|
	ret0, _ := ret[0].([]*model.Knowledge)
 | 
						|
	ret1, _ := ret[1].(int64)
 | 
						|
	ret2, _ := ret[2].(error)
 | 
						|
	return ret0, ret1, ret2
 | 
						|
}
 | 
						|
 | 
						|
// FindKnowledgeByCondition indicates an expected call of FindKnowledgeByCondition.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) FindKnowledgeByCondition(ctx, opts any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FindKnowledgeByCondition", reflect.TypeOf((*MockKnowledgeRepo)(nil).FindKnowledgeByCondition), ctx, opts)
 | 
						|
}
 | 
						|
 | 
						|
// GetByID mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) GetByID(ctx context.Context, id int64) (*model.Knowledge, error) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "GetByID", ctx, id)
 | 
						|
	ret0, _ := ret[0].(*model.Knowledge)
 | 
						|
	ret1, _ := ret[1].(error)
 | 
						|
	return ret0, ret1
 | 
						|
}
 | 
						|
 | 
						|
// GetByID indicates an expected call of GetByID.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) GetByID(ctx, id any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByID", reflect.TypeOf((*MockKnowledgeRepo)(nil).GetByID), ctx, id)
 | 
						|
}
 | 
						|
 | 
						|
// InitTx mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) InitTx() (*gorm.DB, error) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "InitTx")
 | 
						|
	ret0, _ := ret[0].(*gorm.DB)
 | 
						|
	ret1, _ := ret[1].(error)
 | 
						|
	return ret0, ret1
 | 
						|
}
 | 
						|
 | 
						|
// InitTx indicates an expected call of InitTx.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) InitTx() *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InitTx", reflect.TypeOf((*MockKnowledgeRepo)(nil).InitTx))
 | 
						|
}
 | 
						|
 | 
						|
// MGetByID mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) MGetByID(ctx context.Context, ids []int64) ([]*model.Knowledge, error) {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "MGetByID", ctx, ids)
 | 
						|
	ret0, _ := ret[0].([]*model.Knowledge)
 | 
						|
	ret1, _ := ret[1].(error)
 | 
						|
	return ret0, ret1
 | 
						|
}
 | 
						|
 | 
						|
// MGetByID indicates an expected call of MGetByID.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) MGetByID(ctx, ids any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByID", reflect.TypeOf((*MockKnowledgeRepo)(nil).MGetByID), ctx, ids)
 | 
						|
}
 | 
						|
 | 
						|
// Update mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) Update(ctx context.Context, knowledge *model.Knowledge) error {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "Update", ctx, knowledge)
 | 
						|
	ret0, _ := ret[0].(error)
 | 
						|
	return ret0
 | 
						|
}
 | 
						|
 | 
						|
// Update indicates an expected call of Update.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) Update(ctx, knowledge any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockKnowledgeRepo)(nil).Update), ctx, knowledge)
 | 
						|
}
 | 
						|
 | 
						|
// UpdateWithTx mocks base method.
 | 
						|
func (m *MockKnowledgeRepo) UpdateWithTx(ctx context.Context, tx *gorm.DB, knowledgeID int64, updateMap map[string]any) error {
 | 
						|
	m.ctrl.T.Helper()
 | 
						|
	ret := m.ctrl.Call(m, "UpdateWithTx", ctx, tx, knowledgeID, updateMap)
 | 
						|
	ret0, _ := ret[0].(error)
 | 
						|
	return ret0
 | 
						|
}
 | 
						|
 | 
						|
// UpdateWithTx indicates an expected call of UpdateWithTx.
 | 
						|
func (mr *MockKnowledgeRepoMockRecorder) UpdateWithTx(ctx, tx, knowledgeID, updateMap any) *gomock.Call {
 | 
						|
	mr.mock.ctrl.T.Helper()
 | 
						|
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWithTx", reflect.TypeOf((*MockKnowledgeRepo)(nil).UpdateWithTx), ctx, tx, knowledgeID, updateMap)
 | 
						|
}
 |