Organize D&D extensions by domain

internal/modules/dnd/extract/spells/test_helpers_test.go

@@ -0,0 +1,86 @@
+package spells
+
+import (
+	"encoding/json"
+	"testing"
+
+	"gitea.maximumdirect.net/eric/notarius/internal/core/source"
+	"gitea.maximumdirect.net/eric/notarius/internal/framework/contracts"
+	"gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/shared"
+)
+
+func promptExtractionRequest() contracts.ExtractionRequest {
+	doc := promptSourceDocument()
+	chunk := &contracts.SourceChunk{
+		ID:          "session-alpha:chunk:0",
+		SourceID:    doc.ID,
+		Index:       0,
+		StartUnitID: doc.Units[0].ID,
+		EndUnitID:   doc.Units[len(doc.Units)-1].ID,
+		Content:     []byte(`{"units":[1,2]}`),
+		MediaType:   "application/json",
+		Units:       append([]source.SourceUnit(nil), doc.Units...),
+		Metadata:    map[string]any{"ignored": "chunk metadata"},
+	}
+	return contracts.ExtractionRequest{
+		Source: doc,
+		Chunk:  chunk,
+	}
+}
+
+func promptSourceDocument() *source.SourceDocument {
+	return &source.SourceDocument{
+		ID:     "session-alpha",
+		Kind:   "transcript",
+		Format: "application/vnd.seriatim.minimal+json",
+		Digest: "sha256:test",
+		Units: []source.SourceUnit{
+			{
+				ID:   1,
+				Kind: "transcript_segment",
+				Text: "Aria raises her hand and casts Cure Wounds.",
+				Metadata: map[string]any{
+					"speaker": "Alice",
+					"start":   json.Number("1.25"),
+					"end":     json.Number("3.5"),
+					"ignored": "not rendered",
+				},
+			},
+			{
+				ID:       2,
+				Kind:     "transcript_segment",
+				Text:     "The fighter's wounds begin to close.",
+				Metadata: map[string]any{"ignored": "not rendered"},
+			},
+		},
+	}
+}
+
+func mustJSON(t *testing.T, value any) string {
+	t.Helper()
+	encoded, err := json.Marshal(value)
+	if err != nil {
+		t.Fatalf("Marshal() error = %v, want nil", err)
+	}
+	return string(encoded)
+}
+
+func responseSourceRefs(sourceID string, startUnitID int, endUnitID int) []shared.SourceRefResponse {
+	return []shared.SourceRefResponse{
+		{
+			SourceID:    sourceID,
+			StartUnitID: shared.UnitRefFromInt(startUnitID),
+			EndUnitID:   shared.UnitRefFromInt(endUnitID),
+		},
+	}
+}
+
+func responseSourceRefsInt(sourceID string, startUnitID int, endUnitID int) []shared.SourceRefResponse {
+	return []shared.SourceRefResponse{
+		{
+			SourceID:    sourceID,
+			StartUnitID: shared.UnitRefFromInt(startUnitID),
+			EndUnitID:   shared.UnitRefFromInt(endUnitID),
+		},
+	}
+}