import pytest

from core.builtin_concepts import BuiltinConcepts, ReturnValueConcept, UserInputConcept
from core.concept import Concept
from evaluators.PrepareEvalEvaluator import PrepareEvalEvaluator

from tests.TestUsingMemoryBasedSheerka import TestUsingMemoryBasedSheerka

r = ReturnValueConcept


class TestPrepareEvalEvaluator(TestUsingMemoryBasedSheerka):

    @pytest.mark.parametrize("ret_val, expected", [
        (r("name", True, UserInputConcept("eval 1 + 1")), True),
        (r("name", True, UserInputConcept("  eval 1 + 1")), True),
        (r("name", True, UserInputConcept("eval")), False),
        (r("name", True, UserInputConcept("1+1")), False),
        (r("name", True, UserInputConcept("")), False),
        (r("name", True, UserInputConcept("eval   ")), False),
        (r("name", True, UserInputConcept([])), False),
        (r("name", True, Concept("foo")), False),
        (r("name", True, "not a concept"), False),
        (r("name", False, UserInputConcept("eval 1 + 1")), False),
        (r("name", False, UserInputConcept("  eval 1 + 1")), False),
    ])
    def test_i_can_match(self, ret_val, expected):
        context = self.get_context()
        assert PrepareEvalEvaluator().matches(context, ret_val) == expected

    @pytest.mark.parametrize("ret_val, expected", [
        (r("name", True, UserInputConcept("eval 1 + 1")), "1 + 1"),
        (r("name", True, UserInputConcept("   eval 1 + 1")), "1 + 1"),
    ])
    def test_i_can_eval(self, ret_val, expected):
        context = self.get_context()
        sheerka = context.sheerka

        prepare_evaluator = PrepareEvalEvaluator()
        prepare_evaluator.matches(context, ret_val)
        res = prepare_evaluator.eval(context, ret_val)

        assert res.status
        assert sheerka.isinstance(res.body, BuiltinConcepts.USER_INPUT)
        assert res.body.body == expected

        assert BuiltinConcepts.EVAL_BODY_REQUESTED in context.global_hints
        assert BuiltinConcepts.CONCEPT_VALUE_REQUESTED in context.global_hints