80 lines
3.8 KiB
Python
80 lines
3.8 KiB
Python
from core.builtin_concepts import BuiltinConcepts
|
|
from core.sheerka.services.SheerkaDebugManager import NullDebugLogger
|
|
from core.sheerka.services.SheerkaOut import SheerkaOut
|
|
from core.sheerka.services.SheerkaRuleManager import FormatAstList, FormatAstVariable, FormatAstDict, FormatAstMulti
|
|
from out.DeveloperVisitor import DeveloperVisitor
|
|
|
|
from tests.TestUsingMemoryBasedSheerka import TestUsingMemoryBasedSheerka
|
|
|
|
|
|
class TestDeveloperVisitor(TestUsingMemoryBasedSheerka):
|
|
def test_i_can_develop_list(self):
|
|
sheerka, context = self.init_concepts()
|
|
service_out = sheerka.services[SheerkaOut.NAME]
|
|
dev_visitor = DeveloperVisitor(service_out, NullDebugLogger(), set(), 0)
|
|
|
|
bag = {"to_format": ["a", "b", "c"]}
|
|
|
|
res = dev_visitor.visit(context, FormatAstList("to_format"), bag)
|
|
assert res == FormatAstList(variable="to_format", items=[
|
|
FormatAstVariable(name="__item", index=0, value="a"),
|
|
FormatAstVariable(name="__item", index=1, value="b"),
|
|
FormatAstVariable(name="__item", index=2, value="c"),
|
|
])
|
|
|
|
def test_i_can_develop_list_of_list(self):
|
|
sheerka, context = self.init_concepts()
|
|
service_out = sheerka.services[SheerkaOut.NAME]
|
|
dev_visitor = DeveloperVisitor(service_out, NullDebugLogger(), set(), 0)
|
|
|
|
bag = {"to_format": [["a1", "a2"], ["b1"]]}
|
|
|
|
res = dev_visitor.visit(context, FormatAstList("to_format"), bag)
|
|
assert res == FormatAstList(variable="to_format", items=[
|
|
FormatAstList(variable="__item", index=0, debug=True, prefix='[', suffix=']', items=[
|
|
FormatAstVariable(name="__item", index=0, debug=True, value="a1"),
|
|
FormatAstVariable(name="__item", index=1, debug=True, value="a2"),
|
|
]),
|
|
FormatAstList(variable="__item", index=1, debug=True, prefix='[', suffix=']', items=[
|
|
FormatAstVariable(name="__item", index=0, debug=True, value="b1"),
|
|
]),
|
|
])
|
|
|
|
def test_i_can_develop_dict(self):
|
|
sheerka, context = self.init_concepts()
|
|
service_out = sheerka.services[SheerkaOut.NAME]
|
|
dev_visitor = DeveloperVisitor(service_out, NullDebugLogger(), set(), 0)
|
|
|
|
bag = {"to_format": {"a": "value1", "b": 3.14}}
|
|
|
|
res = dev_visitor.visit(context, FormatAstDict("to_format"), bag)
|
|
|
|
assert res == FormatAstDict(variable='to_format', items=[
|
|
(FormatAstVariable(name='__key', index=0, value='a'),
|
|
FormatAstVariable(name='__value', index='a', value='value1')),
|
|
(FormatAstVariable(name='__key', index=1, value='b'),
|
|
FormatAstVariable(name='__value', index='b', value=3.14))])
|
|
|
|
def test_i_can_develop_multi(self):
|
|
sheerka, context, *rules = self.init_format_rules(
|
|
("isinstance(to_format, BuiltinConcepts.TO_DICT)", "dict(to_format)"))
|
|
service_out = sheerka.services[SheerkaOut.NAME]
|
|
dev_visitor = DeveloperVisitor(service_out, NullDebugLogger(), set(), 0)
|
|
|
|
item1 = sheerka.new(BuiltinConcepts.TO_DICT, body={"a1": "value1"})
|
|
item2 = sheerka.new(BuiltinConcepts.TO_DICT, body={"a2": "value2"})
|
|
|
|
bag = {"to_format": sheerka.new(BuiltinConcepts.TO_MULTI, body=[item1, item2])}
|
|
|
|
res = dev_visitor.visit(context, FormatAstMulti("to_format"), bag)
|
|
|
|
assert res == FormatAstMulti("to_format", items=[
|
|
FormatAstDict(variable='to_format', items=[
|
|
(FormatAstVariable(name='__key', index=0, value='a1'),
|
|
FormatAstVariable(name='__value', index='a1', value='value1'))]),
|
|
|
|
FormatAstDict(variable='to_format', items=[
|
|
(FormatAstVariable(name='__key', index=0, value='a2'),
|
|
FormatAstVariable(name='__value', index='a2', value='value2'))])
|
|
])
|