Add node changelog events for cascade delete mutations

Author: ogenstad

backend/infrahub/core/manager.py

@@ -1315,7 +1315,7 @@ async def delete(
         nodes: list[Node],
         branch: Optional[Union[Branch, str]] = None,
         at: Optional[Union[Timestamp, str]] = None,
-    ) -> list[Any]:
+    ) -> list[Node]:
         """Returns list of deleted nodes because of cascading deletes"""
         branch = await registry.get_branch(branch=branch, db=db)
         node_delete_validator = NodeDeleteValidator(db=db, branch=branch)

backend/infrahub/graphql/mutations/ipam.py

@@ -402,7 +402,7 @@ async def mutate_delete(
         info: GraphQLResolveInfo,
         data: InputObjectType,
         branch: Branch,
-    ) -> tuple[Node, Self]:
+    ) -> tuple[Node, Self, list[Node]]:
         graphql_context: GraphqlContext = info.context
         db = graphql_context.db
 
@@ -431,4 +431,4 @@ async def mutate_delete(
 
         ok = True
 
-        return reconciled_prefix, cls(ok=ok)
+        return reconciled_prefix, cls(ok=ok), []

backend/infrahub/graphql/mutations/main.py

@@ -64,6 +64,7 @@ async def mutate(cls, root: dict, info: GraphQLResolveInfo, data: InputObjectTyp
         obj = None
         mutation = None
         action = MutationAction.UNDEFINED
+        deleted: list[Node] = []
 
         if "Create" in cls.__name__:
             obj, mutation = await cls.mutate_create(info=info, branch=graphql_context.branch, data=data, **kwargs)
@@ -86,7 +87,9 @@ async def mutate(cls, root: dict, info: GraphQLResolveInfo, data: InputObjectTyp
             else:
                 action = MutationAction.UPDATED
         elif "Delete" in cls.__name__:
-            obj, mutation = await cls.mutate_delete(info=info, branch=graphql_context.branch, data=data, **kwargs)
+            obj, mutation, deleted = await cls.mutate_delete(
+                info=info, branch=graphql_context.branch, data=data, **kwargs
+            )
             action = MutationAction.DELETED
         else:
             raise ValueError(
@@ -124,18 +127,34 @@ async def mutate(cls, root: dict, info: GraphQLResolveInfo, data: InputObjectTyp
 
             events = [main_event]
 
-            for node_changelog in node_changelogs:
+            deleted_changelogs = [node.node_changelog for node in deleted if node.id != obj.id]
+            deleted_ids = [node.node_id for node in deleted_changelogs]
+
+            for node_changelog in deleted_changelogs:
                 meta = EventMeta.from_parent(parent=main_event)
                 event = NodeMutatedEvent(
                     kind=node_changelog.node_kind,
                     node_id=node_changelog.node_id,
                     data=node_changelog,
-                    action=MutationAction.UPDATED,
+                    action=MutationAction.DELETED,
                     fields=node_changelog.updated_fields,
                     meta=meta,
                 )
                 events.append(event)
 
+            for node_changelog in node_changelogs:
+                if node_changelog.node_id not in deleted_ids:
+                    meta = EventMeta.from_parent(parent=main_event)
+                    event = NodeMutatedEvent(
+                        kind=node_changelog.node_kind,
+                        node_id=node_changelog.node_id,
+                        data=node_changelog,
+                        action=MutationAction.UPDATED,
+                        fields=node_changelog.updated_fields,
+                        meta=meta,
+                    )
+                    events.append(event)
+
             for event in events:
                 graphql_context.background.add_task(graphql_context.active_service.event.send, event)
 
@@ -494,7 +513,7 @@ async def mutate_delete(
         info: GraphQLResolveInfo,
         data: InputObjectType,
         branch: Branch,
-    ) -> tuple[Node, Self]:
+    ) -> tuple[Node, Self, list[Node]]:
         graphql_context: GraphqlContext = info.context
 
         obj = await NodeManager.find_object(
@@ -512,7 +531,7 @@ async def mutate_delete(
 
         ok = True
 
-        return obj, cls(ok=ok)
+        return obj, cls(ok=ok), deleted
 
 
 class InfrahubMutation(InfrahubMutationMixin, Mutation):

backend/infrahub/graphql/mutations/menu.py

@@ -89,7 +89,7 @@ async def mutate_delete(
         info: GraphQLResolveInfo,
         data: InputObjectType,
         branch: Branch,
-    ) -> tuple[Node, Self]:
+    ) -> tuple[Node, Self, list[Node]]:
         graphql_context: GraphqlContext = info.context
         obj = await NodeManager.find_object(
             db=graphql_context.db, kind=CoreMenuItem, id=data.get("id"), hfid=data.get("hfid"), branch=branch

backend/tests/unit/graphql/test_mutation_delete.py

@@ -1,7 +1,16 @@
+from infrahub.auth import AccountSession
+from infrahub.core import registry
+from infrahub.core.branch import Branch
+from infrahub.core.constants import MutationAction, RelationshipDeleteBehavior
 from infrahub.core.manager import NodeManager
 from infrahub.core.node import Node
+from infrahub.core.schema.schema_branch import SchemaBranch
 from infrahub.database import InfrahubDatabase
+from infrahub.events.models import ParentEvent
+from infrahub.events.node_action import NodeMutatedEvent
 from infrahub.graphql.initialization import prepare_graphql_params
+from infrahub.services import InfrahubServices
+from tests.adapters.event import MemoryInfrahubEvent
 from tests.helpers.graphql import graphql
 
 
@@ -36,6 +45,7 @@ async def test_delete_object(db: InfrahubDatabase, default_branch, car_person_sc
     )
 
     assert result.errors is None
+    assert result.data
     assert result.data["TestPersonDelete"]["ok"] is True
 
     assert not await NodeManager.get_one(db=db, id=obj1.id)
@@ -70,6 +80,7 @@ async def test_delete_prevented(
         f"It is linked to mandatory relationship owner on node TestCar '{car_camry_main.id}'"
         in result.errors[0].message
     )
+    assert result.data
     assert result.data["TestPersonDelete"] is None
 
     assert await NodeManager.get_one(db=db, id=person_jane_main.id) is not None
@@ -109,6 +120,7 @@ async def test_delete_allowed_when_peer_rel_optional_on_generic(
     )
 
     assert result.errors is None
+    assert result.data
     assert result.data["TestPersonDelete"]["ok"] is True
 
     updated_dog1 = await NodeManager.get_one(db=db, id=dog1.id)
@@ -151,3 +163,68 @@ async def test_delete_prevented_when_peer_rel_required_on_generic(
     assert result.errors
     assert len(result.errors) == 1
     assert expected_error_message in result.errors[0].message
+
+
+async def test_delete_events_with_cascade(
+    db,
+    default_branch: Branch,
+    dependent_generics_schema: SchemaBranch,
+    enable_broker_config: None,
+    session_first_account: AccountSession,
+) -> None:
+    # set TestPerson.animals to be cascade delete
+    schema_branch = registry.schema.get_schema_branch(name=default_branch.name)
+    for schema_kind in ("TestPerson", "TestHuman", "TestCylon"):
+        schema = schema_branch.get(name=schema_kind, duplicate=False)
+        schema.get_relationship("animals").on_delete = RelationshipDeleteBehavior.CASCADE
+
+    human = await Node.init(db=db, schema="TestHuman", branch=default_branch)
+    await human.new(db=db, name="Jane", height=180)
+    await human.save(db=db)
+    dog = await Node.init(db=db, schema="TestDog", branch=default_branch)
+    await dog.new(db=db, name="Roofus", breed="whocares", weight=50, owner=human)
+    await dog.save(db=db)
+
+    memory_event = MemoryInfrahubEvent()
+    service = await InfrahubServices.new(event=memory_event)
+    gql_params = await prepare_graphql_params(
+        db=db, include_subscription=False, branch=default_branch, service=service, account_session=session_first_account
+    )
+    query = """
+    mutation DeletePerson($human_id: String!){
+        TestHumanDelete(data: {id: $human_id}) {
+            ok
+        }
+    }
+    """
+    result = await graphql(
+        schema=gql_params.schema,
+        source=query,
+        context_value=gql_params.context,
+        root_value=None,
+        variable_values={"human_id": human.id},
+    )
+
+    assert not result.errors
+
+    node_map = await NodeManager.get_many(db=db, ids=[human.id, dog.id])
+    assert node_map == {}
+
+    assert gql_params.context.background
+    await gql_params.context.background()
+    assert len(memory_event.events) == 2
+    primary = memory_event.events[0]
+    secondary = memory_event.events[1]
+    assert isinstance(primary, NodeMutatedEvent)
+    assert isinstance(secondary, NodeMutatedEvent)
+    assert primary.kind == "TestHuman"
+    assert primary.node_id == human.id
+    assert primary.action == MutationAction.DELETED
+    assert primary.meta.has_children
+
+    assert secondary.kind == "TestDog"
+    assert secondary.node_id == dog.id
+    assert secondary.action == MutationAction.DELETED
+    assert not secondary.meta.has_children
+    assert secondary.meta.parent == primary.meta.id
+    assert secondary.meta.ancestors == [ParentEvent(id=primary.get_id(), name=primary.get_name())]