| #!/usr/bin/python -u |
| import sys |
| import libxml2 |
| |
| # Memory debug specific |
| libxml2.debugMemory(1) |
| |
| def foo(ctx, x): |
| return x + 1 |
| |
| def bar(ctx, x): |
| return "%d" % (x + 2) |
| |
| doc = libxml2.parseFile("tst.xml") |
| ctxt = doc.xpathNewContext() |
| res = ctxt.xpathEval("//*") |
| if len(res) != 2: |
| print("xpath query: wrong node set size") |
| sys.exit(1) |
| if res[0].name != "doc" or res[1].name != "foo": |
| print("xpath query: wrong node set value") |
| sys.exit(1) |
| |
| libxml2.registerXPathFunction(ctxt._o, "foo", None, foo) |
| libxml2.registerXPathFunction(ctxt._o, "bar", None, bar) |
| i = 10000 |
| while i > 0: |
| res = ctxt.xpathEval("foo(1)") |
| if res != 2: |
| print("xpath extension failure") |
| sys.exit(1) |
| i = i - 1 |
| i = 10000 |
| while i > 0: |
| res = ctxt.xpathEval("bar(1)") |
| if res != "3": |
| print("xpath extension failure got %s expecting '3'") |
| sys.exit(1) |
| i = i - 1 |
| doc.freeDoc() |
| ctxt.xpathFreeContext() |
| |
| # Memory debug specific |
| libxml2.cleanupParser() |
| if libxml2.debugMemory(1) == 0: |
| print("OK") |
| else: |
| print("Memory leak %d bytes" % (libxml2.debugMemory(1))) |
| libxml2.dumpMemory() |