Function TestLineageGraph_RemoveMemory

pkg/memory/lineage_test.go:212–239

(t *testing.T)

}

func TestLineageGraph_RemoveMemory(t *testing.T) {
	lg := NewLineageGraph()

	// Create parent-child relationship
	lg.TrackMemory("parent", &LineageMetadata{
		ID:        "parent",
		CreatedAt: time.Now().Unix(),
	})

	lg.TrackMemory("child", &LineageMetadata{
		ID:             "child",
		DerivedFromIDs: []string{"parent"},
		CreatedAt:      time.Now().Unix(),
	})

	// Remove child
	lg.RemoveMemory("child")

	// Child should be gone
	if lg.memoryMetadata["child"] != nil {
		t.Error("Child memory should be removed from metadata")
	}

	// Parent should no longer have child in its children list
	if len(lg.parentToChildren["parent"]) != 0 {
		t.Error("Parent should have no children after removal")
	}
}

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