Lib/test/test_pyexpat.py

   1 # Very simple test - Parse a file and print what happens
   2
   3 # XXX TypeErrors on calling handlers, or on bad return values from a
   4 # handler, are obscure and unhelpful.
   5
   6 import pyexpat
   7 from xml.parsers import expat
   8
   9 from test_support import sortdict
  10
  11 class Outputter:
  12     def StartElementHandler(self, name, attrs):
  13         print 'Start element:\n\t', repr(name), sortdict(attrs)
  14
  15     def EndElementHandler(self, name):
  16         print 'End element:\n\t', repr(name)
  17
  18     def CharacterDataHandler(self, data):
  19         data = data.strip()
  20         if data:
  21             print 'Character data:'
  22             print '\t', repr(data)
  23
  24     def ProcessingInstructionHandler(self, target, data):
  25         print 'PI:\n\t', repr(target), repr(data)
  26
  27     def StartNamespaceDeclHandler(self, prefix, uri):
  28         print 'NS decl:\n\t', repr(prefix), repr(uri)
  29
  30     def EndNamespaceDeclHandler(self, prefix):
  31         print 'End of NS decl:\n\t', repr(prefix)
  32
  33     def StartCdataSectionHandler(self):
  34         print 'Start of CDATA section'
  35
  36     def EndCdataSectionHandler(self):
  37         print 'End of CDATA section'
  38
  39     def CommentHandler(self, text):
  40         print 'Comment:\n\t', repr(text)
  41
  42     def NotationDeclHandler(self, *args):
  43         name, base, sysid, pubid = args
  44         print 'Notation declared:', args
  45
  46     def UnparsedEntityDeclHandler(self, *args):
  47         entityName, base, systemId, publicId, notationName = args
  48         print 'Unparsed entity decl:\n\t', args
  49
  50     def NotStandaloneHandler(self, userData):
  51         print 'Not standalone'
  52         return 1
  53
  54     def ExternalEntityRefHandler(self, *args):
  55         context, base, sysId, pubId = args
  56         print 'External entity ref:', args[1:]
  57         return 1
  58
  59     def DefaultHandler(self, userData):
  60         pass
  61
  62     def DefaultHandlerExpand(self, userData):
  63         pass
  64
  65
  66 def confirm(ok):
  67     if ok:
  68         print "OK."
  69     else:
  70         print "Not OK."
  71
  72 out = Outputter()
  73 parser = expat.ParserCreate(namespace_separator='!')
  74
  75 # Test getting/setting returns_unicode
  76 parser.returns_unicode = 0; confirm(parser.returns_unicode == 0)
  77 parser.returns_unicode = 1; confirm(parser.returns_unicode == 1)
  78 parser.returns_unicode = 2; confirm(parser.returns_unicode == 1)
  79 parser.returns_unicode = 0; confirm(parser.returns_unicode == 0)
  80
  81 # Test getting/setting ordered_attributes
  82 parser.ordered_attributes = 0; confirm(parser.ordered_attributes == 0)
  83 parser.ordered_attributes = 1; confirm(parser.ordered_attributes == 1)
  84 parser.ordered_attributes = 2; confirm(parser.ordered_attributes == 1)
  85 parser.ordered_attributes = 0; confirm(parser.ordered_attributes == 0)
  86
  87 # Test getting/setting specified_attributes
  88 parser.specified_attributes = 0; confirm(parser.specified_attributes == 0)
  89 parser.specified_attributes = 1; confirm(parser.specified_attributes == 1)
  90 parser.specified_attributes = 2; confirm(parser.specified_attributes == 1)
  91 parser.specified_attributes = 0; confirm(parser.specified_attributes == 0)
  92
  93 HANDLER_NAMES = [
  94     'StartElementHandler', 'EndElementHandler',
  95     'CharacterDataHandler', 'ProcessingInstructionHandler',
  96     'UnparsedEntityDeclHandler', 'NotationDeclHandler',
  97     'StartNamespaceDeclHandler', 'EndNamespaceDeclHandler',
  98     'CommentHandler', 'StartCdataSectionHandler',
  99     'EndCdataSectionHandler',
 100     'DefaultHandler', 'DefaultHandlerExpand',
 101     #'NotStandaloneHandler',
 102     'ExternalEntityRefHandler'
 103     ]
 104 for name in HANDLER_NAMES:
 105     setattr(parser, name, getattr(out, name))
 106
 107 data = '''\
 108 <?xml version="1.0" encoding="iso-8859-1" standalone="no"?>
 109 <?xml-stylesheet href="stylesheet.css"?>
 110 <!-- comment data -->
 111 <!DOCTYPE quotations SYSTEM "quotations.dtd" [
 112 <!ELEMENT root ANY>
 113 <!NOTATION notation SYSTEM "notation.jpeg">
 114 <!ENTITY acirc "&#226;">
 115 <!ENTITY external_entity SYSTEM "entity.file">
 116 <!ENTITY unparsed_entity SYSTEM "entity.file" NDATA notation>
 117 %unparsed_entity;
 118 ]>
 119
 120 <root attr1="value1" attr2="value2&#8000;">
 121 <myns:subelement xmlns:myns="http://www.python.org/namespace">
 122      Contents of subelements
 123 </myns:subelement>
 124 <sub2><![CDATA[contents of CDATA section]]></sub2>
 125 &external_entity;
 126 </root>
 127 '''
 128
 129 # Produce UTF-8 output
 130 parser.returns_unicode = 0
 131 try:
 132     parser.Parse(data, 1)
 133 except expat.error:
 134     print '** Error', parser.ErrorCode, expat.ErrorString(parser.ErrorCode)
 135     print '** Line', parser.ErrorLineNumber
 136     print '** Column', parser.ErrorColumnNumber
 137     print '** Byte', parser.ErrorByteIndex
 138
 139 # Try the parse again, this time producing Unicode output
 140 parser = expat.ParserCreate(namespace_separator='!')
 141 parser.returns_unicode = 1
 142
 143 for name in HANDLER_NAMES:
 144     setattr(parser, name, getattr(out, name))
 145 try:
 146     parser.Parse(data, 1)
 147 except expat.error:
 148     print '** Error', parser.ErrorCode, expat.ErrorString(parser.ErrorCode)
 149     print '** Line', parser.ErrorLineNumber
 150     print '** Column', parser.ErrorColumnNumber
 151     print '** Byte', parser.ErrorByteIndex
 152
 153 # Try parsing a file
 154 parser = expat.ParserCreate(namespace_separator='!')
 155 parser.returns_unicode = 1
 156
 157 for name in HANDLER_NAMES:
 158     setattr(parser, name, getattr(out, name))
 159 import StringIO
 160 file = StringIO.StringIO(data)
 161 try:
 162     parser.ParseFile(file)
 163 except expat.error:
 164     print '** Error', parser.ErrorCode, expat.ErrorString(parser.ErrorCode)
 165     print '** Line', parser.ErrorLineNumber
 166     print '** Column', parser.ErrorColumnNumber
 167     print '** Byte', parser.ErrorByteIndex
 168
 169
 170 # Tests that make sure we get errors when the namespace_separator value
 171 # is illegal, and that we don't for good values:
 172 print
 173 print "Testing constructor for proper handling of namespace_separator values:"
 174 expat.ParserCreate()
 175 expat.ParserCreate(namespace_separator=None)
 176 expat.ParserCreate(namespace_separator=' ')
 177 print "Legal values tested o.k."
 178 try:
 179     expat.ParserCreate(namespace_separator=42)
 180 except TypeError, e:
 181     print "Caught expected TypeError:"
 182     print e
 183 else:
 184     print "Failed to catch expected TypeError."
 185
 186 try:
 187     expat.ParserCreate(namespace_separator='too long')
 188 except ValueError, e:
 189     print "Caught expected ValueError:"
 190     print e
 191 else:
 192     print "Failed to catch expected ValueError."
 193
 194 # ParserCreate() needs to accept a namespace_separator of zero length
 195 # to satisfy the requirements of RDF applications that are required
 196 # to simply glue together the namespace URI and the localname.  Though
 197 # considered a wart of the RDF specifications, it needs to be supported.
 198 #
 199 # See XML-SIG mailing list thread starting with
 200 # http://mail.python.org/pipermail/xml-sig/2001-April/005202.html
 201 #
 202 expat.ParserCreate(namespace_separator='') # too short