diff --git a/tests/test_connection.py b/tests/test_connection.py index 9143571f..43943ed9 100644 --- a/tests/test_connection.py +++ b/tests/test_connection.py @@ -296,6 +296,15 @@ class ConnectionTest(unittest.TestCase): conn = get_connection('t2') self.assertFalse(get_tz_awareness(conn)) + def test_write_concern(self): + """Ensure write concern can be specified in connect() via + a kwarg or as part of the connection URI. + """ + conn1 = connect(alias='conn1', host='mongodb://localhost/testing?w=1&j=true') + conn2 = connect('testing', alias='conn2', w=1, j=True) + self.assertEqual(conn1.write_concern.document, {'j': True, 'w': 1}) + self.assertEqual(conn2.write_concern.document, {'j': True, 'w': 1}) + def test_datetime(self): connect('mongoenginetest', tz_aware=True) d = datetime.datetime(2010, 5, 5, tzinfo=utc)