package research_contracts_test

import (
	"testing"

	"git.cin.su/m/Assistant/contracts/epistemics"
)

func TestResearchContractsRequireExactEvidence(t *testing.T) {
	plan := epistemics.ResearchPlan{ID: "plan-1", Question: "What changed?", SourceClasses: []string{"repository"}, FreshnessPolicy: "same-day"}
	if err := plan.Validate(); err != nil {
		t.Fatal(err)
	}
	snapshot := epistemics.SourceSnapshot{ID: "snapshot-1", SourceID: "source-1", ContentDigest: "sha256:abc", CapturedAtUnixMillis: 1, Freshness: epistemics.FreshnessCurrent}
	if err := snapshot.Validate(); err != nil {
		t.Fatal(err)
	}
	excerpt := epistemics.SourceExcerpt{ID: "excerpt-1", SnapshotID: snapshot.ID, Text: "exact evidence", Start: 0, End: 14}
	if err := excerpt.Validate(); err != nil {
		t.Fatal(err)
	}
	citation := epistemics.Citation{ID: "citation-1", ExcerptID: excerpt.ID, Evidence: epistemics.EvidenceObserved, Coverage: epistemics.CoverageComplete, Conflict: epistemics.ConflictNone, ResolvedAtUnixMillis: 2}
	if err := citation.Validate(); err != nil {
		t.Fatal(err)
	}
}

func TestResearchContractsPreserveUnknownAndAssumedStates(t *testing.T) {
	snapshot := epistemics.SourceSnapshot{ID: "snapshot-unknown", SourceID: "source-1", ContentDigest: "sha256:abc", CapturedAtUnixMillis: 1, Freshness: epistemics.FreshnessUnknown}
	if err := snapshot.Validate(); err != nil {
		t.Fatal(err)
	}
	citation := epistemics.Citation{ID: "citation-assumed", ExcerptID: "excerpt-1", Evidence: epistemics.EvidenceAssumed, Coverage: epistemics.CoveragePartial, Conflict: epistemics.ConflictPresent, ResolvedAtUnixMillis: 2}
	if err := citation.Validate(); err != nil {
		t.Fatal(err)
	}
}

func TestResolveCitationRequiresMatchingSourceChain(t *testing.T) {
	source := epistemics.Source{ID: "source-1", Locator: "https://example.test/report"}
	snapshot := epistemics.SourceSnapshot{ID: "snapshot-1", SourceID: source.ID, ContentDigest: "sha256:abc", CapturedAtUnixMillis: 1, Freshness: epistemics.FreshnessStale}
	excerpt := epistemics.SourceExcerpt{ID: "excerpt-1", SnapshotID: snapshot.ID, Text: "exact evidence", Start: 0, End: 14}
	citation, err := epistemics.ResolveCitation(source, snapshot, excerpt, epistemics.EvidenceObserved, epistemics.CoveragePartial, epistemics.ConflictPresent, 2)
	if err != nil || citation.ExcerptID != excerpt.ID || citation.Conflict != epistemics.ConflictPresent {
		t.Fatalf("unexpected citation: %#v %v", citation, err)
	}
	wrong := snapshot
	wrong.SourceID = "other-source"
	if _, err := epistemics.ResolveCitation(source, wrong, excerpt, epistemics.EvidenceObserved, epistemics.CoverageComplete, epistemics.ConflictNone, 2); err == nil {
		t.Fatal("expected source-chain mismatch")
	}
}