Fixed django test class

refs hmarr/mongoengine#506
2012-06-18 21:18:40 +01:00 · 2012-06-18 21:18:40 +01:00 · 89220c142b
commit 89220c142b
parent c73ce3d220
1 changed files with 1 additions and 1 deletions
--- a/mongoengine/django/tests.py
+++ b/mongoengine/django/tests.py
@ -10,7 +10,7 @@ class MongoTestCase(TestCase):
    """
    db_name = 'test_%s' % settings.MONGO_DATABASE_NAME
    def __init__(self, methodName='runtest'):
-        self.db = connect(self.db_name)
+        self.db = connect(self.db_name).get_db()
        super(MongoTestCase, self).__init__(methodName)

    def _post_teardown(self):