From db81e57654dfea9c54cb2924cff96c7363f0f0c3 Mon Sep 17 00:00:00 2001
From: antoniofilipovic <filipovicantonio1998@gmail.com>
Date: Tue, 5 Sep 2023 14:26:50 +0200
Subject: [PATCH] add trigger initial test

---
 tests/e2e/triggers/procedures/write.py | 48 ++++++++++++++++++++------
 1 file changed, 37 insertions(+), 11 deletions(-)

diff --git a/tests/e2e/triggers/procedures/write.py b/tests/e2e/triggers/procedures/write.py
index 1b4b7ad99..297f8362d 100644
--- a/tests/e2e/triggers/procedures/write.py
+++ b/tests/e2e/triggers/procedures/write.py
@@ -9,8 +9,11 @@
 # by the Apache License, Version 2.0, included in the file
 # licenses/APL.txt.
 
+from typing import Union
+
 import mgp
 
+
 @mgp.write_proc
 def create_vertex(ctx: mgp.ProcCtx, id: mgp.Any) -> mgp.Record(v=mgp.Any):
     v = None
@@ -36,15 +39,14 @@ def detach_delete_vertex(ctx: mgp.ProcCtx, v: mgp.Any) -> mgp.Record():
 
 
 @mgp.write_proc
-def create_edge(ctx: mgp.ProcCtx, from_vertex: mgp.Vertex,
-                to_vertex: mgp.Vertex,
-                edge_type: str) -> mgp.Record(e=mgp.Any):
+def create_edge(
+    ctx: mgp.ProcCtx, from_vertex: mgp.Vertex, to_vertex: mgp.Vertex, edge_type: str
+) -> mgp.Record(e=mgp.Any):
     e = None
     try:
-        e = ctx.graph.create_edge(
-            from_vertex, to_vertex, mgp.EdgeType(edge_type))
-        e.properties.set("id", 1);
-        e.properties.set("tbd", 0);
+        e = ctx.graph.create_edge(from_vertex, to_vertex, mgp.EdgeType(edge_type))
+        e.properties.set("id", 1)
+        e.properties.set("tbd", 0)
     except RuntimeError as ex:
         return mgp.Record(e=str(ex))
     return mgp.Record(e=e)
@@ -61,19 +63,43 @@ def set_property(ctx: mgp.ProcCtx, object: mgp.Any) -> mgp.Record():
     object.properties.set("id", 2)
     return mgp.Record()
 
+
 @mgp.write_proc
 def remove_property(ctx: mgp.ProcCtx, object: mgp.Any) -> mgp.Record():
     object.properties.set("tbd", None)
     return mgp.Record()
 
+
 @mgp.write_proc
-def add_label(ctx: mgp.ProcCtx, object: mgp.Any,
-              name: str) -> mgp.Record(o=mgp.Any):
+def add_label(ctx: mgp.ProcCtx, object: mgp.Any, name: str) -> mgp.Record(o=mgp.Any):
     object.add_label(name)
     return mgp.Record(o=object)
 
+
 @mgp.write_proc
-def remove_label(ctx: mgp.ProcCtx, object: mgp.Any,
-                 name: str) -> mgp.Record(o=mgp.Any):
+def remove_label(ctx: mgp.ProcCtx, object: mgp.Any, name: str) -> mgp.Record(o=mgp.Any):
     object.remove_label(name)
     return mgp.Record(o=object)
+
+
+@mgp.write_proc
+def access_deleted(ctx: mgp.ProcCtx, event: mgp.Any) -> mgp.Record(status=bool):
+    def process_obj(obj: Union[mgp.Vertex, mgp.Edge]) -> None:
+        for label in obj.labels:
+            assert label
+        for prop in obj.properties:
+            assert prop
+
+    try:
+        if "event_type" not in event:
+            return False
+        if event["event_type"] == "deleted_vertex":
+            process_obj(event["vertex"])
+        if event["event_type"] == "deleted_edge":
+            e = event["edge"]
+            process_obj(e)
+            process_obj(e.from_vertex)
+            process_obj(e.to_vertex)
+    except Exception as e:
+        return mgp.Record(status=False)
+    return mgp.Record(status=True)