95 lines
3.1 KiB
Go
95 lines
3.1 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"yingce/backend/internal/model"
|
|
)
|
|
|
|
// 线上复现："invalid Agent checkpoint (runtime validation): Agent runtime has multiple active tasks"。
|
|
// 批准媒体生成后，媒体任务还在跑（MediaTaskID 已设置），运行时又请求下一步模型调用；
|
|
// 调度模型步骤会同时设置 ActiveTaskID，检查点校验拒绝，整轮失败。
|
|
// 修复后模型步骤必须先等媒体结果回写、释放 MediaTaskID，再入队。
|
|
func TestCloudAgentModelStepWaitsForPendingMediaTask(t *testing.T) {
|
|
s, db, a := agentMediaFixture(t)
|
|
run, state := agentMediaRun(t, s, a, "auto")
|
|
if err := s.advanceCloudAgentTool(run, &state); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
run, err := s.repo.CloudAgent("user", run.ID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
state, err = cloudAgentDecode(run)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// 媒体生成在所有模式下都要先审批；批准后才提交媒体任务。
|
|
if state.Approval == nil {
|
|
t.Fatalf("media did not request approval (status=%s)", run.Status)
|
|
}
|
|
s.disablePiRuntime = true
|
|
if err := s.DecideCloudAgentApproval("user", run.ID, state.Approval.ID, "approve", ""); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if run, err = s.repo.CloudAgent("user", run.ID); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if state, err = cloudAgentDecode(run); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if state.MediaTaskID == "" {
|
|
t.Fatalf("fixture did not submit a media task (status=%s failure=%q)", run.Status, run.FailureMessage)
|
|
}
|
|
mediaTaskID := state.MediaTaskID
|
|
|
|
// 模拟媒体任务稍后才结束。
|
|
go func() {
|
|
time.Sleep(300 * time.Millisecond)
|
|
_ = db.Model(&model.Task{}).Where("id = ?", mediaTaskID).Updates(map[string]any{"status": model.TaskStatusFailed, "error": "HTTP 502 Bad Gateway"}).Error
|
|
}()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
messages := []map[string]any{{"role": "user", "content": "继续"}}
|
|
done := make(chan error, 1)
|
|
go func() {
|
|
_, _, stepErr := s.runCloudAgentModelStep(ctx, "user", run.ID, messages, "off")
|
|
done <- stepErr
|
|
}()
|
|
|
|
// 模型步骤在等待期间，检查点里绝不能同时出现两个活动任务。
|
|
deadline := time.Now().Add(8 * time.Second)
|
|
scheduled := false
|
|
for time.Now().Before(deadline) && !scheduled {
|
|
latest, readErr := s.repo.CloudAgent("user", run.ID)
|
|
if readErr != nil {
|
|
t.Fatal(readErr)
|
|
}
|
|
current, decodeErr := cloudAgentDecode(latest)
|
|
if decodeErr != nil {
|
|
t.Fatalf("checkpoint became invalid while waiting for media: %v", decodeErr)
|
|
}
|
|
if current.ActiveTaskID != "" && current.MediaTaskID != "" {
|
|
t.Fatalf("model step scheduled while media task %s still active", current.MediaTaskID)
|
|
}
|
|
if current.ActiveTaskID != "" {
|
|
scheduled = true
|
|
if current.MediaTaskID != "" {
|
|
t.Fatal("media task not released before model step")
|
|
}
|
|
}
|
|
if latest.Status == "failed" {
|
|
t.Fatalf("run failed while waiting for media: %q", latest.FailureMessage)
|
|
}
|
|
time.Sleep(50 * time.Millisecond)
|
|
}
|
|
if !scheduled {
|
|
t.Fatal("model step was never scheduled after the media task finished")
|
|
}
|
|
cancel()
|
|
<-done
|
|
}
|