Function TestLineageManager_DeleteMemoryWithLineage

pkg/memory/lineage_test.go:266–320

(t *testing.T)

}

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

	// Create lineage
	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)
	}

	ctx := context.Background()

	// Delete with cascade
	deleted, err := lm.DeleteMemoryWithLineage(ctx, "mem-1", true)
	if err != nil {
		t.Fatalf("DeleteMemoryWithLineage() error = %v", err)
	}

	// Should delete mem-1, mem-2, mem-3
	if len(deleted) != 3 {
		t.Fatalf("Deleted %d memories, want 3", len(deleted))
	}

	expectedDeleted := map[string]bool{
		"mem-1": false,
		"mem-2": false,
		"mem-3": false,
	}

	for _, id := range deleted {
		if _, exists := expectedDeleted[id]; exists {
			expectedDeleted[id] = true
		}
	}

	for id, found := range expectedDeleted {
		if !found {
			t.Errorf("Expected memory %v to be deleted", id)
		}
	}

	// Verify memories are removed from graph
	if lm.graph.memoryMetadata["mem-1"] != nil {
		t.Error("mem-1 should be removed from graph")
	}
}

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