From 05a22d5a54b8bd7a6a411732db90464fbbb29371 Mon Sep 17 00:00:00 2001 From: Nuno Rodrigues Date: Mon, 22 Jul 2019 08:42:17 +0000 Subject: [PATCH] Extract test to its own method --- tests/test_connection.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/tests/test_connection.py b/tests/test_connection.py index 57141958..b7dc9268 100644 --- a/tests/test_connection.py +++ b/tests/test_connection.py @@ -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