241 lines
11 KiB
Go
241 lines
11 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
"unicode/utf8"
|
||
)
|
||
|
||
func TestStoryboardCinematicQualityContractIncludesRequestedCountAndDuration(t *testing.T) {
|
||
contract := storyboardCinematicQualityContract(30, 7)
|
||
if !strings.Contains(contract, "严格等于 30 秒") {
|
||
t.Fatalf("contract does not include requested duration: %s", contract)
|
||
}
|
||
if !strings.Contains(contract, "严格输出 7 个镜头") {
|
||
t.Fatalf("contract does not include requested shot count: %s", contract)
|
||
}
|
||
}
|
||
|
||
func TestStoryboardCinematicQualityContractIncludesCameraLanguageGuide(t *testing.T) {
|
||
contract := storyboardCinematicQualityContract(0, 0)
|
||
for _, term := range []string{"先确定观众此刻跟随谁", "大远景只承担空间", "一个镜头只保留一个主运镜", "明确人物左右位置", "避免每镜都推进"} {
|
||
if !strings.Contains(contract, term) {
|
||
t.Fatalf("camera language guide is missing %q: %s", term, contract)
|
||
}
|
||
}
|
||
if !strings.Contains(contract, "1 到 12 个镜头") || !strings.Contains(contract, "所需的最少镜头数") {
|
||
t.Fatalf("automatic shot count contract is not bounded: %s", contract)
|
||
}
|
||
}
|
||
|
||
func TestStoryboardOutputTokenLimitTracksRequestedShotCount(t *testing.T) {
|
||
if got := storyboardOutputTokenLimit(1); got != 2800 {
|
||
t.Fatalf("one-shot token limit = %d, want 2800", got)
|
||
}
|
||
if got := storyboardOutputTokenLimit(10); got != 10000 {
|
||
t.Fatalf("ten-shot token limit = %d, want 10000", got)
|
||
}
|
||
if got := storyboardOutputTokenLimit(0); got != 12000 {
|
||
t.Fatalf("automatic token limit = %d, want 12000", got)
|
||
}
|
||
}
|
||
|
||
func TestStoryboardRepairContextRejectsInsufficientBudget(t *testing.T) {
|
||
ctx, cancel := context.WithTimeout(context.Background(), storyboardFinalizeReserve+storyboardMinimumRepairTime-time.Second)
|
||
defer cancel()
|
||
if _, _, err := storyboardRepairContext(ctx); err == nil {
|
||
t.Fatal("expected insufficient repair budget to fail")
|
||
}
|
||
}
|
||
|
||
func TestStoryboardPromptsLeaveAspectRatioToVideoNode(t *testing.T) {
|
||
contract := storyboardCinematicQualityContract(0, 0)
|
||
if !strings.Contains(contract, "不要讨论画幅配置") {
|
||
t.Fatalf("contract does not delegate aspect ratio: %s", contract)
|
||
}
|
||
if strings.Contains(defaultStoryboardPromptTemplate(), "2.39:1") {
|
||
t.Fatal("default storyboard prompt still hard-codes 2.39:1")
|
||
}
|
||
prompt := buildStoryboardVideoPrompt("【风格组合】仙侠·宏大叙事·3D 动漫", "高清、明亮、空间层次清楚", agentStoryboardShot{
|
||
Duration: 8, Description: "剑仙踏云进入画面", VisualPrompt: "剑仙位于画面左侧", VideoPrompt: "向前踏出一步后停住", Camera: "平视中景", Motion: "缓慢推近后停住", TimeBeats: "0-4秒：进入；4-8秒：停住",
|
||
})
|
||
if strings.Contains(prompt, "2.39:1") || strings.Contains(prompt, "画幅") {
|
||
t.Fatal("compiled storyboard video prompt still mentions output aspect ratio")
|
||
}
|
||
}
|
||
|
||
func TestRemoveFixedMediaRestrictionsKeepsCurrentProjectMedium(t *testing.T) {
|
||
template := "保留这一行\n- 不要在 visualPrompt 或 videoPrompt 中使用 3D动漫、动画、二次元、游戏CG、卡通渲染。\n继续保留"
|
||
filtered := removeFixedMediaRestrictions(template)
|
||
if strings.Contains(filtered, "不要在 visualPrompt") {
|
||
t.Fatal("fixed 3D/cartoon restriction was not removed")
|
||
}
|
||
if !strings.Contains(filtered, "保留这一行") || !strings.Contains(filtered, "继续保留") {
|
||
t.Fatal("unrelated template content was removed")
|
||
}
|
||
}
|
||
|
||
func TestValidateStoryboardShotCount(t *testing.T) {
|
||
plan := agentStoryboardPlan{Shots: make([]agentStoryboardShot, 3)}
|
||
if err := validateStoryboardShotCount(plan, 3); err != nil {
|
||
t.Fatalf("expected matching shot count to pass: %v", err)
|
||
}
|
||
if err := validateStoryboardShotCount(plan, 2); err == nil {
|
||
t.Fatal("expected mismatched shot count to fail")
|
||
}
|
||
if err := validateStoryboardShotCount(plan, 0); err != nil {
|
||
t.Fatalf("expected automatic shot count to pass: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestValidateStoryboardComplexityRejectsOverloadedShot(t *testing.T) {
|
||
plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{
|
||
Duration: 5, CharacterIDs: []string{"a", "b", "c"}, MustHave: []string{"一", "二", "三", "四"}, TimeBeats: "0-1秒：起，1-2秒：承，2-4秒：转，4-5秒：合", Motion: "航拍推进并升降",
|
||
}}}
|
||
if err := validateStoryboardComplexity(plan); err == nil {
|
||
t.Fatal("expected overloaded shot to fail complexity validation")
|
||
}
|
||
}
|
||
|
||
func TestValidateStoryboardPlanTreatsComplexityAsAdvisory(t *testing.T) {
|
||
plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{
|
||
Duration: 8,
|
||
CharacterIDs: []string{"a", "b", "c"},
|
||
MustHave: []string{"一", "二", "三", "四"},
|
||
Motion: "航拍推进并升降",
|
||
}}}
|
||
characters := []storyboardCharacterCard{{AssetID: "a", VersionID: "va", Name: "甲"}, {AssetID: "b", VersionID: "vb", Name: "乙"}, {AssetID: "c", VersionID: "vc", Name: "丙"}}
|
||
|
||
if err := validateStoryboardPlan(&plan, 0, 0, characters, nil); err != nil {
|
||
t.Fatalf("complexity should be advisory for an otherwise valid plan: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestValidateStoryboardContextAcceptsNameOnlyCharacter(t *testing.T) {
|
||
style := storyboardProjectStyle{PresetID: "style", Title: "都市", Prompt: "写实电影"}
|
||
characters := []storyboardCharacterCard{{Name: "张天昊", Definition: map[string]any{"role": "主角"}}}
|
||
if err := validateStoryboardContext(style, characters); err != nil {
|
||
t.Fatalf("name-only character should be accepted: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestValidateStoryboardContextRejectsPartialAssetIdentity(t *testing.T) {
|
||
style := storyboardProjectStyle{PresetID: "style", Title: "都市", Prompt: "写实电影"}
|
||
characters := []storyboardCharacterCard{{AssetID: "character-1", Name: "张天昊"}}
|
||
if err := validateStoryboardContext(style, characters); err == nil {
|
||
t.Fatal("character with only asset id should be rejected")
|
||
}
|
||
}
|
||
|
||
func TestNormalizeStoryboardCharacterReferencesFallsBackToNames(t *testing.T) {
|
||
plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{CharacterIDs: []string{"character-1", "张振天", "张天昊", "路人甲", "张天昊"}}}}
|
||
characters := []storyboardCharacterCard{
|
||
{AssetID: "character-1", VersionID: "version-1", Name: "张振天"},
|
||
{Name: "张天昊"},
|
||
}
|
||
|
||
if err := validateStoryboardPlan(&plan, 0, 0, characters, nil); err != nil {
|
||
t.Fatalf("character names should not invalidate the storyboard plan: %v", err)
|
||
}
|
||
|
||
shot := plan.Shots[0]
|
||
if len(shot.CharacterIDs) != 1 || shot.CharacterIDs[0] != "character-1" {
|
||
t.Fatalf("confirmed character should stay bound to asset: %#v", shot.CharacterIDs)
|
||
}
|
||
if len(shot.CharacterNames) != 2 || shot.CharacterNames[0] != "张天昊" || shot.CharacterNames[1] != "路人甲" {
|
||
t.Fatalf("pending and unknown characters should fall back to names: %#v", shot.CharacterNames)
|
||
}
|
||
if prompt := buildStoryboardImagePrompt("写实电影", "都市夜景", shot); !strings.Contains(prompt, "镜头角色：张振天、张天昊、路人甲") {
|
||
t.Fatalf("image prompt should keep character names after fallback: %s", prompt)
|
||
}
|
||
if prompt := buildStoryboardVideoPrompt("写实电影", "都市夜景", shot); !strings.Contains(prompt, "镜头角色：张振天、张天昊、路人甲") {
|
||
t.Fatalf("video prompt should keep character names after fallback: %s", prompt)
|
||
}
|
||
rows := storyboardRowCharacters(shot, characters)
|
||
if len(rows) != 3 {
|
||
t.Fatalf("expected three character references, got %#v", rows)
|
||
}
|
||
if rows[0]["characterAssetId"] != "character-1" || rows[1]["characterName"] != "张天昊" || rows[2]["characterName"] != "路人甲" {
|
||
t.Fatalf("unexpected character rows: %#v", rows)
|
||
}
|
||
if _, exists := rows[1]["characterAssetId"]; exists {
|
||
t.Fatalf("name-only character must not fabricate an asset id: %#v", rows[1])
|
||
}
|
||
}
|
||
|
||
func TestValidateStoryboardComplexityAcceptsSingleDirectedMove(t *testing.T) {
|
||
plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{
|
||
Duration: 8, CharacterIDs: []string{"a", "b"}, MustHave: []string{"身份稳定", "动作落点", "结尾状态"}, TimeBeats: "0-2秒：建立；2-6秒：动作；6-8秒：反应", Motion: "缓慢推近后停住", Dialogue: "你终于来了。",
|
||
}}}
|
||
if err := validateStoryboardComplexity(plan); err != nil {
|
||
t.Fatalf("expected focused shot to pass complexity validation: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestNormalizeAutomaticStoryboardDurationsPreservesLongDialogue(t *testing.T) {
|
||
plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{
|
||
Duration: 8,
|
||
Dialogue: strings.Repeat("字", 50),
|
||
}}}
|
||
|
||
normalizeAutomaticStoryboardDurations(&plan, 0)
|
||
|
||
if plan.Shots[0].Duration < 10 {
|
||
t.Fatalf("duration = %d, want at least 10", plan.Shots[0].Duration)
|
||
}
|
||
if err := validateStoryboardComplexity(plan); err != nil {
|
||
t.Fatalf("normalized automatic duration should pass: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestResolveStoryboardAssetsUsesStableCanvasNodeIDs(t *testing.T) {
|
||
assets := []storyboardAsset{
|
||
{ID: "environment", Title: "天宫", Tags: []string{"场景:天宫"}},
|
||
{ID: "character-node", Title: "青莲剑仙", CharacterAssetID: "character-1", CharacterVersionID: "version-3"},
|
||
}
|
||
resolved := resolveStoryboardAssets(assets, []storyboardAssetRef{{NodeID: "character-node", Role: "character", Priority: 100}, {NodeID: "environment", Role: "environment", Priority: 80}})
|
||
if len(resolved) != 2 || resolved[0].ID != "character-node" || resolved[1].ID != "environment" {
|
||
t.Fatalf("expected stable asset order from model references: %#v", resolved)
|
||
}
|
||
}
|
||
|
||
func TestValidateStoryboardAssetRefsRejectsInventedNodeID(t *testing.T) {
|
||
plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{AssetRefs: []storyboardAssetRef{{NodeID: "invented", Role: "prop", Priority: 50}}}}}
|
||
err := validateStoryboardAssetRefs(plan, []storyboardAsset{{ID: "real", Title: "真实道具"}})
|
||
if err == nil || !strings.Contains(err.Error(), "不在当前画布资产目录") {
|
||
t.Fatalf("expected invented node id to fail, got %v", err)
|
||
}
|
||
}
|
||
|
||
func TestExtractStoryboardAssetsIncludesMediaAndExcludesShotOutputs(t *testing.T) {
|
||
snapshot := map[string]any{"nodes": []interface{}{
|
||
map[string]interface{}{"id": "image", "type": "image", "title": "角色参考", "metadata": map[string]interface{}{"content": "resource:image", "assetCategory": "character"}},
|
||
map[string]interface{}{"id": "video", "type": "video", "title": "动作参考", "metadata": map[string]interface{}{"content": "resource:video"}},
|
||
map[string]interface{}{"id": "audio", "type": "audio", "title": "环境声", "metadata": map[string]interface{}{"content": "resource:audio"}},
|
||
map[string]interface{}{"id": "output", "type": "video", "title": "旧镜头", "metadata": map[string]interface{}{"content": "resource:output", "workflowKind": "shot"}},
|
||
}}
|
||
assets := extractStoryboardAssets(snapshot)
|
||
if len(assets) != 3 || assets[0].ID != "image" || assets[1].ID != "video" || assets[2].ID != "audio" {
|
||
t.Fatalf("unexpected assets: %#v", assets)
|
||
}
|
||
}
|
||
|
||
func TestNormalizeStoryboardAssetsBoundsAndDeduplicatesInput(t *testing.T) {
|
||
assets := normalizeStoryboardAssets([]storyboardAsset{
|
||
{ID: " asset-1 ", Title: " 场景图 ", Type: "IMAGE", Tags: []string{"雨夜", "雨夜", strings.Repeat("长", 80)}, Prompt: strings.Repeat("镜", 700)},
|
||
{ID: "asset-1", Title: "重复", Type: "image"},
|
||
{ID: "text-1", Title: "文本", Type: "text"},
|
||
})
|
||
if len(assets) != 1 {
|
||
t.Fatalf("expected one normalized asset, got %#v", assets)
|
||
}
|
||
if assets[0].ID != "asset-1" || assets[0].Type != "image" || len(assets[0].Tags) != 2 {
|
||
t.Fatalf("unexpected normalized asset: %#v", assets[0])
|
||
}
|
||
if utf8.RuneCountInString(assets[0].Prompt) != 600 || utf8.RuneCountInString(assets[0].Tags[1]) != 64 {
|
||
t.Fatalf("expected bounded prompt and tag lengths: %#v", assets[0])
|
||
}
|
||
}
|