minor improvements

2019-04-25 22:11:43 +02:00
parent b1e28d02f7
commit 565e1dc0ed
4 changed files with 18 additions and 8 deletions
--- a/tests/test_context_managers.py
+++ b/tests/test_context_managers.py
@@ -37,14 +37,15 @@ class ContextManagersTest(unittest.TestCase):

    def test_switch_collection_context_manager(self):
        connect('mongoenginetest')
-        register_connection('testdb-1', 'mongoenginetest2')
+        register_connection(alias='testdb-1', db='mongoenginetest2')

        class Group(Document):
            name = StringField()

-        Group.drop_collection()
+        Group.drop_collection()         # drops in default
+
        with switch_collection(Group, 'group1') as Group:
-            Group.drop_collection()
+            Group.drop_collection()     # drops in group1

        Group(name="hello - group").save()
        self.assertEqual(1, Group.objects.count())