72 lines
2.0 KiB
Go
72 lines
2.0 KiB
Go
package app
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"yingce/backend/internal/model"
|
|
"yingce/backend/internal/repository"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// 目录始终来自系统渠道；空目录也要显式返回数组，不能被会话初始化判成畸形响应。
|
|
func TestModelCatalogAlwaysSerializesCollectionsAsArrays(t *testing.T) {
|
|
for _, test := range []struct {
|
|
name string
|
|
frontendEnabled bool
|
|
wantSource string
|
|
}{
|
|
{name: "逻辑模型开关不改变目录来源", frontendEnabled: true, wantSource: "system"},
|
|
{name: "空的系统渠道目录", frontendEnabled: false, wantSource: "system"},
|
|
} {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
svc, db := newModelCatalogTestService(t)
|
|
if test.frontendEnabled {
|
|
if err := db.Create(&model.SystemSetting{Key: featureAvailabilitySettingKey, ValueJSON: `{"frontendModelsEnabled":true}`}).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
catalog, err := svc.ModelCatalog(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if string(catalog.Source) != test.wantSource {
|
|
t.Fatalf("source = %q, want %q", catalog.Source, test.wantSource)
|
|
}
|
|
|
|
encoded, err := json.Marshal(catalog)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var payload map[string]json.RawMessage
|
|
if err := json.Unmarshal(encoded, &payload); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, key := range []string{"models", "channels"} {
|
|
raw, present := payload[key]
|
|
if !present || string(raw) == "null" {
|
|
t.Fatalf("%s 集合没有序列化成数组：%s", key, encoded)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func newModelCatalogTestService(t *testing.T) (*Service, *gorm.DB) {
|
|
t.Helper()
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if sqlDB, err := db.DB(); err == nil {
|
|
sqlDB.SetMaxOpenConns(1)
|
|
}
|
|
if err := db.AutoMigrate(&model.SystemSetting{}, &model.AdminAuditEvent{}, &model.ModelChannel{}, &model.ChannelModel{}, &model.LogicalModel{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return New(repository.New(db), t.TempDir()), db
|
|
}
|