Provide a proper cleanup hook and use it.

zopefoundation · Apr 3, 2012 · 9684e5f · 9684e5f
1 parent 48fa9e0
commit 9684e5f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/mongopersist/testing.py b/src/mongopersist/testing.py
@@ -17,7 +17,7 @@
 import pymongo
 import re
 import transaction
-from zope.testing import module, renormalizing
+from zope.testing import cleanup, module, renormalizing
 
 from mongopersist import datamanager, serialize
 
@@ -54,8 +54,13 @@ def tearDown(test):
     transaction.abort()
     test.globs['conn'].drop_database(test.globs['DBNAME'])
     test.globs['conn'].disconnect()
+    resetCaches()
+
+def resetCaches():
     serialize.SERIALIZERS.__init__()
     serialize.OID_CLASS_LRU.__init__(20000)
     serialize.COLLECTIONS_WITH_TYPE.__init__()
     serialize.AVAILABLE_NAME_MAPPINGS.__init__()
     serialize.PATH_RESOLVE_CACHE = {}
+
+cleanup.addCleanUp(resetCaches)