Move D&D spell assets into extractor module
This commit is contained in:
@@ -64,47 +64,3 @@ func TestRenderUserSystemIncludesHardeningText(t *testing.T) {
|
||||
t.Fatalf("expected rendered system prompt to include hardening text: %q", system)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRenderDNDSpellsPromptIncludesHardeningText(t *testing.T) {
|
||||
system, user, metadata, err := RenderUserSystem(DNDSpellsPromptID, map[string]any{
|
||||
"SourceID": "session-alpha",
|
||||
"HasChunk": true,
|
||||
"ChunkID": "session-alpha:chunk:0",
|
||||
"ChunkIndex": 0,
|
||||
"Units": []map[string]any{
|
||||
{
|
||||
"ID": "seg-001",
|
||||
"Text": "Aria casts Cure Wounds.",
|
||||
"Metadata": []map[string]string{
|
||||
{"Key": "speaker", "Value": "Alice"},
|
||||
},
|
||||
},
|
||||
},
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("RenderUserSystem: %v", err)
|
||||
}
|
||||
|
||||
hardening := strings.TrimSpace(HardeningText())
|
||||
if hardening == "" {
|
||||
t.Fatalf("expected hardening text")
|
||||
}
|
||||
if !strings.Contains(system, hardening) {
|
||||
t.Fatalf("expected rendered system prompt to include hardening text: %q", system)
|
||||
}
|
||||
for _, want := range []string{"session-alpha", "session-alpha:chunk:0", "seg-001", "Aria casts Cure Wounds.", "speaker: Alice"} {
|
||||
if !strings.Contains(user, want) {
|
||||
t.Fatalf("rendered user prompt = %q, want substring %q", user, want)
|
||||
}
|
||||
}
|
||||
if metadata.PromptID != DNDSpellsPromptID {
|
||||
t.Fatalf("unexpected metadata: %+v", metadata)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRenderDNDSpellsPromptMissingTemplateDataReturnsError(t *testing.T) {
|
||||
_, _, _, err := RenderUserSystem(DNDSpellsPromptID, map[string]any{})
|
||||
if err == nil || !strings.Contains(err.Error(), "SourceID") {
|
||||
t.Fatalf("expected missing SourceID error, got %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user