Introduction to XSL

from xml.dom import pulldom
import unittest

SMALL_SAMPLE = """<?xml version="1.0"?>
<html >
<!-- A comment -->
<title>Introduction to XSL</title>
<hr/>
<p></p>
</html>"""

class Test(unittest.TestCase):
    def test_comment(self):
        items = pulldom.parseString(SMALL_SAMPLE)
        for evt, node in items:
            if evt == pulldom.COMMENT:
                break
        else:
            self.fail("No comment was encountered")
            
    def test_end_document(self):
        items = pulldom.parseString(SMALL_SAMPLE)
        # Read all of the nodes up to and including </html>:
        for evt, node in items:
            if evt == pulldom.END_ELEMENT and node.tagName == "html":
                break
        try:
            # Assert that the next node is END_DOCUMENT:
            evt, node = next(items)
            self.assertEqual(pulldom.END_DOCUMENT, evt)
        except StopIteration as stop:
            self.fail(
                "Ran out of events, but should have received END_DOCUMENT")


if __name__ == "__main__":
    unittest.main()