.

wcfs/testprog/wcfs_commit_tree_deltas.py

@@ -46,6 +46,8 @@ from __future__ import print_function, absolute_import
 import sys
 from ZODB import DB
 import transaction
+
+import zodbtools
 from wendelin.wcfs.internal import xbtree
 
 # init initializes ...
@@ -106,7 +108,7 @@ def treedeltaGenAllStructs(zstor, kv1, kv2, n):
                 return k
         raise KeyError("%r not found in value registry" % (vobj,))
 
-
+@func
 def main():
     if len(sys.argv) != 5:
         print("Usage: %s <n> <zurl> <kv1> <kv2>" % sys.argv[0], file=sys.stderr)
@@ -115,6 +117,9 @@ def main():
     n = int(sys.argv[1])
     zurl, kv1, kv2 = sys.argv[2:]   # XXX kv decode
 
+    zstor = zodbtools.storageFromURL(zurl)
+    defer(zstor.close)
+
 
 if __name__ == '__main__':
     main()