From da0a1bbe9f63059bd745d378e723c184c079c90c Mon Sep 17 00:00:00 2001
From: Ross Lawley <ross.lawley@gmail.com>
Date: Thu, 26 Jun 2014 17:13:21 +0100
Subject: [PATCH] Fix test_using

---
 tests/queryset/queryset.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py
index 649a3582..b97f9519 100644
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@@ -16,7 +16,7 @@ from bson import ObjectId
 from mongoengine import *
 from mongoengine.connection import get_connection
 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,
                                   MultipleObjectsReturned, DoesNotExist,
                                   queryset_manager)
@@ -3035,8 +3035,10 @@ class QuerySetTest(unittest.TestCase):
             n = IntField()
 
         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.switch_db('test2')
             t.save()