112 lines
3.1 KiB
Go
112 lines
3.1 KiB
Go
package dnd
|
|
|
|
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)
|
|
}
|
|
|
|
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
|
|
}
|