Organize D&D extensions by domain

internal/modules/dnd/shared/unit_refs_test.go

@@ -0,0 +1,114 @@
+package shared
+
+import (
+	"encoding/json"
+	"strings"
+	"testing"
+
+	"gitea.maximumdirect.net/eric/notarius/internal/core/source"
+)
+
+func TestUnitRefUnmarshalAcceptsIntegerAndNumericString(t *testing.T) {
+	var integerRef UnitRef
+	if err := json.Unmarshal([]byte(`12`), &integerRef); err != nil {
+		t.Fatalf("Unmarshal(integer) error = %v, want nil", err)
+	}
+	if got := integerRef.String(); got != "12" {
+		t.Fatalf("integer ref = %q, want 12", got)
+	}
+	if got := integerRef.Int(); got != 12 {
+		t.Fatalf("integer ref value = %d, want 12", got)
+	}
+
+	var stringRef UnitRef
+	if err := json.Unmarshal([]byte(`"12"`), &stringRef); err != nil {
+		t.Fatalf("Unmarshal(string) error = %v, want nil", err)
+	}
+	if got := stringRef.String(); got != "12" {
+		t.Fatalf("string ref = %q, want 12", got)
+	}
+}
+
+func TestUnitRefUnmarshalRejectsNonIntegerValues(t *testing.T) {
+	for _, raw := range []string{`true`, `null`, `1.5`, `{}`, `"seg-001"`, `" 1 "`, `0`, `-1`} {
+		t.Run(raw, func(t *testing.T) {
+			var ref UnitRef
+			err := json.Unmarshal([]byte(raw), &ref)
+			if err == nil {
+				t.Fatal("Unmarshal() error = nil, want error")
+			}
+		})
+	}
+}
+
+func TestResolveUnitIDReturnsExistingIntegerSourceUnitID(t *testing.T) {
+	doc := unitRefSourceDocument(2, 10)
+
+	got, err := ResolveUnitID(doc, "start_unit_id", UnitRefFromInt(2))
+	if err != nil {
+		t.Fatalf("ResolveUnitID() error = %v, want nil", err)
+	}
+	if got != 2 {
+		t.Fatalf("ResolveUnitID() = %d, want exact source unit ID", got)
+	}
+}
+
+func TestResolveUnitIDDoesNotFallbackToOneBasedUnitNumber(t *testing.T) {
+	doc := unitRefSourceDocument(10, 20)
+
+	_, err := ResolveUnitID(doc, "end_unit_id", UnitRefFromInt(2))
+	if err == nil {
+		t.Fatal("ResolveUnitID() error = nil, want missing source-unit ID")
+	}
+}
+
+func TestResolveUnitIDRejectsMissingUnit(t *testing.T) {
+	doc := unitRefSourceDocument(1)
+
+	_, err := ResolveUnitID(doc, "start_unit_id", UnitRefFromInt(9))
+	if err == nil {
+		t.Fatal("ResolveUnitID() error = nil, want error")
+	}
+	if !strings.Contains(err.Error(), "start_unit_id 9") {
+		t.Fatalf("ResolveUnitID() error = %q, want field and value context", err.Error())
+	}
+}
+
+func TestSourceRefCandidateCanonicalizesValidRefsAndPreservesInvalidRefs(t *testing.T) {
+	doc := unitRefSourceDocument(1, 2)
+
+	valid := SourceRefCandidate(doc, SourceRefResponse{
+		SourceID:    " session-alpha ",
+		StartUnitID: UnitRefFromInt(1),
+		EndUnitID:   UnitRefFromInt(2),
+	})
+	if valid != (source.SourceRef{SourceID: "session-alpha", StartUnitID: 1, EndUnitID: 2}) {
+		t.Fatalf("valid candidate = %#v, want canonical source ref", valid)
+	}
+
+	invalid := SourceRefCandidate(doc, SourceRefResponse{
+		SourceID:    "session-alpha",
+		StartUnitID: UnitRefFromInt(9),
+		EndUnitID:   UnitRefFromString("missing"),
+	})
+	if invalid != (source.SourceRef{SourceID: "session-alpha", StartUnitID: 9, EndUnitID: 0}) {
+		t.Fatalf("invalid candidate = %#v, want unresolved values for validator", invalid)
+	}
+}
+
+func unitRefSourceDocument(ids ...int) *source.SourceDocument {
+	doc := &source.SourceDocument{
+		ID:     "session-alpha",
+		Kind:   "transcript",
+		Format: "application/json",
+		Digest: "sha256:test",
+	}
+	for _, id := range ids {
+		doc.Units = append(doc.Units, source.SourceUnit{
+			ID:   id,
+			Kind: "transcript_segment",
+			Text: "text",
+		})
+	}
+	return doc
+}