1 files changed, 155 insertions, 0 deletions
diff --git a/python/tests/reader3.py b/python/tests/reader3.py
new file mode 100755
index 0000000..2458c89
--- /dev/null
+++ b/python/tests/reader3.py
@@ -0,0 +1,155 @@
+#!/usr/bin/python -u
+#
+# this tests the entities substitutions with the XmlTextReader interface
+#
+import sys
+import StringIO
+import libxml2
+
+docstr="""<?xml version='1.0'?>
+<!DOCTYPE doc [
+<!ENTITY tst "<p>test</p>">
+]>
+<doc>&tst;</doc>"""
+
+# Memory debug specific
+libxml2.debugMemory(1)
+
+#
+# First test, normal don't substitute entities.
+#
+f = StringIO.StringIO(docstr)
+input = libxml2.inputBuffer(f)
+reader = input.newTextReader("test_noent")
+ret = reader.Read()
+if ret != 1:
+    print "Error reading to root"
+    sys.exit(1)
+if reader.Name() == "doc" or reader.NodeType() == 10:
+    ret = reader.Read()
+if ret != 1:
+    print "Error reading to root"
+    sys.exit(1)
+if reader.Name() != "doc" or reader.NodeType() != 1:
+    print "test_normal: Error reading the root element"
+    sys.exit(1)
+ret = reader.Read()
+if ret != 1:
+    print "test_normal: Error reading to the entity"
+    sys.exit(1)
+if reader.Name() != "tst" or reader.NodeType() != 5:
+    print "test_normal: Error reading the entity"
+    sys.exit(1)
+ret = reader.Read()
+if ret != 1:
+    print "test_normal: Error reading to the end of root"
+    sys.exit(1)
+if reader.Name() != "doc" or reader.NodeType() != 15:
+    print "test_normal: Error reading the end of the root element"
+    sys.exit(1)
+ret = reader.Read()
+if ret != 0:
+    print "test_normal: Error detecting the end"
+    sys.exit(1)
+
+#
+# Second test, completely substitute the entities.
+#
+f = StringIO.StringIO(docstr)
+input = libxml2.inputBuffer(f)
+reader = input.newTextReader("test_noent")
+reader.SetParserProp(libxml2.PARSER_SUBST_ENTITIES, 1)
+ret = reader.Read()
+if ret != 1:
+    print "Error reading to root"
+    sys.exit(1)
+if reader.Name() == "doc" or reader.NodeType() == 10:
+    ret = reader.Read()
+if ret != 1:
+    print "Error reading to root"
+    sys.exit(1)
+if reader.Name() != "doc" or reader.NodeType() != 1:
+    print "test_noent: Error reading the root element"
+    sys.exit(1)
+ret = reader.Read()
+if ret != 1:
+    print "test_noent: Error reading to the entity content"
+    sys.exit(1)
+if reader.Name() != "p" or reader.NodeType() != 1:
+    print "test_noent: Error reading the p element from entity"
+    sys.exit(1)
+ret = reader.Read()
+if ret != 1:
+    print "test_noent: Error reading to the text node"
+    sys.exit(1)
+if reader.NodeType() != 3 or reader.Value() != "test":
+    print "test_noent: Error reading the text node"
+    sys.exit(1)
+ret = reader.Read()
+if ret != 1:
+    print "test_noent: Error reading to the end of p element"
+    sys.exit(1)
+if reader.Name() != "p" or reader.NodeType() != 15:
+    print "test_noent: Error reading the end of the p element"
+    sys.exit(1)
+ret = reader.Read()
+if ret != 1:
+    print "test_noent: Error reading to the end of root"
+    sys.exit(1)
+if reader.Name() != "doc" or reader.NodeType() != 15:
+    print "test_noent: Error reading the end of the root element"
+    sys.exit(1)
+ret = reader.Read()
+if ret != 0:
+    print "test_noent: Error detecting the end"
+    sys.exit(1)
+
+#
+# third test, crazy stuff about empty element in external parsed entities
+#
+s = """<!DOCTYPE struct [
+<!ENTITY simplestruct2.ent SYSTEM "simplestruct2.ent">
+]>
+<struct>&simplestruct2.ent;</struct>
+"""
+expect="""10 struct 0 0
+1 struct 0 0
+1 descr 1 1
+15 struct 0 0
+"""
+res=""
+simplestruct2_ent="""<descr/>"""
+
+def myResolver(URL, ID, ctxt):
+    if URL == "simplestruct2.ent":
+        return(StringIO.StringIO(simplestruct2_ent))
+    return None
+
+libxml2.setEntityLoader(myResolver)
+
+input = libxml2.inputBuffer(StringIO.StringIO(s))
+reader = input.newTextReader("test3")
+reader.SetParserProp(libxml2.PARSER_SUBST_ENTITIES,1)
+while reader.Read() == 1:
+    res = res + "%s %s %d %d\n" % (reader.NodeType(),reader.Name(),
+                                   reader.Depth(),reader.IsEmptyElement())
+
+if res != expect:
+    print "test3 failed: unexpected output"
+    print res
+    sys.exit(1)
+
+#
+# cleanup
+#
+del f
+del input
+del reader
+
+# Memory debug specific
+libxml2.cleanupParser()
+if libxml2.debugMemory(1) == 0:
+    print "OK"
+else:
+    print "Memory leak %d bytes" % (libxml2.debugMemory(1))
+    libxml2.dumpMemory()