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 | # Memory debug specific |
| 18 | libxml2.debugMemory(1) |
| 19 | |
| 20 | schema="""<?xml version="1.0" encoding="iso-8859-1"?> |
| 21 | <schema xmlns = "http://www.w3.org/2001/XMLSchema"> |
| 22 | <element name = "Customer"> |
| 23 | <complexType> |
| 24 | <sequence> |
| 25 | <element name = "FirstName" type = "string" /> |
| 26 | <element name = "MiddleInitial" type = "string" /> |
| 27 | <element name = "LastName" type = "string" /> |
| 28 | </sequence> |
| 29 | <attribute name = "customerID" type = "integer" /> |
| 30 | </complexType> |
| 31 | </element> |
| 32 | </schema>""" |
| 33 | |
| 34 | valid="""<?xml version="1.0" encoding="iso-8859-1"?> |
| 35 | <Customer customerID = "24332"> |
| 36 | <FirstName>Raymond</FirstName> |
| 37 | <MiddleInitial>G</MiddleInitial> |
| 38 | <LastName>Bayliss</LastName> |
| 39 | </Customer> |
| 40 | """ |
| 41 | |
| 42 | invalid="""<?xml version="1.0" encoding="iso-8859-1"?> |
| 43 | <Customer customerID = "24332"> |
| 44 | <MiddleInitial>G</MiddleInitial> |
| 45 | <LastName>Bayliss</LastName> |
| 46 | </Customer> |
| 47 | """ |
| 48 | |
| 49 | e = ErrorHandler() |
| 50 | ctxt_parser = libxml2.schemaNewMemParserCtxt(schema, len(schema)) |
| 51 | ctxt_schema = ctxt_parser.schemaParse() |
| 52 | ctxt_valid = ctxt_schema.schemaNewValidCtxt() |
| 53 | ctxt_valid.setValidityErrorHandler(e.handler, e.handler, ARG) |
| 54 | |
| 55 | # Test valid document |
| 56 | doc = libxml2.parseDoc(valid) |
| 57 | ret = doc.schemaValidateDoc(ctxt_valid) |
| 58 | if ret != 0 or e.errors: |
| 59 | print "error doing schema validation" |
| 60 | sys.exit(1) |
| 61 | doc.freeDoc() |
| 62 | |
| 63 | # Test invalid document |
| 64 | doc = libxml2.parseDoc(invalid) |
| 65 | ret = doc.schemaValidateDoc(ctxt_valid) |
| 66 | if ret == 0 or not e.errors: |
| 67 | print "Error: document supposer to be schema invalid" |
| 68 | sys.exit(1) |
| 69 | doc.freeDoc() |
| 70 | |
| 71 | del ctxt_parser |
| 72 | del ctxt_schema |
| 73 | del ctxt_valid |
| 74 | libxml2.schemaCleanupTypes() |
| 75 | |
| 76 | # Memory debug specific |
| 77 | libxml2.cleanupParser() |
| 78 | if libxml2.debugMemory(1) == 0: |
| 79 | print "OK" |
| 80 | else: |
| 81 | print "Memory leak %d bytes" % (libxml2.debugMemory(1)) |
| 82 | libxml2.dumpMemory() |
| 83 | |