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_connection.py
+++ b/tests/test_connection.py
@@ -249,9 +249,15 @@ class ConnectionTest(unittest.TestCase):
        connect('mongoenginetest')

        self.assertEqual(len(connections), 1)
-        self.assertEqual(len(dbs), 1)
+        self.assertEqual(len(dbs), 0)
        self.assertEqual(len(connection_settings), 1)

+        class TestDoc(Document):
+            pass
+
+        TestDoc.drop_collection()  # triggers the db
+        self.assertEqual(len(dbs), 1)
+
        disconnect()
        self.assertEqual(len(connections), 0)
        self.assertEqual(len(dbs), 0)