hide helper classes

dgreisen · Jun 21, 2011 · 638f549 · 638f549
1 parent 2d8307d
commit 638f549
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/django_couchdb_utils/tests/__init__.py b/django_couchdb_utils/tests/__init__.py
@@ -19,17 +19,9 @@
 
 from datetime import datetime, timedelta
 
-from couchdbkit.ext.django.loading import get_db
-
 from django_couchdb_utils.auth import User
 from django_couchdb_utils.sessions import Session, cleanup_sessions
-from django_couchdb_utils.tests.utils import TestHelper
-
-
-class DbTester(TestHelper):
-    def setUp(self):
-        db = get_db('django_couchdb_utils')
-        db.flush()
+from django_couchdb_utils.tests.utils import DbTester
 
 
 class AuthTests(DbTester):

diff --git a/django_couchdb_utils/tests/utils.py b/django_couchdb_utils/tests/utils.py
@@ -1,5 +1,7 @@
 from django.test import TestCase
 
+from couchdbkit.ext.django.loading import get_db
+
 
 class TestHelper(TestCase):
     def assertExcMsg(self, exc, msg, callable, *args, **kw):
@@ -14,3 +16,10 @@ def assertExcMsg(self, exc, msg, callable, *args, **kw):
         with self.assertRaises(exc) as cm:
             callable(*args, **kw)
         self.assertEqual(cm.exception.message, msg)
+
+
+class DbTester(TestHelper):
+    '''Keep separate from TestHelper to make it subclassable as library code'''
+    def setUp(self):
+        db = get_db('django_couchdb_utils')
+        db.flush()