Clean up mango test user docsclean-up-after-mango-tests

Tests based on class `UsersDbTests` don't clean up the user docs it
puts in the `_users` db. This uses the classmethod `tearDownClass` to
delete those docs.

2 files changed, 8 insertions, 0 deletions

diff --git a/src/mango/test/mango.py b/src/mango/test/mango.py
index b87189446..7a404e253 100644
--- a/src/mango/test/mango.py
+++ b/src/mango/test/mango.py
@@ -314,6 +314,10 @@ class UsersDbTests(unittest.TestCase):
         klass.db = Database("_users")
         user_docs.setup_users(klass.db)
 
+    @classmethod
+    def tearDownClass(klass):
+        user_docs.teardown_users(klass.db)
+
     def setUp(self):
         self.db = self.__class__.db
 
diff --git a/src/mango/test/user_docs.py b/src/mango/test/user_docs.py
index 679360171..40e9cf931 100644
--- a/src/mango/test/user_docs.py
+++ b/src/mango/test/user_docs.py
@@ -59,6 +59,10 @@ def setup_users(db, **kwargs):
     db.save_docs(copy.deepcopy(USERS_DOCS))
 
 
+def teardown_users(db):
+    [db.delete_doc(doc['_id']) for doc in USERS_DOCS]
+
+
 def setup(db, index_type="view", **kwargs):
     db.recreate()
     if index_type == "view":