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]))