jaclang 0.7.16__py3-none-any.whl → 0.7.18__py3-none-any.whl

jaclang/plugin/tests/test_jaseci.py

@@ -7,6 +7,8 @@ import sys
 from jaclang.cli import cli
 from jaclang.utils.test import TestCase
 
+session = ""
+
 
 class TestJaseciPlugin(TestCase):
     """Test jaseci plugin."""
@@ -30,22 +32,27 @@ class TestJaseciPlugin(TestCase):
         sys.stdout = sys.__stdout__
 
     def _del_session(self, session: str) -> None:
-        if os.path.exists(session):
-            os.remove(session)
+        path = os.path.dirname(session)
+        prefix = os.path.basename(session)
+        for file in os.listdir(path):
+            if file.startswith(prefix):
+                os.remove(f"{path}/{file}")
 
     def test_walker_simple_persistent(self) -> None:
         """Test simple persistent object."""
         session = self.fixture_abs_path("test_walker_simple_persistent.session")
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_persistent.jac"),
             session=session,
-            walker="create",
+            entrypoint="create",
+            args=[],
         )
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_persistent.jac"),
             session=session,
-            walker="traverse",
+            entrypoint="traverse",
+            args=[],
         )
         output = self.capturedOutput.getvalue().strip()
         self.assertEqual(output, "node a\nnode b")
@@ -54,18 +61,24 @@ class TestJaseciPlugin(TestCase):
     def test_entrypoint_root(self) -> None:
         """Test entrypoint being root."""
         session = self.fixture_abs_path("test_entrypoint_root.session")
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_persistent.jac"),
             session=session,
-            walker="create",
+            entrypoint="create",
+            args=[],
+        )
+        obj = cli.get_object(
+            filename=self.fixture_abs_path("simple_persistent.jac"),
+            id="root",
+            session=session,
         )
-        obj = cli.get_object(session=session, id="root")
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_persistent.jac"),
             session=session,
-            node=str(obj["_jac_"]["id"]),
-            walker="traverse",
+            entrypoint="traverse",
+            args=[],
+            node=str(obj["id"]),
         )
         output = self.capturedOutput.getvalue().strip()
         self.assertEqual(output, "node a\nnode b")
@@ -74,20 +87,34 @@ class TestJaseciPlugin(TestCase):
     def test_entrypoint_non_root(self) -> None:
         """Test entrypoint being non root node."""
         session = self.fixture_abs_path("test_entrypoint_non_root.session")
-        cli.run(
+        cli.enter(
+            filename=self.fixture_abs_path("simple_persistent.jac"),
+            session=session,
+            entrypoint="create",
+            args=[],
+        )
+        obj = cli.get_object(
+            filename=self.fixture_abs_path("simple_persistent.jac"),
+            id="root",
+            session=session,
+        )
+        edge_obj = cli.get_object(
+            filename=self.fixture_abs_path("simple_persistent.jac"),
+            id=obj["edges"][0].id.hex,
+            session=session,
+        )
+        a_obj = cli.get_object(
             filename=self.fixture_abs_path("simple_persistent.jac"),
+            id=edge_obj["target"].id.hex,
             session=session,
-            walker="create",
         )
-        obj = cli.get_object(session=session, id="root")
-        edge_obj = cli.get_object(session=session, id=str(obj["_jac_"]["edge_ids"][0]))
-        a_obj = cli.get_object(session=session, id=str(edge_obj["_jac_"]["target_id"]))
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_persistent.jac"),
             session=session,
-            node=str(a_obj["_jac_"]["id"]),
-            walker="traverse",
+            entrypoint="traverse",
+            node=str(a_obj["id"]),
+            args=[],
         )
         output = self.capturedOutput.getvalue().strip()
         self.assertEqual(output, "node a\nnode b")
@@ -100,16 +127,33 @@ class TestJaseciPlugin(TestCase):
             filename=self.fixture_abs_path("simple_node_connection.jac"),
             session=session,
         )
-        obj = cli.get_object(session=session, id="root")
-        self.assertEqual(len(obj["_jac_"]["edge_ids"]), 2)
+        obj = cli.get_object(
+            filename=self.fixture_abs_path("simple_node_connection.jac"),
+            session=session,
+            id="root",
+        )
+        self.assertEqual(len(obj["edges"]), 2)
         edge_objs = [
-            cli.get_object(session=session, id=str(e_id))
-            for e_id in obj["_jac_"]["edge_ids"]
+            cli.get_object(
+                filename=self.fixture_abs_path("simple_node_connection.jac"),
+                session=session,
+                id=e.id.hex,
+            )
+            for e in obj["edges"]
+        ]
+        node_ids = [obj["target"].id.hex for obj in edge_objs]
+        node_objs = [
+            cli.get_object(
+                filename=self.fixture_abs_path("simple_node_connection.jac"),
+                session=session,
+                id=str(n_id),
+            )
+            for n_id in node_ids
         ]
-        node_ids = [obj["_jac_"]["target_id"] for obj in edge_objs]
-        node_objs = [cli.get_object(session=session, id=str(n_id)) for n_id in node_ids]
         self.assertEqual(len(node_objs), 2)
-        self.assertEqual({obj["tag"] for obj in node_objs}, {"first", "second"})
+        self.assertEqual(
+            {obj["architype"].tag for obj in node_objs}, {"first", "second"}
+        )
         self._del_session(session)
 
     def test_filter_on_edge_get_edge(self) -> None:
@@ -120,10 +164,11 @@ class TestJaseciPlugin(TestCase):
             session=session,
         )
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_node_connection.jac"),
             session=session,
-            walker="filter_on_edge_get_edge",
+            entrypoint="filter_on_edge_get_edge",
+            args=[],
         )
         self.assertEqual(
             self.capturedOutput.getvalue().strip(), "[simple_edge(index=1)]"
@@ -138,10 +183,11 @@ class TestJaseciPlugin(TestCase):
             session=session,
         )
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_node_connection.jac"),
             session=session,
-            walker="filter_on_edge_get_node",
+            entrypoint="filter_on_edge_get_node",
+            args=[],
         )
         self.assertEqual(
             self.capturedOutput.getvalue().strip(), "[simple(tag='second')]"
@@ -156,10 +202,11 @@ class TestJaseciPlugin(TestCase):
             session=session,
         )
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_node_connection.jac"),
             session=session,
-            walker="filter_on_node_get_node",
+            entrypoint="filter_on_node_get_node",
+            args=[],
         )
         self.assertEqual(
             self.capturedOutput.getvalue().strip(), "[simple(tag='second')]"
@@ -174,10 +221,11 @@ class TestJaseciPlugin(TestCase):
             session=session,
         )
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_node_connection.jac"),
             session=session,
-            walker="filter_on_edge_visit",
+            entrypoint="filter_on_edge_visit",
+            args=[],
         )
         self.assertEqual(self.capturedOutput.getvalue().strip(), "simple(tag='first')")
         self._del_session(session)
@@ -190,10 +238,11 @@ class TestJaseciPlugin(TestCase):
             session=session,
         )
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_node_connection.jac"),
             session=session,
-            walker="filter_on_node_visit",
+            entrypoint="filter_on_node_visit",
+            args=[],
         )
         self.assertEqual(self.capturedOutput.getvalue().strip(), "simple(tag='first')")
         self._del_session(session)
@@ -201,19 +250,342 @@ class TestJaseciPlugin(TestCase):
     def test_indirect_reference_node(self) -> None:
         """Test reference node indirectly without visiting."""
         session = self.fixture_abs_path("test_indirect_reference_node.session")
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_persistent.jac"),
             session=session,
-            walker="create",
+            entrypoint="create",
+            args=[],
         )
         self._output2buffer()
-        cli.run(
+        cli.enter(
             filename=self.fixture_abs_path("simple_persistent.jac"),
             session=session,
-            walker="indirect_ref",
+            entrypoint="indirect_ref",
+            args=[],
         )
         self.assertEqual(
             self.capturedOutput.getvalue().strip(),
-            "[b(name='node b')]\n[GenericEdge]",
+            "[b(name='node b')]\n[GenericEdge()]",
         )
         self._del_session(session)
+
+    def trigger_access_validation_test(
+        self, give_access_to_full_graph: bool, via_all: bool = False
+    ) -> None:
+        """Test different access validation."""
+        self._output2buffer()
+
+        ##############################################
+        #              ALLOW READ ACCESS             #
+        ##############################################
+
+        node_1 = "" if give_access_to_full_graph else self.nodes[0]
+        node_2 = "" if give_access_to_full_graph else self.nodes[1]
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="allow_other_root_access",
+            args=[self.roots[1], 0, via_all],
+            session=session,
+            root=self.roots[0],
+            node=node_1,
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="allow_other_root_access",
+            args=[self.roots[0], 0, via_all],
+            session=session,
+            root=self.roots[1],
+            node=node_2,
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_node",
+            args=[20],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_node",
+            args=[10],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+        archs = self.capturedOutput.getvalue().strip().split("\n")
+        self.assertTrue(len(archs) == 2)
+
+        # --------- NO UPDATE SHOULD HAPPEN -------- #
+
+        self.assertTrue(archs[0], "A(val=2)")
+        self.assertTrue(archs[1], "A(val=1)")
+
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="disallow_other_root_access",
+            args=[self.roots[1], via_all],
+            session=session,
+            root=self.roots[0],
+            node=node_1,
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="disallow_other_root_access",
+            args=[self.roots[0], via_all],
+            session=session,
+            root=self.roots[1],
+            node=node_2,
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+        self.assertFalse(self.capturedOutput.getvalue().strip())
+
+        ##############################################
+        #             ALLOW WRITE ACCESS             #
+        ##############################################
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="allow_other_root_access",
+            args=[self.roots[1], "WRITE", via_all],
+            session=session,
+            root=self.roots[0],
+            node=node_1,
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="allow_other_root_access",
+            args=[self.roots[0], "WRITE", via_all],
+            session=session,
+            root=self.roots[1],
+            node=node_2,
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_node",
+            args=[20],
+            root=self.roots[0],
+            node=self.nodes[1],
+            session=session,
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="update_node",
+            args=[10],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+        archs = self.capturedOutput.getvalue().strip().split("\n")
+        self.assertTrue(len(archs) == 2)
+
+        # --------- UPDATE SHOULD HAPPEN -------- #
+
+        self.assertTrue(archs[0], "A(val=20)")
+        self.assertTrue(archs[1], "A(val=10)")
+
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="disallow_other_root_access",
+            args=[self.roots[1], via_all],
+            session=session,
+            root=self.roots[0],
+            node=node_1,
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="disallow_other_root_access",
+            args=[self.roots[0], via_all],
+            session=session,
+            root=self.roots[1],
+            node=node_2,
+        )
+
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[0],
+            node=self.nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=self.roots[1],
+            node=self.nodes[0],
+        )
+        self.assertFalse(self.capturedOutput.getvalue().strip())
+
+    def test_other_root_access(self) -> None:
+        """Test filtering on node, then visit."""
+        global session
+        session = self.fixture_abs_path("other_root_access.session")
+
+        ##############################################
+        #                CREATE ROOTS                #
+        ##############################################
+
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="create_other_root",
+            args=[],
+            session=session,
+        )
+        root1 = self.capturedOutput.getvalue().strip()
+
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="create_other_root",
+            args=[],
+            session=session,
+        )
+        root2 = self.capturedOutput.getvalue().strip()
+
+        ##############################################
+        #           CREATE RESPECTIVE NODES          #
+        ##############################################
+
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="create_node",
+            args=[1],
+            session=session,
+            root=root1,
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="create_node",
+            args=[2],
+            session=session,
+            root=root2,
+        )
+        nodes = self.capturedOutput.getvalue().strip().split("\n")
+        self.assertTrue(len(nodes) == 2)
+
+        ##############################################
+        #           VISIT RESPECTIVE NODES           #
+        ##############################################
+
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=root1,
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=root2,
+        )
+        archs = self.capturedOutput.getvalue().strip().split("\n")
+        self.assertEqual(2, len(archs))
+        self.assertTrue(archs[0], "A(val=1)")
+        self.assertTrue(archs[1], "A(val=2)")
+
+        ##############################################
+        #              SWAP TARGET NODE              #
+        #                  NO ACCESS                 #
+        ##############################################
+
+        self._output2buffer()
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=root1,
+            node=nodes[1],
+        )
+        cli.enter(
+            filename=self.fixture_abs_path("other_root_access.jac"),
+            entrypoint="check_node",
+            args=[],
+            session=session,
+            root=root2,
+            node=nodes[0],
+        )
+        self.assertFalse(self.capturedOutput.getvalue().strip())
+
+        ##############################################
+        #        TEST DIFFERENT ACCESS OPTIONS       #
+        ##############################################
+
+        self.roots = [root1, root2]
+        self.nodes = nodes
+
+        self.trigger_access_validation_test(give_access_to_full_graph=False)
+        self.trigger_access_validation_test(give_access_to_full_graph=True)
+
+        self.trigger_access_validation_test(
+            give_access_to_full_graph=False, via_all=True
+        )
+        self.trigger_access_validation_test(
+            give_access_to_full_graph=True, via_all=True
+        )
+
+        self._del_session(session)