import pytest
from core.builtin_concepts import BuiltinConcepts, ReturnValueConcept, ParserResultConcept
from core.concept import Concept, DoNotResolve, ConceptParts, Property, InfiniteRecursionResolved, CB, NotInit
from parsers.PythonParser import PythonNode

from tests.TestUsingMemoryBasedSheerka import TestUsingMemoryBasedSheerka


class TestSheerkaEvaluateConcept(TestUsingMemoryBasedSheerka):

    @pytest.mark.parametrize("body, expected", [
        (None, BuiltinConcepts.NOT_INITIALIZED),
        ("", ""),
        ("1", 1),
        ("1+1", 2),
        ("'one'", "one"),
        ("'one' + 'two'", "onetwo"),
        ("True", True),
        ("1 > 2", False),
    ])
    def test_i_can_evaluate_a_concept_with_simple_body(self, body, expected):
        sheerka, context, concept = self.init_concepts(Concept("foo", body=body), eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.key == concept.key
        assert evaluated.body == expected
        assert evaluated.metadata.body == body
        assert evaluated.metadata.pre is None
        assert evaluated.metadata.post is None
        assert evaluated.metadata.where is None
        assert evaluated.variables() == {}
        assert evaluated.metadata.is_evaluated
        assert len(evaluated.values) == 0 if body is None else 1

    @pytest.mark.parametrize("expr, expected", [
        ("", ""),
        ("1", 1),
        ("1+1", 2),
        ("'one'", "one"),
        ("'one' + 'two'", "onetwo"),
        ("True", True),
        ("1 > 2", False),
    ])
    def test_i_can_evaluate_the_other_metadata(self, expr, expected):
        """
        I only test PRE, it's the same for the others
        :param expr:
        :param expected:
        :return:
        """

        sheerka, context, concept = self.init_concepts(Concept("foo", pre=expr))

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.key == concept.key
        assert evaluated.metadata.body is None
        assert evaluated.metadata.pre == expr
        assert evaluated.metadata.post is None
        assert evaluated.metadata.where is None
        assert evaluated.get_value(ConceptParts.PRE) == expected
        assert evaluated.variables() == {}
        assert not evaluated.metadata.is_evaluated
        assert len(evaluated.values) == 0 if expr is None else 1

    @pytest.mark.parametrize("expr, expected", [
        (None, NotInit),
        ("", ""),
        ("1", 1),
        ("1+1", 2),
        ("'one'", "one"),
        ("'one' + 'two'", "onetwo"),
        ("True", True),
        ("1 > 2", False),
    ])
    def test_i_can_evaluate_a_concept_with_prop(self, expr, expected):
        sheerka, context, concept = self.init_concepts(Concept("foo").def_var("a", expr), eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.key == concept.key
        assert evaluated.metadata.pre is None
        assert evaluated.metadata.pre is None
        assert evaluated.metadata.post is None
        assert evaluated.metadata.where is None
        assert evaluated.variables() == {"a": Property("a", expected)}
        assert evaluated.metadata.is_evaluated

    def test_i_can_evaluate_metadata_using_do_not_resolve(self):
        sheerka, context, concept = self.init_concepts(Concept("foo"), eval_body=True)
        concept.compiled[ConceptParts.BODY] = DoNotResolve("do not resolve")

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.body == "do not resolve"
        assert evaluated.metadata.is_evaluated

    def test_i_can_evaluate_variable_using_do_not_resolve(self):
        sheerka, context, concept = self.init_concepts(Concept("foo").def_var("a"), eval_body=True)
        concept.compiled["a"] = DoNotResolve("do not resolve")

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.get_value("a") == "do not resolve"
        assert evaluated.metadata.is_evaluated

    def test_original_value_is_overridden_when_using_do_no_resolve(self):
        concept = Concept("foo", body="original value").def_var("a", "original value")
        sheerka, context, concept = self.init_concepts(concept, eval_body=True)
        concept.compiled["a"] = DoNotResolve("do not resolve")
        concept.compiled[ConceptParts.BODY] = DoNotResolve("do not resolve")

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.body == "do not resolve"
        assert evaluated.get_value("a") == "do not resolve"
        assert evaluated.metadata.is_evaluated

    def test_variables_are_evaluated_before_body(self):
        sheerka, context, concept = self.init_concepts(Concept("foo", body="a+1").def_var("a", "10"), eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated == CB(concept, 11)

    def test_i_can_evaluate_when_another_concept_is_referenced(self):
        sheerka, context, concept_a, concept = self.init_concepts(
            Concept("a"),
            Concept("foo", body="a"),
            eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated == CB("foo", CB("a", BuiltinConcepts.NOT_INITIALIZED))
        assert evaluated.metadata.is_evaluated
        assert evaluated.body.metadata.is_evaluated

    def test_i_can_evaluate_when_the_referenced_concept_has_a_body(self):
        sheerka, context, concept_a, concept = self.init_concepts(
            Concept("a", body="1"),
            Concept("foo", body="a"),
            eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.key == concept.key
        assert evaluated.body == CB("a", 1)
        assert not concept_a.metadata.is_evaluated
        assert evaluated.metadata.is_evaluated

    def test_i_can_evaluate_concept_of_concept_when_the_leaf_has_a_body(self):
        sheerka, context, concept_a, concept_b, concept_c, concept_d = self.init_concepts(
            Concept("a", body="'a'"),
            Concept("b", body="a"),
            Concept("c", body="b"),
            Concept("d", body="c"),
            eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept_d)

        assert evaluated.key == concept_d.key
        expected = CB("c", CB("b", CB("a", "a")))
        assert evaluated.body == expected
        assert sheerka.objvalue(evaluated) == 'a'
        assert evaluated.metadata.is_evaluated

    def test_i_can_evaluate_concept_of_concept_does_not_have_a_body(self):
        sheerka, context, concept_a, concept_b, concept_c, concept_d = self.init_concepts(
            Concept("a"),
            Concept("b", body="a"),
            Concept("c", body="b"),
            Concept("d", body="c"),
            eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept_d)

        assert evaluated.key == concept_d.key
        expected = CB("c", CB("b", CB("a", BuiltinConcepts.NOT_INITIALIZED)))
        assert evaluated.body == expected
        assert sheerka.objvalue(evaluated) == CB("a", BuiltinConcepts.NOT_INITIALIZED)
        assert evaluated.metadata.is_evaluated

    def test_i_can_evaluate_concept_when_variables_reference_others_concepts(self):
        sheerka, context, concept_a, concept = self.init_concepts(
            Concept("a"),
            Concept("foo", body="a").def_var("a", "a"),
            eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept)

        # first, prop a is evaluated to concept_a
        # then body is evaluated to prop a -> concept_a
        assert evaluated.key == concept.key
        assert evaluated.body == concept_a

    def test_i_can_evaluate_concept_when_variables_reference_others_concepts_2(self):
        """
        Same test,
        but the name of the property and the name of the concept are different
        :return:
        """
        sheerka, context, concept_a = self.init_concepts(Concept(name="a"), eval_body=True)

        concept = Concept("foo", body="concept_a").def_var("concept_a", "a")
        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.key == concept.key
        assert evaluated.body == concept_a

    def test_i_can_evaluate_concept_when_variables_reference_others_concepts_with_body(self):
        sheerka, context, *concepts = self.init_concepts(
            Concept(name="a", body="1"),
            Concept(name="b", body="2"),
            eval_body=True
        )

        concept = Concept("foo", body="propA + propB").def_var("propA", "a").def_var("propB", "b")
        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.key == concept.key
        assert evaluated.body == 3

    def test_i_can_evaluate_concept_when_variables_is_a_concept(self):
        sheerka, context, concept_a = self.init_concepts(Concept(name="a", body="'a'"), eval_body=True)

        concept = Concept("foo").def_var("a")
        concept.compiled["a"] = concept_a
        evaluated = sheerka.evaluate_concept(context, concept)

        assert evaluated.key == concept.key
        assert evaluated.get_value("a") == CB("a", "a")

    def test_i_can_evaluate_concept_when_variable_is_a_concept_token(self):
        sheerka, context, concept_a, concept_b = self.init_concepts(
            "a",
            Concept("b").def_var("var_name", "c:a:"),  # c:a: means concept 'a'
            eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept_b)

        assert evaluated.key == concept_b.key
        assert evaluated.get_value("var_name") == concept_a

    def test_i_can_evaluate_concept_when_body_isa_concept_token(self):
        sheerka, context, concept_a, concept_b = self.init_concepts(
            "a",
            Concept("b", body="c:a:"),  # c:a: means concept 'a'
            eval_body=True)

        evaluated = sheerka.evaluate_concept(context, concept_b)

        assert evaluated.key == concept_b.key
        assert evaluated.body == concept_a

    def test_i_can_evaluate_when_variable_asts_is_a_list(self):
        sheerka = self.get_sheerka()
        foo = Concept("foo", body="1")

        concept = Concept("to_eval").def_var("prop")
        concept.compiled["prop"] = [foo, DoNotResolve("1")]
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, True), concept)

        variables = evaluated.get_value("prop")
        assert len(variables) == 2
        assert variables[0] == CB("foo", 1)
        assert variables[1] == "1"

    def test_i_can_evaluate_when_compiled_is_set_up_with_return_value(self):
        sheerka = self.get_sheerka()

        python_node = PythonNode("1 +1 ")
        parser_result = ParserResultConcept(parser="who", value=python_node)

        concept = Concept("to_eval").def_var("prop")
        concept.compiled["prop"] = [ReturnValueConcept("who", True, parser_result)]
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, True), concept)
        assert evaluated.get_value("prop") == 2

        # also works when only one return value
        concept = Concept("to_eval").def_var("prop")
        concept.compiled["prop"] = ReturnValueConcept("who", True, parser_result)
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, True), concept)
        assert evaluated.get_value("prop") == 2

    def test_i_can_reference_sheerka(self):
        sheerka = self.get_sheerka()

        concept = Concept("foo", body="sheerka.test()").init_key()
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, True), concept)

        assert evaluated.key == concept.key
        assert evaluated.body == sheerka.test()

    def test_variables_values_takes_precedence_over_the_outside_world(self):
        sheerka, context, concept_a, concept_b = self.init_concepts(
            Concept(name="a", body="'concept_a'"),
            Concept(name="b", body="'concept_b'"),
            eval_body=True
        )

        concept = Concept("foo", body="a")
        evaluated = sheerka.evaluate_concept(context, concept)
        assert evaluated.key == concept.key
        assert evaluated.body == CB("a", "concept_a")  # this test was already done

        # so check this one.
        concept = Concept("foo", body="a").def_var("a", "'property_a'")
        evaluated = sheerka.evaluate_concept(context, concept)
        assert evaluated.key == concept.key
        assert evaluated.body == 'property_a'

        # or this one.
        concept = Concept("foo", body="a").def_var("a", "b")
        evaluated = sheerka.evaluate_concept(context, concept)
        assert evaluated.key == concept.key
        assert evaluated.body == CB("b", "concept_b")

    def test_variables_values_takes_precedence(self):
        sheerka, context, concept_a, concept_b = self.init_concepts(
            Concept(name="a", body="'concept_a'"),
            Concept(name="b", body="'concept_b'"),
            eval_body=True
        )

        concept = Concept("foo", body="a + b").def_var("a", "'prop_a'").init_key()
        evaluated = sheerka.evaluate_concept(context, concept)
        assert evaluated.key == concept.key
        assert evaluated.body == 'prop_aconcept_b'

    def test_i_can_reference_sub_property_of_a_variable(self):
        sheerka, context, concept_a = self.init_concepts(
            Concept(name="concept_a").def_var("subProp", "'sub_a'"),
            eval_body=True
        )

        concept = Concept("foo", body="a.subProp").def_var("a", "concept_a")
        evaluated = sheerka.evaluate_concept(context, concept)
        assert evaluated == CB(concept.key, "sub_a")

    def test_i_cannot_evaluate_concept_if_variable_is_in_error(self):
        sheerka = self.get_sheerka()

        concept = Concept(name="concept_a").def_var("subProp", "undef_concept")
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, True), concept)
        assert sheerka.isinstance(evaluated, BuiltinConcepts.CONCEPT_EVAL_ERROR)

    def test_key_is_initialized_by_evaluation(self):
        sheerka = self.get_sheerka()

        concept = Concept("foo")
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, True), concept)

        assert evaluated.key == concept.init_key().key

    @pytest.mark.parametrize("where_clause, expected", [
        ("True", True),
        ("False", False),
        ("self < 10", False),
        ("self < 11", True),
        ("a < 20", False),
        ("a > 19", True),
        ("a + self > 20", True),
    ])
    def test_i_can_evaluate_simple_where(self, where_clause, expected):
        sheerka, context, concept = self.init_concepts(
            Concept("foo", body="10", where=where_clause).def_var("a", "20"),
        )

        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, False, True), concept)

        if expected:
            assert evaluated.key == concept.key
        else:
            assert sheerka.isinstance(evaluated, BuiltinConcepts.WHERE_CLAUSE_FAILED)
            assert evaluated.body == concept

    def test_i_can_evaluate_where_when_using_other_concept(self):
        sheerka, context, foo_true, foo_false = self.init_concepts(
            Concept("foo_true", body="True"),
            Concept("foo_false", body="False"),
        )

        concept = Concept("foo", where="foo_true").init_key()
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, False, True), concept)
        assert evaluated.key == concept.key

        concept = Concept("foo", where="foo_false")
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, False, True), concept)
        assert sheerka.isinstance(evaluated, BuiltinConcepts.WHERE_CLAUSE_FAILED)
        assert evaluated.body == concept

        concept = Concept("foo", where="foo_false").init_key()
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, False, False), concept)
        assert evaluated.key == concept.key

    def test_i_can_evaluate_disable_where_clause_evaluation(self):
        sheerka, context, concept = self.init_concepts(
            Concept("foo", body="10", where="a > 10").def_var("a", None),
        )
        evaluated = sheerka.evaluate_concept(context, concept)
        assert evaluated.key == concept.key

        context.local_hints.add(BuiltinConcepts.EVAL_WHERE_REQUESTED)
        evaluated = sheerka.evaluate_concept(context, concept)
        assert sheerka.isinstance(evaluated, BuiltinConcepts.CONCEPT_EVAL_ERROR)

    def test_i_can_detect_infinite_recursion_with_numeric_constant(self):
        sheerka, context, one_str, one_digit = self.init_concepts(
            Concept("one", body="1"),
            Concept("1", body="one"),
            eval_body=True
        )

        sheerka.add_in_cache(one_str)
        sheerka.add_in_cache(one_digit)

        evaluated = sheerka.evaluate_concept(context, one_digit)
        assert evaluated.key == one_digit.key
        assert evaluated.body == InfiniteRecursionResolved(1)

        evaluated = sheerka.evaluate_concept(context, one_str)
        assert evaluated.key == one_str.key
        assert evaluated.body == InfiniteRecursionResolved(1)

    def test_i_can_detect_infinite_recursion_with_boolean_constant(self):
        sheerka, context, true_str, true_bool = self.init_concepts(
            Concept("true", body="True"),
            Concept("True", body="true"),
            eval_body=True
        )

        evaluated = sheerka.evaluate_concept(context, true_str)
        assert evaluated.key == true_str.key
        assert evaluated.body == InfiniteRecursionResolved(True)

        evaluated = sheerka.evaluate_concept(context, true_bool)
        assert evaluated.key == true_bool.key
        assert evaluated.body == InfiniteRecursionResolved(True)

    def test_i_can_detect_infinite_recursion_with_constant_with_more_concepts(self):
        sheerka, context, c1, c2, c3, c4 = self.init_concepts(
            Concept("one", body="1"),
            Concept("1", body="2"),
            Concept("2", body="3"),
            Concept("3", body="one"),
            eval_body=True
        )

        for concept in (c1, c2, c3, c4):
            evaluated = sheerka.evaluate_concept(context, concept)
            assert evaluated.key == concept.key
            assert evaluated.body == InfiniteRecursionResolved(3)

    def test_i_can_detect_infinite_recursion_when_no_constant(self):
        sheerka, context, foo, bar, baz, qux = self.init_concepts(
            Concept("foo", body="bar"),
            Concept("bar", body="baz"),
            Concept("baz", body="qux"),
            Concept("qux", body="foo"),
            eval_body=True
        )

        evaluated = sheerka.evaluate_concept(context, foo)
        assert sheerka.isinstance(evaluated, BuiltinConcepts.CHICKEN_AND_EGG)
        assert evaluated.body == {foo, bar, baz, qux}

        evaluated = sheerka.evaluate_concept(context, bar)
        assert sheerka.isinstance(evaluated, BuiltinConcepts.CHICKEN_AND_EGG)
        assert evaluated.body == {foo, bar, baz, qux}

        evaluated = sheerka.evaluate_concept(context, baz)
        assert sheerka.isinstance(evaluated, BuiltinConcepts.CHICKEN_AND_EGG)
        assert evaluated.body == {foo, bar, baz, qux}

        evaluated = sheerka.evaluate_concept(context, qux)
        assert sheerka.isinstance(evaluated, BuiltinConcepts.CHICKEN_AND_EGG)
        assert evaluated.body == {foo, bar, baz, qux}

    def test_i_can_detect_auto_recursion(self):
        sheerka, context, foo = self.init_concepts(
            Concept("foo", body="foo"),
            eval_body=True
        )

        evaluated = sheerka.evaluate_concept(context, foo)
        assert sheerka.isinstance(evaluated, BuiltinConcepts.CHICKEN_AND_EGG)
        assert evaluated.body == {foo}

    def test_i_can_manage_auto_recursion_when_constant(self):
        sheerka = self.get_sheerka()

        one = Concept("1", body="1")
        evaluated = sheerka.evaluate_concept(self.get_context(sheerka, True), one)
        assert evaluated.key == one.key
        assert evaluated.body == 1

    def test_i_can_evaluate_thousand_concept(self):
        sheerka, context, thousand, number, forties, forty, one = self.init_concepts(
            Concept("thousand", body="number * 1000").def_var("number"),
            Concept("number"),
            Concept("forties", body="forty + number").def_var("forty").def_var("number"),
            Concept("forty", body="40"),
            Concept("one", body="1"),
            eval_body=True
        )

        one = sheerka.new("one")
        number2 = sheerka.new("number")
        number2.compiled["one"] = one
        number2.compiled[ConceptParts.BODY] = one
        forties = sheerka.new("forties")
        forties.compiled["forty"] = sheerka.new("forty")
        forties.compiled["number"] = number2

        number1 = sheerka.new("number")
        number1.compiled["forties"] = forties
        number1.compiled[ConceptParts.BODY] = forties

        forty_one_thousand = sheerka.new("thousand")
        forty_one_thousand.compiled["number"] = number1

        evaluated = sheerka.evaluate_concept(context, forty_one_thousand)

        assert evaluated.body == 41000