senlinai/agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: shorten model and table prefixes

Browse Source
This commit is contained in:
yanweidong
2026-07-23 14:30:17 +08:00
parent 6e7be2f3a4
commit 0ca2898ac0
71 changed files with 707 additions and 638 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
backend/internal/initdb/expert_test.go
View File

@@ -16,10 +16,10 @@ func TestNewInitializesUserAndExpertCatalog(t *testing.T) {
require.NoError(t, New(database))
requireExpertCatalogCounts(t, database)
var user models.SenlinAgentUser
var user models.SaUser
require.NoError(t, database.Where("email = ?", rootUsername).First(&user).Error)
require.Equal(t, rootRole, user.Role)
var expert models.SenlinAgentAIExpertItem
var expert models.SaAIExpertItem
require.NoError(t, database.Preload("ExpertCategory").Where("slug = ?", "academic-geographer").First(&expert).Error)
require.Equal(t, "academic", expert.ExpertCategory.Slug)
require.Equal(t, "学术研究", expert.ExpertCategory.Name)
@@ -30,7 +30,7 @@ func TestInitExpertOnlyInsertsMissingRecords(t *testing.T) {
database := newExpertTestDatabase(t)
require.NoError(t, InitExpert(database))
var preserved models.SenlinAgentAIExpertItem
var preserved models.SaAIExpertItem
require.NoError(t, database.Where("slug = ?", "academic-geographer").First(&preserved).Error)
require.NoError(t, database.Model(&preserved).Updates(map[string]any{
"name": "保留名称",
@@ -38,7 +38,7 @@ func TestInitExpertOnlyInsertsMissingRecords(t *testing.T) {
"category_identity": "",
"reference": "",
}).Error)
require.NoError(t, database.Where("slug = ?", "academic-historian").Delete(&models.SenlinAgentAIExpertItem{}).Error)
require.NoError(t, database.Where("slug = ?", "academic-historian").Delete(&models.SaAIExpertItem{}).Error)
require.NoError(t, InitExpert(database))
requireExpertCatalogCounts(t, database)
@@ -48,7 +48,7 @@ func TestInitExpertOnlyInsertsMissingRecords(t *testing.T) {
require.NotZero(t, preserved.CategoryID)
require.NotEmpty(t, preserved.CategoryIdentity)
require.Equal(t, "https://ao.aiolaola.com/experts", preserved.Reference)
var restored models.SenlinAgentAIExpertItem
var restored models.SaAIExpertItem
require.NoError(t, database.Where("slug = ?", "academic-historian").First(&restored).Error)
}
@@ -58,9 +58,9 @@ func newExpertTestDatabase(t *testing.T) *gorm.DB {
database, err := gorm.Open(sqlite.Open(fmt.Sprintf("file:%s?mode=memory&cache=shared", t.Name())), &gorm.Config{})
require.NoError(t, err)
require.NoError(t, database.AutoMigrate(
&models.SenlinAgentUser{},
&models.SenlinAgentAIExpertCategory{},
&models.SenlinAgentAIExpertItem{},
&models.SaUser{},
&models.SaAIExpertCategory{},
&models.SaAIExpertItem{},
))
return database
}
@@ -69,9 +69,9 @@ func requireExpertCatalogCounts(t *testing.T, database *gorm.DB) {
t.Helper()
var categoryCount int64
require.NoError(t, database.Model(&models.SenlinAgentAIExpertCategory{}).Count(&categoryCount).Error)
require.NoError(t, database.Model(&models.SaAIExpertCategory{}).Count(&categoryCount).Error)
require.Equal(t, int64(19), categoryCount)
var expertCount int64
require.NoError(t, database.Model(&models.SenlinAgentAIExpertItem{}).Count(&expertCount).Error)
require.NoError(t, database.Model(&models.SaAIExpertItem{}).Count(&expertCount).Error)
require.Equal(t, int64(267), expertCount)
}