l_ai_knowledge/internal/application/service/common_test.go

28 lines
443 B
Go

package service
import (
"context"
"knowlege-lsxd/internal/types"
"knowlege-lsxd/internal/types/interfaces"
"github.com/stretchr/testify/mock"
)
type MockKnowledgeBaseRepo struct {
mock.Mock
interfaces.KnowledgeBaseRepository
}
func (m *MockKnowledgeBaseRepo) Create(ctx context.Context, kb *types.KnowledgeBase) error {
args := m.Called(ctx, kb)
return args.Error(0)
}
func must(err error) {
if err != nil {
panic(err)
}
}