69 lines
3.0 KiB
Python
69 lines
3.0 KiB
Python
import pytest
|
|
from core.builtin_concepts import ReturnValueConcept, BuiltinConcepts
|
|
from core.concept import Concept
|
|
from core.sheerka.services.SheerkaFunctionsParametersHistory import SheerkaFunctionsParametersHistory
|
|
from evaluators.UpdateFunctionsParametersEvaluator import UpdateFunctionsParametersEvaluator
|
|
from parsers.PythonParser import PythonNode
|
|
|
|
from tests.BaseTest import BaseTest
|
|
from tests.TestUsingMemoryBasedSheerka import TestUsingMemoryBasedSheerka
|
|
|
|
r = ReturnValueConcept
|
|
|
|
|
|
class TestUpdateFunctionsParametersEvaluator(TestUsingMemoryBasedSheerka):
|
|
|
|
@pytest.mark.parametrize("return_value, expected_result", [
|
|
(r("evaluators.Python", True), True),
|
|
(r("evaluators.Python", False), False),
|
|
(r("other_name", True), False),
|
|
(r("other_name", False), False),
|
|
])
|
|
def test_i_can_match(self, return_value, expected_result):
|
|
sheerka, context = self.init_concepts()
|
|
evaluator = UpdateFunctionsParametersEvaluator()
|
|
|
|
assert evaluator.matches(context, return_value) == expected_result
|
|
|
|
@pytest.mark.parametrize("return_value", [
|
|
r("evaluators.Python", True),
|
|
r("evaluators.Python", True, parents=[]),
|
|
r("evaluators.Python", True, parents=[BaseTest.pretval(Concept(), who="notPythonParser")]),
|
|
])
|
|
def test_i_cannot_eval_if_original_parser_is_not_found(self, return_value):
|
|
sheerka, context = self.init_concepts()
|
|
evaluator = UpdateFunctionsParametersEvaluator()
|
|
|
|
res = evaluator.eval(context, return_value)
|
|
|
|
assert res == return_value
|
|
assert not res.parents[-1].status
|
|
assert sheerka.isinstance(res.parents[-1].body, BuiltinConcepts.NOT_FOUND)
|
|
assert res.parents[-1].body.body == "source code"
|
|
|
|
@pytest.mark.parametrize("source, func_name, param_number, expected", [
|
|
("func()", "func", 0, []),
|
|
("func(10)", "func", 0, ["10"]),
|
|
("func(10, True, 'some string')", "func", 0, ["10"]),
|
|
("func(10, True, 'some string')", "func", 1, ["True"]),
|
|
("func(10, True, 'some string')", "func", 2, ["'some string'"]),
|
|
("func1(10) | func2(20)", "func2", 0, ["20"]),
|
|
("func1(10, func2(20), 'string')", "func1", 0, ["10"]),
|
|
("func1(10, func2(20), 'string')", "func1", 1, ["func2(20)"]),
|
|
("func1(10, func2(20), 'string')", "func1", 2, ["'string'"]),
|
|
("func1(10, func2(20), 'string')", "func2", 0, ["20"]),
|
|
])
|
|
def test_i_can_record_functions_parameters(self, source, func_name, param_number, expected):
|
|
sheerka, context = self.init_concepts()
|
|
|
|
parser_ret_val = self.pretval(PythonNode(source), who="parsers.Python", source=source)
|
|
return_value = r("evaluators.Python", True, parents=[parser_ret_val])
|
|
evaluator = UpdateFunctionsParametersEvaluator()
|
|
|
|
res = evaluator.eval(context, return_value)
|
|
|
|
assert res == return_value
|
|
|
|
service = sheerka.services[SheerkaFunctionsParametersHistory.NAME]
|
|
assert service.get_function_parameters(func_name, param_number) == expected
|