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

52
backend/internal/logic/inbox/service_test.go
View File

@@ -21,9 +21,9 @@ import (
type inboxTestFixture struct {
database *gorm.DB
owner models.SenlinAgentUser
other models.SenlinAgentUser
project models.SenlinAgentProject
owner models.SaUser
other models.SaUser
project models.SaProject
}
type inboxSuggestionResponse struct {
@@ -38,7 +38,7 @@ type inboxAnalyzeResponse struct {
}
func TestStaticAnalyzerBuildsReviewableDraftsFromInboxContent(t *testing.T) {
item := models.SenlinAgentInboxItem{Title: "客户访谈", Body: "客户希望下周确认交付计划。"}
item := models.SaInboxItem{Title: "客户访谈", Body: "客户希望下周确认交付计划。"}
suggestions, err := (StaticAnalyzer{}).Analyze(item, 7)
@@ -70,7 +70,7 @@ func TestInboxCaptureUsesOwnedUUIDv7AndCamelCaseDTO(t *testing.T) {
require.Equal(t, "open", payload["status"])
requireUUIDv7(t, payload["id"].(string))
var item models.SenlinAgentInboxItem
var item models.SaInboxItem
require.NoError(t, fixture.database.Where("identity = ?", payload["id"]).First(&item).Error)
require.Equal(t, fixture.owner.ID, item.CreatedBy)
require.Equal(t, fixture.owner.Identity, item.CreatedByIdentity)
@@ -87,7 +87,7 @@ func TestInboxCaptureRejectsNumericAndUnownedProjectIdentities(t *testing.T) {
})
require.Equal(t, http.StatusBadRequest, numeric.Code)
otherProject := models.SenlinAgentProject{OwnerID: fixture.other.ID, Name: "Other", Identifier: "OTHER"}
otherProject := models.SaProject{OwnerID: fixture.other.ID, Name: "Other", Identifier: "OTHER"}
require.NoError(t, fixture.database.Create(&otherProject).Error)
unowned := performInboxJSON(t, router, http.MethodPost, "/api/v1/projects/"+otherProject.Identity+"/inbox", gin.H{
"sourceType": "text", "body": "not mine",
@@ -95,7 +95,7 @@ func TestInboxCaptureRejectsNumericAndUnownedProjectIdentities(t *testing.T) {
require.Equal(t, http.StatusNotFound, unowned.Code)
var count int64
require.NoError(t, fixture.database.Model(&models.SenlinAgentInboxItem{}).Count(&count).Error)
require.NoError(t, fixture.database.Model(&models.SaInboxItem{}).Count(&count).Error)
require.Zero(t, count)
}
@@ -116,7 +116,7 @@ func TestAnalyzeUsesInboxIdentityAndCreatesNoFormalObjects(t *testing.T) {
requireUUIDv7(t, analysis.Suggestions[0].ID)
requireFormalObjectCounts(t, fixture.database, 0, 0, 0)
var reloaded models.SenlinAgentInboxItem
var reloaded models.SaInboxItem
require.NoError(t, fixture.database.First(&reloaded, item.ID).Error)
require.Equal(t, "open", reloaded.Status)
}
@@ -148,21 +148,21 @@ func TestConfirmCreatesOnlySelectedSavedSuggestionsWithInboxIdentity(t *testing.
require.Equal(t, 3, result.CreatedCount)
requireFormalObjectCounts(t, fixture.database, 1, 1, 1)
var task models.SenlinAgentTask
var task models.SaTask
require.NoError(t, fixture.database.First(&task).Error)
require.Equal(t, "跟进报价", task.Title)
require.Equal(t, fixture.project.ID, task.ProjectID)
require.Equal(t, fixture.owner.ID, task.CreatedBy)
requireSourceInboxIdentity(t, item, task.SourceInboxItemID, task.SourceInboxItemIdentity)
var note models.SenlinAgentNote
var note models.SaNote
require.NoError(t, fixture.database.First(&note).Error)
require.Equal(t, "会议纪要", note.Title)
require.Equal(t, fixture.project.ID, note.ProjectID)
require.Equal(t, fixture.owner.ID, note.CreatedBy)
requireSourceInboxIdentity(t, item, note.SourceInboxItemID, note.SourceInboxItemIdentity)
var source models.SenlinAgentSource
var source models.SaSource
require.NoError(t, fixture.database.First(&source).Error)
require.Equal(t, "背景资料", source.Title)
require.Equal(t, "text", source.Kind)
@@ -173,9 +173,9 @@ func TestConfirmCreatesOnlySelectedSavedSuggestionsWithInboxIdentity(t *testing.
requireSourceInboxIdentity(t, item, source.SourceInboxItemID, source.SourceInboxItemIdentity)
var skipped int64
require.NoError(t, fixture.database.Model(&models.SenlinAgentTask{}).Where("title = ?", "不创建的任务").Count(&skipped).Error)
require.NoError(t, fixture.database.Model(&models.SaTask{}).Where("title = ?", "不创建的任务").Count(&skipped).Error)
require.Zero(t, skipped)
var reloaded models.SenlinAgentInboxItem
var reloaded models.SaInboxItem
require.NoError(t, fixture.database.First(&reloaded, item.ID).Error)
require.Equal(t, "processed", reloaded.Status)
}
@@ -234,7 +234,7 @@ func TestConfirmIsTransactionalWhenASelectedWriteFails(t *testing.T) {
const callbackName = "test:fail_inbox_source_create"
require.NoError(t, fixture.database.Callback().Create().Before("gorm:create").Register(callbackName, func(tx *gorm.DB) {
if tx.Statement.Schema != nil && tx.Statement.Schema.Name == "SenlinAgentSource" {
if tx.Statement.Schema != nil && tx.Statement.Schema.Name == "SaSource" {
tx.AddError(errors.New("simulated source write failure"))
}
}))
@@ -246,7 +246,7 @@ func TestConfirmIsTransactionalWhenASelectedWriteFails(t *testing.T) {
require.Equal(t, http.StatusInternalServerError, response.Code)
requireFormalObjectCounts(t, fixture.database, 0, 0, 0)
var reloaded models.SenlinAgentInboxItem
var reloaded models.SaInboxItem
require.NoError(t, fixture.database.First(&reloaded, item.ID).Error)
require.Equal(t, "open", reloaded.Status)
}
@@ -268,7 +268,7 @@ func TestRepeatedConfirmReturnsPersistedResultAfterRelatedObjectsChange(t *testi
requireFormalObjectCounts(t, fixture.database, 1, 1, 0)
sourceInboxItemID := item.ID
extraSource := models.SenlinAgentSource{
extraSource := models.SaSource{
ProjectID: item.ProjectID, CreatedBy: fixture.owner.ID, SourceInboxItemID: &sourceInboxItemID,
Kind: "text", Title: "后续关联资料", ContentText: "不得改变历史确认结果",
}
@@ -277,7 +277,7 @@ func TestRepeatedConfirmReturnsPersistedResultAfterRelatedObjectsChange(t *testi
require.Equal(t, http.StatusOK, second.Code, second.Body.String())
requireConfirmCreatedCount(t, second, 2)
var createdTask models.SenlinAgentTask
var createdTask models.SaTask
require.NoError(t, fixture.database.Where("source_inbox_item_id = ?", item.ID).First(&createdTask).Error)
require.NoError(t, fixture.database.Delete(&createdTask).Error)
require.NoError(t, fixture.database.Delete(&extraSource).Error)
@@ -297,9 +297,9 @@ func (fixture inboxTestFixture) router(userID uint, analyzer Analyzer) http.Hand
)
}
func (fixture inboxTestFixture) createInbox(t *testing.T, userID, projectID uint) models.SenlinAgentInboxItem {
func (fixture inboxTestFixture) createInbox(t *testing.T, userID, projectID uint) models.SaInboxItem {
t.Helper()
item := models.SenlinAgentInboxItem{
item := models.SaInboxItem{
ProjectID: projectID, CreatedBy: userID, SourceType: "text", Title: "待整理", Body: "需要整理的原始内容", Status: "open",
}
require.NoError(t, fixture.database.Create(&item).Error)
@@ -309,11 +309,11 @@ func (fixture inboxTestFixture) createInbox(t *testing.T, userID, projectID uint
func newInboxTestFixture(t *testing.T) inboxTestFixture {
t.Helper()
database := newTestDB(t)
owner := models.SenlinAgentUser{Email: t.Name() + "-owner@example.com", DisplayName: "Owner", PasswordHash: "hash"}
other := models.SenlinAgentUser{Email: t.Name() + "-other@example.com", DisplayName: "Other", PasswordHash: "hash"}
owner := models.SaUser{Email: t.Name() + "-owner@example.com", DisplayName: "Owner", PasswordHash: "hash"}
other := models.SaUser{Email: t.Name() + "-other@example.com", DisplayName: "Other", PasswordHash: "hash"}
require.NoError(t, database.Create(&owner).Error)
require.NoError(t, database.Create(&other).Error)
project := models.SenlinAgentProject{OwnerID: owner.ID, Name: "Inbox Project", Identifier: "INBOX"}
project := models.SaProject{OwnerID: owner.ID, Name: "Inbox Project", Identifier: "INBOX"}
require.NoError(t, database.Create(&project).Error)
return inboxTestFixture{database: database, owner: owner, other: other, project: project}
}
@@ -348,9 +348,9 @@ func requireFormalObjectCounts(t *testing.T, database *gorm.DB, tasks, notes, so
model any
want int64
}{
{model: &models.SenlinAgentTask{}, want: tasks},
{model: &models.SenlinAgentNote{}, want: notes},
{model: &models.SenlinAgentSource{}, want: sources},
{model: &models.SaTask{}, want: tasks},
{model: &models.SaNote{}, want: notes},
{model: &models.SaSource{}, want: sources},
} {
var count int64
require.NoError(t, database.Model(check.model).Count(&count).Error)
@@ -365,7 +365,7 @@ func requireConfirmCreatedCount(t *testing.T, response *httptest.ResponseRecorde
require.Equal(t, want, result.CreatedCount)
}
func requireSourceInboxIdentity(t *testing.T, item models.SenlinAgentInboxItem, sourceID *uint, sourceIdentity *string) {
func requireSourceInboxIdentity(t *testing.T, item models.SaInboxItem, sourceID *uint, sourceIdentity *string) {
t.Helper()
require.NotNil(t, sourceID)
require.Equal(t, item.ID, *sourceID)