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')
