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)
	}
}