pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix test_using

Browse Source
This commit is contained in:
Ross Lawley 2014-06-26 17:13:21 +01:00
parent bc66fb33e9
commit da0a1bbe9f
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/queryset/queryset.py
View File

@ -16,7 +16,7 @@ from bson import ObjectId
from mongoengine import * from mongoengine import *
from mongoengine.connection import get_connection from mongoengine.connection import get_connection
from mongoengine.python_support import PY3 from mongoengine.python_support import PY3
from mongoengine.context_managers import query_counter from mongoengine.context_managers import query_counter, switch_db
from mongoengine.queryset import (QuerySet, QuerySetManager, from mongoengine.queryset import (QuerySet, QuerySetManager,
MultipleObjectsReturned, DoesNotExist, MultipleObjectsReturned, DoesNotExist,
queryset_manager) queryset_manager)
@ -3035,8 +3035,10 @@ class QuerySetTest(unittest.TestCase):
n = IntField() n = IntField()
Number2.drop_collection() Number2.drop_collection()
with switch_db(Number2, 'test2') as Number2:
Number2.drop_collection()
for i in xrange(1, 10): for i in range(1, 10):
t = Number2(n=i) t = Number2(n=i)
t.switch_db('test2') t.switch_db('test2')
t.save() t.save()