test(projection): Add test for attribute and function projection attribute

test/unittest_parser.py

@@ -329,6 +329,19 @@
         tree = self.parse("EUser X;")
         self.assertEqual(tree.as_string(), "Any X WHERE X is EUser")
 
+    def test_projection_with_attribute_to_get(self):
+        tree = self.parse("Any X.my_attribute Where X is CWUser;")
+        varref = tree.children[0].selection[0]
+        self.assertTrue(isinstance(varref, nodes.VariableRefAttributeAccess))
+        self.assertEqual(varref.attribute_to_access, "my_attribute")
+
+    def test_projection_with_function_to_apply(self):
+        tree = self.parse('Any X.my_function("loutre") Where X is CWUser;')
+        varref = tree.children[0].selection[0]
+        self.assertTrue(isinstance(varref, nodes.VariableRefMethodCall))
+        self.assertEqual(varref.method_to_call, "my_function")
+        self.assertEqual(varref.args_for_method_to_call[0].value, "loutre")
+
     def test_spec(self):
         """test all RQL string found in the specification and test they are well parsed"""
         for rql in SPEC_QUERIES: