def handle_starttag(self, tag, attrs):
if tag == "body":
self.body = 1
self.dom = DOMElement(tag='html')
self.currentElement = self.dom
if self.body == 1:
elem = DOMElement(tag=tag)
elem.set_attrs(dict(attrs))
self.currentElement.add_children(elem)
self.currentElement = elem
def setUp(self):
self.de = DOMElement()
self.div = DOMElement(tag='div')
p = DOMElement(tag='p')
subdiv = DOMElement(tag='div', classes='divClass1')
subdiv.add_children(p)
self.root = DOMElement(tag='div', classes='divClass divClass1')
self.root.add_children(p)
class TestDOMElement(unittest.TestCase):
def setUp(self):
self.de = DOMElement()
self.div = DOMElement(tag='div')
p = DOMElement(tag='p')
subdiv = DOMElement(tag='div', classes='divClass1')
subdiv.add_children(p)
self.root = DOMElement(tag='div', classes='divClass divClass1')
self.root.add_children(p)
def test_createwithtag(self):
de = DOMElement(tag='div')
self.assertEqual("<div></div>", str(de))
def test_createwithattr(self):
de = DOMElement(tag='div', id='divId', classes=["divClass1", "divClass2"], name="divName")
self.assertEqual("<div id='divId' name='divName' class='divClass1 divClass2'></div>", str(de))
def test_add_tag(self):
self.de.set_tag('html')
self.assertEqual("<html></html>", str(self.de))
def test_get_tag(self):
self.assertEqual('div', self.div.get_tag())
def test_add_id(self):
self.div.set_id('divId')
self.assertEqual("<div id='divId'></div>", str(self.div))
def test_get_id(self):
self.div.set_id('divId')
self.assertEqual('divId', self.div.get_id())
def test_set_class(self):
self.div.set_class('divClass')
self.assertEqual("<div class='divClass'></div>", str(self.div))
def test_add_class(self):
self.div.set_class('divClass1')
self.div.add_class('divClass2')
self.assertEqual("<div class='divClass1 divClass2'></div>", str(self.div))
def test_get_class1(self):
self.div.set_class('divClass')
self.assertEqual(['divClass'], self.div.get_class())
def test_get_class2(self):
self.div.set_class('divClass1')
self.div.add_class('divClass2')
self.assertEqual(['divClass1', 'divClass2'], self.div.get_class())
def test_set_name(self):
self.div.set_name('divName')
self.assertEqual("<div name='divName'></div>", str(self.div))
def test_get_name(self):
self.div.set_name('divName')
self.assertEqual('divName', self.div.get_name())
def test_add_children1(self):
self.div.set_class('divClass1')
self.de.set_tag('div')
self.de.set_class('divClass2')
self.div.add_children(self.de)
self.assertEqual("<div class='divClass1'><div class='divClass2'></div></div>", str(self.div))
def test_add_children2(self):
self.div.set_class('divClass1')
self.div.set_id('divId')
self.de.set_tag('div')
self.de.set_class('divClass2')
p = DOMElement(tag='p', name='pName')
self.de.add_children(p)
self.div.add_children(self.de)
self.assertEqual("<div class='divClass1' id='divId'><div class='divClass2'><p name='pName'></p></div></div>",
str(self.div))
def test_getElementByTag1(self):
self.assertEqual("<p></p>", str(self.root.get_elements_by_tag('p')[0]))
def handle_startendtag(self, tag, attrs):
if self.body == 1:
elem = DOMElement(tag=tag)
elem.set_attrs(dict(attrs))
self.currentElement.add_children(elem)
