Daniel Veillard | 0e460da | 2005-03-30 22:47:10 +0000 | [diff] [blame] | 1 | #!/usr/bin/python -u |
| 2 | import libxml2 |
| 3 | import sys |
| 4 | |
| 5 | ARG = 'test string' |
| 6 | |
| 7 | class ErrorHandler: |
| 8 | |
| 9 | def __init__(self): |
| 10 | self.errors = [] |
| 11 | |
| 12 | def handler(self, msg, data): |
| 13 | if data != ARG: |
| 14 | raise Exception, "Error handler did not receive correct argument" |
| 15 | self.errors.append(msg) |
| 16 | |
| 17 | |
| 18 | # Memory debug specific |
| 19 | libxml2.debugMemory(1) |
| 20 | |
| 21 | dtd="""<!ELEMENT foo EMPTY>""" |
| 22 | valid="""<?xml version="1.0"?> |
| 23 | <foo></foo>""" |
| 24 | |
| 25 | invalid="""<?xml version="1.0"?> |
| 26 | <foo><bar/></foo>""" |
| 27 | |
| 28 | dtd = libxml2.parseDTD(None, 'test.dtd') |
| 29 | ctxt = libxml2.newValidCtxt() |
| 30 | e = ErrorHandler() |
| 31 | ctxt.setValidityErrorHandler(e.handler, e.handler, ARG) |
| 32 | |
| 33 | # Test valid document |
| 34 | doc = libxml2.parseDoc(valid) |
| 35 | ret = doc.validateDtd(ctxt, dtd) |
| 36 | if ret != 1 or e.errors: |
| 37 | print "error doing DTD validation" |
| 38 | sys.exit(1) |
| 39 | doc.freeDoc() |
| 40 | |
| 41 | # Test invalid document |
| 42 | doc = libxml2.parseDoc(invalid) |
| 43 | ret = doc.validateDtd(ctxt, dtd) |
| 44 | if ret != 0 or not e.errors: |
| 45 | print "Error: document supposed to be invalid" |
| 46 | doc.freeDoc() |
| 47 | |
| 48 | dtd.freeDtd() |
| 49 | del dtd |
| 50 | del ctxt |
| 51 | |
| 52 | # Memory debug specific |
| 53 | libxml2.cleanupParser() |
| 54 | if libxml2.debugMemory(1) == 0: |
| 55 | print "OK" |
| 56 | else: |
| 57 | print "Memory leak %d bytes" % (libxml2.debugMemory(1)) |
| 58 | libxml2.dumpMemory() |
| 59 | |