#!/usr/bin/python -u # # this tests the entities substitutions with the XmlTextReader interface # import sys import libxml2 try:

import StringIO
str_io = StringIO.StringIO

except:

import io
str_io = io.StringIO

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 = str_io(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 = str_io(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(str_io(simplestruct2_ent))
return None

libxml2.setEntityLoader(myResolver)

input = libxml2.inputBuffer(str_io(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()