#!/usr/bin/python -u## this tests the entities substitutions with the XmlTextReader interface#import sysimport StringIOimport libxml2 docstr="""<?xml version='1.0'?><!DOCTYPE doc [<!ENTITY tst "<p>test</p>">]><doc>&tst;</doc>""" # Memory debug specificlibxml2.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 01 struct 0 01 descr 1 115 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 fdel inputdel reader # Memory debug specificlibxml2.cleanupParser()if libxml2.debugMemory(1) == 0:    print "OK"else:    print "Memory leak %d bytes" % (libxml2.debugMemory(1))    libxml2.dumpMemory()