def test_visit_expression():
col1 = Column("al", "t1", "c1")
literal1 = Literal("al2", "test")
mapping = Column("al2", "t1", "tags")
key = Literal(None, "myTag")
tag = SubscriptableReference(None, mapping, key)
f1 = FunctionCall("al3", "f1", [col1, literal1, tag])
col2 = Column("al4", "t1", "c2")
literal2 = Literal("al5", "test2")
f2 = FunctionCall("al6", "f2", [col2, literal2])
curried = CurriedFunctionCall("al7", f2, [f1])
visitor = DummyVisitor()
ret = curried.accept(visitor)
expected = [
curried,
f2,
col2,
literal2,
f1,
col1,
literal1,
tag,
mapping,
key,
]
# Tests the state changes on the Visitor
assert visitor.get_visited_nodes() == expected
# Tests the return value of the visitor
assert ret == expected
def test_visit_expression():
col1 = Column("al", "c1", "t1")
literal1 = Literal("al2", "test")
f1 = FunctionCall("al3", "f1", [col1, literal1])
col2 = Column("al4", "c2", "t1")
literal2 = Literal("al5", "test2")
f2 = FunctionCall("al6", "f2", [col2, literal2])
curried = CurriedFunctionCall("al7", f2, [f1])
visitor = DummyVisitor()
ret = curried.accept(visitor)
expected = [curried, f2, col2, literal2, f1, col1, literal1]
# Tests the state changes on the Visitor
assert visitor.get_visited_nodes() == expected
# Tests the return value of the visitor
assert ret == expected