Added no_dereference context manager (#82)
Reorganised the context_managers as well
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
from __future__ import with_statement
|
||||
import datetime
|
||||
import pymongo
|
||||
import unittest
|
||||
@@ -8,6 +9,7 @@ from bson.tz_util import utc
|
||||
|
||||
from mongoengine import *
|
||||
from mongoengine.connection import get_db, get_connection, ConnectionError
|
||||
from mongoengine.context_managers import switch_db
|
||||
|
||||
|
||||
class ConnectionTest(unittest.TestCase):
|
||||
@@ -105,7 +107,7 @@ class ConnectionTest(unittest.TestCase):
|
||||
Group(name="hello - default").save()
|
||||
self.assertEqual(1, Group.objects.count())
|
||||
|
||||
with SwitchDB(Group, 'testdb-1') as Group:
|
||||
with switch_db(Group, 'testdb-1') as Group:
|
||||
|
||||
self.assertEqual(0, Group.objects.count())
|
||||
|
||||
|
||||
Reference in New Issue
Block a user