Extract test to its own method

This commit is contained in:
Nuno Rodrigues 2019-07-22 08:42:17 +00:00
parent 2424ece0c5
commit 05a22d5a54

View File

@ -271,16 +271,25 @@ class ConnectionTest(unittest.TestCase):
conn = get_connection("testdb7")
self.assertIsInstance(conn, mongomock.MongoClient)
def test_default_database_with_mocking(self):
"""Ensure that the default database is correctly set when using mongomock.
"""
try:
import mongomock
except ImportError:
raise SkipTest("you need mongomock installed to run this testcase")
disconnect_all()
class SomeDocument(Document):
pass
conn = connect(host="mongomock://localhost:27017/mongoenginetest8")
conn = connect(host="mongomock://localhost:27017/mongoenginetest")
some_document = SomeDocument()
# database won't exist until we save a document
some_document.save()
self.assertEqual(conn.get_default_database().name, "mongoenginetest8")
self.assertEqual(conn.database_names()[0], "mongoenginetest8")
self.assertEqual(conn.get_default_database().name, "mongoenginetest")
self.assertEqual(conn.database_names()[0], "mongoenginetest")
def test_connect_with_host_list(self):
"""Ensure that the connect() method works when host is a list