t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
49 lines
1.4 KiB
Go
49 lines
1.4 KiB
Go
package aiagent
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/models"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func TestSyncSchemaCreatesEveryRegisteredTableWithModulePrefix(t *testing.T) {
|
|
hostDB, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{TablePrefix: "host_", SingularTable: true},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("open database: %v", err)
|
|
}
|
|
if err := SyncSchema(hostDB, "iot_ai_"); err != nil {
|
|
t.Fatalf("SyncSchema() error = %v", err)
|
|
}
|
|
|
|
var tableNames []string
|
|
if err := hostDB.Raw("SELECT name FROM sqlite_master WHERE type = 'table' AND name LIKE 'iot_ai_%'").Scan(&tableNames).Error; err != nil {
|
|
t.Fatalf("list AI Agent tables: %v", err)
|
|
}
|
|
if len(tableNames) != len(models.Models) {
|
|
t.Fatalf("AI Agent table count = %d, registered model count = %d", len(tableNames), len(models.Models))
|
|
}
|
|
|
|
for _, forbidden := range []string{
|
|
"iot_ai_migration",
|
|
"iot_ai_user",
|
|
"iot_ai_admin",
|
|
"iot_ai_role",
|
|
"iot_ai_permission",
|
|
"iot_ai_token",
|
|
} {
|
|
var count int64
|
|
if err := hostDB.Raw("SELECT COUNT(1) FROM sqlite_master WHERE type = 'table' AND name = ?", forbidden).Scan(&count).Error; err != nil {
|
|
t.Fatalf("check forbidden table %s: %v", forbidden, err)
|
|
}
|
|
if count != 0 {
|
|
t.Fatalf("forbidden table %s must not be created", forbidden)
|
|
}
|
|
}
|
|
}
|