346 lines
12 KiB
Python
346 lines
12 KiB
Python
|
import io
|
||
|
|
import unittest
|
||
|
|
import sys
|
||
|
|
import xml.sax
|
||
|
|
|
||
|
|
from xml.sax.xmlreader import AttributesImpl
|
||
|
|
from xml.dom import pulldom
|
||
|
|
|
||
|
|
from test.support import run_unittest, findfile
|
||
|
|
|
||
|
|
|
||
|
|
tstfile = findfile("test.xml", subdir="xmltestdata")
|
||
|
|
|
||
|
|
# A handy XML snippet, containing attributes, a namespace prefix, and a
|
||
|
|
# self-closing tag:
|
||
|
|
SMALL_SAMPLE = """<?xml version="1.0"?>
|
||
|
|
<html xmlns="http://www.w3.org/1999/xhtml" xmlns:xdc="http://www.xml.com/books">
|
||
|
|
<!-- A comment -->
|
||
|
|
<title>Introduction to XSL</title>
|
||
|
|
<hr/>
|
||
|
|
<p><xdc:author xdc:attrib="prefixed attribute" attrib="other attrib">A. Namespace</xdc:author></p>
|
||
|
|
</html>"""
|
||
|
|
|
||
|
|
|
||
|
|
class PullDOMTestCase(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_parse(self):
|
||
|
|
"""Minimal test of DOMEventStream.parse()"""
|
||
|
|
|
||
|
|
# This just tests that parsing from a stream works. Actual parser
|
||
|
|
# semantics are tested using parseString with a more focused XML
|
||
|
|
# fragment.
|
||
|
|
|
||
|
|
# Test with a filename:
|
||
|
|
list(pulldom.parse(tstfile))
|
||
|
|
|
||
|
|
# Test with a file object:
|
||
|
|
with open(tstfile, "rb") as fin:
|
||
|
|
list(pulldom.parse(fin))
|
||
|
|
|
||
|
|
def test_parse_semantics(self):
|
||
|
|
"""Test DOMEventStream parsing semantics."""
|
||
|
|
|
||
|
|
items = pulldom.parseString(SMALL_SAMPLE)
|
||
|
|
evt, node = next(items)
|
||
|
|
# Just check the node is a Document:
|
||
|
|
self.assertTrue(hasattr(node, "createElement"))
|
||
|
|
self.assertEqual(pulldom.START_DOCUMENT, evt)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.START_ELEMENT, evt)
|
||
|
|
self.assertEqual("html", node.tagName)
|
||
|
|
self.assertEqual(2, len(node.attributes))
|
||
|
|
self.assertEqual(node.attributes.getNamedItem("xmlns:xdc").value,
|
||
|
|
"http://www.xml.com/books")
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.CHARACTERS, evt) # Line break
|
||
|
|
evt, node = next(items)
|
||
|
|
# XXX - A comment should be reported here!
|
||
|
|
# self.assertEqual(pulldom.COMMENT, evt)
|
||
|
|
# Line break after swallowed comment:
|
||
|
|
self.assertEqual(pulldom.CHARACTERS, evt)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual("title", node.tagName)
|
||
|
|
title_node = node
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.CHARACTERS, evt)
|
||
|
|
self.assertEqual("Introduction to XSL", node.data)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.END_ELEMENT, evt)
|
||
|
|
self.assertEqual("title", node.tagName)
|
||
|
|
self.assertTrue(title_node is node)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.CHARACTERS, evt)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.START_ELEMENT, evt)
|
||
|
|
self.assertEqual("hr", node.tagName)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.END_ELEMENT, evt)
|
||
|
|
self.assertEqual("hr", node.tagName)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.CHARACTERS, evt)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.START_ELEMENT, evt)
|
||
|
|
self.assertEqual("p", node.tagName)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.START_ELEMENT, evt)
|
||
|
|
self.assertEqual("xdc:author", node.tagName)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.CHARACTERS, evt)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.END_ELEMENT, evt)
|
||
|
|
self.assertEqual("xdc:author", node.tagName)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.END_ELEMENT, evt)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.CHARACTERS, evt)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(pulldom.END_ELEMENT, evt)
|
||
|
|
# XXX No END_DOCUMENT item is ever obtained:
|
||
|
|
#evt, node = next(items)
|
||
|
|
#self.assertEqual(pulldom.END_DOCUMENT, evt)
|
||
|
|
|
||
|
|
def test_expandItem(self):
|
||
|
|
"""Ensure expandItem works as expected."""
|
||
|
|
items = pulldom.parseString(SMALL_SAMPLE)
|
||
|
|
# Loop through the nodes until we get to a "title" start tag:
|
||
|
|
for evt, item in items:
|
||
|
|
if evt == pulldom.START_ELEMENT and item.tagName == "title":
|
||
|
|
items.expandNode(item)
|
||
|
|
self.assertEqual(1, len(item.childNodes))
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
self.fail("No \"title\" element detected in SMALL_SAMPLE!")
|
||
|
|
# Loop until we get to the next start-element:
|
||
|
|
for evt, node in items:
|
||
|
|
if evt == pulldom.START_ELEMENT:
|
||
|
|
break
|
||
|
|
self.assertEqual("hr", node.tagName,
|
||
|
|
"expandNode did not leave DOMEventStream in the correct state.")
|
||
|
|
# Attempt to expand a standalone element:
|
||
|
|
items.expandNode(node)
|
||
|
|
self.assertEqual(next(items)[0], pulldom.CHARACTERS)
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(node.tagName, "p")
|
||
|
|
items.expandNode(node)
|
||
|
|
next(items) # Skip character data
|
||
|
|
evt, node = next(items)
|
||
|
|
self.assertEqual(node.tagName, "html")
|
||
|
|
with self.assertRaises(StopIteration):
|
||
|
|
next(items)
|
||
|
|
items.clear()
|
||
|
|
self.assertIsNone(items.parser)
|
||
|
|
self.assertIsNone(items.stream)
|
||
|
|
|
||
|
|
@unittest.expectedFailure
|
||
|
|
def test_comment(self):
|
||
|
|
"""PullDOM does not receive "comment" events."""
|
||
|
|
items = pulldom.parseString(SMALL_SAMPLE)
|
||
|
|
for evt, _ in items:
|
||
|
|
if evt == pulldom.COMMENT:
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
self.fail("No comment was encountered")
|
||
|
|
|
||
|
|
@unittest.expectedFailure
|
||
|
|
def test_end_document(self):
|
||
|
|
"""PullDOM does not receive "end-document" events."""
|
||
|
|
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:
|
||
|
|
self.fail(
|
||
|
|
"Ran out of events, but should have received END_DOCUMENT")
|
||
|
|
|
||
|
|
|
||
|
|
class ThoroughTestCase(unittest.TestCase):
|
||
|
|
"""Test the hard-to-reach parts of pulldom."""
|
||
|
|
|
||
|
|
def test_thorough_parse(self):
|
||
|
|
"""Test some of the hard-to-reach parts of PullDOM."""
|
||
|
|
self._test_thorough(pulldom.parse(None, parser=SAXExerciser()))
|
||
|
|
|
||
|
|
@unittest.expectedFailure
|
||
|
|
def test_sax2dom_fail(self):
|
||
|
|
"""SAX2DOM can"t handle a PI before the root element."""
|
||
|
|
pd = SAX2DOMTestHelper(None, SAXExerciser(), 12)
|
||
|
|
self._test_thorough(pd)
|
||
|
|
|
||
|
|
def test_thorough_sax2dom(self):
|
||
|
|
"""Test some of the hard-to-reach parts of SAX2DOM."""
|
||
|
|
pd = SAX2DOMTestHelper(None, SAX2DOMExerciser(), 12)
|
||
|
|
self._test_thorough(pd, False)
|
||
|
|
|
||
|
|
def _test_thorough(self, pd, before_root=True):
|
||
|
|
"""Test some of the hard-to-reach parts of the parser, using a mock
|
||
|
|
parser."""
|
||
|
|
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.START_DOCUMENT, evt)
|
||
|
|
# Just check the node is a Document:
|
||
|
|
self.assertTrue(hasattr(node, "createElement"))
|
||
|
|
|
||
|
|
if before_root:
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.COMMENT, evt)
|
||
|
|
self.assertEqual("a comment", node.data)
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.PROCESSING_INSTRUCTION, evt)
|
||
|
|
self.assertEqual("target", node.target)
|
||
|
|
self.assertEqual("data", node.data)
|
||
|
|
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.START_ELEMENT, evt)
|
||
|
|
self.assertEqual("html", node.tagName)
|
||
|
|
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.COMMENT, evt)
|
||
|
|
self.assertEqual("a comment", node.data)
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.PROCESSING_INSTRUCTION, evt)
|
||
|
|
self.assertEqual("target", node.target)
|
||
|
|
self.assertEqual("data", node.data)
|
||
|
|
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.START_ELEMENT, evt)
|
||
|
|
self.assertEqual("p", node.tagName)
|
||
|
|
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.CHARACTERS, evt)
|
||
|
|
self.assertEqual("text", node.data)
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.END_ELEMENT, evt)
|
||
|
|
self.assertEqual("p", node.tagName)
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.END_ELEMENT, evt)
|
||
|
|
self.assertEqual("html", node.tagName)
|
||
|
|
evt, node = next(pd)
|
||
|
|
self.assertEqual(pulldom.END_DOCUMENT, evt)
|
||
|
|
|
||
|
|
|
||
|
|
class SAXExerciser(object):
|
||
|
|
"""A fake sax parser that calls some of the harder-to-reach sax methods to
|
||
|
|
ensure it emits the correct events"""
|
||
|
|
|
||
|
|
def setContentHandler(self, handler):
|
||
|
|
self._handler = handler
|
||
|
|
|
||
|
|
def parse(self, _):
|
||
|
|
h = self._handler
|
||
|
|
h.startDocument()
|
||
|
|
|
||
|
|
# The next two items ensure that items preceding the first
|
||
|
|
# start_element are properly stored and emitted:
|
||
|
|
h.comment("a comment")
|
||
|
|
h.processingInstruction("target", "data")
|
||
|
|
|
||
|
|
h.startElement("html", AttributesImpl({}))
|
||
|
|
|
||
|
|
h.comment("a comment")
|
||
|
|
h.processingInstruction("target", "data")
|
||
|
|
|
||
|
|
h.startElement("p", AttributesImpl({"class": "paraclass"}))
|
||
|
|
h.characters("text")
|
||
|
|
h.endElement("p")
|
||
|
|
h.endElement("html")
|
||
|
|
h.endDocument()
|
||
|
|
|
||
|
|
def stub(self, *args, **kwargs):
|
||
|
|
"""Stub method. Does nothing."""
|
||
|
|
pass
|
||
|
|
setProperty = stub
|
||
|
|
setFeature = stub
|
||
|
|
|
||
|
|
|
||
|
|
class SAX2DOMExerciser(SAXExerciser):
|
||
|
|
"""The same as SAXExerciser, but without the processing instruction and
|
||
|
|
comment before the root element, because S2D can"t handle it"""
|
||
|
|
|
||
|
|
def parse(self, _):
|
||
|
|
h = self._handler
|
||
|
|
h.startDocument()
|
||
|
|
h.startElement("html", AttributesImpl({}))
|
||
|
|
h.comment("a comment")
|
||
|
|
h.processingInstruction("target", "data")
|
||
|
|
h.startElement("p", AttributesImpl({"class": "paraclass"}))
|
||
|
|
h.characters("text")
|
||
|
|
h.endElement("p")
|
||
|
|
h.endElement("html")
|
||
|
|
h.endDocument()
|
||
|
|
|
||
|
|
|
||
|
|
class SAX2DOMTestHelper(pulldom.DOMEventStream):
|
||
|
|
"""Allows us to drive SAX2DOM from a DOMEventStream."""
|
||
|
|
|
||
|
|
def reset(self):
|
||
|
|
self.pulldom = pulldom.SAX2DOM()
|
||
|
|
# This content handler relies on namespace support
|
||
|
|
self.parser.setFeature(xml.sax.handler.feature_namespaces, 1)
|
||
|
|
self.parser.setContentHandler(self.pulldom)
|
||
|
|
|
||
|
|
|
||
|
|
class SAX2DOMTestCase(unittest.TestCase):
|
||
|
|
|
||
|
|
def confirm(self, test, testname="Test"):
|
||
|
|
self.assertTrue(test, testname)
|
||
|
|
|
||
|
|
def test_basic(self):
|
||
|
|
"""Ensure SAX2DOM can parse from a stream."""
|
||
|
|
with io.StringIO(SMALL_SAMPLE) as fin:
|
||
|
|
sd = SAX2DOMTestHelper(fin, xml.sax.make_parser(),
|
||
|
|
len(SMALL_SAMPLE))
|
||
|
|
for evt, node in sd:
|
||
|
|
if evt == pulldom.START_ELEMENT and node.tagName == "html":
|
||
|
|
break
|
||
|
|
# Because the buffer is the same length as the XML, all the
|
||
|
|
# nodes should have been parsed and added:
|
||
|
|
self.assertGreater(len(node.childNodes), 0)
|
||
|
|
|
||
|
|
def testSAX2DOM(self):
|
||
|
|
"""Ensure SAX2DOM expands nodes as expected."""
|
||
|
|
sax2dom = pulldom.SAX2DOM()
|
||
|
|
sax2dom.startDocument()
|
||
|
|
sax2dom.startElement("doc", {})
|
||
|
|
sax2dom.characters("text")
|
||
|
|
sax2dom.startElement("subelm", {})
|
||
|
|
sax2dom.characters("text")
|
||
|
|
sax2dom.endElement("subelm")
|
||
|
|
sax2dom.characters("text")
|
||
|
|
sax2dom.endElement("doc")
|
||
|
|
sax2dom.endDocument()
|
||
|
|
|
||
|
|
doc = sax2dom.document
|
||
|
|
root = doc.documentElement
|
||
|
|
(text1, elm1, text2) = root.childNodes
|
||
|
|
text3 = elm1.childNodes[0]
|
||
|
|
|
||
|
|
self.assertIsNone(text1.previousSibling)
|
||
|
|
self.assertIs(text1.nextSibling, elm1)
|
||
|
|
self.assertIs(elm1.previousSibling, text1)
|
||
|
|
self.assertIs(elm1.nextSibling, text2)
|
||
|
|
self.assertIs(text2.previousSibling, elm1)
|
||
|
|
self.assertIsNone(text2.nextSibling)
|
||
|
|
self.assertIsNone(text3.previousSibling)
|
||
|
|
self.assertIsNone(text3.nextSibling)
|
||
|
|
|
||
|
|
self.assertIs(root.parentNode, doc)
|
||
|
|
self.assertIs(text1.parentNode, root)
|
||
|
|
self.assertIs(elm1.parentNode, root)
|
||
|
|
self.assertIs(text2.parentNode, root)
|
||
|
|
self.assertIs(text3.parentNode, elm1)
|
||
|
|
doc.unlink()
|
||
|
|
|
||
|
|
|
||
|
|
def test_main():
|
||
|
|
run_unittest(PullDOMTestCase, ThoroughTestCase, SAX2DOMTestCase)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_main()
|