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)
|
|
}
|