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

Various test adjustments to improve stability independantly of execution order

Browse Source
This commit is contained in:
Matthieu Rigal 2015-06-22 16:41:36 +02:00
parent 9063b559c4
commit 5c807f3dc8
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/test_connection.py
View File

@ -54,11 +54,10 @@ class ConnectionTest(unittest.TestCase):
def test_sharing_connections(self): def test_sharing_connections(self):
"""Ensure that connections are shared when the connection settings are exactly the same """Ensure that connections are shared when the connection settings are exactly the same
""" """
connect('mongoenginetest', alias='testdb1') connect('mongoenginetests', alias='testdb1')
expected_connection = get_connection('testdb1') expected_connection = get_connection('testdb1')
connect('mongoenginetest', alias='testdb2') connect('mongoenginetests', alias='testdb2')
actual_connection = get_connection('testdb2') actual_connection = get_connection('testdb2')
# Handle PyMongo 3+ Async Connection # Handle PyMongo 3+ Async Connection
@ -96,8 +95,7 @@ class ConnectionTest(unittest.TestCase):
c.mongoenginetest.system.users.remove({}) c.mongoenginetest.system.users.remove({})
def test_connect_uri_without_db(self): def test_connect_uri_without_db(self):
"""Ensure that the connect() method works properly with uri's """Ensure connect() method works properly with uri's without database_name
without database_name
""" """
c = connect(db='mongoenginetest', alias='admin') c = connect(db='mongoenginetest', alias='admin')
c.admin.system.users.remove({}) c.admin.system.users.remove({})
@ -130,28 +128,27 @@ class ConnectionTest(unittest.TestCase):
# Create users # Create users
c = connect('mongoenginetest') c = connect('mongoenginetest')
c.admin.system.users.remove({}) c.admin.system.users.remove({})
c.admin.add_user('username', 'password') c.admin.add_user('username2', 'password')
# Authentication fails without "authSource" # Authentication fails without "authSource"
if IS_PYMONGO_3: if IS_PYMONGO_3:
test_conn = connect('mongoenginetest', alias='test2', test_conn = connect('mongoenginetest', alias='test1',
host='mongodb://username:password@localhost/mongoenginetest') host='mongodb://username2:password@localhost/mongoenginetest')
self.assertRaises(OperationFailure, test_conn.server_info) self.assertRaises(OperationFailure, test_conn.server_info)
else: else:
self.assertRaises( self.assertRaises(
ConnectionError, connect, 'mongoenginetest', alias='test1', ConnectionError, connect, 'mongoenginetest', alias='test1',
host='mongodb://username:password@localhost/mongoenginetest' host='mongodb://username2:password@localhost/mongoenginetest'
) )
self.assertRaises(ConnectionError, get_db, 'test1') self.assertRaises(ConnectionError, get_db, 'test1')
# Authentication succeeds with "authSource" # Authentication succeeds with "authSource"
test_conn2 = connect( test_conn2 = connect(
'mongoenginetest', alias='test2', 'mongoenginetest', alias='test2',
host=('mongodb://username:password@localhost/' host=('mongodb://username2:password@localhost/'
'mongoenginetest?authSource=admin') 'mongoenginetest?authSource=admin')
) )
# This will fail starting from MongoDB 2.6+ # This will fail starting from MongoDB 2.6+
# test_conn2.server_info()
db = get_db('test2') db = get_db('test2')
self.assertTrue(isinstance(db, pymongo.database.Database)) self.assertTrue(isinstance(db, pymongo.database.Database))
self.assertEqual(db.name, 'mongoenginetest') self.assertEqual(db.name, 'mongoenginetest')