t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package dashboard
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/models"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/mlogclub/simple/sqls"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestBuildAIAgentResponseExposesPublishedRevision(t *testing.T) {
|
|
setupAIAgentHandlerTestDB(t)
|
|
|
|
published := buildAIAgentResponse(&models.AIAgent{PublishedRevisionID: 12})
|
|
if published.PublishedRevisionID != 12 {
|
|
t.Fatalf("published revision = %d, want 12", published.PublishedRevisionID)
|
|
}
|
|
|
|
rollout := buildAIAgentResponse(&models.AIAgent{RolloutPercent: 20, PreviousRolloutPercent: 100})
|
|
if rollout.RolloutPercent != 20 || rollout.PreviousRolloutPercent != 100 {
|
|
t.Fatalf("unexpected rollout response: %#v", rollout)
|
|
}
|
|
}
|
|
|
|
func setupAIAgentHandlerTestDB(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)
|
|
}
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
t.Fatalf("get sqlite db: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
if err := sqlDB.Close(); err != nil {
|
|
t.Fatalf("close sqlite db: %v", err)
|
|
}
|
|
})
|
|
if err := db.AutoMigrate(
|
|
&models.AIConfig{},
|
|
&models.AgentTeam{},
|
|
&models.KnowledgeBase{},
|
|
); err != nil {
|
|
t.Fatalf("auto migrate: %v", err)
|
|
}
|
|
sqls.SetDB(db)
|
|
}
|