78 lines
2.7 KiB
Go
78 lines
2.7 KiB
Go
package app
|
||
|
||
import (
|
||
"strings"
|
||
"testing"
|
||
|
||
"infinite-canvas/backend/internal/model"
|
||
)
|
||
|
||
func TestCloudAgentContinuationReplySplitsFactsFromAssistantText(t *testing.T) {
|
||
task := &model.Task{Status: model.TaskStatusSucceeded, ResultJSON: `{"text":"已完成"}`}
|
||
run := &CloudAgentRun{
|
||
ID: "ag1",
|
||
Status: "failed",
|
||
Events: []CloudAgentEvent{
|
||
{Type: "assistant_message", Payload: map[string]any{"text": "先看画布"}},
|
||
{Type: "tool_completed", Payload: map[string]any{"toolName": "canvas_get_state", "nodeId": "n1"}},
|
||
{Type: "run_failed", Payload: map[string]any{"reason": "cancelled"}},
|
||
},
|
||
}
|
||
reply, context, err := cloudAgentContinuationReply(task, run)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if reply != "先看画布" {
|
||
t.Fatalf("reply 应是上一轮 assistant 正文，got %q", reply)
|
||
}
|
||
if context == "" {
|
||
t.Fatal("失败轮次必须留下收束摘要")
|
||
}
|
||
if strings.Contains(reply, "上一轮已结束") || strings.Contains(reply, "canvas_get_state") {
|
||
t.Fatal("摘要不能拼进 assistant 历史")
|
||
}
|
||
if strings.Contains(context, "canvas_get_state") || strings.Contains(context, `"event"`) {
|
||
t.Fatalf("摘要不得把工具流水当成本轮目标：%s", context)
|
||
}
|
||
if !strings.Contains(context, "上一轮已结束（failed）") {
|
||
t.Fatalf("失败摘要不对：%s", context)
|
||
}
|
||
}
|
||
|
||
func TestCloudAgentContinuationOmitsQuietCompletedTurns(t *testing.T) {
|
||
task := &model.Task{Status: model.TaskStatusSucceeded, ResultJSON: `{"text":"好"}`}
|
||
run := &CloudAgentRun{
|
||
Status: "completed",
|
||
Events: []CloudAgentEvent{
|
||
{Type: "assistant_message", Payload: map[string]any{"text": "好"}},
|
||
{Type: "tool_completed", Payload: map[string]any{"toolName": "canvas_get_state"}},
|
||
},
|
||
}
|
||
reply, context, err := cloudAgentContinuationReply(task, run)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if reply != "好" || context != "" {
|
||
t.Fatalf("安静完成的一轮不应再塞执行流水：reply=%q context=%q", reply, context)
|
||
}
|
||
}
|
||
|
||
func TestCloudAgentContinuationKeepsSubmittedTaskIDs(t *testing.T) {
|
||
task := &model.Task{Status: model.TaskStatusSucceeded}
|
||
run := &CloudAgentRun{
|
||
Status: "completed",
|
||
Events: []CloudAgentEvent{
|
||
{Type: "assistant_message", Payload: map[string]any{"text": "已提交"}},
|
||
{Type: "generation_task_created", Payload: map[string]any{"taskId": "task-1"}},
|
||
{Type: "tool_completed", Payload: map[string]any{"result": map[string]any{"taskId": "task-2", "taskSubmitted": true}}},
|
||
},
|
||
}
|
||
_, context, err := cloudAgentContinuationReply(task, run)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if !strings.Contains(context, "task-1") || !strings.Contains(context, "task-2") || !strings.Contains(context, "已提交生成任务") {
|
||
t.Fatalf("已提交任务应留下防重发提示：%s", context)
|
||
}
|
||
}
|