#!/usr/bin/python -u

import sys, unittest

import libxml2

class TestCase(unittest.TestCase):

    def runTest(self):

        self.test1()

        self.test2()

    def setUp(self):

        libxml2.debugMemory(1)

    def tearDown(self):

        libxml2.cleanupParser()

        if libxml2.debugMemory(1) != 0:

            libxml2.dumpMemory() 

            self.fail("Memory leak %d bytes" % (libxml2.debugMemory(1),))

        else:

            print("OK")

    def failUnlessXmlError(self,f,args,exc,domain,code,message,level,file,line):

        """Run function f, with arguments args and expect an exception exc;

        when the exception is raised, check the libxml2.lastError for

        expected values."""

        # disable the default error handler

        libxml2.registerErrorHandler(None,None)

        try:

            f(*args)

        except exc:

            e = libxml2.lastError()

            if e is None:

                self.fail("lastError not set")

            if 0:

                print("domain = ",e.domain())

                print("code = ",e.code())

                print("message =",repr(e.message()))

                print("level =",e.level())

                print("file =",e.file())

                print("line =",e.line())

                print()

            self.failUnlessEqual(domain,e.domain())

            self.failUnlessEqual(code,e.code())

            self.failUnlessEqual(message,e.message())

            self.failUnlessEqual(level,e.level())

            self.failUnlessEqual(file,e.file())

            self.failUnlessEqual(line,e.line())

        else:

            self.fail("exception %s should have been raised" % exc)

    def test1(self):

        """Test readFile with a file that does not exist"""

        self.failUnlessXmlError(libxml2.readFile,

                        ("dummy.xml",None,0),

                        libxml2.treeError,

                        domain=libxml2.XML_FROM_IO,

                        code=libxml2.XML_IO_LOAD_ERROR,

                        message='failed to load external entity "dummy.xml"\n',

                        level=libxml2.XML_ERR_WARNING,

                        file=None,

                        line=0)

    def test2(self):

        """Test a well-formedness error: we get the last error only"""

        s = "<x>\n\n</x>"

        self.failUnlessXmlError(libxml2.readMemory,

                        (s,len(s),"dummy.xml",None,0),

                        libxml2.treeError,

                        domain=libxml2.XML_FROM_PARSER,

                        code=libxml2.XML_ERR_TAG_NOT_FINISHED,

                        message='Premature end of data in tag x line 1\n',

                        level=libxml2.XML_ERR_FATAL,

                        file='dummy.xml',

                        line=3)

if __name__ == "__main__":

    test = TestCase()

    test.setUp()

    test.test1()

    test.test2()

    test.tearDown()