def test_getElementsByAttr(self):
html = """<html> <head> <title> Hello </title> </head>
<body>
<div cheese="cheddar" id="cheddar1" >
<span> Hello </span>
</div>
<div cheese="bologna" id="not_really_cheese">
<span cheese="cheddar" id="cheddar2" > Goodbye </span>
</div>
</body>
</html>"""
parser = AdvancedHTMLParser()
parser.parseStr(html)
elements = parser.getElementsByAttr('cheese', 'cheddar')
assert len(elements) == 2
foundCheese1 = foundCheese2 = False
for element in elements:
myID = element.getAttribute('id')
if myID == 'cheddar1':
foundCheese1 = True
elif myID == 'cheddar2':
foundCheese2 = True
assert foundCheese1
assert foundCheese2
