tt/ttstd-infinite-canvas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: commit workspace updates

Browse Source
This commit is contained in:
ddCat
2026-09-14 01:11:56 +08:00
parent 01a2d171f5
commit 9727656b2b
491 changed files with 12826 additions and 72249 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
backend/internal/app/task_security_test.go
View File

@@ -34,7 +34,7 @@ func TestNormalizeTaskInputMakesTypedProviderConfigBillable(t *testing.T) {
if err != nil {
t.Fatal(err)
}
order, err := svc.taskBillingOrder("user-1", &model.Task{ID: "task-1", Type: "agent_storyboard"}, input)
order, err := svc.taskBillingOrder("user-1", &model.Task{ID: "task-1", Type: "canvas_text"}, input)
if err != nil {
t.Fatal(err)
}
@@ -75,33 +75,3 @@ func TestTaskInputRejectsInlineMedia(t *testing.T) {
t.Fatal("containsInlineMediaDataURL() = false")
}
}
func TestCreateSessionRemovesDraftWhenTaskCreationFails(t *testing.T) {
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
if err != nil {
t.Fatal(err)
}
if err := db.AutoMigrate(&model.SystemSetting{}, &model.Asset{}, &model.CanvasProject{}, &model.Session{}, &model.Message{}, &model.Task{}, &model.TaskLog{}, &model.Result{}, &model.ApiCallLog{}); err != nil {
t.Fatal(err)
}
for range 5 {
if err := db.Create(&model.Task{ID: newID(), UserID: "user-1", Status: model.TaskStatusQueued, Prompt: "queued"}).Error; err != nil {
t.Fatal(err)
}
}
svc := &Service{repo: repository.New(db), dataDir: t.TempDir()}
if _, err := svc.CreateSession("user-1", CreateSessionRequest{Prompt: "new session"}); err == nil {
t.Fatal("CreateSession() error = nil")
}
var sessionCount int64
var messageCount int64
if err := db.Model(&model.Session{}).Where("user_id = ?", "user-1").Count(&sessionCount).Error; err != nil {
t.Fatal(err)
}
if err := db.Model(&model.Message{}).Where("user_id = ?", "user-1").Count(&messageCount).Error; err != nil {
t.Fatal(err)
}
if sessionCount != 0 || messageCount != 0 {
t.Fatalf("draft counts = sessions:%d messages:%d", sessionCount, messageCount)
}
}