def get_connection(): """ :return: :rtype: pymongo.collection.Connection """ return _get_connection()
def test_connect(self): """Ensure that the connect() method works properly. """ connect('mongoenginetest') conn = _get_connection() self.assertTrue(isinstance(conn, pymongo.connection.Connection)) db = _get_db() self.assertTrue(isinstance(db, pymongo.database.Database)) self.assertEqual(db.name, 'mongoenginetest')
def test_register_connection(self): """Ensure that connections with different aliases may be registered. """ register_connection('testdb', 'mongoenginetest2') self.assertRaises(ConnectionError, _get_connection) conn = _get_connection('testdb') self.assertTrue(isinstance(conn, pymongo.connection.Connection)) db = _get_db('testdb') self.assertTrue(isinstance(db, pymongo.database.Database)) self.assertEqual(db.name, 'mongoenginetest2')
def tearDownClass(cls): connection = _get_connection() connection.drop_database('test_model')
def tearDownClass(cls): c = _get_connection() c.drop_database('test_auth')
def tearDown(self): connection = _get_connection() connection.drop_database(_get_db())
def clearDB(): db = _get_db() conn = _get_connection() conn.drop_database(db) return 'DB {} cleared'.format(db.name), 200
def tearDownClass(cls): c = _get_connection() c.drop_database('test_pagination')