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

feat(workflow): 工作流提供方 - 集成 RunningHub 与 ComfyUI Bridge

Browse Source
This commit is contained in:
yepwait
2026-08-26 15:02:24 +08:00
committed by ddCat
parent 6141bd688c
commit 86fc3fda2f
82 changed files with 11405 additions and 248 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
backend/internal/service/task_cancellation_test.go
View File

@@ -12,7 +12,7 @@ import (
"gorm.io/gorm"
)
func TestCancelTaskRejectsRunningTask(t *testing.T) {
func TestCancelTaskCancelsRunningTask(t *testing.T) {
db, err := gorm.Open(sqlite.Open("file:"+newID()+"?mode=memory&cache=shared"), &gorm.Config{})
if err != nil {
t.Fatal(err)
@@ -33,20 +33,24 @@ func TestCancelTaskRejectsRunningTask(t *testing.T) {
}
svc := &Service{repo: repository.New(db)}
if _, err := svc.CancelTask(context.Background(), task.UserID, task.ID); err == nil || err.Error() != "任务已发起，无法取消，请等待任务完成" {
cancelled, err := svc.CancelTask(context.Background(), task.UserID, task.ID)
if err != nil {
t.Fatalf("CancelTask() error = %v", err)
}
if cancelled.Status != model.TaskStatusCancelled {
t.Fatalf("CancelTask() status = %s, want cancelled", cancelled.Status)
}
stored, err := svc.repo.TaskForUser(task.UserID, task.ID)
if err != nil {
t.Fatal(err)
}
if stored.Status != model.TaskStatusRunning || stored.CompletedAt != nil {
t.Fatalf("running task changed after cancellation attempt: status=%s completedAt=%v", stored.Status, stored.CompletedAt)
if stored.Status != model.TaskStatusCancelled || stored.CompletedAt == nil {
t.Fatalf("running task was not cancelled: status=%s completedAt=%v", stored.Status, stored.CompletedAt)
}
}
func TestCancelTaskRejectsQueuedTaskWithoutChangingBillingState(t *testing.T) {
func TestCancelTaskCancelsQueuedTask(t *testing.T) {
db, err := gorm.Open(sqlite.Open("file:"+newID()+"?mode=memory&cache=shared"), &gorm.Config{})
if err != nil {
t.Fatal(err)
@@ -60,16 +64,20 @@ func TestCancelTaskRejectsQueuedTaskWithoutChangingBillingState(t *testing.T) {
}
svc := &Service{repo: repository.New(db)}
if _, err := svc.CancelTask(context.Background(), task.UserID, task.ID); err == nil || err.Error() != "任务已发起，无法取消，请等待任务完成" {
cancelled, err := svc.CancelTask(context.Background(), task.UserID, task.ID)
if err != nil {
t.Fatalf("CancelTask() error = %v", err)
}
if cancelled.Status != model.TaskStatusCancelled {
t.Fatalf("CancelTask() status = %s, want cancelled", cancelled.Status)
}
storedTask, err := svc.repo.TaskForUser(task.UserID, task.ID)
if err != nil {
t.Fatal(err)
}
if storedTask.Status != model.TaskStatusQueued || storedTask.CompletedAt != nil {
t.Fatalf("queued task changed after cancellation attempt: status=%s completedAt=%v", storedTask.Status, storedTask.CompletedAt)
if storedTask.Status != model.TaskStatusCancelled || storedTask.CompletedAt == nil {
t.Fatalf("queued task was not cancelled: status=%s completedAt=%v", storedTask.Status, storedTask.CompletedAt)
}
}
@@ -92,7 +100,7 @@ func TestCancelTaskRejectsTaskWithProviderRequestID(t *testing.T) {
}
svc := &Service{repo: repository.New(db)}
if _, err := svc.CancelTask(context.Background(), task.UserID, task.ID); err == nil || err.Error() != "任务已发起，无法取消，请等待任务完成" {
if _, err := svc.CancelTask(context.Background(), task.UserID, task.ID); err == nil || err.Error() != "任务当前状态为 failed，无法取消" {
t.Fatalf("CancelTask() error = %v", err)
}