t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
102 lines
3.6 KiB
Go
102 lines
3.6 KiB
Go
package services
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/models"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/dto/request"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/enums"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/repositories"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/mlogclub/simple/sqls"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestBuildKnowledgeBaseModelUsesLowerDefaultScoreThreshold(t *testing.T) {
|
|
item, err := KnowledgeBaseService.buildKnowledgeBaseModel(request.CreateKnowledgeBaseRequest{})
|
|
if err != nil {
|
|
t.Fatalf("build knowledge base model failed: %v", err)
|
|
}
|
|
if item.DefaultScoreThreshold != 0.2 {
|
|
t.Fatalf("expected default score threshold 0.2, got %v", item.DefaultScoreThreshold)
|
|
}
|
|
}
|
|
|
|
func TestDeleteKnowledgeBaseCascadesContentWhenNotReferenced(t *testing.T) {
|
|
setupKnowledgeBaseServiceTestDB(t)
|
|
kb := createKnowledgeBaseServiceTestBase(t, "Delete KB")
|
|
document := &models.KnowledgeDocument{
|
|
KnowledgeBaseID: kb.ID,
|
|
Title: "Doc",
|
|
ContentType: enums.KnowledgeDocumentContentTypeMarkdown,
|
|
Content: "content",
|
|
Status: enums.StatusOk,
|
|
IndexStatus: enums.KnowledgeDocumentIndexStatusIndexed,
|
|
}
|
|
if err := repositories.KnowledgeDocumentRepository.Create(sqls.DB(), document); err != nil {
|
|
t.Fatalf("create document: %v", err)
|
|
}
|
|
faq := &models.KnowledgeFAQ{
|
|
KnowledgeBaseID: kb.ID,
|
|
Question: "Question",
|
|
Answer: "Answer",
|
|
Status: enums.StatusOk,
|
|
IndexStatus: enums.KnowledgeDocumentIndexStatusIndexed,
|
|
}
|
|
if err := repositories.KnowledgeFAQRepository.Create(sqls.DB(), faq); err != nil {
|
|
t.Fatalf("create faq: %v", err)
|
|
}
|
|
if err := repositories.KnowledgeChunkRepository.BatchCreate(sqls.DB(), []models.KnowledgeChunk{
|
|
{KnowledgeBaseID: kb.ID, DocumentID: document.ID, ChunkNo: 1, Status: enums.StatusOk},
|
|
{KnowledgeBaseID: kb.ID, FaqID: faq.ID, ChunkNo: 1, Status: enums.StatusOk},
|
|
}); err != nil {
|
|
t.Fatalf("create chunks: %v", err)
|
|
}
|
|
|
|
if err := KnowledgeBaseService.DeleteKnowledgeBase(kb.ID); err != nil {
|
|
t.Fatalf("DeleteKnowledgeBase() error = %v", err)
|
|
}
|
|
|
|
assertKnowledgeBaseServiceTestCount(t, &models.KnowledgeBase{}, "id = ?", kb.ID, 0)
|
|
assertKnowledgeBaseServiceTestCount(t, &models.KnowledgeDocument{}, "knowledge_base_id = ?", kb.ID, 0)
|
|
assertKnowledgeBaseServiceTestCount(t, &models.KnowledgeFAQ{}, "knowledge_base_id = ?", kb.ID, 0)
|
|
assertKnowledgeBaseServiceTestCount(t, &models.KnowledgeChunk{}, "knowledge_base_id = ?", kb.ID, 0)
|
|
}
|
|
|
|
func setupKnowledgeBaseServiceTestDB(t *testing.T) {
|
|
t.Helper()
|
|
db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatalf("open sqlite db: %v", err)
|
|
}
|
|
if err := db.AutoMigrate(&models.KnowledgeBase{}, &models.KnowledgeDocument{}, &models.KnowledgeFAQ{}, &models.KnowledgeChunk{}, &models.AIAgent{}); err != nil {
|
|
t.Fatalf("auto migrate: %v", err)
|
|
}
|
|
sqls.SetDB(db)
|
|
}
|
|
|
|
func createKnowledgeBaseServiceTestBase(t *testing.T, name string) *models.KnowledgeBase {
|
|
t.Helper()
|
|
item := &models.KnowledgeBase{
|
|
Name: name,
|
|
KnowledgeType: string(enums.KnowledgeBaseTypeDocument),
|
|
Status: enums.StatusOk,
|
|
}
|
|
if err := repositories.KnowledgeBaseRepository.Create(sqls.DB(), item); err != nil {
|
|
t.Fatalf("create knowledge base: %v", err)
|
|
}
|
|
return item
|
|
}
|
|
|
|
func assertKnowledgeBaseServiceTestCount(t *testing.T, model any, query string, arg any, want int64) {
|
|
t.Helper()
|
|
var count int64
|
|
if err := sqls.DB().Model(model).Where(query, arg).Count(&count).Error; err != nil {
|
|
t.Fatalf("count %T: %v", model, err)
|
|
}
|
|
if count != want {
|
|
t.Fatalf("count %T = %d, want %d", model, count, want)
|
|
}
|
|
}
|