34 lines
1.1 KiB
Go
34 lines
1.1 KiB
Go
package flow
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"ai-agent/workflow/model/entity"
|
|
)
|
|
|
|
func TestPlanSegmentResumePositionBased(t *testing.T) {
|
|
params := []map[string]any{{"a": 1}, {"a": 2}, {"a": 3}}
|
|
saved := map[int]entity.SegmentRef{0: {Key: "video", URL: "url0"}, 2: {Key: "video", URL: "url2"}}
|
|
idxList, needGen := planSegmentResume(params, saved)
|
|
if len(idxList) != 3 || len(needGen) != 3 {
|
|
t.Fatalf("长度应等于段数")
|
|
}
|
|
if needGen[0] || !needGen[1] || needGen[2] {
|
|
t.Fatalf("位置0/2已保存应复用，位置1应生成: %v", needGen)
|
|
}
|
|
}
|
|
|
|
func TestMergeSegmentOutputsOrder(t *testing.T) {
|
|
idxList := []int{0, 1, 2}
|
|
needGen := []bool{false, true, false}
|
|
newRes := [][]map[string]any{nil, {{"video": "new1"}}, nil}
|
|
saved := map[int]entity.SegmentRef{0: {Key: "video", URL: "url0"}, 2: {Key: "video", URL: "url2"}}
|
|
merged := mergeSegmentOutputs(idxList, needGen, newRes, saved)
|
|
if len(merged) != 3 {
|
|
t.Fatalf("应合并 3 条: %v", merged)
|
|
}
|
|
if merged[0]["video"] != "url0" || merged[1]["video"] != "new1" || merged[2]["video"] != "url2" {
|
|
t.Fatalf("合并顺序应为 url0,new1,url2: %v", merged)
|
|
}
|
|
}
|