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

View File

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

View File

@ -65,6 +65,16 @@ class ConnectionTest(unittest.TestCase):
self.assertTrue(isinstance(db, pymongo.database.Database))
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__':
unittest.main()