52 lines
1.7 KiB
Go
52 lines
1.7 KiB
Go
package pipeline
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/notarius/internal/core/source"
|
|
)
|
|
|
|
func TestDebugSourceDocumentPreservesUnitReferences(t *testing.T) {
|
|
doc := validSourceDocument()
|
|
envelope := debugSourceDocumentEnvelope(doc)
|
|
encoded, err := json.Marshal(envelope)
|
|
if err != nil {
|
|
t.Fatalf("Marshal(debug source document) error = %v, want nil", err)
|
|
}
|
|
|
|
var decoded debugSourceDocument
|
|
if err := json.Unmarshal(encoded, &decoded); err != nil {
|
|
t.Fatalf("Unmarshal(debug source document) error = %v, want nil", err)
|
|
}
|
|
if got, want := decoded.Units[0].Ref, (source.SourceRef{SourceID: "source-1", StartUnitID: 1, EndUnitID: 1}); got != want {
|
|
t.Fatalf("debug source unit ref = %#v, want %#v", got, want)
|
|
}
|
|
|
|
envelope.Units[0].Ref.SourceID = "mutated"
|
|
if got := doc.Units[0].Ref.SourceID; got != "source-1" {
|
|
t.Fatalf("source document ref = %q after debug mutation, want source-1", got)
|
|
}
|
|
}
|
|
|
|
func TestDebugSourceChunkPreservesReference(t *testing.T) {
|
|
doc := validSourceDocument()
|
|
chunk := source.Chunk{
|
|
ID: "chunk-1", SourceID: doc.ID, Index: 0,
|
|
Ref: source.SourceRef{SourceID: doc.ID, StartUnitID: 1, EndUnitID: 1},
|
|
Content: []byte("chunk content"), MediaType: "text/plain", Units: doc.Units[:1],
|
|
}
|
|
envelope := debugSourceChunkEnvelope(chunk)
|
|
encoded, err := json.Marshal(envelope)
|
|
if err != nil {
|
|
t.Fatalf("Marshal(debug source chunk) error = %v, want nil", err)
|
|
}
|
|
var decoded debugSourceChunk
|
|
if err := json.Unmarshal(encoded, &decoded); err != nil {
|
|
t.Fatalf("Unmarshal(debug source chunk) error = %v, want nil", err)
|
|
}
|
|
if decoded.Ref != chunk.Ref {
|
|
t.Fatalf("debug chunk ref = %#v, want %#v", decoded.Ref, chunk.Ref)
|
|
}
|
|
}
|