python/tests/reader3.py

   1 #!/usr/bin/python -u
   2 #
   3 # this tests the entities substitutions with the XmlTextReader interface
   4 #
   5 import sys
   6 import StringIO
   7 import libxml2
   8
   9 docstr="""<?xml version='1.0'?>
  10 <!DOCTYPE doc [
  11 <!ENTITY tst "<p>test</p>">
  12 ]>
  13 <doc>&tst;</doc>"""
  14
  15 # Memory debug specific
  16 libxml2.debugMemory(1)
  17
  18 #
  19 # First test, normal don't substitute entities.
  20 #
  21 f = StringIO.StringIO(docstr)
  22 input = libxml2.inputBuffer(f)
  23 reader = input.newTextReader("test_noent")
  24 ret = reader.Read()
  25 if ret != 1:
  26     print "Error reading to root"
  27     sys.exit(1)
  28 if reader.Name() == "doc" or reader.NodeType() == 10:
  29     ret = reader.Read()
  30 if ret != 1:
  31     print "Error reading to root"
  32     sys.exit(1)
  33 if reader.Name() != "doc" or reader.NodeType() != 1:
  34     print "test_normal: Error reading the root element"
  35     sys.exit(1)
  36 ret = reader.Read()
  37 if ret != 1:
  38     print "test_normal: Error reading to the entity"
  39     sys.exit(1)
  40 if reader.Name() != "tst" or reader.NodeType() != 5:
  41     print "test_normal: Error reading the entity"
  42     sys.exit(1)
  43 ret = reader.Read()
  44 if ret != 1:
  45     print "test_normal: Error reading to the end of root"
  46     sys.exit(1)
  47 if reader.Name() != "doc" or reader.NodeType() != 15:
  48     print "test_normal: Error reading the end of the root element"
  49     sys.exit(1)
  50 ret = reader.Read()
  51 if ret != 0:
  52     print "test_normal: Error detecting the end"
  53     sys.exit(1)
  54
  55 #
  56 # Second test, completely substitute the entities.
  57 #
  58 f = StringIO.StringIO(docstr)
  59 input = libxml2.inputBuffer(f)
  60 reader = input.newTextReader("test_noent")
  61 reader.SetParserProp(libxml2.PARSER_SUBST_ENTITIES, 1)
  62 ret = reader.Read()
  63 if ret != 1:
  64     print "Error reading to root"
  65     sys.exit(1)
  66 if reader.Name() == "doc" or reader.NodeType() == 10:
  67     ret = reader.Read()
  68 if ret != 1:
  69     print "Error reading to root"
  70     sys.exit(1)
  71 if reader.Name() != "doc" or reader.NodeType() != 1:
  72     print "test_noent: Error reading the root element"
  73     sys.exit(1)
  74 ret = reader.Read()
  75 if ret != 1:
  76     print "test_noent: Error reading to the entity content"
  77     sys.exit(1)
  78 if reader.Name() != "p" or reader.NodeType() != 1:
  79     print "test_noent: Error reading the p element from entity"
  80     sys.exit(1)
  81 ret = reader.Read()
  82 if ret != 1:
  83     print "test_noent: Error reading to the text node"
  84     sys.exit(1)
  85 if reader.NodeType() != 3 or reader.Value() != "test":
  86     print "test_noent: Error reading the text node"
  87     sys.exit(1)
  88 ret = reader.Read()
  89 if ret != 1:
  90     print "test_noent: Error reading to the end of p element"
  91     sys.exit(1)
  92 if reader.Name() != "p" or reader.NodeType() != 15:
  93     print "test_noent: Error reading the end of the p element"
  94     sys.exit(1)
  95 ret = reader.Read()
  96 if ret != 1:
  97     print "test_noent: Error reading to the end of root"
  98     sys.exit(1)
  99 if reader.Name() != "doc" or reader.NodeType() != 15:
 100     print "test_noent: Error reading the end of the root element"
 101     sys.exit(1)
 102 ret = reader.Read()
 103 if ret != 0:
 104     print "test_noent: Error detecting the end"
 105     sys.exit(1)
 106
 107 #
 108 # third test, crazy stuff about empty element in external parsed entities
 109 #
 110 s = """<!DOCTYPE struct [
 111 <!ENTITY simplestruct2.ent SYSTEM "simplestruct2.ent">
 112 ]>
 113 <struct>&simplestruct2.ent;</struct>
 114 """
 115 expect="""10 struct 0 0
 116 1 struct 0 0
 117 1 descr 1 1
 118 15 struct 0 0
 119 """
 120 res=""
 121 simplestruct2_ent="""<descr/>"""
 122
 123 def myResolver(URL, ID, ctxt):
 124     if URL == "simplestruct2.ent":
 125         return(StringIO.StringIO(simplestruct2_ent))
 126     return None
 127
 128 libxml2.setEntityLoader(myResolver)
 129
 130 input = libxml2.inputBuffer(StringIO.StringIO(s))
 131 reader = input.newTextReader("test3")
 132 reader.SetParserProp(libxml2.PARSER_SUBST_ENTITIES,1)
 133 while reader.Read() == 1:
 134     res = res + "%s %s %d %d\n" % (reader.NodeType(),reader.Name(),
 135                                    reader.Depth(),reader.IsEmptyElement())
 136
 137 if res != expect:
 138     print "test3 failed: unexpected output"
 139     print res
 140     sys.exit(1)
 141
 142 #
 143 # cleanup
 144 #
 145 del f
 146 del input
 147 del reader
 148
 149 # Memory debug specific
 150 libxml2.cleanupParser()
 151 if libxml2.debugMemory(1) == 0:
 152     print "OK"
 153 else:
 154     print "Memory leak %d bytes" % (libxml2.debugMemory(1))
 155     libxml2.dumpMemory()