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

corrected connection test for PyMongo3+

Browse Source
This commit is contained in:
Matthieu Rigal 2015-05-07 10:47:28 +02:00
parent 1005c99e9c
commit c41dd6495d
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/test_connection.py
View File

@ -1,5 +1,6 @@
import sys import sys
import datetime import datetime
from pymongo.errors import OperationFailure
sys.path[0:0] = [""] sys.path[0:0] = [""]
@ -124,6 +125,7 @@ class ConnectionTest(unittest.TestCase):
def test_connect_uri_with_authsource(self): def test_connect_uri_with_authsource(self):
"""Ensure that the connect() method works well with """Ensure that the connect() method works well with
the option `authSource` in URI. the option `authSource` in URI.
This feature was introduced in MongoDB 2.4 and removed in 2.6
""" """
# Create users # Create users
c = connect('mongoenginetest') c = connect('mongoenginetest')
@ -131,18 +133,25 @@ class ConnectionTest(unittest.TestCase):
c.admin.add_user('username', 'password') c.admin.add_user('username', 'password')
# Authentication fails without "authSource" # Authentication fails without "authSource"
self.assertRaises( if IS_PYMONGO_3:
ConnectionError, connect, 'mongoenginetest', alias='test1', test_conn = connect('mongoenginetest', alias='test2',
host='mongodb://username:password@localhost/mongoenginetest' host='mongodb://username:password@localhost/mongoenginetest')
) self.assertRaises(OperationFailure, test_conn.server_info)
self.assertRaises(ConnectionError, get_db, 'test1') else:
self.assertRaises(
ConnectionError, connect, 'mongoenginetest', alias='test1',
host='mongodb://username:password@localhost/mongoenginetest'
)
self.assertRaises(ConnectionError, get_db, 'test1')
# Authentication succeeds with "authSource" # Authentication succeeds with "authSource"
connect( test_conn2 = connect(
'mongoenginetest', alias='test2', 'mongoenginetest', alias='test2',
host=('mongodb://username:password@localhost/' host=('mongodb://username:password@localhost/'
'mongoenginetest?authSource=admin') 'mongoenginetest?authSource=admin')
) )
# 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')