#!/usr/bin/python -u

import libxml2

import sys

ARG = 'test string'

class ErrorHandler:

    def __init__(self):

        self.errors = []

    def handler(self, msg, data):

        if data != ARG:

            raise Exception("Error handler did not receive correct argument")

        self.errors.append(msg)

# Memory debug specific

libxml2.debugMemory(1)

dtd=""""""

valid="""

<foo></foo>"""

invalid="""

<foo><bar/></foo>"""

dtd = libxml2.parseDTD(None, 'test.dtd')

ctxt = libxml2.newValidCtxt()

e = ErrorHandler()

ctxt.setValidityErrorHandler(e.handler, e.handler, ARG)

# Test valid document

doc = libxml2.parseDoc(valid)

ret = doc.validateDtd(ctxt, dtd)

if ret != 1 or e.errors:

    print("error doing DTD validation")

    sys.exit(1)

doc.freeDoc()

# Test invalid document

doc = libxml2.parseDoc(invalid)

ret = doc.validateDtd(ctxt, dtd)

if ret != 0 or not e.errors:

    print("Error: document supposed to be invalid")

doc.freeDoc()

dtd.freeDtd()

del dtd

del ctxt

# Memory debug specific

libxml2.cleanupParser()

if libxml2.debugMemory(1) == 0:

    print("OK")

else:

    print("Memory leak %d bytes" % (libxml2.debugMemory(1)))

    libxml2.dumpMemory()