Fixed #51 : Added unit tests

tests/core/test_SheerkaRuleManager.py

@@ -7,12 +7,14 @@ from core.concept import Concept, DEFINITION_TYPE_DEF, DoNotResolve
 from core.global_symbols import RULE_COMPARISON_CONTEXT, NotFound, EVENT_RULE_DELETED
 from core.rule import Rule, ACTION_TYPE_PRINT, ACTION_TYPE_EXEC
 from core.sheerka.Sheerka import RECOGNIZED_BY_ID, RECOGNIZED_BY_NAME
+from core.sheerka.services.SheerkaEvaluateRules import SheerkaEvaluateRules
 from core.sheerka.services.SheerkaExecute import ParserInput
 from core.sheerka.services.SheerkaRuleManager import SheerkaRuleManager, FormatRuleActionParser, \
     FormatAstRawText, FormatAstVariable, FormatAstSequence, FormatAstFunction, \
     FormatRuleSyntaxError, FormatAstList, UnexpectedEof, FormatAstColor, RuleCompiledPredicate, FormatAstDict, \
     FormatAstMulti, \
-    PythonCodeEmitter, NoConditionFound, FormatAstNode, ReteConditionExprVisitor
+    PythonCodeEmitter, NoConditionFound, FormatAstNode, ReteConditionExprVisitor, PythonConditionExprVisitor, \
+    CompiledCondition
 from core.sheerka.services.sheerka_service import FailedToCompileError
 from core.tokenizer import Token, TokenKind
 from parsers.BaseNodeParser import SourceCodeWithConceptNode, SourceCodeNode
@@ -1240,6 +1242,7 @@ isinstance(var, Concept) and var.key == 'hello __var__0'""" + \
         matches = list(network.matches)
         assert len(matches) == 1
 
+    @pytest.mark.skip("I am not sure yet of what I want to get")
     @pytest.mark.parametrize("expression, expected_as_str", [
         (
             "eval(__ret.body, 'foo' starts with 'f')",
@@ -1287,6 +1290,74 @@ isinstance(var, Concept) and var.key == 'hello __var__0'""" + \
         matches = list(network.matches)
         assert len(matches) == 1
 
+    def test_i_can_get_compiled_conditions_when_testing_data_existence(self):
+        sheerka, context = self.init_test().unpack()
+        expression = "__ret"
+
+        parser = ExpressionParser()
+        error_sink = ErrorSink()
+        parser_input = ParserInput(expression)
+        parser.reset_parser_input(parser_input, error_sink)
+        parsed = parser.parse_input(context, parser_input, error_sink)
+
+        visitor = PythonConditionExprVisitor(context)
+        conditions = visitor.get_conditions(parsed)
+
+        assert len(conditions) == 1
+        assert isinstance(conditions[0], CompiledCondition)
+        assert conditions[0].evaluator_type is None
+        assert conditions[0].return_value is None
+        assert conditions[0].concept is None
+        assert conditions[0].variables == {"__ret"}
+
+        # check against SheerkaEvaluateRules
+        evaluate_rules_service = sheerka.services[SheerkaEvaluateRules.NAME]
+        bag = {"__ret": ReturnValueConcept("Test", True, None)}
+        rule = Rule(name="test_i_can_get_compiled_conditions", predicate=expression)
+        rule.compiled_conditions = conditions
+        res = evaluate_rules_service.evaluate_rule(context, rule, bag)
+        assert res.status
+        assert self.sheerka.is_success(self.sheerka.objvalue(res))
+
+    @pytest.mark.parametrize("expression", [
+        "__ret",
+        "__ret.status == True",
+        "__ret.status",
+        "__ret and __ret.status",
+    ])
+    def test_i_can_get_compiled_conditions(self, expression):
+        sheerka, context = self.init_test().unpack()
+
+        parser = ExpressionParser()
+        error_sink = ErrorSink()
+        parser_input = ParserInput(expression)
+        parser.reset_parser_input(parser_input, error_sink)
+        parsed = parser.parse_input(context, parser_input, error_sink)
+
+        visitor = PythonConditionExprVisitor(context)
+        conditions = visitor.get_conditions(parsed)
+
+        ast_ = ast.parse(expression, "<source>", 'eval')
+        expected_python_node = PythonNode(expression, ast_)
+
+        assert len(conditions) == 1
+        assert isinstance(conditions[0], CompiledCondition)
+        assert conditions[0].evaluator_type == PYTHON_EVALUATOR_NAME
+        assert sheerka.isinstance(conditions[0].return_value, BuiltinConcepts.RETURN_VALUE)
+        assert sheerka.objvalue(conditions[0].return_value) == expected_python_node
+        assert conditions[0].concept is None
+        assert conditions[0].variables == {"__ret"}
+
+        # check against SheerkaEvaluateRules
+        evaluate_rules_service = sheerka.services[SheerkaEvaluateRules.NAME]
+        bag = {"__ret": ReturnValueConcept("Test", True, None)}
+        rule = Rule(name="test_i_can_get_compiled_conditions", predicate=expression)
+        rule.compiled_conditions = conditions
+        res = evaluate_rules_service.evaluate_rule(context, rule, bag)
+        assert res.status
+        assert self.sheerka.is_success(self.sheerka.objvalue(res))
+
+
 class TestSheerkaRuleManagerUsingFileBasedSheerka(TestUsingFileBasedSheerka):
     def test_rules_are_initialized_at_startup(self):
         sheerka, context, *rules = self.init_test().with_rules(