228 lines
9.3 KiB
Go
228 lines
9.3 KiB
Go
package config
|
|
|
|
import (
|
|
"path/filepath"
|
|
"reflect"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestArtifactFamiliesExpandCanonicalCharactersDeterministically(t *testing.T) {
|
|
dir := t.TempDir()
|
|
writePartyResolutionFile(t, dir+"/party.yml", `schema_version: narratio.party.v1
|
|
characters:
|
|
zeta:
|
|
player: {name: Zara}
|
|
character:
|
|
name: Zeta
|
|
alias: [Z]
|
|
classes: [{name: wizard, level: 8}, {name: fighter}]
|
|
alpha:
|
|
player: {name: Ada}
|
|
character:
|
|
name: Alpha
|
|
alias: [A, The First]
|
|
classes: [{name: ranger}]
|
|
`)
|
|
pipelinePath, campaignPath, sessionPath := writeArtifactFamilyConfig(t, dir, artifactFamilyPipelineYAML(`
|
|
character_items:
|
|
enabled: false
|
|
for_each: party.characters
|
|
prompt_id: dnd.character_items
|
|
profile_id: production
|
|
output_path_pattern: artifacts/characters/{character_id}/items.md
|
|
inputs:
|
|
transcript: {source: narratio.transcript.final_trimmed, required: true}
|
|
member_vars:
|
|
id: character_id
|
|
player: player.name
|
|
name: character.name
|
|
classes: character.class_summary
|
|
aliases: character.alias_summary
|
|
vars:
|
|
static_flag: true
|
|
character_meta:
|
|
enabled: true
|
|
for_each: party.characters
|
|
prompt_id: dnd.character_meta
|
|
profile_id: production
|
|
output_path_pattern: artifacts/characters/{character_id}/meta.md
|
|
inputs:
|
|
transcript: {source: narratio.transcript.final_trimmed, required: true}
|
|
`))
|
|
|
|
cfg, err := LoadWithSessionOptions(pipelinePath, campaignPath, sessionPath, SessionLoadOptions{})
|
|
if err != nil {
|
|
t.Fatalf("LoadWithSessionOptions() error = %v", err)
|
|
}
|
|
if err := Validate(cfg); err != nil {
|
|
t.Fatalf("Validate() error = %v", err)
|
|
}
|
|
if len(cfg.Pipeline.Scriptorium.ArtifactFamilies) != 0 {
|
|
t.Fatalf("adapter-facing artifact families = %#v, want none", cfg.Pipeline.Scriptorium.ArtifactFamilies)
|
|
}
|
|
keys := sortedArtifactKeys(cfg.Pipeline.Scriptorium.Artifacts)
|
|
wantKeys := []string{"character_items_alpha", "character_items_zeta", "character_meta_alpha", "character_meta_zeta"}
|
|
if !reflect.DeepEqual(keys, wantKeys) {
|
|
t.Fatalf("artifact keys = %#v, want %#v", keys, wantKeys)
|
|
}
|
|
alpha := cfg.Pipeline.Scriptorium.Artifacts["character_items_alpha"]
|
|
if alpha.Enabled || alpha.OutputPath != "artifacts/characters/alpha/items.md" {
|
|
t.Fatalf("alpha artifact = %#v", alpha)
|
|
}
|
|
if !reflect.DeepEqual(alpha.Vars, map[string]any{
|
|
"static_flag": true, "id": "alpha", "player": "Ada", "name": "Alpha", "classes": "ranger", "aliases": "A, The First",
|
|
}) {
|
|
t.Fatalf("alpha vars = %#v", alpha.Vars)
|
|
}
|
|
zeta := cfg.Pipeline.Scriptorium.Artifacts["character_items_zeta"]
|
|
if got := zeta.Vars["classes"]; got != "wizard 8 / fighter" {
|
|
t.Fatalf("zeta class summary = %#v", got)
|
|
}
|
|
catalog := ArtifactFamilies(cfg.Pipeline)
|
|
if got := catalog.Families["character_meta"].Members; !reflect.DeepEqual(got, []string{"character_meta_alpha", "character_meta_zeta"}) {
|
|
t.Fatalf("meta members = %#v", got)
|
|
}
|
|
if got := catalog.Members["character_items_zeta"]; got.Family != "character_items" || got.CharacterID != "zeta" || got.Source == "" {
|
|
t.Fatalf("zeta origin = %#v", got)
|
|
}
|
|
|
|
firstDigest := EffectivePipelineDigest(cfg.Pipeline)
|
|
writePartyResolutionFile(t, dir+"/party.yml", `schema_version: narratio.party.v1
|
|
characters:
|
|
alpha:
|
|
player: {name: Ada}
|
|
character:
|
|
name: Alpha
|
|
alias: [A, The First]
|
|
classes: [{name: ranger}]
|
|
zeta:
|
|
player: {name: Zara}
|
|
character:
|
|
name: Zeta
|
|
alias: [Z]
|
|
classes: [{name: wizard, level: 8}, {name: fighter}]
|
|
`)
|
|
again, err := LoadWithSessionOptions(pipelinePath, campaignPath, sessionPath, SessionLoadOptions{})
|
|
if err != nil {
|
|
t.Fatalf("reordered LoadWithSessionOptions() error = %v", err)
|
|
}
|
|
if !reflect.DeepEqual(cfg.Pipeline.Scriptorium.Artifacts, again.Pipeline.Scriptorium.Artifacts) || !reflect.DeepEqual(ArtifactFamilies(cfg.Pipeline), ArtifactFamilies(again.Pipeline)) || firstDigest != EffectivePipelineDigest(again.Pipeline) {
|
|
t.Fatalf("reordered source maps changed expansion")
|
|
}
|
|
}
|
|
|
|
func TestArtifactFamiliesRejectInvalidExpansionDeclarations(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
party string
|
|
families string
|
|
wantError string
|
|
}{
|
|
{name: "legacy party", party: "legacy: party\n", families: validFamilyYAML, wantError: "requires a canonical campaign party"},
|
|
{name: "invalid iteration", party: canonicalPartyFixture, families: strings.Replace(validFamilyYAML, "party.characters", "party.players", 1), wantError: ".for_each must be"},
|
|
{name: "invalid token", party: canonicalPartyFixture, families: strings.Replace(validFamilyYAML, "{character_id}", "{character_name}", 1), wantError: "must contain exactly one"},
|
|
{name: "repeated token", party: canonicalPartyFixture, families: strings.Replace(validFamilyYAML, "meta.md", "{character_id}.md", 1), wantError: "must contain exactly one"},
|
|
{name: "member selector", party: canonicalPartyFixture, families: strings.Replace(validFamilyYAML, "character.name", "character.unknown", 1), wantError: "selector"},
|
|
{name: "static variable collision", party: canonicalPartyFixture, families: strings.Replace(validFamilyYAML, "member_vars:\n character_name", "vars:\n character_name: static\n member_vars:\n character_name", 1), wantError: "conflicts with static vars"},
|
|
{name: "unsafe output", party: canonicalPartyFixture, families: strings.Replace(validFamilyYAML, "artifacts/characters/{character_id}/meta.md", "../{character_id}.md", 1), wantError: "must not contain path traversal"},
|
|
}
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
dir := t.TempDir()
|
|
writePartyResolutionFile(t, dir+"/party.yml", test.party)
|
|
pipelinePath, campaignPath, sessionPath := writeArtifactFamilyConfig(t, dir, artifactFamilyPipelineYAML(test.families))
|
|
_, err := LoadWithSessionOptions(pipelinePath, campaignPath, sessionPath, SessionLoadOptions{})
|
|
if err == nil || !strings.Contains(err.Error(), test.wantError) {
|
|
t.Fatalf("LoadWithSessionOptions() error = %v, want %q", err, test.wantError)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestArtifactFamilyStrictDecodeRejectsUnknownFields(t *testing.T) {
|
|
dir := t.TempDir()
|
|
pipelinePath, _, _ := writeArtifactFamilyConfig(t, dir, artifactFamilyPipelineYAML(validFamilyYAML+" unsupported: true\n"))
|
|
_, err := LoadPipeline(pipelinePath)
|
|
if err == nil || !strings.Contains(err.Error(), "field unsupported not found") {
|
|
t.Fatalf("LoadPipeline() error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestArtifactFamiliesRejectGeneratedKeyAndOutputCollisions(t *testing.T) {
|
|
document, err := ParseParty([]byte(canonicalPartyFixture))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
base := ScriptoriumArtifactFamilyConfig{
|
|
ForEach: partyCharactersFamilySource, PromptID: "dnd.character_meta",
|
|
OutputPathPattern: "artifacts/characters/{character_id}/meta.md",
|
|
}
|
|
for _, test := range []struct {
|
|
name string
|
|
artifacts map[string]ScriptoriumArtifactConfig
|
|
wantError string
|
|
}{
|
|
{
|
|
name: "generated key", artifacts: map[string]ScriptoriumArtifactConfig{
|
|
"character_meta_arannis": {OutputPath: "artifacts/other.md"},
|
|
}, wantError: "generates artifact key",
|
|
},
|
|
{
|
|
name: "generated output", artifacts: map[string]ScriptoriumArtifactConfig{
|
|
"existing": {OutputPath: "artifacts/characters/arannis/meta.md"},
|
|
}, wantError: "generates output path",
|
|
},
|
|
} {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
pipeline := &PipelineConfig{
|
|
Scriptorium: &ScriptoriumConfig{Artifacts: test.artifacts},
|
|
resolution: &pipelineResolutionMetadata{artifactFamilies: map[string]ScriptoriumArtifactFamilyConfig{"character_meta": base}},
|
|
}
|
|
err := expandPipelineArtifactFamilies(pipeline, ResolvedParty{Mode: PartyModeCanonical, Canonical: document.Canonical})
|
|
if err == nil || !strings.Contains(err.Error(), test.wantError) {
|
|
t.Fatalf("expandPipelineArtifactFamilies() error = %v, want %q", err, test.wantError)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
const validFamilyYAML = `
|
|
character_meta:
|
|
enabled: true
|
|
for_each: party.characters
|
|
prompt_id: dnd.character_meta
|
|
profile_id: production
|
|
output_path_pattern: artifacts/characters/{character_id}/meta.md
|
|
member_vars:
|
|
character_name: character.name
|
|
`
|
|
|
|
func artifactFamilyPipelineYAML(families string) string {
|
|
lines := strings.Split(strings.TrimPrefix(families, "\n"), "\n")
|
|
for index, line := range lines {
|
|
if line != "" {
|
|
lines[index] = " " + line
|
|
}
|
|
}
|
|
return "workspace:\n root: /tmp/narratio-work\nwhisperx:\n transcribe_url: https://example.test/transcribe\nnotification:\n mode: noop\nscriptorium:\n binary: scriptorium\n artifact_families:\n" + strings.Join(lines, "\n")
|
|
}
|
|
|
|
func writeArtifactFamilyConfig(t *testing.T, dir, pipeline string) (string, string, string) {
|
|
t.Helper()
|
|
pipelinePath := writePartyResolutionFile(t, filepath.Join(dir, "pipeline.yml"), pipeline)
|
|
campaignPath := writePartyResolutionFile(t, filepath.Join(dir, "campaign.yml"), "campaign_id: campaign\ninputs:\n speakers_file: speakers.yml\n autocorrect_file: autocorrect.yml\n glossary_file: glossary.yml\n party_file: party.yml\n")
|
|
sessionPath := writePartyResolutionFile(t, filepath.Join(dir, "session.yml"), "session_id: session\ncampaign: campaign\ninputs:\n audio_dir: audio\n")
|
|
return pipelinePath, campaignPath, sessionPath
|
|
}
|
|
|
|
func sortedArtifactKeys(artifacts map[string]ScriptoriumArtifactConfig) []string {
|
|
keys := make([]string, 0, len(artifacts))
|
|
for key := range artifacts {
|
|
keys = append(keys, key)
|
|
}
|
|
sort.Strings(keys)
|
|
return keys
|
|
}
|