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

fix for #494

Browse Source
This commit is contained in:
Anthony Nemitz 2012-05-09 02:54:08 -07:00
parent be1c28fc45
commit dd786d6fc4
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mongoengine/connection.py
View File

@ -65,6 +65,8 @@ def register_connection(alias, name, host='localhost', port=27017,
}) })
if "replicaSet" in host: if "replicaSet" in host:
conn_settings['replicaSet'] = True conn_settings['replicaSet'] = True
conn_settings.update(kwargs)
_connection_settings[alias] = conn_settings _connection_settings[alias] = conn_settings

10
tests/connection.py
View File

@ -65,6 +65,16 @@ class ConnectionTest(unittest.TestCase):
self.assertTrue(isinstance(db, pymongo.database.Database)) self.assertTrue(isinstance(db, pymongo.database.Database))
self.assertEqual(db.name, 'mongoenginetest2') self.assertEqual(db.name, 'mongoenginetest2')
def test_connection_kwargs(self):
"""Ensure that connection kwargs get passed to pymongo.
"""
connect('mongoenginetest', alias='t1', tz_aware=True)
conn = get_connection('t1')
self.assertTrue(conn.tz_aware)
connect('mongoenginetest2', alias='t2')
conn = get_connection('t2')
self.assertFalse(conn.tz_aware)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()