def TestMatchOnlyCommonAttributes():
""" Test Matching only common attributes."""
s1 = S.Selector(value='0', asdf='Haha', match_only_common_attributes=False)
s2 = S.Selector(value='0', asdf='Haha', match_only_common_attributes=True)
l1 = list(S.RunSelector(s1, tree))
l2 = list(S.RunSelector(s2, tree))
assert (len(l1) == 0)
assert (len(l2) == 1)
assert (l2[0].GetName() == 'Constant')
return
def TestEmptySelector():
""" Test the Empty Selector (Should return all nodes)."""
s = S.Selector()
l = list(S.RunSelector(s, tree))
# print l, len(l)
assert (len(l) == 18) # Number of nodes in the tree.
return
def TestOr():
""" Test Or selector combinator."""
s1 = S.Selector(node_type=N.CallFunction)
s2 = S.Selector(node_type=N.DefineFunction)
l1 = list(S.RunSelector(S.Or(s1, s2), tree))
assert (len(l1) == 2)
return
def TestAnd():
""" Test And selector combinator."""
s1 = S.Selector(node_type='Constant')
s2 = S.Selector(value='1')
l1 = list(S.RunSelector(S.And(s1, s2), tree))
assert (len(l1) == 2)
return
def TestCommonAttributesFunction1():
""" Test match common attributes for lambda function - success."""
s1 = S.Selector(function=lambda x: x.value == '0',
match_only_common_attributes=True)
l1 = list(S.RunSelector(s1, tree))
assert (len(l1) == 1)
assert (l1[0].GetName() == 'Constant')
return
def TestSelectorNodeType():
""" Test a Basic Selector that takes a node type as a class."""
s1 = S.Selector(node_type=N.Constant)
l1 = list(S.RunSelector(s1, tree))
assert (len(l1) == 3)
assert (l1[0].GetName() == 'Constant')
assert (l1[1].GetName() == 'Constant')
assert (l1[2].GetName() == 'Constant')
return
def TestHasDescendant():
""" Test HasDescendant selector combinator."""
s1 = S.Selector(node_type=N.Return)
s2 = S.Selector(node_type=N.CallFunction)
l1 = list(S.RunSelector(S.HasDescendant(s1, s2), tree))
assert (len(l1) == 1)
assert (l1[0].GetName() == 'Return')
return
def TestHasChild():
""" Test HasChild selector combinator."""
s1 = S.Selector(node_type=N.CallFunction)
s2 = S.Selector(node_type=N.ReferVariable)
l1 = list(S.RunSelector(S.HasChild(s1, s2), tree))
assert (len(l1) == 1)
assert (l1[0].GetName() == 'CallFunction')
return
def TestHasAncestor():
""" Test HasAncestor selector combinator."""
s1 = S.Selector(node_type=N.Return)
s2 = S.Selector(node_type=N.CallFunction)
l1 = list(S.RunSelector(S.HasAncestor(s2, s1), tree))
assert (len(l1) == 1)
assert (l1[0].GetName() == 'CallFunction')
return
def TestHasParent():
""" Test HasParent selector combinator."""
s1 = S.Selector(node_type=N.CallFunction)
s2 = S.Selector(node_type=N.ReferVariable)
l1 = list(S.RunSelector(S.HasParent(s2, s1), tree))
assert (len(l1) == 1)
assert (l1[0].GetName() == 'ReferVariable')
return
def TestAttributeMatcherFunction():
""" Test Attribute Matcher Function."""
s = S.Selector(attr_matcher=lambda x, y: x.lower() == y.lower(),
name='fAcToRiaL')
l = list(S.RunSelector(s, tree))
assert (len(l) == 2)
# Depends on order of traversal
assert (l[0].GetName() == 'DefineFunction')
assert (l[1].GetName() == 'ReferVariable')
return
def TestSelector():
""" Test Basic Selectors - function, attributes, node_type"""
s1 = S.Selector(node_type='DefineFunction')
s2 = S.Selector(value='0')
s3 = S.Selector(function=lambda x: hasattr(x, 'value') and x.value == '1')
l1 = list(S.RunSelector(s1, tree))
l2 = list(S.RunSelector(s2, tree))
l3 = list(S.RunSelector(s3, tree))
assert (len(l1) == 1 and len(l2) == 1 and len(l3) == 2)
r1, r2, r3, r4 = l1[0], l2[0], l3[0], l3[1]
assert (r1.name == 'Factorial')
assert (r2.GetName() == 'Constant')
assert (r3.GetName() == 'Constant')
assert (r4.GetName() == 'Constant')
return
def TestCommonAttributesFunction2():
""" Test match common attributes for lambda function - failure."""
s1 = S.Selector(function=lambda x: x.value == '0' and x.asdf == 'Haha',
match_only_common_attributes=False)
l1 = list(S.RunSelector(s1, tree))
return