92 lines
2.9 KiB
Go
92 lines
2.9 KiB
Go
package repository
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
|
||
"yingce/backend/internal/model"
|
||
|
||
"gorm.io/driver/sqlite"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// 验证 text-replay 闭环：text_replay 任务可写增量，完成后置为 succeeded 并写入最终正文。
|
||
func TestTextReplayLifecycle(t *testing.T) {
|
||
db, err := gorm.Open(sqlite.Open("file:text-replay-lifecycle?mode=memory&cache=shared"), &gorm.Config{})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := db.AutoMigrate(&model.Task{}, &model.TaskTextDelta{}); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
task := model.Task{
|
||
ID: "replay-1", UserID: "user-1", Type: "text", Prompt: "写一段剧本",
|
||
Status: model.TaskStatusTextReplay, Stage: "文本持久化（前端自管）",
|
||
}
|
||
if err := db.Create(&task).Error; err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
repo := New(db)
|
||
now := time.Now()
|
||
limits := TextReplayLimits{MaxTaskBytes: 1 << 20, MaxUserBytes: 1 << 20, MaxTaskEvents: 100}
|
||
|
||
// text_replay 状态应允许写入增量（未结束）
|
||
if _, err := repo.AppendTaskTextDelta("user-1", "replay-1", "第一段内容", now.Add(time.Hour), limits); err != nil {
|
||
t.Fatalf("text_replay 任务写入增量失败: %v", err)
|
||
}
|
||
if _, err := repo.AppendTaskTextDelta("user-1", "replay-1", "第二段内容", now.Add(time.Hour), limits); err != nil {
|
||
t.Fatalf("text_replay 任务写入增量失败: %v", err)
|
||
}
|
||
|
||
// 完成：写入最终正文并置为 succeeded
|
||
resultJSON := `{"mode":"text","text":"第一段内容第二段内容"}`
|
||
completed, err := repo.CompleteTextReplayTask("user-1", "replay-1", resultJSON, now)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if !completed {
|
||
t.Fatal("expected complete to succeed")
|
||
}
|
||
stored, err := repo.Task("replay-1")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if stored.Status != model.TaskStatusSucceeded {
|
||
t.Fatalf("expected status succeeded, got %s", stored.Status)
|
||
}
|
||
if stored.ResultJSON != resultJSON {
|
||
t.Fatalf("unexpected result_json: %s", stored.ResultJSON)
|
||
}
|
||
if stored.CompletedAt == nil {
|
||
t.Fatal("expected completed_at set")
|
||
}
|
||
|
||
// 已完成的 text_replay 任务不能再写入增量（进入 closed）
|
||
if _, err := repo.AppendTaskTextDelta("user-1", "replay-1", "追加", now.Add(time.Hour), limits); err == nil {
|
||
t.Fatal("expected ErrTextReplayClosed after completion")
|
||
}
|
||
|
||
// 重复完成应返回 completed=false（状态已不是 text_replay）
|
||
again, err := repo.CompleteTextReplayTask("user-1", "replay-1", resultJSON, now.Add(time.Minute))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if again {
|
||
t.Fatal("expected second complete to be a no-op")
|
||
}
|
||
|
||
// 非 text_replay 任务不能完成
|
||
other := model.Task{ID: "queued-1", UserID: "user-1", Type: "text", Status: model.TaskStatusQueued}
|
||
if err := db.Create(&other).Error; err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
done, err := repo.CompleteTextReplayTask("user-1", "queued-1", resultJSON, now)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if done {
|
||
t.Fatal("queued task must not be completable as text_replay")
|
||
}
|
||
}
|