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

Extract test to its own method

Browse Source
This commit is contained in:
Nuno Rodrigues 2019-07-22 08:42:17 +00:00
parent 2424ece0c5
commit 05a22d5a54
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_connection.py
View File

@ -271,16 +271,25 @@ class ConnectionTest(unittest.TestCase):
conn = get_connection("testdb7") conn = get_connection("testdb7")
self.assertIsInstance(conn, mongomock.MongoClient) 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() disconnect_all()
class SomeDocument(Document): class SomeDocument(Document):
pass pass
conn = connect(host="mongomock://localhost:27017/mongoenginetest8") conn = connect(host="mongomock://localhost:27017/mongoenginetest")
some_document = SomeDocument() some_document = SomeDocument()
# database won't exist until we save a document
some_document.save() some_document.save()
self.assertEqual(conn.get_default_database().name, "mongoenginetest8") self.assertEqual(conn.get_default_database().name, "mongoenginetest")
self.assertEqual(conn.database_names()[0], "mongoenginetest8") self.assertEqual(conn.database_names()[0], "mongoenginetest")
def test_connect_with_host_list(self): def test_connect_with_host_list(self):
"""Ensure that the connect() method works when host is a list """Ensure that the connect() method works when host is a list