Function TestLineageManager_GetLineageStats

pkg/memory/lineage_test.go:437–481

(t *testing.T)

}

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

	// Create some memories
	p1 := NewProvenance(SourceUserInput, "session-1")
	if err := lm.TrackMemoryCreation("mem-1", p1, nil); err != nil { // Root
		t.Fatalf("TrackMemoryCreation failed: %v", err)
	}

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

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

	stats := lm.GetLineageStats()

	if stats.TotalMemories != 3 {
		t.Errorf("TotalMemories = %v, want 3", stats.TotalMemories)
	}

	if stats.RootMemories != 2 {
		t.Errorf("RootMemories = %v, want 2", stats.RootMemories)
	}

	if stats.DerivedMemories != 1 {
		t.Errorf("DerivedMemories = %v, want 1", stats.DerivedMemories)
	}

	if stats.MaxDepth != 1 {
		t.Errorf("MaxDepth = %v, want 1", stats.MaxDepth)
	}

	if stats.MemoriesBySource["session-1"] != 2 {
		t.Errorf("MemoriesBySource['session-1'] = %v, want 2", stats.MemoriesBySource["session-1"])
	}

	if stats.MemoriesBySource["session-2"] != 1 {
		t.Errorf("MemoriesBySource['session-2'] = %v, want 1", stats.MemoriesBySource["session-2"])
	}
}