#!/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="""

test
">

]>

<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 = """

]>

<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()