class TestDocumentParser(TestCase): def setUp(self): self.parser = DocumentParser() def test_empty(self): self.parser.feed('') doc = self.parser.root self.assertEqual(doc.length, 2) self.assertEqual(doc.nodeType, Node.DOCUMENT_NODE) self.assertEqual(doc.doctype.nodeType, Node.DOCUMENT_TYPE_NODE) self.assertEqual(doc.html.length, 2) def test_single_tag(self): self.parser.feed('<h1>title</h1>') doc = self.parser.root self.assertEqual(doc.html.length, 2) self.assertEqual(doc.body.length, 2) # first child is <script> self.assertEqual(doc.body.lastChild.tagName, 'H1') def test_full_docc(self): self.parser.feed( '<!DOCTYPE html><html><head><title>Title</title></head>' '<body><h1>title</h1></body></html>') doc = self.parser.root self.assertEqual(doc.html.length, 2) self.assertEqual(doc.body.length, 2) # first child is <script> self.assertEqual(doc.body.lastChild.tagName, 'H1')
def setUp(self): self.parser = DocumentParser()