t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
101 lines
3.4 KiB
Go
101 lines
3.4 KiB
Go
package services
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/models"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/dto"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/dto/request"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/enums"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/mlogclub/simple/sqls"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func TestUpdateAIAgentKeepsPublishedRevisionActive(t *testing.T) {
|
|
dbName := strings.NewReplacer("/", "_", " ", "_").Replace(t.Name())
|
|
db, err := gorm.Open(sqlite.Open("file:"+dbName+"?mode=memory&cache=shared"), &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{
|
|
TablePrefix: "t_",
|
|
SingularTable: true,
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("open sqlite: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
sqlDB, dbErr := db.DB()
|
|
if dbErr == nil {
|
|
_ = sqlDB.Close()
|
|
}
|
|
})
|
|
if err := db.AutoMigrate(&models.AIConfig{}, &models.AIAgent{}); err != nil {
|
|
t.Fatalf("auto migrate: %v", err)
|
|
}
|
|
sqls.SetDB(db)
|
|
|
|
config := &models.AIConfig{
|
|
Name: "test", Status: enums.StatusOk, Provider: enums.AIProviderOpenAI,
|
|
ModelType: enums.AIModelTypeLLM, ModelName: "test-model",
|
|
}
|
|
if err := db.Create(config).Error; err != nil {
|
|
t.Fatalf("create ai config: %v", err)
|
|
}
|
|
createdModel, err := AIAgentService.buildAIAgentModel(0, request.CreateAIAgentRequest{
|
|
Name: "fixed ai reception agent",
|
|
Avatar: " https://cdn.example.com/agent.png ",
|
|
AIConfigID: config.ID,
|
|
ServiceMode: enums.IMConversationServiceModeHumanOnly,
|
|
HandoffMode: enums.AIAgentHandoffModeWaitPool,
|
|
FallbackMode: enums.AIAgentFallbackModeNoAnswer,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("build ai agent model: %v", err)
|
|
}
|
|
if createdModel.ServiceMode != enums.IMConversationServiceModeAIFirst {
|
|
t.Fatalf("service mode = %d, want AI first", createdModel.ServiceMode)
|
|
}
|
|
if createdModel.Avatar != "https://cdn.example.com/agent.png" {
|
|
t.Fatalf("avatar = %q, want trimmed avatar URL", createdModel.Avatar)
|
|
}
|
|
agent := &models.AIAgent{
|
|
Name: "published agent", Status: enums.StatusOk, AIConfigID: config.ID,
|
|
ServiceMode: enums.IMConversationServiceModeAIFirst, HandoffMode: enums.AIAgentHandoffModeWaitPool,
|
|
FallbackMode: enums.AIAgentFallbackModeNoAnswer, RolloutPercent: 100, PublishedRevisionID: 18,
|
|
}
|
|
if err := db.Create(agent).Error; err != nil {
|
|
t.Fatalf("create ai agent: %v", err)
|
|
}
|
|
|
|
err = AIAgentService.UpdateAIAgent(request.UpdateAIAgentRequest{
|
|
ID: agent.ID,
|
|
CreateAIAgentRequest: request.CreateAIAgentRequest{
|
|
Name: "updated draft", Avatar: "https://cdn.example.com/updated.png", AIConfigID: config.ID,
|
|
ServiceMode: enums.IMConversationServiceModeAIFirst,
|
|
HandoffMode: enums.AIAgentHandoffModeWaitPool,
|
|
FallbackMode: enums.AIAgentFallbackModeNoAnswer,
|
|
RolloutPercent: 100,
|
|
},
|
|
}, &dto.AuthPrincipal{UserID: 1, Username: "admin"})
|
|
if err != nil {
|
|
t.Fatalf("UpdateAIAgent: %v", err)
|
|
}
|
|
|
|
var updated models.AIAgent
|
|
if err := db.First(&updated, agent.ID).Error; err != nil {
|
|
t.Fatalf("get updated ai agent: %v", err)
|
|
}
|
|
if updated.PublishedRevisionID != agent.PublishedRevisionID {
|
|
t.Fatalf("published revision id = %d, want %d", updated.PublishedRevisionID, agent.PublishedRevisionID)
|
|
}
|
|
if updated.Name != "updated draft" {
|
|
t.Fatalf("draft name = %q, want updated draft", updated.Name)
|
|
}
|
|
if updated.Avatar != "https://cdn.example.com/updated.png" {
|
|
t.Fatalf("draft avatar = %q, want updated avatar", updated.Avatar)
|
|
}
|
|
}
|