83 lines
3.6 KiB
Go
83 lines
3.6 KiB
Go
package service
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestPromptOperationCatalogContainsProductJobs(t *testing.T) {
|
|
want := []string{
|
|
promptOperationChapterAssetsExtract,
|
|
promptOperationStoryboardPlan,
|
|
promptOperationStoryboardRepair,
|
|
promptOperationStoryboardFirstFrame,
|
|
promptOperationStoryboardVideo,
|
|
promptOperationCharacterExtract,
|
|
promptOperationCharacterTurnaround,
|
|
promptOperationShortDramaOutline,
|
|
promptOperationSkillDraft,
|
|
}
|
|
got := make(map[string]bool, len(want))
|
|
for _, definition := range defaultPromptDefinitions() {
|
|
got[definition.Operation] = true
|
|
}
|
|
for _, operation := range want {
|
|
if !got[operation] {
|
|
t.Fatalf("prompt catalog missing operation %s", operation)
|
|
}
|
|
}
|
|
if len(got) != len(want) {
|
|
t.Fatalf("prompt catalog size = %d, want %d; update this lock list when adding an operation", len(got), len(want))
|
|
}
|
|
}
|
|
|
|
func TestRenderShortDramaTemplateSubstitutesVariables(t *testing.T) {
|
|
definition, ok := promptDefinition(promptOperationShortDramaOutline)
|
|
if !ok {
|
|
t.Fatal("short_drama_outline definition missing")
|
|
}
|
|
rendered, err := renderPromptTemplate(definition, definition.DefaultContent, map[string]string{
|
|
"章节数量": "5",
|
|
"叙事结构": "单线推进",
|
|
"每章字数": "800",
|
|
"叙事视角": "第三人称",
|
|
"整体基调": "平稳叙事",
|
|
"角色规模": "3-4 个",
|
|
"章节篇幅": "中",
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("renderPromptTemplate() error = %v", err)
|
|
}
|
|
if !strings.Contains(rendered, "5 个章节") || !strings.Contains(rendered, "单线推进") || strings.Contains(rendered, "{{") {
|
|
t.Fatalf("renderPromptTemplate() = %q, want interpolated short-drama instruction", rendered)
|
|
}
|
|
protected := protectedPromptContext(promptOperationShortDramaOutline, map[string]string{"用户故事": "租客发现房东不是人"})
|
|
if !strings.Contains(protected, "租客发现房东不是人") || !strings.Contains(protected, "short-drama-outline/v1") {
|
|
t.Fatalf("protectedPromptContext() = %q, want story and JSON contract", protected)
|
|
}
|
|
}
|
|
|
|
func TestValidatePromptTemplateResultShortDramaOutline(t *testing.T) {
|
|
ok := `{"title":"夜灯","synopsis":"租客发现房东不是人","chapters":[{"title":"入住","content":"林夏搬进旧公寓。"}]}`
|
|
if err := validatePromptTemplateResult(promptOperationShortDramaOutline, map[string]interface{}{"text": "```json\n" + ok + "\n```"}); err != nil {
|
|
t.Fatalf("valid outline error = %v", err)
|
|
}
|
|
if err := validatePromptTemplateResult(promptOperationShortDramaOutline, map[string]interface{}{"text": `{"title":"夜灯","synopsis":"简介","chapters":[]}`}); err == nil || !strings.Contains(err.Error(), "没有生成任何章节") {
|
|
t.Fatalf("empty chapters error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestValidatePromptTemplateResultSkillDraft(t *testing.T) {
|
|
ok := `{"skillName":"分镜节奏","tag":"drama","description":"把章节拆成可执行镜头","instruction":"角色设定：分镜导演。"}`
|
|
if err := validatePromptTemplateResult(promptOperationSkillDraft, map[string]interface{}{"text": ok}); err != nil {
|
|
t.Fatalf("valid skill draft error = %v", err)
|
|
}
|
|
if err := validatePromptTemplateResult(promptOperationSkillDraft, map[string]interface{}{"text": `{"skillName":"分镜节奏","tag":"drama","description":"简介"}`}); err == nil || !strings.Contains(err.Error(), "instruction") {
|
|
t.Fatalf("missing instruction error = %v", err)
|
|
}
|
|
protected := protectedPromptContext(promptOperationSkillDraft, map[string]string{"用户想法": "做一个电商主图技能"})
|
|
if !strings.Contains(protected, "做一个电商主图技能") || !strings.Contains(protected, "skill-draft/v1") {
|
|
t.Fatalf("protectedPromptContext() = %q, want idea and JSON contract", protected)
|
|
}
|
|
}
|