Function TestLineageManager_GetLineageDepth

pkg/memory/lineage_test.go:393–435

(t *testing.T)

}

func TestLineageManager_GetLineageDepth(t *testing.T) {
	lm := NewLineageManager()

	// Create a chain: mem-1 -> mem-2 -> mem-3 -> mem-4
	p1 := NewProvenance(SourceUserInput, "session-1")
	if err := lm.TrackMemoryCreation("mem-1", p1, nil); err != nil {
		t.Fatalf("TrackMemoryCreation failed: %v", err)
	}

	p2 := NewProvenance(SourceUserInput, "session-1")
	if err := lm.TrackMemoryCreation("mem-2", p2, []string{"mem-1"}); err != nil {
		t.Fatalf("TrackMemoryCreation failed: %v", err)
	}

	p3 := NewProvenance(SourceUserInput, "session-1")
	if err := lm.TrackMemoryCreation("mem-3", p3, []string{"mem-2"}); err != nil {
		t.Fatalf("TrackMemoryCreation failed: %v", err)
	}

	p4 := NewProvenance(SourceUserInput, "session-1")
	if err := lm.TrackMemoryCreation("mem-4", p4, []string{"mem-3"}); err != nil {
		t.Fatalf("TrackMemoryCreation failed: %v", err)
	}

	tests := []struct {
		memoryID  string
		wantDepth int
	}{
		{"mem-1", 0},
		{"mem-2", 1},
		{"mem-3", 2},
		{"mem-4", 3},
	}

	for _, tt := range tests {
		t.Run(tt.memoryID, func(t *testing.T) {
			depth := lm.GetLineageDepth(tt.memoryID)
			if depth != tt.wantDepth {
				t.Errorf("GetLineageDepth(%v) = %v, want %v", tt.memoryID, depth, tt.wantDepth)
			}
		})
	}
}

func TestLineageManager_GetLineageStats(t *testing.T) {
	lm := NewLineageManager()