import pytest

from core.builtin_concepts import ReturnValueConcept, BuiltinConcepts
from core.concept import Concept
from core.sheerka import Sheerka, ExecutionContext
from evaluators.EvalEvaluator import EvalEvaluator
from sdp.sheerkaDataProvider import Event


def get_context():
    sheerka = Sheerka(skip_builtins_in_db=True)
    sheerka.initialize("mem://")
    return ExecutionContext("test", Event(), sheerka)


def r(value, status=True):
    return ReturnValueConcept("some_name", status, value)


eval_requested = ReturnValueConcept("some_name", True, Concept(key=BuiltinConcepts.CONCEPT_EVAL_REQUESTED))


def test_i_can_match_and_eval():
    context = get_context()

    to_eval1 = ReturnValueConcept("some_name", True, Concept(name="2", body="to eval"))
    to_eval2 = ReturnValueConcept("some_name", True, Concept(name="3", body="also to eval"))

    return_values = [
        ReturnValueConcept("some_name", True, "not to eval"),
        ReturnValueConcept("some_name", True, Concept(name="not to eval")),
        ReturnValueConcept("some_name", False, Concept(name="1", body="not to eval")),
        to_eval1,
        to_eval2,
        eval_requested
    ]

    evaluator = EvalEvaluator()
    assert evaluator.matches(context, return_values)

    evaluated = evaluator.eval(context, return_values)
    assert len(evaluated) == 2
    assert evaluated[0].value == to_eval1.body.body
    assert evaluated[0].parents == [to_eval1, eval_requested]

    assert evaluated[1].value == to_eval2.body.body
    assert evaluated[1].parents == [to_eval2, eval_requested]


@pytest.mark.parametrize("return_values, expected", [
    ([r(Concept("foo", body="bar")), eval_requested], True),
    ([r(Concept("status is false", body="bar"), False), eval_requested], False),
    ([r("string_value"), eval_requested], False),
    ([r(Concept("no body")), eval_requested], False),
    ([r(Concept("eval requested missing", body="bar"))], False),
])
def test_i_cannot_match_if_eval_request_is_not_present(return_values, expected):
    context = get_context()
    assert EvalEvaluator().matches(context, return_values) == expected