Extract test to its own method
This commit is contained in:
		@@ -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
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user